how about this

Author: smoreinis

src/agentex/lib/testing/sessions/sync.py

@@ -147,8 +147,8 @@ def send_message_streaming(self, content: str):
             first_chunk = next(response_generator, None)
             if first_chunk and hasattr(first_chunk, 'result'):
                 result = first_chunk.result
-                if hasattr(result, 'task_id') and getattr(result, 'task_id'):
-                    self.task_id = getattr(result, 'task_id')
+                if hasattr(result, 'task_id') and result.task_id: # type: ignore
+                    self.task_id = result.task_id # type: ignore
                     logger.debug(f"Task auto-created from stream: {self.task_id}")
                 # Check if result has parent_task_message with task_id
                 elif hasattr(result, 'parent_task_message') and result.parent_task_message:

src/agentex/lib/testing/type_utils.py

@@ -64,15 +64,15 @@ def extract_agent_response(response, agent_id: str):  # type: ignore[no-untyped-
 
         # SendMessageResponse: result.content
         if hasattr(result, "content"):
-            content = getattr(result, "content")
+            content = result.content # type: ignore
             if isinstance(content, TextContent):
                 return content
 
         # SendEventResponse: result.message.content
-        if hasattr(result, "message") and getattr(result, "message"):
-            message = getattr(result, "message")
+        if hasattr(result, "message") and result.message: # type: ignore
+            message = result.message # type: ignore
             if hasattr(message, "content"):
-                content = getattr(message, "content")
+                content = message.content
                 if isinstance(content, TextContent):
                     return content