mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-05-24 17:59:31 +08:00
95b5acad66
对多个模块进行了重构，以改进错误处理和日志记录机制。 主要变更包括： - 在 `gemini_routes` 中，现在会返回更具体的错误信息，包括错误码和错误消息，而不仅仅是异常的字符串表示。 - 在 `api_client` 中，简化了 Gemini API 客户端的错误处理逻辑，移除了冗余的 `try...except` 块，让异常直接向上抛出。 - 在多个服务（如 `openai_chat_service`, `embedding_service`, `tts_service` 等）中，增加了根据配置项 `ERROR_LOG_RECORD_REQUEST_BODY` 来决定是否记录请求体的逻辑，以增强隐私和性能控制。 - 在前端 `keys_status.js` 中，更新了密钥验证结果的处理逻辑，以适应后端返回的新的错误对象结构（包含 `error_code` 和 `error_message`），并移除了冗余的 `executeVerifyAllKeys` 函数。
87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
import datetime
|
|
import time
|
|
from typing import List, Union
|
|
|
|
import openai
|
|
from openai import APIStatusError
|
|
from openai.types import CreateEmbeddingResponse
|
|
|
|
from app.config.config import settings
|
|
from app.database.services import add_error_log, add_request_log
|
|
from app.log.logger import get_embeddings_logger
|
|
|
|
logger = get_embeddings_logger()
|
|
|
|
|
|
class EmbeddingService:
|
|
|
|
async def create_embedding(
|
|
self, input_text: Union[str, List[str]], model: str, api_key: str
|
|
) -> CreateEmbeddingResponse:
|
|
"""Create embeddings using OpenAI API with database logging"""
|
|
start_time = time.perf_counter()
|
|
request_datetime = datetime.datetime.now()
|
|
is_success = False
|
|
status_code = None
|
|
response = None
|
|
error_log_msg = ""
|
|
if isinstance(input_text, list):
|
|
request_msg_log = {
|
|
"input_truncated": [
|
|
str(item)[:100] + "..." if len(str(item)) > 100 else str(item)
|
|
for item in input_text[:5]
|
|
]
|
|
}
|
|
if len(input_text) > 5:
|
|
request_msg_log["input_truncated"].append("...")
|
|
else:
|
|
request_msg_log = {
|
|
"input_truncated": (
|
|
input_text[:1000] + "..." if len(input_text) > 1000 else input_text
|
|
)
|
|
}
|
|
|
|
try:
|
|
client = openai.OpenAI(api_key=api_key, base_url=settings.BASE_URL)
|
|
response = client.embeddings.create(input=input_text, model=model)
|
|
is_success = True
|
|
status_code = 200
|
|
return response
|
|
except APIStatusError as e:
|
|
is_success = False
|
|
status_code = e.status_code
|
|
error_log_msg = f"OpenAI API error: {e}"
|
|
logger.error(f"Error creating embedding (APIStatusError): {error_log_msg}")
|
|
raise e
|
|
except Exception as e:
|
|
is_success = False
|
|
status_code = 500
|
|
error_log_msg = f"Generic error: {e}"
|
|
logger.error(f"Error creating embedding (Exception): {error_log_msg}")
|
|
raise e
|
|
finally:
|
|
end_time = time.perf_counter()
|
|
latency_ms = int((end_time - start_time) * 1000)
|
|
if not is_success:
|
|
await add_error_log(
|
|
gemini_key=api_key,
|
|
model_name=model,
|
|
error_type="openai-embedding",
|
|
error_log=error_log_msg,
|
|
error_code=status_code,
|
|
request_msg=(
|
|
request_msg_log
|
|
if settings.ERROR_LOG_RECORD_REQUEST_BODY
|
|
else None
|
|
),
|
|
request_datetime=request_datetime,
|
|
)
|
|
await add_request_log(
|
|
model_name=model,
|
|
api_key=api_key,
|
|
is_success=is_success,
|
|
status_code=status_code,
|
|
latency_ms=latency_ms,
|
|
request_time=request_datetime,
|
|
)
|