scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(e2e): annotate _call return type in user budget across-keys test

Browse source
This commit is contained in:
Yassin Kortam 2026-07-17 09:17:07 -07:00
parent ab4392cd36
commit 9c878f2ac1
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/e2e/quota_management/budgets/test_user_budget_across_keys_e2e.py
View file

@ -14,7 +14,7 @@ import pytest
from budget_client import BudgetClient, is_budget_block
from e2e_config import unique_marker
from e2e_http import require_successful_call
from e2e_http import StreamingResponse, require_successful_call
from lifecycle import ResourceManager
pytestmark = pytest.mark.e2e
@ -24,7 +24,7 @@ TINY_CAP = 3e-6
RECORDED_SPEND_DEADLINE_SECONDS = 90
def _call(client: BudgetClient, key: str):
def _call(client: BudgetClient, key: str) -> StreamingResponse:
return client.chat(key, MODEL, f"across {unique_marker()}", max_tokens=16)