adding the new image

Author: RoxyFarhad

examples/tutorials/10_async/10_temporal/010_agent_chat/tests/test_agent.py

@@ -156,7 +156,7 @@ async def test_multi_turn_conversation(self, client: AsyncAgentex, agent_id: str
             agent_id=agent_id,
             task_id=task.id,
             user_message=user_message_1,
-            timeout=20,
+            timeout=30,
             sleep_interval=1.0,
         ):
             assert isinstance(message, TaskMessage)
@@ -215,14 +215,16 @@ async def test_send_event_and_stream_with_reasoning(self, client: AsyncAgentex,
         # Check for user message and agent response
         user_message_found = False
         agent_response_found = False
+        reasoning_found = False
 
         async def stream_messages() -> None:  # noqa: ANN101
-            nonlocal user_message_found, agent_response_found
+            nonlocal user_message_found, agent_response_found, reasoning_found
             async for event in stream_agent_response(
                 client=client,
                 task_id=task.id,
-                timeout=60,
+                timeout=90,  # Increased timeout for CI environments
             ):
+                print(event)
                 msg_type = event.get("type")
                 if msg_type == "full":
                     task_message_update = StreamTaskMessageFull.model_validate(event)
@@ -241,22 +243,40 @@ async def stream_messages() -> None:  # noqa: ANN101
                         ):
                             agent_response_found = True
                         elif finished_message.content and finished_message.content.type == "reasoning":
-                            tool_response_found = True
+                            reasoning_found = True
+
+                    # Exit early if we have what we need
+                    if user_message_found and agent_response_found:
+                        break
+
                 elif msg_type == "done":
                     task_message_update = StreamTaskMessageDone.model_validate(event)
                     if task_message_update.parent_task_message and task_message_update.parent_task_message.id:
                         finished_message = await client.messages.retrieve(task_message_update.parent_task_message.id)
                         if finished_message.content and finished_message.content.type == "reasoning":
+                            reasoning_found = True
+                        elif (
+                            finished_message.content
+                            and finished_message.content.type == "text"
+                            and finished_message.content.author == "agent"
+                        ):
                             agent_response_found = True
-                    continue
+
+                    # Exit early if we have what we need
+                    if user_message_found and agent_response_found:
+                        break
 
         stream_task = asyncio.create_task(stream_messages())
 
         event_content = TextContentParam(type="text", author="user", content=user_message)
         await client.agents.send_event(agent_id=agent_id, params={"task_id": task.id, "content": event_content})
 
-        # Wait for streaming to complete
-        await stream_task
+        # Wait for streaming to complete with timeout
+        try:
+            await asyncio.wait_for(stream_task, timeout=120)  # Overall timeout for CI
+        except asyncio.TimeoutError:
+            stream_task.cancel()
+            pytest.fail("Test timed out waiting for streaming response")
 
         assert user_message_found, "User message not found in stream"
         assert agent_response_found, "Agent response not found in stream"

examples/tutorials/test_utils/async_utils.py

@@ -125,16 +125,10 @@ async def poll_messages(
 
             if yield_updates:
                 # For streaming: track content changes
-                content_str = message.content.content if message.content and hasattr(message.content, 'content') else ""
-                # Ensure content_str is always a string to avoid concatenation errors
-                if isinstance(content_str, list):
-                    # If it's a list, convert to string representation
-                    content_str = str(content_str)
-                elif content_str is None:
-                    content_str = ""
-                elif not isinstance(content_str, str):
-                    # Handle any other non-string types
-                    content_str = str(content_str)
+                # Use getattr to safely extract content and convert to string
+                # This handles various content structures at runtime
+                raw_content = getattr(message.content, 'content', message.content) if message.content else None
+                content_str = str(raw_content) if raw_content is not None else ""
 
                 # Ensure streaming_status is also properly converted to string
                 streaming_status_str = str(message.streaming_status) if message.streaming_status is not None else ""