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

fix: 修复生成配置字段名称以符合API要求

Browse Source
This commit is contained in:
snaily
2025-07-19 00:40:44 +08:00
parent dd1fa35c73
commit 5a98a701cb
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/router/gemini_routes.py
+2 -2
View File
@@ -329,7 +329,7 @@ async def verify_key(api_key: str, chat_service: GeminiChatService = Depends(get
parts=[{"text": "hi"}], parts=[{"text": "hi"}],
) )
], ],
generation_config={"temperature": 0.7, "top_p": 1.0, "max_output_tokens": 10} generation_config={"temperature": 0.7, "topP": 1.0, "maxOutputTokens": 10}
) )
response = await chat_service.generate_content( response = await chat_service.generate_content(
@@ -374,7 +374,7 @@ async def verify_selected_keys(
try: try:
gemini_request = GeminiRequest( gemini_request = GeminiRequest(
contents=[GeminiContent(role="user", parts=[{"text": "hi"}])], contents=[GeminiContent(role="user", parts=[{"text": "hi"}])],
generation_config={"temperature": 0.7, "top_p": 1.0, "max_output_tokens": 10} generation_config={"temperature": 0.7, "topP": 1.0, "maxOutputTokens": 10}
) )
await chat_service.generate_content( await chat_service.generate_content(
settings.TEST_MODEL, settings.TEST_MODEL,
app/service/chat/gemini_chat_service.py
+2 -1
View File
@@ -161,7 +161,8 @@ def _build_payload(model: str, request: GeminiRequest) -> Dict[str, Any]:
if request.generationConfig: if request.generationConfig:
if request.generationConfig.maxOutputTokens is None: if request.generationConfig.maxOutputTokens is None:
# 如果未指定最大输出长度，则不传递该字段，解决截断的问题 # 如果未指定最大输出长度，则不传递该字段，解决截断的问题
request_dict["generationConfig"].pop("maxOutputTokens") if "maxOutputTokens" in request_dict["generationConfig"]:
request_dict["generationConfig"].pop("maxOutputTokens")
# 检查是否为TTS模型 # 检查是否为TTS模型
is_tts_model = "tts" in model.lower() is_tts_model = "tts" in model.lower()