agentex-sdk 0.4.6__py3-none-any.whl → 0.4.8__py3-none-any.whl

agentex/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "agentex"
-__version__ = "0.4.6"  # x-release-please-version
+__version__ = "0.4.8"  # x-release-please-version

agentex/lib/cli/templates/sync/project/acp.py.j2

@@ -2,7 +2,7 @@ from typing import AsyncGenerator, Union
 from agentex.lib.sdk.fastacp.fastacp import FastACP
 from agentex.lib.types.acp import SendMessageParams
 
-from agentex.lib.types.task_message_updates import TaskMessageUpdate
+from agentex.types.task_message_update import TaskMessageUpdate
 from agentex.types.task_message_content import TaskMessageContent
 from agentex.types.text_content import TextContent
 from agentex.lib.utils.logging import make_logger

agentex/lib/core/services/adk/acp/acp.py

@@ -1,4 +1,4 @@
-from typing import Any, cast
+from typing import Any, List, cast
 
 from agentex import AsyncAgentex
 from agentex.lib.core.tracing.tracer import AsyncTracer
@@ -78,7 +78,7 @@ class ACPService:
         task_name: str | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
-    ) -> TaskMessage:
+    ) -> List[TaskMessage]:
         trace = self._tracer.trace(trace_id=trace_id)
         async with trace.span(
             parent_id=parent_span_id,
@@ -115,10 +115,17 @@ class ACPService:
             else:
                 raise ValueError("Either agent_name or agent_id must be provided")
 
-            task_message = TaskMessage.model_validate(json_rpc_response.result)
+            task_messages: List[TaskMessage] = []
+            if isinstance(json_rpc_response.result, list):
+                for message in json_rpc_response.result:
+                    task_message = TaskMessage.model_validate(message)
+                    task_messages.append(task_message)
+            else:
+                task_messages = [TaskMessage.model_validate(json_rpc_response.result)]
+
             if span:
-                span.output = task_message.model_dump()
-            return task_message
+                span.output = [task_message.model_dump() for task_message in task_messages]
+            return task_messages
 
     async def event_send(
         self,

agentex/lib/core/services/adk/messages.py

@@ -4,7 +4,7 @@ from typing import Any, Coroutine, cast
 from agentex import AsyncAgentex
 from agentex.lib.core.services.adk.streaming import StreamingService
 from agentex.lib.core.tracing.tracer import AsyncTracer
-from agentex.lib.types.task_message_updates import StreamTaskMessageFull, TaskMessageUpdate
+from agentex.types.task_message_update import StreamTaskMessageFull, TaskMessageUpdate
 from agentex.types.task_message import TaskMessage, TaskMessageContent
 from agentex.lib.utils.logging import make_logger
 from agentex.lib.utils.temporal import heartbeat_if_in_workflow

agentex/lib/core/services/adk/providers/litellm.py

@@ -8,11 +8,11 @@ from agentex.lib.types.llm_messages import (
     Completion,
     LLMConfig,
 )
-from agentex.lib.types.task_message_updates import (
+from agentex.types.task_message_update import (
     StreamTaskMessageDelta,
     StreamTaskMessageFull,
-    TextDelta,
 )
+from agentex.types.task_message_delta import TextDelta
 from agentex.types.task_message import TaskMessage
 from agentex.types.task_message_content import TextContent
 from agentex.lib.utils import logging
@@ -112,6 +112,7 @@ class LiteLLMService:
                         update=StreamTaskMessageFull(
                             parent_task_message=streaming_context.task_message,
                             content=final_content,
+                            type="full",
                         ),
                     )
                 else:
@@ -221,7 +222,8 @@ class LiteLLMService:
                             await streaming_context.stream_update(
                                 update=StreamTaskMessageDelta(
                                     parent_task_message=streaming_context.task_message,
-                                    delta=TextDelta(text_delta=delta),
+                                    delta=TextDelta(text_delta=delta, type="text"),
+                                    type="delta",
                                 ),
                             )
                             heartbeat_if_in_workflow("content chunk streamed")
@@ -244,6 +246,7 @@ class LiteLLMService:
                         update=StreamTaskMessageFull(
                             parent_task_message=streaming_context.task_message,
                             content=final_content,
+                            type="full",
                         ),
                     )
 

agentex/lib/core/services/adk/providers/openai.py

@@ -11,7 +11,12 @@ from openai.types.responses import (
     ResponseCompletedEvent,
     ResponseFunctionToolCall,
     ResponseOutputItemDoneEvent,
+    ResponseReasoningSummaryPartDoneEvent,
     ResponseTextDeltaEvent,
+    ResponseReasoningSummaryTextDeltaEvent,
+    ResponseReasoningSummaryTextDoneEvent,
+    ResponseReasoningTextDeltaEvent,
+    ResponseReasoningTextDoneEvent,
 )
 from pydantic import BaseModel
 
@@ -22,14 +27,18 @@ from agentex.lib.core.services.adk.streaming import (
     StreamingTaskMessageContext,
 )
 from agentex.lib.core.tracing.tracer import AsyncTracer
-from agentex.lib.types.task_message_updates import (
+from agentex.types.task_message_update import (
     StreamTaskMessageDelta,
     StreamTaskMessageFull,
+)
+from agentex.types.task_message_delta import (
     TextDelta,
+    ReasoningSummaryDelta,
+    ReasoningContentDelta,
 )
 from agentex.types.task_message_content import (
-    TextContent,
     ReasoningContent,
+    TextContent,
     ToolRequestContent,
     ToolResponseContent,
 )
@@ -317,6 +326,7 @@ class OpenAIService:
                                 update=StreamTaskMessageFull(
                                     parent_task_message=streaming_context.task_message,
                                     content=text_content,
+                                    type="full",
                                 ),
                             )
 
@@ -341,6 +351,7 @@ class OpenAIService:
                                 update=StreamTaskMessageFull(
                                     parent_task_message=streaming_context.task_message,
                                     content=tool_request_content,
+                                    type="full",
                                 ),
                             )
 
