From 29f73c66e010ceb6a90b0893dd07fc03a9ab21ca Mon Sep 17 00:00:00 2001
From: Eran Cohen <eranco@redhat.com>
Date: Wed, 18 Jun 2025 22:11:03 +0300
Subject: [PATCH] feat: Add Gemini 2.0 and 2.5 models

This commit expands the set of known Gemini models by introducing:
- `gemini/gemini-2.0-flash`
- `gemini/gemini-2.5-flash`
- `gemini/gemini-2.5-pro`

Signed-off-by: Eran Cohen <eranco@redhat.com>
---
 .../providers/remote/inference/gemini/models.py   |  3 +++
 llama_stack/templates/starter/run.yaml            | 15 +++++++++++++++
 2 files changed, 18 insertions(+)

diff --git a/llama_stack/providers/remote/inference/gemini/models.py b/llama_stack/providers/remote/inference/gemini/models.py
index 1d7b47315..ef1cf339f 100644
--- a/llama_stack/providers/remote/inference/gemini/models.py
+++ b/llama_stack/providers/remote/inference/gemini/models.py
@@ -12,6 +12,9 @@ from llama_stack.providers.utils.inference.model_registry import (
 LLM_MODEL_IDS = [
     "gemini/gemini-1.5-flash",
     "gemini/gemini-1.5-pro",
+    "gemini/gemini-2.0-flash",
+    "gemini/gemini-2.5-flash",
+    "gemini/gemini-2.5-pro",
 ]
 
 
diff --git a/llama_stack/templates/starter/run.yaml b/llama_stack/templates/starter/run.yaml
index 319ababe5..960e96d01 100644
--- a/llama_stack/templates/starter/run.yaml
+++ b/llama_stack/templates/starter/run.yaml
@@ -731,6 +731,21 @@ models:
   provider_id: gemini
   provider_model_id: gemini/gemini-1.5-pro
   model_type: llm
+- metadata: {}
+  model_id: gemini/gemini-2.0-flash
+  provider_id: gemini
+  provider_model_id: gemini/gemini-2.0-flash
+  model_type: llm
+- metadata: {}
+  model_id: gemini/gemini-2.5-flash
+  provider_id: gemini
+  provider_model_id: gemini/gemini-2.5-flash
+  model_type: llm
+- metadata: {}
+  model_id: gemini/gemini-2.5-pro
+  provider_id: gemini
+  provider_model_id: gemini/gemini-2.5-pro
+  model_type: llm
 - metadata:
     embedding_dimension: 768
     context_length: 2048