From 70f488b0db87cec6ca93df89596328906a6cceb2 Mon Sep 17 00:00:00 2001 From: Roland Tannous Date: Mon, 30 Mar 2026 04:07:52 +0000 Subject: [PATCH] feat: rename google -> gemini, refresh model list to 2.5 series --- studio/backend/core/inference/providers.py | 5 +++-- studio/backend/tests/test_providers_api.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/studio/backend/core/inference/providers.py b/studio/backend/core/inference/providers.py index 654d9672c1..36f50a28a7 100644 --- a/studio/backend/core/inference/providers.py +++ b/studio/backend/core/inference/providers.py @@ -42,20 +42,21 @@ PROVIDER_REGISTRY: dict[str, dict[str, Any]] = { "auth_header": "Authorization", "auth_prefix": "Bearer ", }, - "google": { + "gemini": { "display_name": "Google Gemini", "base_url": "https://generativelanguage.googleapis.com/v1beta/openai", "default_models": [ "gemini-2.5-flash", "gemini-2.5-pro", "gemini-2.5-flash-lite", + "gemini-2.0-flash", ], "supports_streaming": True, "supports_vision": True, "supports_tool_calling": True, "auth_header": "Authorization", "auth_prefix": "Bearer ", - "notes": "OpenAI compatibility layer (beta). Uses native Google API key as Bearer token.", + "notes": "OpenAI-compatible endpoint. API key from https://aistudio.google.com/apikey.", }, "cohere": { "display_name": "Cohere", diff --git a/studio/backend/tests/test_providers_api.py b/studio/backend/tests/test_providers_api.py index 9482c14c69..12a5d0bcad 100644 --- a/studio/backend/tests/test_providers_api.py +++ b/studio/backend/tests/test_providers_api.py @@ -43,7 +43,7 @@ PASSWORD = os.getenv("STUDIO_TEST_PASSWORD", "") _PROVIDER_CONFIGS: dict[str, tuple[str, str]] = { "openai": ("OPENAI_API_KEY", "gpt-4o-mini"), "mistral": ("MISTRAL_API_KEY", "mistral-small-latest"), - "google": ("GOOGLE_API_KEY", "gemini-2.5-flash"), + "gemini": ("GEMINI_API_KEY", "gemini-2.5-flash"), "cohere": ("COHERE_API_KEY", "command-a-03-2025"), "together": ("TOGETHER_API_KEY", "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8"), "fireworks": ("FIREWORKS_API_KEY", "accounts/fireworks/models/llama4-maverick-instruct-basic"),