@@ -357,13 +368,14 @@ class OpenAIService:
                         async with (
                             self.streaming_service.streaming_task_message_context(
                                 task_id=task_id,
-                                initial_content=tool_response_content,
+                                initial_content=tool_response_content
                             ) as streaming_context
                         ):
                             await streaming_context.stream_update(
                                 update=StreamTaskMessageFull(
                                     parent_task_message=streaming_context.task_message,
                                     content=tool_response_content,
+                                    type="full",
                                 ),
                             )
 
@@ -619,6 +631,7 @@ class OpenAIService:
                                         update=StreamTaskMessageFull(
                                             parent_task_message=streaming_context.task_message,
                                             content=tool_request_content,
+                                            type="full",
                                         ),
                                     )
 
@@ -638,7 +651,7 @@ class OpenAIService:
                                 async with (
                                     self.streaming_service.streaming_task_message_context(
                                         task_id=task_id,
-                                        initial_content=tool_response_content,
+                                        initial_content=tool_response_content
                                     ) as streaming_context
                                 ):
                                     # The message has already been persisted, but we still need to send an update
@@ -646,33 +659,7 @@ class OpenAIService:
                                         update=StreamTaskMessageFull(
                                             parent_task_message=streaming_context.task_message,
                                             content=tool_response_content,
-                                        ),
-                                    )
-                            
-                            elif event.item.type == "reasoning_item":
-                                # Handle reasoning items
-                                reasoning_item = event.item.raw_item
-                                
-                                reasoning_content = ReasoningContent(
-                                    author="agent",
-                                    style="static",
-                                    summary=[summary.text for summary in reasoning_item.summary],
-                                    content=[content.text for content in reasoning_item.content] if hasattr(reasoning_item, "content") and reasoning_item.content else None,
-                                    type="reasoning",
-                                )
-
-                                # Create reasoning content using streaming context (immediate completion)
-                                async with (
-                                    self.streaming_service.streaming_task_message_context(
-                                        task_id=task_id,
-                                        initial_content=reasoning_content,
-                                    ) as streaming_context
-                                ):
-                                    # The message has already been persisted, but we still need to send an update
-                                    await streaming_context.stream_update(
-                                        update=StreamTaskMessageFull(
-                                            parent_task_message=streaming_context.task_message,
-                                            content=reasoning_content,
+                                            type="full",
                                         ),
                                     )
 
@@ -705,10 +692,117 @@ class OpenAIService:
                                 await streaming_context.stream_update(
                                     update=StreamTaskMessageDelta(
                                         parent_task_message=streaming_context.task_message,
-                                        delta=TextDelta(text_delta=event.data.delta),
+                                        delta=TextDelta(text_delta=event.data.delta, type="text"),
+                                        type="delta",
+                                    ),
+                                )
+
+                            elif isinstance(event.data, ResponseReasoningSummaryTextDeltaEvent):
+                                # Handle reasoning summary text delta
+                                item_id = event.data.item_id
+                                summary_index = event.data.summary_index
+
+                                # Check if we already have a streaming context for this reasoning item
+                                if item_id not in item_id_to_streaming_context:
+                                    # Create a new streaming context for this reasoning item
+                                    streaming_context = self.streaming_service.streaming_task_message_context(
+                                        task_id=task_id,
+                                        initial_content=ReasoningContent(
+                                            author="agent",
+                                            summary=[],
+                                            content=[],
+                                            type="reasoning",
+                                            style="active",
+                                        ),
+                                    )
+                                    # Open the streaming context
+                                    item_id_to_streaming_context[
+                                        item_id
+                                    ] = await streaming_context.open()
+                                    unclosed_item_ids.add(item_id)
+                                else:
+                                    streaming_context = item_id_to_streaming_context[
+                                        item_id
+                                    ]
+
+                                # Stream the summary delta through the streaming service
+                                await streaming_context.stream_update(
+                                    update=StreamTaskMessageDelta(
+                                        parent_task_message=streaming_context.task_message,
+                                        delta=ReasoningSummaryDelta(
+                                            summary_index=summary_index,
+                                            summary_delta=event.data.delta,
+                                            type="reasoning_summary",
+                                        ),
+                                        type="delta",
+                                    ),
+                                )
+
+                            elif isinstance(event.data, ResponseReasoningTextDeltaEvent):
+                                # Handle reasoning content text delta
+                                item_id = event.data.item_id
+                                content_index = event.data.content_index
+
+                                # Check if we already have a streaming context for this reasoning item
+                                if item_id not in item_id_to_streaming_context:
+                                    # Create a new streaming context for this reasoning item
+                                    streaming_context = self.streaming_service.streaming_task_message_context(
+                                        task_id=task_id,
+                                        initial_content=ReasoningContent(
+                                            author="agent",
+                                            summary=[],
+                                            content=[],
+                                            type="reasoning",
+                                            style="active",
+                                        ),
+                                    )
+                                    # Open the streaming context
+                                    item_id_to_streaming_context[
+                                        item_id
+                                    ] = await streaming_context.open()
+                                    unclosed_item_ids.add(item_id)
+                                else:
+                                    streaming_context = item_id_to_streaming_context[
+                                        item_id
+                                    ]
+
+                                # Stream the content delta through the streaming service
+                                await streaming_context.stream_update(
+                                    update=StreamTaskMessageDelta(
+                                        parent_task_message=streaming_context.task_message,
+                                        delta=ReasoningContentDelta(
+                                            content_index=content_index,
+                                            content_delta=event.data.delta,
+                                            type="reasoning_content",
+                                        ),
+                                        type="delta",
                                     ),
                                 )
 
