diff --git a/cucyuqing/cmd/embedding.py b/cucyuqing/cmd/embedding.py index 3fdd1e6..ddbbf86 100644 --- a/cucyuqing/cmd/embedding.py +++ b/cucyuqing/cmd/embedding.py @@ -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, )