This commit is contained in:
2024-09-19 14:35:34 +08:00
parent 71fd688227
commit 1e1f92461a

View File

@@ -10,14 +10,14 @@ import hashlib
from pydantic import BaseModel
from typing import Any, Literal
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
EmbeddingModel = Literal["acge-large-zh", "text-embedding-3-small"]
embedding_client = openai.AsyncOpenAI(
api_key=OPENAI_API_KEY,
base_url=OPENAI_BASE_URL,
api_key=OPENAI_EMBEDDING_API_KEY,
base_url=OPENAI_EMBEDDING_BASE_URL,
)