+                            elif isinstance(event.data, ResponseReasoningSummaryTextDoneEvent):
+                                # Handle reasoning summary text completion
+                                item_id = event.data.item_id
+                                summary_index = event.data.summary_index
+
+                                # We do NOT close the streaming context here as there can be multiple
+                                # reasoning summaries. The context will be closed when the entire
+                                # output item is done (ResponseOutputItemDoneEvent)
+
+                                # You would think they would use the event ResponseReasoningSummaryPartDoneEvent
+                                # to close the streaming context, but they do!!!
+                                # They output both a ResponseReasoningSummaryTextDoneEvent and a ResponseReasoningSummaryPartDoneEvent
+                                # I have no idea why they do this.
+    
+                            elif isinstance(event.data, ResponseReasoningTextDoneEvent):
+                                # Handle reasoning content text completion
+                                item_id = event.data.item_id
+                                content_index = event.data.content_index
+
+                                # We do NOT close the streaming context here as there can be multiple
+                                # reasoning content texts. The context will be closed when the entire
+                                # output item is done (ResponseOutputItemDoneEvent)
+
+
                             elif isinstance(event.data, ResponseOutputItemDoneEvent):
                                 # Handle item completion
                                 item_id = event.data.item.id
@@ -719,23 +813,32 @@ class OpenAIService:
                                         item_id
                                     ]
                                     await streaming_context.close()
-                                    unclosed_item_ids.remove(item_id)
+                                    if item_id in unclosed_item_ids:
+                                        unclosed_item_ids.remove(item_id)
 
                             elif isinstance(event.data, ResponseCompletedEvent):
                                 # All items complete, finish all remaining streaming contexts for this session
-                                for item_id in unclosed_item_ids:
-                                    streaming_context = item_id_to_streaming_context[
-                                        item_id
-                                    ]
-                                    await streaming_context.close()
-                                    unclosed_item_ids.remove(item_id)
+                                # Create a copy to avoid modifying set during iteration
+                                remaining_items = list(unclosed_item_ids)
+                                for item_id in remaining_items:
+                                    if (item_id in unclosed_item_ids and
+                                            item_id in item_id_to_streaming_context):  # Check if still unclosed
+                                        streaming_context = item_id_to_streaming_context[
+                                            item_id
+                                        ]
+                                        await streaming_context.close()
+                                        unclosed_item_ids.discard(item_id)
 
                 finally:
                     # Cleanup: ensure all streaming contexts for this session are properly finished
