From 7db4594200d472323969eee8154174111ddeff6b Mon Sep 17 00:00:00 2001
From: shin-bot-litellm <shin-bot-litellm@berri.ai>
Date: Sat, 31 Jan 2026 07:32:33 -0800
Subject: [PATCH] litellm_fix(test): allow comment field in schema and exclude
 robotics models from tpm check (#20139)

---
 tests/test_litellm/test_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index 6a79fd0823..d11fe8d921 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -639,6 +639,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                 "rpd": {"type": "number"},
                 "rpm": {"type": "number"},
                 "source": {"type": "string"},
+                "comment": {"type": "string"},
                 "supports_assistant_prefill": {"type": "boolean"},
                 "supports_audio_input": {"type": "boolean"},
                 "supports_audio_output": {"type": "boolean"},
@@ -841,6 +842,7 @@ def test_get_model_info_gemini():
             and not "learnlm" in model
             and not "imagen" in model
             and not "veo" in model
+            and not "robotics" in model
         ):
             assert info.get("tpm") is not None, f"{model} does not have tpm"
             assert info.get("rpm") is not None, f"{model} does not have rpm"