From 1edc6562589d7b1f954d9633ec762519576a23cf Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Fri, 25 Sep 2026 00:22:20 +0000
Subject: [PATCH] test(proxy): type the fake tag batch helper parameters

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 .../proxy/hooks/test_parallel_request_limiter_v3.py        | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py b/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py
index 3c51297c5c3..bec62155f5f 100644
--- a/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py
+++ b/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py
@@ -5097,8 +5097,13 @@ async def test_resolve_tag_rate_limits_from_db_reads_budget_row(monkeypatch):
     from litellm.proxy import proxy_server
     from litellm.proxy.auth import auth_checks
     from litellm.proxy.hooks.parallel_request_limiter_v3 import resolve_tag_rate_limits_from_db
+    from litellm.proxy.utils import PrismaClient
 
-    async def fake_batch(tag_names, prisma_client, user_api_key_cache, **kwargs):
+    async def fake_batch(
+        tag_names: Sequence[str],
+        prisma_client: PrismaClient | None,
+        user_api_key_cache: DualCache,
+    ) -> dict[str, LiteLLM_TagTable]:
         return {
             "limited": LiteLLM_TagTable(
                 tag_name="limited",