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

feat: 将 Gemini API 调用迁移至 ChatService 并支持 API Key 验证

Browse Source
This commit is contained in:
yinpeng
2024-12-18 21:35:49 +08:00
parent 914922624e
commit 8dfe617468
5 changed files with 64 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/schemas/gemini_models.py
+4 -4
View File
@@ -3,7 +3,7 @@ from pydantic import BaseModel
class SafetySetting(BaseModel):
category: Optional[Literal["HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT"]] = None
threshold: Optional[Literal["HARM_BLOCK", "HARM_FLAG", "HARM_UNSPECIFIED"]] = None
threshold: Optional[Literal["HARM_BLOCK_THRESHOLD_UNSPECIFIED", "BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE","BLOCK_ONLY_HIGH","BLOCK_NONE","OFF"]] = None
class GenerationConfig(BaseModel):
@@ -33,7 +33,7 @@ class GeminiContent(BaseModel):
class GeminiRequest(BaseModel):
contents: List[GeminiContent]
# tools: Optional[List[Dict[str, Any]]] = None
# safetySettings: Optional[List[SafetySetting]] = None
tools: Optional[List[Dict[str, Any]]] = []
safetySettings: Optional[List[SafetySetting]] = None
generationConfig: Optional[GenerationConfig] = None
# systemInstruction: Optional[SystemInstruction] = None
systemInstruction: Optional[SystemInstruction] = None