GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-05-11 18:09:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 根据模型类型选择不同的API密钥处理聊天请求

Browse Source
This commit is contained in:
yinpeng
2025-02-11 04:20:28 +08:00
parent 34a98389f5
commit 97b9b99235
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/api/openai_routes.py
View File

@@ -54,7 +54,11 @@ async def chat_completion(
logger.info(f"Request: \n{request.model_dump_json(indent=2)}")
logger.info(f"Using API key: {api_key}")
try:
response = await chat_service.create_image_chat_completion(request=request)
# 如果model是imagen3,使用paid_key
if request.model == f"{settings.CREATE_IMAGE_MODEL}-chat":
response = await chat_service.create_image_chat_completion(request=request)
else:
response = await chat_service.create_chat_completion(request,api_key)
# 处理流式响应
if request.stream:
return StreamingResponse(response, media_type="text/event-stream")