mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-05-30 20:59:45 +08:00
cd45f4b5ab
- 将 `_build_payload`、`_build_tools`、`_get_safety_settings` 和 `_has_image_parts` 函数从 `OpenAIChatService` 和 `GeminiChatService` 类中提取为独立的函数。 - 将 `_handle_stream_response` 和 `_handle_normal_response` 函数从 `GeminiResponseHandler` 和 `OpenAIResponseHandler` 类中提取为独立的函数。 - 将 `_extract_text` 函数从 `OpenAIResponseHandler` 类中提取为独立的函数, 并在 `GeminiResponseHandler` 中复用。 - 将 `_convert_image` 函数从 `OpenAIMessageConverter` 类中提取为独立的函数。 - 优化 `OpenAIChatService` 和 `GeminiChatService` 中的代码结构, 使其更清晰。 - 优化 `app/api/openai_routes.py` 和 `app/api/gemini_routes.py` 中的路由函数, 移除不必要的参数。
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
from typing import List, Optional, Dict, Any, Literal
|
|
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", "HARM_CATEGORY_CIVIC_INTEGRITY"]] = 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):
|
|
stopSequences: Optional[List[str]] = None
|
|
responseMimeType: Optional[str] = None
|
|
responseSchema: Optional[Dict[str, Any]] = None
|
|
candidateCount: Optional[int] = 1
|
|
maxOutputTokens: Optional[int] = None
|
|
temperature: Optional[float] = None
|
|
topP: Optional[float] = None
|
|
topK: Optional[int] = None
|
|
presencePenalty: Optional[float] = None
|
|
frequencyPenalty: Optional[float] = None
|
|
responseLogprobs: Optional[bool] = None
|
|
logprobs: Optional[int] = None
|
|
|
|
|
|
class SystemInstruction(BaseModel):
|
|
role: str = "system"
|
|
parts: List[Dict[str, Any]]
|
|
|
|
|
|
class GeminiContent(BaseModel):
|
|
role: str
|
|
parts: List[Dict[str, Any]]
|
|
|
|
|
|
class GeminiRequest(BaseModel):
|
|
contents: List[GeminiContent]
|
|
tools: Optional[List[Dict[str, Any]]] = []
|
|
safetySettings: Optional[List[SafetySetting]] = None
|
|
generationConfig: Optional[GenerationConfig] = None
|
|
systemInstruction: Optional[SystemInstruction] = None
|