test(harness): narrow content types in the Gemini CLI tests for pyright

pyright 1.1.399 (the locked version `scripts/lint` runs) reported six
errors in the Gemini CLI tests: attribute access on a content union after
a list-comprehension filter, which does not narrow the indexed element,
and aclose() on the tap's AsyncIterator return type. Assert the concrete
content type before reading it, and cast the tap to AsyncGenerator where
the test closes it. Test behaviour is unchanged.

Author: michaelxu2288

tests/lib/adk/test_gemini_cli_sync.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 import json
-from typing import Any, AsyncIterator
+from typing import Any, AsyncIterator, AsyncGenerator, cast
 
 from agentex.types.text_content import TextContent
 from agentex.types.task_message_delta import TextDelta
@@ -135,6 +135,7 @@ async def test_error_tool_result_sets_is_error_and_uses_message(self):
         ]
         out = await _collect(convert_gemini_cli_to_agentex_events(_aiter(events)))
         full = [e for e in out if isinstance(e, StreamTaskMessageFull)][0]
+        assert isinstance(full.content, ToolResponseContent)
         assert full.content.content == {"result": "denied", "is_error": True}
 
     async def test_missing_tool_id_gets_a_synthetic_one(self):
@@ -233,7 +234,7 @@ async def __anext__(self):
             async def aclose(self) -> None:
                 closed["v"] = True
 
-        gen = convert_gemini_cli_to_agentex_events(_Src())
+        gen = cast(AsyncGenerator[Any, None], convert_gemini_cli_to_agentex_events(_Src()))
         await gen.__anext__()
         await gen.aclose()
         assert closed["v"] is True

tests/lib/core/harness/test_harness_gemini_cli_sync.py

@@ -65,6 +65,7 @@ async def test_tool_request_precedes_tool_response_then_text(self) -> None:
         kinds = [type(e).__name__ for e in out]
         assert kinds.index("StreamTaskMessageFull") > kinds.index("StreamTaskMessageStart")
         req = [e for e in out if isinstance(e, StreamTaskMessageStart) and isinstance(e.content, ToolRequestContent)][0]
+        assert isinstance(req.content, ToolRequestContent)
         res = [e for e in out if isinstance(e, StreamTaskMessageFull)][0]
         assert isinstance(res.content, ToolResponseContent)
         assert req.content.tool_call_id == res.content.tool_call_id == "call-1"
@@ -97,8 +98,8 @@ async def test_failed_turn_reaches_the_caller(self) -> None:
             },
         ]
         out, turn = await _run_yield_turn(events)
-        errors = [e for e in out if isinstance(e, StreamTaskMessageFull) and isinstance(e.content, TextContent)]
-        assert [e.content.content for e in errors] == ["Error: Gemini CLI turn failed: Reached max session turns"]
+        errors = [e.content.content for e in out if isinstance(e, StreamTaskMessageFull) and isinstance(e.content, TextContent)]
+        assert errors == ["Error: Gemini CLI turn failed: Reached max session turns"]
         starts = {e.index for e in out if isinstance(e, StreamTaskMessageStart)}
         dones = {e.index for e in out if isinstance(e, StreamTaskMessageDone)}
         assert starts == dones