Initial fix

Author: IdkwhatImD0ing

examples/tutorials/10_agentic/10_temporal/000_hello_acp/project/workflow.py

@@ -33,7 +33,6 @@ def __init__(self):
     @workflow.signal(name=SignalName.RECEIVE_EVENT)
     @override
     async def on_task_event_send(self, params: SendEventParams) -> None:
-        logger.info(f"Received task message instruction: {params}")
 
         # 2. Echo back the client's message to show it in the UI. This is not done by default so the agent developer has full control over what is shown to the user.
         await adk.messages.create(task_id=params.task.id, content=params.event.content)

examples/tutorials/10_agentic/10_temporal/010_agent_chat/dev.ipynb

@@ -48,11 +48,11 @@ class StateModel(BaseModel):
     turn_number: int
 
 
-MCP_SERVERS = [
-    StdioServerParameters(
-        command="npx",
-        args=["-y", "@modelcontextprotocol/server-sequential-thinking"],
-    ),
+MCP_SERVERS = [ # No longer needed due to reasoning
+    # StdioServerParameters(
+    #     command="npx",
+    #     args=["-y", "@modelcontextprotocol/server-sequential-thinking"],
+    # ),
     StdioServerParameters(
         command="uvx",
         args=["openai-websearch-mcp"],
@@ -163,7 +163,6 @@ def __init__(self):
     @workflow.signal(name=SignalName.RECEIVE_EVENT)
     @override
     async def on_task_event_send(self, params: SendEventParams) -> None:
-        logger.info(f"Received task message instruction: {params}")
 
         if not params.event.content:
             return

examples/tutorials/10_agentic/10_temporal/010_agent_chat/project/workflow.py

@@ -387,7 +387,6 @@ def __init__(self):
     @workflow.signal(name=SignalName.RECEIVE_EVENT)
     @override
     async def on_task_event_send(self, params: SendEventParams) -> None:
-        logger.info(f"Received task message instruction: {params}")
 
         if not params.event.content:
             return

examples/tutorials/10_agentic/10_temporal/050_agent_chat_guardrails/project/workflow.py

@@ -9,7 +9,7 @@ description = "An AgentEx agentthat streams multiturn tool-enabled chat with tra
 readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
-    "agentex-sdk",
+    "agentex-sdk @ file://../../../../../",
     "debugpy>=1.8.15",
     "scale-gp",
 ]
@@ -22,6 +22,9 @@ dev = [
     "flake8",
 ]
 
+[tool.hatch.metadata]
+allow-direct-references = true
+
 [tool.hatch.build.targets.wheel]
 packages = ["project"]
 

examples/tutorials/10_agentic/10_temporal/050_agent_chat_guardrails/pyproject.toml

@@ -208,6 +208,35 @@ async def open(self) -> "StreamingTaskMessageContext":
 
         return self
 
+    # Temp function to just send the full content
+    async def send_full_content(self) -> TaskMessage:
+        """Send the full content message to the repository without updating the task message."""
+        if not self.task_message:
+            raise ValueError("Context not properly initialized - no task message")
+
+        if self._is_closed:
+            return self.task_message  # Already done
+
+        # Update the task message with the final content from 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()
+
+        # Send the FULL event with the complete content
+        full_event = StreamTaskMessageFull(
+            parent_task_message=self.task_message,
+            content=self.task_message.content,
+            type="full",
+        )
+        await self._streaming_service.stream_update(full_event)
+
+        return self.task_message
+
+
     async def close(self) -> TaskMessage:
         """Close the streaming context."""
         if not self.task_message:

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

@@ -50,8 +50,8 @@ def print_task_message(
 
     # 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)
+        has_summary = bool(message.content.summary) and any(s for s in message.content.summary if s)
+        has_content = bool(message.content.content) and any(c for c in message.content.content if c) if message.content.content is not None else False
         if not has_summary and not has_content:
             return
 
@@ -136,18 +136,19 @@ def print_task_message(
 
     if rich_print and console:
         author_color = "bright_cyan" if message.content.author == "user" else "green"
-        title = f"[bold {author_color}]{message.content.author.upper()}[/bold {author_color}] [{timestamp}]"
 
-        # Use different border styles for tool messages
+        # Use different border styles and colors for different content types
         if content_type == "tool_request":
             border_style = "yellow"
         elif content_type == "tool_response":
             border_style = "bright_green"
         elif content_type == "reasoning":
             border_style = "bright_magenta"
+            author_color = "bright_magenta"  # Also make the author text magenta
         else:
             border_style = author_color
-            
+        
+        title = f"[bold {author_color}]{message.content.author.upper()}[/bold {author_color}] [{timestamp}]"
         panel = Panel(Markdown(content), title=title, border_style=border_style, width=80)
         console.print(panel)
     else:
@@ -330,6 +331,7 @@ def subscribe_to_async_task_messages(
 
                             # Deserialize the discriminated union TaskMessageUpdate based on the "type" field
                             message_type = task_message_update_data.get("type", "unknown")
+
 
                             # Handle different message types for streaming progress
                             if message_type == "start":
@@ -360,11 +362,20 @@ def subscribe_to_async_task_messages(
                                 if index in active_spinners:
                                     active_spinners[index].stop()
                                     del active_spinners[index]
+                                    # Ensure clean line after spinner
+                                    if print_messages:
+                                        print()
 
-                                if task_message_update.parent_task_message and task_message_update.parent_task_message.id:
-                                    finished_message = client.messages.retrieve(task_message_update.parent_task_message.id)
-                                    messages_to_return.append(finished_message)
-                                    print_task_message(finished_message, print_messages, rich_print)
+                                # Use the content directly from the stream event, not from re-fetching
+                                # The stream has the full content, but re-fetching sometimes returns empty arrays
+                                if task_message_update.parent_task_message:
+                                    # Create a temporary message object with the stream content for display
+                                    display_message = task_message_update.parent_task_message
+                                    # Override the content with the stream's full content
+                                    display_message.content = task_message_update.content
+                                    
+                                    messages_to_return.append(display_message)
+                                    print_task_message(display_message, print_messages, rich_print)
 
                             elif message_type == "done":
                                 task_message_update = StreamTaskMessageDone.model_validate(task_message_update_data)
@@ -374,12 +385,24 @@ def subscribe_to_async_task_messages(
                                 if index in active_spinners:
                                     active_spinners[index].stop()
                                     del active_spinners[index]
+                                    # Ensure clean line after spinner
+                                    if print_messages:
+                                        print()
 
                                 if task_message_update.parent_task_message and task_message_update.parent_task_message.id:
                                     finished_message = client.messages.retrieve(task_message_update.parent_task_message.id)
                                     messages_to_return.append(finished_message)
                                     print_task_message(finished_message, print_messages, rich_print)
 
+                                # Clean stream termination - break on done event
+                                if print_messages:
+                                    console = Console(width=80) if rich_print else None
+                                    if console:
+                                        console.print("[dim]Stream ended - received 'done' event[/dim]")
+                                    else:
+                                        print("Stream ended - received 'done' event")
+                                break
+                                
                             # Ignore "connected" message type
                             elif message_type == "connected":
                                 pass