ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: rename google -> gemini, refresh model list to 2.5 series

Browse source
This commit is contained in:
Roland Tannous 2026-03-30 04:07:52 +00:00
commit 70f488b0db
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
studio/backend/core/inference/providers.py
View file

@ -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",

2
studio/backend/tests/test_providers_api.py
View file

@ -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"),