From 1e1f92461a3d30b9717ea5d1bbd1fb1fc8d76d24 Mon Sep 17 00:00:00 2001 From: heimoshuiyu Date: Thu, 19 Sep 2024 14:35:34 +0800 Subject: [PATCH] typo --- cucyuqing/cmd/embedding.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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, )