diff --git a/cortex/model_registry.py b/cortex/model_registry.py index 2620949..13c7bb2 100644 --- a/cortex/model_registry.py +++ b/cortex/model_registry.py @@ -90,10 +90,14 @@ ANTHROPIC_CATALOG: list[dict] = [ ] GOOGLE_CATALOG: list[dict] = [ - {"id": "gemini-2.5-pro", "label": "Gemini 2.5 Pro", "context_k": 1000}, - {"id": "gemini-2.5-flash", "label": "Gemini 2.5 Flash", "context_k": 1000}, - {"id": "gemini-2.0-flash", "label": "Gemini 2.0 Flash", "context_k": 1000}, - {"id": "gemini-1.5-pro", "label": "Gemini 1.5 Pro", "context_k": 2000}, + # Stable / generally available + {"id": "gemini-2.5-pro", "label": "Gemini 2.5 Pro", "context_k": 1000}, + {"id": "gemini-2.5-flash", "label": "Gemini 2.5 Flash", "context_k": 1000}, + {"id": "gemini-2.5-flash-lite", "label": "Gemini 2.5 Flash-Lite", "context_k": 1000}, + # Preview + {"id": "gemini-3.1-pro-preview", "label": "Gemini 3.1 Pro (preview)", "context_k": 1000}, + {"id": "gemini-3-flash-preview", "label": "Gemini 3 Flash (preview)", "context_k": 1000}, + {"id": "gemini-3.1-flash-lite-preview", "label": "Gemini 3.1 Flash-Lite (preview)", "context_k": 1000}, ]