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

fix: ensure generationConfig is not None in payload

Browse Source
This commit is contained in:
ripper
2025-07-09 10:17:32 +08:00
parent b3da021803
commit 858df0548e
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/service/chat/gemini_chat_service.py
+4
View File
@@ -114,6 +114,10 @@ def _build_payload(model: str, request: GeminiRequest) -> Dict[str, Any]:
"systemInstruction": request_dict.get("systemInstruction"), "systemInstruction": request_dict.get("systemInstruction"),
} }
# 确保 generationConfig 不为 None
if payload["generationConfig"] is None:
payload["generationConfig"] = {}
if model.endswith("-image") or model.endswith("-image-generation"): if model.endswith("-image") or model.endswith("-image-generation"):
payload.pop("systemInstruction") payload.pop("systemInstruction")
payload["generationConfig"]["responseModalities"] = ["Text", "Image"] payload["generationConfig"]["responseModalities"] = ["Text", "Image"]