fix linting errors

Signed-off-by: Jordan Dubrick <jdubrick@redhat.com>

Author: Jdubrick

tests/integration/endpoints/test_stream_interrupt_integration.py

@@ -34,6 +34,7 @@ async def test_stream_interrupt_full_round_trip(
     """Full lifecycle: register, interrupt, then verify deregistration."""
     # test_config loads configuration so @authorize on the handler can resolve.
     _ = test_config
+
     async def pending_stream() -> None:
         await asyncio.sleep(10)
 

tests/unit/pydantic_ai_lightspeed/llamastack/test_transport.py

@@ -3,6 +3,7 @@
 # pylint: disable=protected-access
 
 import json
+from collections.abc import AsyncGenerator
 from typing import Any
 
 import httpx
@@ -45,7 +46,7 @@ class TestAsyncByteStream:
     async def test_iterates_chunks(self) -> None:
         """Test that _AsyncByteStream yields all chunks from the wrapped generator."""
 
-        async def gen():
+        async def gen() -> AsyncGenerator[bytes, None]:
             yield b"chunk1"
             yield b"chunk2"
             yield b"chunk3"
@@ -59,7 +60,7 @@ async def gen():
     async def test_empty_generator(self) -> None:
         """Test that _AsyncByteStream handles an empty generator gracefully."""
 
-        async def gen():
+        async def gen() -> AsyncGenerator[bytes, None]:
             return
             yield  # pragma: no cover
 
@@ -134,7 +135,7 @@ async def test_streaming_request(
             content=json.dumps(body).encode("utf-8"),
         )
 
-        async def mock_stream_result():
+        async def mock_stream_result() -> AsyncGenerator[dict[str, int], None]:
             yield {"chunk": 1}
             yield {"chunk": 2}
 
@@ -317,7 +318,7 @@ async def test_produces_sse_format(
     ) -> None:
         """Test that streaming responses produce SSE-formatted byte chunks."""
 
-        async def mock_stream():
+        async def mock_stream() -> AsyncGenerator[dict[str, str], None]:
             yield {"delta": "hello"}
             yield {"delta": "world"}