Skip to content

Commit 4521e4d

Browse files
fix: update default VLM model from deprecated gemini-2.0-flash to gemini-2.5-flash
The gemini-2.0-flash model has been deprecated by Google and returns 404 NOT_FOUND for new users. This updates all default references to gemini-2.5-flash across config, provider, tests, examples, and docs. Fixes #44
1 parent a6b8ace commit 4521e4d

File tree

7 files changed

+9
-9
lines changed

7 files changed

+9
-9
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ PaperBanana supports multiple VLM and image generation providers:
137137
|-----------|----------|-------|-------|
138138
| VLM (planning, critique) | OpenAI | `gpt-5.2` | Default |
139139
| Image Generation | OpenAI | `gpt-image-1.5` | Default |
140-
| VLM | Google Gemini | `gemini-2.0-flash` | Free tier |
140+
| VLM | Google Gemini | `gemini-2.5-flash` | Free tier |
141141
| Image Generation | Google Gemini | `gemini-3-pro-image-preview` | Free tier |
142142
| VLM / Image | OpenRouter | Any supported model | Flexible routing |
143143

configs/config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
# Provider settings
55
vlm:
66
provider: gemini
7-
model: gemini-2.0-flash
7+
model: gemini-2.5-flash
88

99
image:
1010
provider: google_imagen

configs/provider/vlm/gemini.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ capabilities:
77
- vision_input
88
- json_mode
99
config:
10-
model: gemini-2.0-flash
10+
model: gemini-2.5-flash
1111
api_key: ${GOOGLE_API_KEY}
1212
temperature: 1.0
1313
max_tokens: 4096

examples/generate_diagram.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ async def main():
3939

4040
settings = Settings(
4141
vlm_provider="gemini",
42-
vlm_model="gemini-2.0-flash",
42+
vlm_model="gemini-2.5-flash",
4343
image_provider="google_imagen",
4444
image_model="gemini-3-pro-image-preview",
4545
refinement_iterations=2,

paperbanana/core/config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ class VLMConfig(BaseSettings):
1616
"""VLM provider configuration."""
1717

1818
provider: str = "gemini"
19-
model: str = "gemini-2.0-flash"
19+
model: str = "gemini-2.5-flash"
2020

2121

2222
class ImageConfig(BaseSettings):
@@ -57,7 +57,7 @@ class Settings(BaseSettings):
5757

5858
# Provider settings
5959
vlm_provider: str = Field(default="gemini", alias="VLM_PROVIDER")
60-
vlm_model: str = Field(default="gemini-2.0-flash", alias="VLM_MODEL")
60+
vlm_model: str = Field(default="gemini-2.5-flash", alias="VLM_MODEL")
6161
image_provider: str = Field(default="google_imagen", alias="IMAGE_PROVIDER")
6262
image_model: str = Field(default="gemini-3-pro-image-preview", alias="IMAGE_MODEL")
6363

paperbanana/providers/vlm/gemini.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ class GeminiVLM(VLMProvider):
2020
Free tier: https://makersuite.google.com/app/apikey
2121
"""
2222

23-
def __init__(self, api_key: Optional[str] = None, model: str = "gemini-2.0-flash"):
23+
def __init__(self, api_key: Optional[str] = None, model: str = "gemini-2.5-flash"):
2424
self._api_key = api_key
2525
self._model = model
2626
self._client = None

tests/test_providers/test_registry.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,12 @@ def test_create_gemini_vlm():
1212
"""Test creating a Gemini VLM provider."""
1313
settings = Settings(
1414
vlm_provider="gemini",
15-
vlm_model="gemini-2.0-flash",
15+
vlm_model="gemini-2.5-flash",
1616
google_api_key="test-key",
1717
)
1818
vlm = ProviderRegistry.create_vlm(settings)
1919
assert vlm.name == "gemini"
20-
assert vlm.model_name == "gemini-2.0-flash"
20+
assert vlm.model_name == "gemini-2.5-flash"
2121

2222

2323
def test_create_google_imagen_gen():

0 commit comments

Comments
 (0)