From df88492113c3e044c8af875548211d3b126272d8 Mon Sep 17 00:00:00 2001 From: yangtb2024 Date: Tue, 18 Mar 2025 15:21:29 +0800 Subject: [PATCH] =?UTF-8?q?=E5=B0=86chat-bison-001=E3=80=81text-bison-001?= =?UTF-8?q?=E5=92=8Cembedding-gecko-001=E6=B7=BB=E5=8A=A0=E5=88=B0FILTERED?= =?UTF-8?q?=5FMODELS=E5=88=97=E8=A1=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .env.example | 2 +- app/core/config.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.env.example b/.env.example index 080d9b4..357eb69 100644 --- a/.env.example +++ b/.env.example @@ -1,7 +1,7 @@ API_KEYS=["AIzaSyxxxxxxxxxxxxxxxxxxx","AIzaSyxxxxxxxxxxxxxxxxxxx"] ALLOWED_TOKENS=["sk-123456"] # AUTH_TOKEN=sk-123456 -FILTERED_MODELS=["gemini-1.0-pro-vision-latest", "gemini-pro-vision"] +FILTERED_MODELS=["gemini-1.0-pro-vision-latest", "gemini-pro-vision", "chat-bison-001", "text-bison-001", "embedding-gecko-001"] MODEL_SEARCH=["gemini-2.0-flash-exp","gemini-2.0-pro-exp"] MODEL_IMAGE=["gemini-2.0-flash-exp"] TOOLS_CODE_EXECUTION_ENABLED=false diff --git a/app/core/config.py b/app/core/config.py index 05031b2..daa9cdf 100644 --- a/app/core/config.py +++ b/app/core/config.py @@ -8,7 +8,7 @@ class Settings(BaseSettings): BASE_URL: str = "https://generativelanguage.googleapis.com/v1beta" MODEL_SEARCH: List[str] = ["gemini-2.0-flash-exp"] MODEL_IMAGE: List[str] = ["gemini-2.0-flash-exp"] - FILTERED_MODELS: List[str] = ["gemini-1.0-pro-vision-latest", "gemini-pro-vision"] + FILTERED_MODELS: List[str] = ["gemini-1.0-pro-vision-latest", "gemini-pro-vision", "chat-bison-001", "text-bison-001", "embedding-gecko-001"] TOOLS_CODE_EXECUTION_ENABLED: bool = False SHOW_SEARCH_LINK: bool = True SHOW_THINKING_PROCESS: bool = True