fix: 修复 OpenAI 和 Gemini API 调用重试逻辑及日志记录

This commit is contained in:
yinpeng
2024-12-21 02:27:59 +08:00
parent 80bcaf5cd4
commit 33a5f9d89f
9 changed files with 192 additions and 168 deletions
+5 -2
View File
@@ -1,5 +1,8 @@
from typing import Union, List
import openai
from typing import Union, List, Dict, Any
from openai.types import CreateEmbeddingResponse
from app.core.logger import get_embeddings_logger
logger = get_embeddings_logger()
@@ -11,7 +14,7 @@ class EmbeddingService:
async def create_embedding(
self, input_text: Union[str, List[str]], model: str, api_key: str
) -> Dict[str, Any]:
) -> CreateEmbeddingResponse:
"""Create embeddings using OpenAI API"""
try:
client = openai.OpenAI(api_key=api_key, base_url=self.base_url)