fix lint errors

Author: michael-chou359

src/agentex/lib/adk/_modules/_pydantic_ai_sync.py

@@ -184,7 +184,10 @@ async def convert_pydantic_ai_to_agentex_events(
                 # for deltas.
                 initial_args: dict[str, Any] = {}
                 if isinstance(event.part.args, dict):
-                    initial_args = event.part.args
+                    # dict(...) materializes a fresh dict[str, Any]; pydantic-ai's
+                    # ToolCallPart.args includes TypedDict-style variants that
+                    # pyright doesn't narrow to plain dict[str, Any] via isinstance.
+                    initial_args = dict(event.part.args)
                 yield StreamTaskMessageStart(
                     type="start",
                     index=message_index,

tests/lib/adk/test_pydantic_ai_async.py

@@ -402,7 +402,7 @@ async def test_retry_prompt_part_surfaces_as_tool_response(
         assert isinstance(content, ToolResponseContent)
         assert content.tool_call_id == "c1"
         # RetryPromptPart.content stringifies to the error description
-        assert "bad arguments" in content.content
+        assert "bad arguments" in str(content.content)
 
 
 class TestContextLifecycle:

tests/lib/adk/test_pydantic_ai_sync.py

@@ -100,6 +100,7 @@ async def test_text_with_initial_content_emits_delta(self):
         ]
         out = await _collect(convert_pydantic_ai_to_agentex_events(_aiter(events)))
         assert isinstance(out[0], StreamTaskMessageStart)
+        assert isinstance(out[0].content, TextContent)
         assert out[0].content.content == ""
         assert isinstance(out[1], StreamTaskMessageDelta)
         assert isinstance(out[1].delta, TextDelta)
@@ -132,6 +133,7 @@ async def test_thinking_with_initial_content_emits_delta(self):
         out = await _collect(convert_pydantic_ai_to_agentex_events(_aiter(events)))
         assert isinstance(out[0], StreamTaskMessageStart)
         assert isinstance(out[1], StreamTaskMessageDelta)
+        assert isinstance(out[1].delta, ReasoningContentDelta)
         assert out[1].delta.content_delta == "seed reasoning"
 
     async def test_thinking_delta_skipped_when_empty(self):
@@ -197,6 +199,7 @@ async def test_tool_call_with_full_args_at_start(self):
         ]
         out = await _collect(convert_pydantic_ai_to_agentex_events(_aiter(events)))
         assert isinstance(out[0], StreamTaskMessageStart)
+        assert isinstance(out[0].content, ToolRequestContent)
         assert out[0].content.arguments == {"query": "weather"}
         # No deltas emitted — args were already complete.
         assert len(out) == 2
@@ -216,8 +219,10 @@ async def test_tool_call_with_full_args_string_at_start(self):
         ]
         out = await _collect(convert_pydantic_ai_to_agentex_events(_aiter(events)))
         assert isinstance(out[0], StreamTaskMessageStart)
+        assert isinstance(out[0].content, ToolRequestContent)
         assert out[0].content.arguments == {}
         assert isinstance(out[1], StreamTaskMessageDelta)
+        assert isinstance(out[1].delta, ToolRequestDelta)
         assert out[1].delta.arguments_delta == '{"query":"weather"}'
 
     async def test_tool_call_dict_args_delta_serialized(self):
@@ -319,7 +324,9 @@ async def test_text_then_tool_then_text_assigns_distinct_indices(self):
             if isinstance(e, StreamTaskMessageDelta) and isinstance(e.delta, TextDelta) and e.index == text2_start.index
         ]
         assert len(text2_deltas) == 1
-        assert text2_deltas[0].delta.text_delta == "It's sunny."
+        text2_delta = text2_deltas[0].delta
+        assert isinstance(text2_delta, TextDelta)
+        assert text2_delta.text_delta == "It's sunny."
 
 
 class TestIgnoredEvents: