This commit is contained in:
minecraft1024a
2025-08-19 18:48:17 +08:00

View File

@@ -37,7 +37,7 @@ class APIProvider(ValidatedConfigBase):
@classmethod @classmethod
def validate_client_type(cls, v): def validate_client_type(cls, v):
"""验证客户端类型""" """验证客户端类型"""
allowed_types = ["openai", "gemini"] allowed_types = ["openai", "gemini","aiohttp_gemini"]
if v not in allowed_types: if v not in allowed_types:
raise ValueError(f"客户端类型必须是以下之一: {allowed_types}") raise ValueError(f"客户端类型必须是以下之一: {allowed_types}")
return v return v
@@ -89,9 +89,9 @@ class TaskConfig(ValidatedConfigBase):
"""任务配置类""" """任务配置类"""
model_list: List[str] = Field(..., description="任务使用的模型列表") model_list: List[str] = Field(..., description="任务使用的模型列表")
max_tokens: int = Field(default=None, ge=1, le=100000, description="任务最大输出token数") max_tokens: int = Field(default=800, description="任务最大输出token数")
temperature: float = Field(default=None, ge=0.0, le=2.0, description="模型温度") temperature: float = Field(default=0.7, description="模型温度")
concurrency_count: int = Field(default=None, ge=1, le=10, description="并发请求数量") concurrency_count: int = Field(default=1, description="并发请求数量")
@field_validator('model_list') @field_validator('model_list')
@classmethod @classmethod