From a725dab8517c1eb7155a895e56af84fab6e204e2 Mon Sep 17 00:00:00 2001
From: shin-bot-litellm <shin-bot-litellm@users.noreply.github.com>
Date: Sat, 31 Jan 2026 14:51:40 +0000
Subject: [PATCH] litellm_fix(ci): add comment field to schema and add rpm/tpm
 to gemini-robotics model

- Add 'comment' field to model_prices JSON schema validation to allow
  documentation comments in model entries
- Add missing rpm (10) and tpm (4000000) fields to gemini/gemini-robotics-er-1.5-preview
  model entry to fix test_get_model_info_gemini test failure

Fixes:
- test_aaamodel_prices_and_context_window_json_is_valid
- test_get_model_info_gemini
---
 model_prices_and_context_window.json | 2 ++
 tests/test_litellm/test_utils.py     | 1 +
 2 files changed, 3 insertions(+)

diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index 0f84bba941d..4b3c2b5a83d 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -13652,7 +13652,9 @@
         "mode": "chat",
         "output_cost_per_token": 2.5e-06,
         "output_cost_per_reasoning_token": 2.5e-06,
+        "rpm": 10,
         "source": "https://ai.google.dev/gemini-api/docs/models#gemini-robotics-er-1-5-preview",
+        "tpm": 4000000,
         "supported_endpoints": [
             "/v1/chat/completions",
             "/v1/completions"
diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index 6a79fd0823b..57170e9da11 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -695,6 +695,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                     "additionalProperties": False,
                 },
                 "citation_cost_per_token": {"type": "number"},
+                "comment": {"type": "string"},
                 "supported_modalities": {
                     "type": "array",
                     "items": {