From 58c3d04733f2bebfbc15e8f1f6dd702a37c6e2f6 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Sat, 5 Sep 2026 23:24:47 -0700
Subject: [PATCH] test: cover is_batch_retrieve_call_type in router batch utils

---
 .../router_unit_tests/test_router_batch_utils.py  | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/tests/router_unit_tests/test_router_batch_utils.py b/tests/router_unit_tests/test_router_batch_utils.py
index c9f19731372..e274ac61a01 100644
--- a/tests/router_unit_tests/test_router_batch_utils.py
+++ b/tests/router_unit_tests/test_router_batch_utils.py
@@ -317,3 +317,18 @@ def test_replace_model_in_jsonl_with_embedded_newlines():
         == "This is a message\nwith multiple\nlines"
     )
     assert result_json["custom_id"] == "test123"
+
+
+def test_is_batch_retrieve_call_type_matches_only_batch_retrieves():
+    from litellm.router_utils.batch_utils import is_batch_retrieve_call_type
+    from litellm.types.utils import CallTypes
+
+    assert is_batch_retrieve_call_type(CallTypes.aretrieve_batch.value) is True
+    assert is_batch_retrieve_call_type(CallTypes.retrieve_batch.value) is True
+
+    for call_type in CallTypes:
+        if call_type in (CallTypes.aretrieve_batch, CallTypes.retrieve_batch):
+            continue
+        assert is_batch_retrieve_call_type(call_type.value) is False
+
+    assert is_batch_retrieve_call_type(None) is False