-                    for item_id in unclosed_item_ids:
-                        streaming_context = item_id_to_streaming_context[item_id]
-                        await streaming_context.close()
-                        unclosed_item_ids.remove(item_id)
+                    # Create a copy to avoid modifying set during iteration
+                    remaining_items = list(unclosed_item_ids)
+                    for item_id in remaining_items:
+                        if (item_id in unclosed_item_ids and
+                                item_id in item_id_to_streaming_context):  # Check if still unclosed
+                            streaming_context = item_id_to_streaming_context[item_id]
+                            await streaming_context.close()
+                            unclosed_item_ids.discard(item_id)
 
                 if span:
                     span.output = {

agentex/lib/core/services/adk/streaming.py

@@ -1,21 +1,24 @@
 import json
-from typing import Literal, cast
+from typing import Literal
 
 from agentex import AsyncAgentex
 from agentex.lib.core.adapters.streams.port import StreamRepository
-from agentex.lib.types.task_message_updates import (
+from agentex.types.task_message_update import (
     TaskMessageDelta, 
     TaskMessageUpdate,
-    TextDelta,
-    DataDelta,
-    ToolRequestDelta,
-    ToolResponseDelta,
-    StreamTaskMessage,
     StreamTaskMessageStart,
     StreamTaskMessageDelta,
     StreamTaskMessageFull,
     StreamTaskMessageDone,
 )
+from agentex.types.task_message_delta import (
+    TextDelta,
+    DataDelta,
+    ToolRequestDelta,
+    ToolResponseDelta,
+    ReasoningSummaryDelta,
+    ReasoningContentDelta,
+)
 from agentex.lib.utils.logging import make_logger
 from agentex.types.data_content import DataContent
 from agentex.types.task_message import (
@@ -25,6 +28,7 @@ from agentex.types.task_message import (
 from agentex.types.text_content import TextContent
 from agentex.types.tool_request_content import ToolRequestContent
 from agentex.types.tool_response_content import ToolResponseContent
+from agentex.types.reasoning_content import ReasoningContent
 
 logger = make_logger(__name__)
 
@@ -36,7 +40,10 @@ def _get_stream_topic(task_id: str) -> str:
 class DeltaAccumulator:
     def __init__(self):
         self._accumulated_deltas: list[TaskMessageDelta] = []
-        self._delta_type: Literal["text", "data", "tool_request", "tool_response"] | None = None
+        self._delta_type: Literal["text", "data", "tool_request", "tool_response", "reasoning"] | None = None
+        # For reasoning, we need to track both summary and content deltas
+        self._reasoning_summaries: dict[int, str] = {}
+        self._reasoning_contents: dict[int, str] = {}
 
     def add_delta(self, delta: TaskMessageDelta):
         if self._delta_type is None:
@@ -48,15 +55,35 @@ class DeltaAccumulator:
                 self._delta_type = "tool_request"
             elif delta.type == "tool_response":
                 self._delta_type = "tool_response"
+            elif delta.type in ["reasoning_summary", "reasoning_content"]:
+                self._delta_type = "reasoning"
             else:
                 raise ValueError(f"Unknown delta type: {delta.type}")
         else:
-            if self._delta_type != delta.type:
+            # For reasoning, we allow both summary and content deltas
+            if self._delta_type == "reasoning":
+                if delta.type not in ["reasoning_summary", "reasoning_content"]:
+                    raise ValueError(
+                        f"Expected reasoning delta but got: {delta.type}"
+                    )
+            elif self._delta_type != delta.type:
                 raise ValueError(
                     f"Delta type mismatch: {self._delta_type} != {delta.type}"
                 )
 
-        self._accumulated_deltas.append(delta)
+        # Handle reasoning deltas specially
+        if delta.type == "reasoning_summary":
+            if isinstance(delta, ReasoningSummaryDelta):
+                if delta.summary_index not in self._reasoning_summaries:
+                    self._reasoning_summaries[delta.summary_index] = ""
+                self._reasoning_summaries[delta.summary_index] += delta.summary_delta or ""
+        elif delta.type == "reasoning_content":
+            if isinstance(delta, ReasoningContentDelta):
+                if delta.content_index not in self._reasoning_contents:
+                    self._reasoning_contents[delta.content_index] = ""
+                self._reasoning_contents[delta.content_index] += delta.content_delta or ""
+        else:
+            self._accumulated_deltas.append(delta)
 
     def convert_to_content(self) -> TaskMessageContent:
         if self._delta_type == "text":
@@ -107,7 +134,7 @@ class DeltaAccumulator:
             # Type assertion: we know all deltas are ToolResponseDelta when _delta_type is TOOL_RESPONSE
             tool_response_deltas = [delta for delta in self._accumulated_deltas if isinstance(delta, ToolResponseDelta)]
             tool_response_content_str = "".join(
-                [delta.tool_response_delta or "" for delta in tool_response_deltas]
+                [delta.content_delta or "" for delta in tool_response_deltas]
             )
             return ToolResponseContent(
                 author="agent",
@@ -115,6 +142,27 @@ class DeltaAccumulator:
                 name=tool_response_deltas[0].name,
                 content=tool_response_content_str,
             )
+        elif self._delta_type == "reasoning":
+            # Convert accumulated reasoning deltas to ReasoningContent
+            # Sort by index to maintain order
+            summary_list = [self._reasoning_summaries[i] for i in sorted(self._reasoning_summaries.keys()) if self._reasoning_summaries[i]]
+            content_list = [self._reasoning_contents[i] for i in sorted(self._reasoning_contents.keys()) if self._reasoning_contents[i]]
+            
+            # Only return reasoning content if we have non-empty summaries or content
+            if summary_list or content_list:
+                return ReasoningContent(
+                    author="agent",
+                    summary=summary_list,
+                    content=content_list if content_list else None,
+                    type="reasoning",
+                    style="static",
+                )
+            else:
+                # Return empty text content instead of empty reasoning
+                return TextContent(
+                    author="agent",
+                    content="",
+                )
         else:
             raise ValueError(f"Unknown delta type: {self._delta_type}")
 
@@ -154,6 +202,7 @@ class StreamingTaskMessageContext:
         start_event = StreamTaskMessageStart(
             parent_task_message=self.task_message,
             content=self.initial_content,
+            type="start",
         )
         await self._streaming_service.stream_update(start_event)
 
@@ -168,11 +217,19 @@ class StreamingTaskMessageContext:
             return self.task_message  # Already done
 
         # Send the DONE event
-        done_event = StreamTaskMessageDone(parent_task_message=self.task_message)
+        done_event = StreamTaskMessageDone(
+            parent_task_message=self.task_message,
+            type="done",
+        )
         await self._streaming_service.stream_update(done_event)
 
         # Update the task message with the final content
-        if self._delta_accumulator._accumulated_deltas:
+        has_deltas = (
+            self._delta_accumulator._accumulated_deltas or 
+            self._delta_accumulator._reasoning_summaries or 
+            self._delta_accumulator._reasoning_contents
+        )
+        if has_deltas:
             self.task_message.content = self._delta_accumulator.convert_to_content()
 
         await self._agentex_client.messages.update(
@@ -187,8 +244,8 @@ class StreamingTaskMessageContext:
         return self.task_message
 
     async def stream_update(
-        self, update: StreamTaskMessage
-    ) -> StreamTaskMessage | None:
+        self, update: TaskMessageUpdate
+    ) -> TaskMessageUpdate | None:
         """Stream an update to the repository."""
         if self._is_closed:
             raise ValueError("Context is already done")

agentex/lib/core/temporal/activities/adk/streaming_activities.py

@@ -3,7 +3,7 @@ from enum import Enum
 from temporalio import activity
 
 from agentex.lib.core.services.adk.streaming import StreamingService
-from agentex.lib.types.task_message_updates import TaskMessageUpdate
+from agentex.types.task_message_update import TaskMessageUpdate
 from agentex.lib.utils.logging import make_logger
 from agentex.lib.utils.model_utils import BaseModel
 from agentex.lib.utils.temporal import heartbeat_if_in_workflow

agentex/lib/sdk/fastacp/base/base_acp_server.py

@@ -21,7 +21,7 @@ from agentex.lib.types.acp import (
     SendMessageParams,
 )
 from agentex.lib.types.json_rpc import JSONRPCError, JSONRPCRequest, JSONRPCResponse
-from agentex.lib.types.task_message_updates import StreamTaskMessageFull, TaskMessageUpdate
+from agentex.types.task_message_update import StreamTaskMessageFull, TaskMessageUpdate
 from agentex.types.task_message_content import TaskMessageContent
 from agentex.lib.utils.logging import make_logger
 from agentex.lib.utils.model_utils import BaseModel
@@ -332,7 +332,7 @@ class BaseACPServer(FastAPI):
 
                 async def async_generator(task_message_content_list: list[TaskMessageContent]):
                     for i, task_message_content in enumerate(task_message_content_list):
-                        yield StreamTaskMessageFull(index=i, content=task_message_content)
+                        yield StreamTaskMessageFull(type="full", index=i, content=task_message_content)
 
                 return async_generator(task_message_content_list)
 

agentex/lib/sdk/fastacp/impl/sync_acp.py

@@ -3,12 +3,12 @@ from typing import Any, override
 
 from agentex.lib.sdk.fastacp.base.base_acp_server import BaseACPServer
 from agentex.lib.types.acp import SendMessageParams
-from agentex.lib.types.task_message_updates import (
+from agentex.types.task_message_update import (
     StreamTaskMessageDelta,
     StreamTaskMessageFull,
     TaskMessageUpdate,
-    TextDelta,
 )
+from agentex.types.task_message_delta import TextDelta
 from agentex.types.task_message_content import TaskMessageContent, TextContent
 from agentex.lib.utils.logging import make_logger
 

agentex/lib/utils/dev_tools/async_messages.py

@@ -12,7 +12,7 @@ from typing import List, Optional
 from yaspin.core import Yaspin
 
 from agentex import Agentex
-from agentex.types import Task, TaskMessage, TextContent, ToolRequestContent, ToolResponseContent
+from agentex.types import Task, TaskMessage, TextContent, ToolRequestContent, ToolResponseContent, ReasoningContent
 from agentex.types.task_message_update import (
     TaskMessageUpdate,
     StreamTaskMessageStart,
@@ -48,6 +48,13 @@ def print_task_message(
     if isinstance(message.content, TextContent) and not message.content.content.strip():
         return
     
+    # Skip empty reasoning messages
+    if isinstance(message.content, ReasoningContent):
+        has_summary = message.content.summary and any(s for s in message.content.summary if s)
+        has_content = message.content.content and any(c for c in message.content.content if c)
+        if not has_summary and not has_content:
+            return
+    
     timestamp = message.created_at.strftime("%m/%d/%Y %H:%M:%S") if message.created_at else "N/A"
     
     console = None
@@ -103,6 +110,26 @@ def print_task_message(
                 content = f"✅ **Tool Response: {tool_name}**\n\n{tool_response}"
         
         content_type = "tool_response"
+    elif isinstance(message.content, ReasoningContent):
+        # Format reasoning content
+        reasoning_parts = []
+        
+        # Add summary if available
+        if message.content.summary:
+            # Join summaries with double newline for better formatting
+            summary_text = "\n\n".join(s for s in message.content.summary if s)
+            if summary_text:
+                reasoning_parts.append(summary_text)
+        
+        # Add full reasoning content if available
+        if message.content.content:
+            content_text = "\n\n".join(c for c in message.content.content if c)
+            if content_text:
+                reasoning_parts.append(content_text)
+        
+        # Format reasoning content (we already checked it's not empty at the top)
+        content = "🧠 **Reasoning**\n\n" + "\n\n".join(reasoning_parts)
+        content_type = "reasoning"
     else:
         content = f"{type(message.content).__name__}: {message.content}"
         content_type = "other"
@@ -116,6 +143,8 @@ def print_task_message(
             border_style = "yellow"
         elif content_type == "tool_response":
             border_style = "bright_green"
+        elif content_type == "reasoning":
+            border_style = "bright_magenta"
         else:
             border_style = author_color
             
@@ -123,6 +152,8 @@ def print_task_message(
         console.print(panel)
     else:
         title = f"{message.content.author.upper()} [{timestamp}]"
+        if content_type == "reasoning":
+            title = f"🧠 REASONING [{timestamp}]"
         print(f"{title}\n{content}\n")
 
 

{agentex_sdk-0.4.6.dist-info → agentex_sdk-0.4.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.4.6
+Version: 0.4.8
 Summary: The official Python library for the agentex API
 Project-URL: Homepage, https://github.com/scaleapi/agentex-python
 Project-URL: Repository, https://github.com/scaleapi/agentex-python

{agentex_sdk-0.4.6.dist-info → agentex_sdk-0.4.8.dist-info}/RECORD

@@ -11,7 +11,7 @@ agentex/_resource.py,sha256=S1t7wmR5WUvoDIhZjo_x-E7uoTJBynJ3d8tPJMQYdjw,1106
 agentex/_response.py,sha256=Tb9zazsnemO2rTxWtBjAD5WBqlhli5ZaXGbiKgdu5DE,28794
 agentex/_streaming.py,sha256=FNGJExRCF-vTRUZHFKUfoAWFhDGOB3XbioVCF37Jr7E,10104
 agentex/_types.py,sha256=KyKYySGIfHPod2hho1fPxssk5NuVn8C4MeMTtA-lg80,6198
-agentex/_version.py,sha256=0jOnjx7KMHpb97faXF4SWf9u49iBskkyVhnWoa4wzZc,159
+agentex/_version.py,sha256=B50Ua8_D0D-KO2Gx_YEx60xydpxFLCIU8BwZh9UZhbY,159
 agentex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
 agentex/_utils/_logs.py,sha256=LUjFPc3fweSChBUmjhQD8uYmwQAmFMNDuVFKfjYBQfM,777
@@ -83,7 +83,7 @@ agentex/lib/cli/templates/sync/environments.yaml.j2,sha256=BGprRPca_Y2sPA7kOiSK8
 agentex/lib/cli/templates/sync/manifest.yaml.j2,sha256=V497KXzvA76sHrgIJ5zRJptpIH8sGbSXZaIsEyp5NZ4,3747
 agentex/lib/cli/templates/sync/pyproject.toml.j2,sha256=eyN6dYqJTFzb5WztJMxboy9Wc0XPXVnKYaF5JBxJE7o,507
 agentex/lib/cli/templates/sync/requirements.txt.j2,sha256=iTmO-z8qFkUa1jTctFCs0WYuq7Sqi6VNQAwATakh2fQ,94
-agentex/lib/cli/templates/sync/project/acp.py.j2,sha256=X5RaE9iR4Dp-kPJL0r1QAe6ohfiOTcYizwtwGW2GzHg,1003
+agentex/lib/cli/templates/sync/project/acp.py.j2,sha256=PUVXnN7IikBWYnfqLMgual-iXdl_benAjEfNcRADHBk,998
 agentex/lib/cli/templates/temporal/.dockerignore.j2,sha256=hweGFxw5eDZYsb5EnRHpv27o9M1HF2PEWOxqsfBBcAE,320
 agentex/lib/cli/templates/temporal/Dockerfile-uv.j2,sha256=_FNRUdE8m1jfcmYR3rZAAkm2mqyTCwUXbLs4dm4lZs8,1401
 agentex/lib/cli/templates/temporal/Dockerfile.j2,sha256=N1Z73jb8pnxsjP9zbs-tSyNHO6usVzyOdtWorbR5gVY,1434
@@ -122,16 +122,16 @@ agentex/lib/core/services/adk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5N
 agentex/lib/core/services/adk/agent_task_tracker.py,sha256=7HHxzTezx0FEmUlIcWusoXT0D91QVeggOKowIA919gI,2897
 agentex/lib/core/services/adk/agents.py,sha256=MVklI4Ht8j8EAO18-0reLFRr6S5OIdEIymB1mU8FnKE,1446
 agentex/lib/core/services/adk/events.py,sha256=gzhEfodEKk2bW5e1RJdrk2MW316U5VWdxMynhtFWgTA,1949
-agentex/lib/core/services/adk/messages.py,sha256=aek1g4yq301JC6J9f2-kGSJIKu-oegeNIPAd4I4EMak,6047
+agentex/lib/core/services/adk/messages.py,sha256=pqIYaY11tmGOD8OnU3qoy5lk82AI1FgW5ShmfxidG_I,6042
 agentex/lib/core/services/adk/state.py,sha256=O4hF7dj-uQP5VVeHmx37m36rmIlOwfQ2pL2ROndq7Zg,3830
-agentex/lib/core/services/adk/streaming.py,sha256=zIXxMaoL5iTTC8CvmNoG4lRQO5JHyWQmwT2KyiUQL84,9563
+agentex/lib/core/services/adk/streaming.py,sha256=keZyy7CixM2_3tf2THv-X4gGOOSAk0mEKBk-WReBqJE,12406
 agentex/lib/core/services/adk/tasks.py,sha256=XEX0u4skq81Fn-gjTzSE_-Iv_mKySZHchpotxAsnJwg,2618
 agentex/lib/core/services/adk/tracing.py,sha256=w4mnIoq5m5iwlkDN_NOUjhi6hUpD958cYn1QR6hFgB8,1156
 agentex/lib/core/services/adk/acp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentex/lib/core/services/adk/acp/acp.py,sha256=Ei6r5CcM1RA7DqfTRpAWCmP7Buw97b3ajQuE4DsdSaQ,7559
+agentex/lib/core/services/adk/acp/acp.py,sha256=4KeRPaka8GqnF2PQiNigcijgvQ_rf8Si0EoOdQNqjN4,7926
 agentex/lib/core/services/adk/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentex/lib/core/services/adk/providers/litellm.py,sha256=7RIVqQ5XaCP2cEL1pr4nl5jfG_0FDMvDTZVKu1iaN3M,9864
-agentex/lib/core/services/adk/providers/openai.py,sha256=tR2iKJc2SZo2eLUIWwkyl7W4QIkFnzdxT1OQxaPXQBc,33348
+agentex/lib/core/services/adk/providers/litellm.py,sha256=EKzus_xohNW-85V5hwvd1WqUd3ebv2wc9vDIWO2t1Mw,10044
+agentex/lib/core/services/adk/providers/openai.py,sha256=Ad7iFQDg1iz8KjswO16lzjpzEJCz5FtyXV-fB6qMPLk,39535
 agentex/lib/core/services/adk/providers/sgp.py,sha256=9gm-sPNQ_OSTaBzL0onerKhokPk_ZHndaKNO-z16wyQ,3676
 agentex/lib/core/services/adk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/core/services/adk/utils/templating.py,sha256=eaXSFq31Y9p5pRD6J6SL4QdTFtxy81dilbF2XXc2JYQ,1889
@@ -144,7 +144,7 @@ agentex/lib/core/temporal/activities/adk/agents_activities.py,sha256=q3PGgOLXvcE
 agentex/lib/core/temporal/activities/adk/events_activities.py,sha256=60ygWk1aNG8Qf2IcnatXK1h86ZT32WCYkUAyRiM_p4c,1532
 agentex/lib/core/temporal/activities/adk/messages_activities.py,sha256=fmiNOAt6Ov4nYzGGbkvDcSCMXPPTrsL66xSNud3_3RQ,3053
 agentex/lib/core/temporal/activities/adk/state_activities.py,sha256=2YtAz7C2GMfyZGdp8wje1TvYjFGttdbLNAm9xSIbESA,2590
-agentex/lib/core/temporal/activities/adk/streaming_activities.py,sha256=HtxGBgz4UNPcFSo1izCzg2Yw__C4E6igUNrqhC38FVA,1053
+agentex/lib/core/temporal/activities/adk/streaming_activities.py,sha256=PEXeFU9mNHbGoF3-kk4HNS53NX_MfdwI7LKWBl3--4c,1048
 agentex/lib/core/temporal/activities/adk/tasks_activities.py,sha256=hQjdD-TfSzxsiW9eiJ5ZTvY0Ieu0tE950_Zk2PUVc9g,1442
 agentex/lib/core/temporal/activities/adk/tracing_activities.py,sha256=RF8nSBA9x6Ui7_MD9He5ljlH_xZduIhx_JUTEyKdCxA,1523
 agentex/lib/core/temporal/activities/adk/acp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -181,9 +181,9 @@ agentex/lib/sdk/config/project_config.py,sha256=CGH_r9KbnSFMj2CnBkZnfg41L2o0TeVN
 agentex/lib/sdk/config/validation.py,sha256=QGAlAzlVJiWRlIksqxNS-JSwkk8Z4gXMSFUJc4qPrIQ,8989
 agentex/lib/sdk/fastacp/__init__.py,sha256=UvAdexdnfb4z0F4a2sfXROFyh9EjH89kf3AxHPybzCM,75
 agentex/lib/sdk/fastacp/fastacp.py,sha256=K4D7a9EiJfCgWp2hrE_TbpZBaF4Uc46MfndZK3F9mA0,3061
-agentex/lib/sdk/fastacp/base/base_acp_server.py,sha256=fkEVg5M4Ni2uPt_oiCP7_jfTmRvcC0o2VTg0Oka7mXA,14590
+agentex/lib/sdk/fastacp/base/base_acp_server.py,sha256=vj7jPg8a8z07dTl3P-ZVTdhcuW3ulm4QrptWyYVnTAs,14598
 agentex/lib/sdk/fastacp/impl/agentic_base_acp.py,sha256=LWLAlHrs-2Lc2UICBAEFL8c3JwTA6oxPnzUzW0qQWSA,2694
-agentex/lib/sdk/fastacp/impl/sync_acp.py,sha256=8FEDfBxI31NoVLX9nyckQ8QwA0UV4svC3fhGKgXBIwk,3862
+agentex/lib/sdk/fastacp/impl/sync_acp.py,sha256=0y_cYD-0UJOiVJv-BBMOt6PvElDf5zmc1uvbr81VcSI,3897
 agentex/lib/sdk/fastacp/impl/temporal_acp.py,sha256=hQa_zGG6l_jIHNC75H_KXlPUuVqNBc-zIm-pmCDSZ6w,3759
 agentex/lib/sdk/fastacp/tests/README.md,sha256=HejScemERLCs-wCgU3l1Xo4tcQ4gQy15GgoF-CkNh-0,8270
 agentex/lib/sdk/fastacp/tests/conftest.py,sha256=wU3DCGZNBqY-HcJ4tz_tlphhS_5xGbZ6gSd72fGY2E0,9490
@@ -209,7 +209,6 @@ agentex/lib/types/fastacp.py,sha256=nU4rT823Ckix7bFwvVXtPsk6el3U1E4TH-OEvMqIBr4,
 agentex/lib/types/files.py,sha256=sNxiuR_oEo7Z0YMqjQShErvS3txWyobrZzc4QMMM61U,320
 agentex/lib/types/json_rpc.py,sha256=9sPADVEduOnyTm1yCS8vhZDfjMQVEdzqIWpycY_NrV8,1068
 agentex/lib/types/llm_messages.py,sha256=eXsVJt_aQf3eGg_tPrDOPjFOhkbolb9IN1tU1c-IHYM,10668
-agentex/lib/types/task_message_updates.py,sha256=jt9RUftr13sw0WbvdHOeK0Tidzdq-xfewx1AYqV8gtg,4298
 agentex/lib/types/tracing.py,sha256=0YBdy3aKvxIcpOW0podEAqsFl3yEKW4PswNf-QnMNqo,731
 agentex/lib/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/utils/completions.py,sha256=m9TMwl9-nYK4DOQ8bNyj5YCFrxK3yN3zKvXmdINcIbo,4158
@@ -226,7 +225,7 @@ agentex/lib/utils/regex.py,sha256=Y3VcehCznAqa59D4WTwK_ki722oudHBlFqBk0I930zo,21
 agentex/lib/utils/registration.py,sha256=FAL40KSMRCltHZcDtioGeRNgfB19Twqu6Srp4x7UMSU,3781
 agentex/lib/utils/temporal.py,sha256=sXo8OPMMXiyrF7OSBCJBuN_ufyQOD2bLOXgDbVZoyds,292
 agentex/lib/utils/dev_tools/__init__.py,sha256=oaHxw6ymfhNql-kzXHv3NWVHuqD4fHumasNXJG7kHTU,261
-agentex/lib/utils/dev_tools/async_messages.py,sha256=-alUK1KFltcRb6xtRtIJIRJW9Sf1FwDRgaNPhOn-luw,18105
+agentex/lib/utils/dev_tools/async_messages.py,sha256=dM01spww2Fy6ej_WDnPXs2eG-9yCU7Xc1IMKwJzq6cM,19543
 agentex/resources/__init__.py,sha256=74rMqWBzQ2dSrKQqsrd7-jskPws0O_ogkFltvZO3HoU,3265
 agentex/resources/agents.py,sha256=WLG7wTQcCZCtqvNiyi2wjuDA2u85nEA5o9kUr5pBQXE,46977
 agentex/resources/events.py,sha256=Zc9JhUm3bq2VFnBAolC0M7KZernzj1AjZ_vj0ibP4GY,10412
@@ -301,8 +300,8 @@ agentex/types/messages/batch_update_params.py,sha256=Ug5CThbD49a8j4qucg04OdmVrp_
 agentex/types/messages/batch_update_response.py,sha256=TbSBe6SuPzjXXWSj-nRjT1JHGBooTshHQQDa1AixQA8,278
 agentex/types/shared/__init__.py,sha256=IKs-Qn5Yja0kFh1G1kDqYZo43qrOu1hSoxlPdN-85dI,149
 agentex/types/shared/delete_response.py,sha256=8qH3zvQXaOHYQSHyXi7UQxdR4miTzR7V9K4zXVsiUyk,215
-agentex_sdk-0.4.6.dist-info/METADATA,sha256=u1EUOOWGKnqAOgh82EvTTThHbIh8ZcpRrRCIlqLjTPk,15094
-agentex_sdk-0.4.6.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-agentex_sdk-0.4.6.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
-agentex_sdk-0.4.6.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
-agentex_sdk-0.4.6.dist-info/RECORD,,
+agentex_sdk-0.4.8.dist-info/METADATA,sha256=_ZN2ZmRBA_8REs6q2Q8gsidbWdcI2kqDPPvoDgqkdZQ,15094
+agentex_sdk-0.4.8.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+agentex_sdk-0.4.8.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
+agentex_sdk-0.4.8.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
+agentex_sdk-0.4.8.dist-info/RECORD,,

agentex/lib/types/task_message_updates.py

@@ -1,171 +0,0 @@
-from enum import Enum
-from typing import Annotated, Literal
-
-from pydantic import Field, model_validator
-
-from agentex.types.task_message import TaskMessage
-from agentex.types.task_message_content import TaskMessageContent
-from agentex.lib.utils.model_utils import BaseModel
-
-
-class BaseTaskMessageDelta(BaseModel):
-    """
-    Base class for all delta updates
-
-    Attributes:
-        type: The type of delta update
-    """
-
-    type: Literal["text", "data", "tool_request", "tool_response"]
-
-
-class TextDelta(BaseTaskMessageDelta):
-    """
-    Delta for text updates
-
-    Attributes:
-        type: The type of delta update
-        text_delta: The delta for the text
-    """
-
-    type: Literal["text"] = "text"
-    text_delta: str | None = ""
-
-
-class DataDelta(BaseTaskMessageDelta):
-    """
-    Delta for data updates
-
-    Attributes:
-        type: The type of delta update
-        data_delta: The delta for the data
-    """
-
-    type: Literal["data"] = "data"
-    data_delta: str | None = ""
-
-
-class ToolRequestDelta(BaseTaskMessageDelta):
-    """
-    Delta for tool request updates
-
-    Attributes:
-        type: The type of delta update
-        name: The name of the tool
-        arguments_delta: The delta for the arguments
-    """
-
-    type: Literal["tool_request"] = "tool_request"
-    tool_call_id: str
-    name: str
-    arguments_delta: str | None = ""
-
-
-class ToolResponseDelta(BaseTaskMessageDelta):
-    """
-    Delta for tool response updates
-
-    Attributes:
-        type: The type of delta update
-        tool_response_delta: The delta for the tool response
-    """
-
-    type: Literal["tool_response"] = "tool_response"
-    tool_call_id: str
-    name: str
-    tool_response_delta: str | None = ""
-
-
-TaskMessageDelta = Annotated[
-    TextDelta | DataDelta | ToolRequestDelta | ToolResponseDelta,
-    Field(discriminator="type"),
-]
-
-
-class StreamTaskMessage(BaseModel):
-    """Base class for all task message stream events
-
-    Attributes:
-        type: The type of task message update
-        parent_task_message: The parent task message
-        index: The index of the task message
-    """
-
-    type: Literal["start", "delta", "full", "done"]
-    # Used for streaming chunks to a direct parent_task_message
-    parent_task_message: TaskMessage | None = None
-    # Used to correlate chunks of different task messages with each other
-    # directly in the Sync ACP case
-    index: int | None = None
-
-    @model_validator(mode="after")
-    def validate_message_correlation(self):
-        """Ensure exactly one of index or parent_task_message is set"""
-        has_parent = self.parent_task_message is not None
-        has_index = self.index is not None
-
-        if not has_parent and not has_index:
-            raise ValueError("Either 'index' or 'parent_task_message' must be set")
-
-        if has_parent and has_index:
-            raise ValueError(
-                "Cannot set both 'index' and 'parent_task_message' - only one is allowed"
-            )
-
-        return self
-
-
-# Everything is streamed as a partial json blob, except for text.
-class StreamTaskMessageStart(StreamTaskMessage):
-    """Event for starting a streaming message
-
-    Attributes:
-        type: The type of task message update
-        content: The content of the task message
-    """
-
-    type: Literal["start"] = "start"
-    content: TaskMessageContent
-
-
-class StreamTaskMessageDelta(StreamTaskMessage):
-    """Event for streaming chunks of content
-
-    Attributes:
-        type: The type of task message update
-        delta: The delta of the task message
-    """
-
-    type: Literal["delta"] = "delta"
-    delta: TaskMessageDelta | None = None
-
-
-class StreamTaskMessageFull(StreamTaskMessage):
-    """Event for streaming the full content
-
-    Attributes:
-        type: The type of task message update
-        content: The content of the task message
-    """
-
-    type: Literal["full"] = "full"
-    content: TaskMessageContent
-
-
-class StreamTaskMessageDone(StreamTaskMessage):
-    """Event for indicating the task is done
-
-    Attributes:
-        type: The type of task message update
-    """
-
-    type: Literal["done"] = "done"
-
-
-TaskMessageUpdate = Annotated[
-    StreamTaskMessageStart
-    | StreamTaskMessageDelta
-    | StreamTaskMessageFull
-    | StreamTaskMessageDone,
-    Field(discriminator="type"),
-]