GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-09-07 00:36:36 +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
app/api/openai_routes.py
+5 -1
View File
@@ -54,7 +54,11 @@ async def chat_completion(
logger.info(f"Request: \n{request.model_dump_json(indent=2)}") logger.info(f"Request: \n{request.model_dump_json(indent=2)}")
logger.info(f"Using API key: {api_key}") logger.info(f"Using API key: {api_key}")
try: 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: if request.stream:
return StreamingResponse(response, media_type="text/event-stream") return StreamingResponse(response, media_type="text/event-stream")