text content replace with reasoningcontent in sync

Author: michael-chou359

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

@@ -46,6 +46,7 @@ async def handle_message_send(params):
 )
 
 from agentex.lib.utils.logging import make_logger
+from agentex.types.reasoning_content import ReasoningContent
 from agentex.types.task_message_delta import TextDelta
 from agentex.types.tool_request_delta import ToolRequestDelta
 from agentex.types.task_message_update import (
@@ -109,7 +110,7 @@ async def convert_pydantic_ai_to_agentex_events(
 
     Mapping:
         PartStartEvent(TextPart)       -> StreamTaskMessageStart(TextContent)
-        PartStartEvent(ThinkingPart)   -> StreamTaskMessageStart(TextContent)         [reasoning channel]
+        PartStartEvent(ThinkingPart)   -> StreamTaskMessageStart(ReasoningContent)
         PartStartEvent(ToolCallPart)   -> StreamTaskMessageStart(ToolRequestContent)
         PartDeltaEvent(TextPartDelta)  -> StreamTaskMessageDelta(TextDelta)
         PartDeltaEvent(ThinkingPart..) -> StreamTaskMessageDelta(ReasoningContentDelta)
@@ -171,10 +172,12 @@ async def convert_pydantic_ai_to_agentex_events(
                 yield StreamTaskMessageStart(
                     type="start",
                     index=message_index,
-                    content=TextContent(
-                        type="text",
+                    content=ReasoningContent(
+                        type="reasoning",
                         author="agent",
-                        content="",
+                        summary=[],
+                        content=[],
+                        style="active",
                     ),
                 )
                 if event.part.content:

tests/lib/adk/test_pydantic_ai_sync.py

@@ -23,6 +23,7 @@
     FunctionToolResultEvent,
 )
 
+from agentex.types.reasoning_content import ReasoningContent
 from agentex.types.task_message_delta import TextDelta
 from agentex.types.tool_request_delta import ToolRequestDelta
 from agentex.types.task_message_update import (
@@ -118,6 +119,12 @@ async def test_thinking_emits_reasoning_deltas(self):
         out = await _collect(convert_pydantic_ai_to_agentex_events(_aiter(events)))
 
         assert isinstance(out[0], StreamTaskMessageStart)
+        # Thinking content opens a ReasoningContent start, not a TextContent one,
+        # so the Start's content_type matches the ReasoningContentDelta updates
+        # that follow. Mismatched types here would render thinking as a plain
+        # text bubble (or break server-side accumulators) instead of a
+        # collapsible reasoning block.
+        assert isinstance(out[0].content, ReasoningContent)
         assert isinstance(out[1], StreamTaskMessageDelta)
         assert isinstance(out[1].delta, ReasoningContentDelta)
         assert out[1].delta.content_delta == "step 1..."