GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-09-05 23:56:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Filter out vision-based Gemini models from model list

Browse Source
This commit is contained in:
yangtb2024
2025-03-17 13:56:01 +08:00
parent 8779a5f0b3
commit 016e6e06ee
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/services/model_service.py
+6
View File
@@ -11,6 +11,7 @@ class ModelService:
self.model_search = model_search self.model_search = model_search
self.model_image = model_image self.model_image = model_image
self.base_url = "https://generativelanguage.googleapis.com/v1beta" self.base_url = "https://generativelanguage.googleapis.com/v1beta"
self.filtered_models = ["gemini-1.0-pro-vision-latest", "gemini-pro-vision"]
def get_gemini_models(self, api_key: str) -> Optional[Dict[str, Any]]: def get_gemini_models(self, api_key: str) -> Optional[Dict[str, Any]]:
url = f"{self.base_url}/models?key={api_key}" url = f"{self.base_url}/models?key={api_key}"
@@ -43,6 +44,11 @@ class ModelService:
for model in gemini_models.get("models", []): for model in gemini_models.get("models", []):
model_id = model["name"].split("/")[-1] model_id = model["name"].split("/")[-1]
if model_id in self.filtered_models:
logger.info(f"Filtered out model: {model_id}")
continue
openai_model = { openai_model = {
"id": model_id, "id": model_id,
"object": "model", "object": "model",