From cfb7a263277e9dde7a1359fbe6f31763ffdb1e33 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Sat, 29 Aug 2026 14:38:21 -0700
Subject: [PATCH] fix(registry): add gemma 4 capability flags verified against
 the gemini api

---
 litellm/model_prices_and_context_window_backup.json | 10 ++++++++++
 model_prices_and_context_window.json                | 10 ++++++++++
 2 files changed, 20 insertions(+)

diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json
index 772369fc8ff..7373ffd8597 100644
--- a/litellm/model_prices_and_context_window_backup.json
+++ b/litellm/model_prices_and_context_window_backup.json
@@ -23614,7 +23614,12 @@
         "max_output_tokens": 32768,
         "max_tokens": 32768,
         "mode": "chat",
+        "supports_function_calling": true,
         "supports_reasoning": true,
+        "supports_response_schema": true,
+        "supports_system_messages": true,
+        "supports_tool_choice": true,
+        "supports_vision": true,
         "source": "https://ai.google.dev/gemini-api/docs/pricing"
     },
     "gemini/gemma-4-31b-it": {
@@ -23625,7 +23630,12 @@
         "max_output_tokens": 32768,
         "max_tokens": 32768,
         "mode": "chat",
+        "supports_function_calling": true,
         "supports_reasoning": true,
+        "supports_response_schema": true,
+        "supports_system_messages": true,
+        "supports_tool_choice": true,
+        "supports_vision": true,
         "source": "https://ai.google.dev/gemini-api/docs/pricing"
     },
     "gemini/imagen-3.0-fast-generate-001": {
diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index 772369fc8ff..7373ffd8597 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -23614,7 +23614,12 @@
         "max_output_tokens": 32768,
         "max_tokens": 32768,
         "mode": "chat",
+        "supports_function_calling": true,
         "supports_reasoning": true,
+        "supports_response_schema": true,
+        "supports_system_messages": true,
+        "supports_tool_choice": true,
+        "supports_vision": true,
         "source": "https://ai.google.dev/gemini-api/docs/pricing"
     },
     "gemini/gemma-4-31b-it": {
@@ -23625,7 +23630,12 @@
         "max_output_tokens": 32768,
         "max_tokens": 32768,
         "mode": "chat",
+        "supports_function_calling": true,
         "supports_reasoning": true,
+        "supports_response_schema": true,
+        "supports_system_messages": true,
+        "supports_tool_choice": true,
+        "supports_vision": true,
         "source": "https://ai.google.dev/gemini-api/docs/pricing"
     },
     "gemini/imagen-3.0-fast-generate-001": {