From 9c878f2ac16d614b49f0366c200357cd20f9e77f Mon Sep 17 00:00:00 2001 From: Yassin Kortam Date: Fri, 17 Jul 2026 09:17:07 -0700 Subject: [PATCH] test(e2e): annotate _call return type in user budget across-keys test --- .../budgets/test_user_budget_across_keys_e2e.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/e2e/quota_management/budgets/test_user_budget_across_keys_e2e.py b/tests/e2e/quota_management/budgets/test_user_budget_across_keys_e2e.py index e87d6384406..984a100756c 100644 --- a/tests/e2e/quota_management/budgets/test_user_budget_across_keys_e2e.py +++ b/tests/e2e/quota_management/budgets/test_user_budget_across_keys_e2e.py @@ -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)