From 05762cb6a53b2abb910f65384dab0c1c6751dded Mon Sep 17 00:00:00 2001 From: snaily Date: Thu, 18 Sep 2025 05:24:29 +0800 Subject: [PATCH] =?UTF-8?q?feat(config):=E6=9B=B4=E6=96=B0=E9=BB=98?= =?UTF-8?q?=E8=AE=A4=E6=A8=A1=E5=9E=8B=E5=92=8C=E7=9B=B8=E5=85=B3=E9=85=8D?= =?UTF-8?q?=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 更新默认模型和相关配置： - 将默认测试模型从 gemini-1.5-flash 更新为 gemini-2.5-flash-lite - 更新思考模型列表至 gemini-2.5-flash 和 gemini-2.5-pro - 添加新的图像模型 gemini-2.5-flash-image-preview - 更新搜索模型配置以支持最新的 Gemini 2.5 系列 - 同步更新文档中的模型配置说明 --- .env.example | 10 +++++----- README.md | 6 +++--- README_ZH.md | 6 +++--- app/config/config.py | 4 ++-- app/core/constants.py | 2 +- 5 files changed, 14 insertions(+), 14 deletions(-) diff --git a/.env.example b/.env.example index 1b9543a..d07a9fc 100644 --- a/.env.example +++ b/.env.example @@ -14,11 +14,11 @@ AUTH_TOKEN=sk-123456 VERTEX_API_KEYS=["AQ.Abxxxxxxxxxxxxxxxxxxx"] # For Vertex AI Platform Express API Base URL VERTEX_EXPRESS_BASE_URL=https://aiplatform.googleapis.com/v1beta1/publishers/google -TEST_MODEL=gemini-1.5-flash -THINKING_MODELS=["gemini-2.5-flash-preview-04-17"] -THINKING_BUDGET_MAP={"gemini-2.5-flash-preview-04-17": 4000} -IMAGE_MODELS=["gemini-2.0-flash-exp"] -SEARCH_MODELS=["gemini-2.0-flash-exp","gemini-2.0-pro-exp"] +TEST_MODEL=gemini-2.5-flash-lite +THINKING_MODELS=["gemini-2.5-flash","gemini-2.5-pro"] +THINKING_BUDGET_MAP={"gemini-2.5-flash": -1} +IMAGE_MODELS=["gemini-2.0-flash-exp", "gemini-2.5-flash-image-preview"] +SEARCH_MODELS=["gemini-2.5-flash","gemini-2.5-pro"] FILTERED_MODELS=["gemini-1.0-pro-vision-latest", "gemini-pro-vision", "chat-bison-001", "text-bison-001", "embedding-gecko-001"] # 是否启用网址上下文，默认启用 URL_CONTEXT_ENABLED=false diff --git a/README.md b/README.md index a3d144a..5326ed7 100644 --- a/README.md +++ b/README.md @@ -184,9 +184,9 @@ This endpoint is directly forwarded to official OpenAI Compatible API format end | `ALLOWED_TOKENS` | **Required**, list of access tokens | `[]` | | `AUTH_TOKEN` | Super admin token, defaults to the first of `ALLOWED_TOKENS` | `sk-123456` | | `ADMIN_SESSION_EXPIRE` | Admin session expiration time in seconds (5 minutes to 24 hours) | `3600` | -| `TEST_MODEL` | Model for testing key validity | `gemini-1.5-flash` | -| `IMAGE_MODELS` | Models supporting image generation | `["gemini-2.0-flash-exp"]` | -| `SEARCH_MODELS` | Models supporting web search | `["gemini-2.0-flash-exp"]` | +| `TEST_MODEL` | Model for testing key validity | `gemini-2.5-flash-lite` | +| `IMAGE_MODELS` | Models supporting image generation | `["gemini-2.0-flash-exp", "gemini-2.5-flash-image-preview"]` | +| `SEARCH_MODELS` | Models supporting web search | `["gemini-2.5-flash","gemini-2.5-pro"]` | | `FILTERED_MODELS` | Disabled models | `[]` | | `TOOLS_CODE_EXECUTION_ENABLED` | Enable code execution tool | `false` | | `SHOW_SEARCH_LINK` | Display search result links in response | `true` | diff --git a/README_ZH.md b/README_ZH.md index 5ca3b2c..7156532 100644 --- a/README_ZH.md +++ b/README_ZH.md @@ -184,9 +184,9 @@ app/ | `API_KEYS` | **必填**, Gemini API 密钥列表，用于负载均衡 | `[]` | | `ALLOWED_TOKENS` | **必填**, 允许访问的 Token 列表 | `[]` | | `AUTH_TOKEN` | 超级管理员 Token，不填则使用 `ALLOWED_TOKENS` 的第一个 | `sk-123456` | -| `TEST_MODEL` | 用于测试密钥可用性的模型 | `gemini-1.5-flash` | -| `IMAGE_MODELS` | 支持绘图功能的模型列表 | `["gemini-2.0-flash-exp"]` | -| `SEARCH_MODELS` | 支持搜索功能的模型列表 | `["gemini-2.0-flash-exp"]` | +| `TEST_MODEL` | 用于测试密钥可用性的模型 | `gemini-2.5-flash-lite` | +| `IMAGE_MODELS` | 支持绘图功能的模型列表 | `["gemini-2.0-flash-exp", "gemini-2.5-flash-image-preview"]` | +| `SEARCH_MODELS` | 支持搜索功能的模型列表 | `["gemini-2.5-flash","gemini-2.5-pro"]` | | `FILTERED_MODELS` | 被禁用的模型列表 | `[]` | | `TOOLS_CODE_EXECUTION_ENABLED` | 是否启用代码执行工具 | `false` | | `SHOW_SEARCH_LINK` | 是否在响应中显示搜索结果链接 | `true` | diff --git a/app/config/config.py b/app/config/config.py index df9e1a9..4ae10bd 100644 --- a/app/config/config.py +++ b/app/config/config.py @@ -73,8 +73,8 @@ class Settings(BaseSettings): CUSTOM_HEADERS: Dict[str, str] = {} # 模型相关配置 - SEARCH_MODELS: List[str] = ["gemini-2.0-flash-exp"] - IMAGE_MODELS: List[str] = ["gemini-2.0-flash-exp"] + SEARCH_MODELS: List[str] = ["gemini-2.5-flash", "gemini-2.5-pro"] + IMAGE_MODELS: List[str] = ["gemini-2.0-flash-exp", "gemini-2.5-flash-image-preview"] FILTERED_MODELS: List[str] = DEFAULT_FILTER_MODELS TOOLS_CODE_EXECUTION_ENABLED: bool = False # 是否启用网址上下文 diff --git a/app/core/constants.py b/app/core/constants.py index f7aef94..9041bb4 100644 --- a/app/core/constants.py +++ b/app/core/constants.py @@ -9,7 +9,7 @@ MAX_RETRIES = 3 # 最大重试次数 # 模型相关常量 SUPPORTED_ROLES = ["user", "model", "system"] -DEFAULT_MODEL = "gemini-1.5-flash" +DEFAULT_MODEL = "gemini-2.5-flash-lite" DEFAULT_TEMPERATURE = 0.7 DEFAULT_MAX_TOKENS = 8192 DEFAULT_TOP_P = 0.9