From 1b816943733a8e256d448f003bda99fc8894ccbf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9B=85=E8=AF=BA=E7=8B=90?= <212194964+foxcyber907@users.noreply.github.com> Date: Tue, 19 Aug 2025 18:42:33 +0800 Subject: [PATCH] Add aiohttp_gemini client type and update TaskConfig defaults Added 'aiohttp_gemini' to allowed client types in APIProvider. Updated TaskConfig defaults: max_tokens to 800, temperature to 0.7, and concurrency_count to 1 for improved default behavior. --- src/config/api_ada_configs.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/config/api_ada_configs.py b/src/config/api_ada_configs.py index 030b6dd43..3c7827b81 100644 --- a/src/config/api_ada_configs.py +++ b/src/config/api_ada_configs.py @@ -37,7 +37,7 @@ class APIProvider(ValidatedConfigBase): @classmethod def validate_client_type(cls, v): """验证客户端类型""" - allowed_types = ["openai", "gemini"] + allowed_types = ["openai", "gemini","aiohttp_gemini"] if v not in allowed_types: raise ValueError(f"客户端类型必须是以下之一: {allowed_types}") return v @@ -89,9 +89,9 @@ class TaskConfig(ValidatedConfigBase): """任务配置类""" model_list: List[str] = Field(..., description="任务使用的模型列表") - max_tokens: int = Field(default=None, ge=1, le=100000, description="任务最大输出token数") - temperature: float = Field(default=None, ge=0.0, le=2.0, description="模型温度") - concurrency_count: int = Field(default=None, ge=1, le=10, description="并发请求数量") + max_tokens: int = Field(default=800, description="任务最大输出token数") + temperature: float = Field(default=0.7, description="模型温度") + concurrency_count: int = Field(default=1, description="并发请求数量") @field_validator('model_list') @classmethod