scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

test(proxy): type the fake tag batch helper parameters

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-25 00:22:20 +00:00
parent cc312c30af
commit 1edc656258
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py
View file

@ -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",