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

feat: 根据模型版本调整安全设置逻辑

Browse Source
This commit is contained in:
yinpeng
2025-01-18 23:22:39 +08:00
parent 9f9197f8c8
commit 0a9de39143
2 changed files with 22 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/services/gemini_chat_service.py
+11 -3
View File
@@ -53,7 +53,7 @@ class GeminiChatService:
return { return {
"contents": payload.get("contents", []), "contents": payload.get("contents", []),
"tools": self._build_tools(model, payload), "tools": self._build_tools(model, payload),
"safetySettings": self._get_safety_settings(), "safetySettings": self._get_safety_settings(model),
"generationConfig": payload.get("generationConfig", {}), "generationConfig": payload.get("generationConfig", {}),
"systemInstruction": payload.get("systemInstruction", []) "systemInstruction": payload.get("systemInstruction", [])
} }
@@ -78,12 +78,20 @@ class GeminiChatService:
return True return True
return False return False
def _get_safety_settings(self) -> List[Dict[str, str]]: def _get_safety_settings(self, model: str) -> List[Dict[str, str]]:
"""获取安全设置""" """获取安全设置"""
if "2.0" in model and model != "gemini-2.0-flash-thinking-exp":
return [
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "OFF"},
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "OFF"},
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "OFF"},
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "OFF"},
{"category": "HARM_CATEGORY_CIVIC_INTEGRITY", "threshold": "BLOCK_ONLY_HIGH"}
]
return [ return [
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"}, {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"}, {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"}, {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"}, {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
{"category": "HARM_CATEGORY_CIVIC_INTEGRITY", "threshold": "BLOCK_NONE"} {"category": "HARM_CATEGORY_CIVIC_INTEGRITY", "threshold": "BLOCK_ONLY_HIGH"}
] ]
app/services/openai_chat_service.py
+11 -3
View File
@@ -100,7 +100,7 @@ class OpenAIChatService:
"topK": request.top_k "topK": request.top_k
}, },
"tools": self._build_tools(request, messages), "tools": self._build_tools(request, messages),
"safetySettings": self._get_safety_settings() "safetySettings": self._get_safety_settings(request.model)
} }
def _build_tools(self, request: ChatRequest, messages: List[Dict[str, Any]]) -> List[Dict[str, Any]]: def _build_tools(self, request: ChatRequest, messages: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
@@ -125,12 +125,20 @@ class OpenAIChatService:
return True return True
return False return False
def _get_safety_settings(self) -> List[Dict[str, str]]: def _get_safety_settings(self, model: str) -> List[Dict[str, str]]:
"""获取安全设置""" """获取安全设置"""
if "2.0" in model and model != "gemini-2.0-flash-thinking-exp":
return [
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "OFF"},
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "OFF"},
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "OFF"},
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "OFF"},
{"category": "HARM_CATEGORY_CIVIC_INTEGRITY", "threshold": "BLOCK_ONLY_HIGH"}
]
return [ return [
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"}, {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"}, {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"}, {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"}, {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
{"category": "HARM_CATEGORY_CIVIC_INTEGRITY", "threshold": "BLOCK_NONE"} {"category": "HARM_CATEGORY_CIVIC_INTEGRITY", "threshold": "BLOCK_ONLY_HIGH"}
] ]