From e6290016c1e7b17e91404bf1167767d3c3dbc0d2 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Wed, 28 Oct 2026 13:00:03 -0700
Subject: [PATCH] test(proxy): type the first-chunk poll test parameters

---
 .../proxy/response_polling/test_response_polling_handler.py  | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/unit/proxy/response_polling/test_response_polling_handler.py b/tests/unit/proxy/response_polling/test_response_polling_handler.py
index 6cb9376c0e5..830e3f2585c 100644
--- a/tests/unit/proxy/response_polling/test_response_polling_handler.py
+++ b/tests/unit/proxy/response_polling/test_response_polling_handler.py
@@ -16,6 +16,7 @@ following the OpenAI Response API format.
 
 import asyncio
 import json
+from collections.abc import Mapping
 from datetime import datetime, timezone
 from typing import Any, Dict, Final, Optional
 from unittest.mock import AsyncMock, Mock, patch
@@ -1597,7 +1598,9 @@ class TestBackgroundStreamingTerminalEvents:
             (429, "Too Many Requests", {"code": "rate_limit_exceeded", "message": None}),
         ],
     )
-    async def test_a_first_chunk_error_polls_failed_like_a_late_one(self, status_code, body, expected_error):
+    async def test_a_first_chunk_error_polls_failed_like_a_late_one(
+        self, status_code: int, body: str, expected_error: Mapping[str, str | None]
+    ):
         from fastapi import Response
 
         from litellm.proxy.response_polling.background_streaming import (