heimoshuiyu/cucyuqing
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

typo

Browse Source
This commit is contained in:
黑墨水鱼
2024-09-19 14:35:34 +08:00
parent 71fd688227
commit 1e1f92461a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
cucyuqing/cmd/embedding.py
View File

@@ -10,14 +10,14 @@ import hashlib
from pydantic import BaseModel from pydantic import BaseModel
from typing import Any, Literal from typing import Any, Literal
from cucyuqing.pg import pool, get_cur from cucyuqing.pg import pool, get_cur
from cucyuqing.config import OPENAI_API_KEY, OPENAI_BASE_URL from cucyuqing.config import OPENAI_EMBEDDING_API_KEY, OPENAI_EMBEDDING_BASE_URL
from cucyuqing.utils import print from cucyuqing.utils import print
EmbeddingModel = Literal["acge-large-zh", "text-embedding-3-small"] EmbeddingModel = Literal["acge-large-zh", "text-embedding-3-small"]
embedding_client = openai.AsyncOpenAI( embedding_client = openai.AsyncOpenAI(
api_key=OPENAI_API_KEY, api_key=OPENAI_EMBEDDING_API_KEY,
base_url=OPENAI_BASE_URL, base_url=OPENAI_EMBEDDING_BASE_URL,
) )