More linting fixes

Author: danielmillerp

examples/demos/procurement_agent/project/agents/procurement_agent.py

@@ -61,6 +61,10 @@ async def wait_for_human(recommended_action: str) -> str:
             print(f"[WORKFLOW] Processing human message from queue")
             return human_input
 
+        # If queue became empty after wait_condition succeeded, this shouldn't normally happen
+        workflow.logger.warning("Queue empty after wait condition succeeded")
+        return "No human response available"
+
     except TimeoutError:
         # Human didn't respond within 24 hours
         workflow.logger.warning("Human escalation timed out after 24 hours")

examples/demos/procurement_agent/project/data/database.py

@@ -8,7 +8,7 @@
 from typing import Optional
 from pathlib import Path
 
-import aiosqlite
+import aiosqlite  # type: ignore[import-untyped]
 
 from agentex.lib.utils.logging import make_logger
 

examples/demos/procurement_agent/project/utils/learning_extraction.py

@@ -62,6 +62,7 @@ def get_new_wait_for_human_context(
         # Reverse back to chronological order
         new_context.reverse()
         logger.info(f"Extracted {len(new_context)} messages of new context")
+        assert new_wait_for_human_call_id is not None, "call_id should be set when found_new_wait_for_human is True"
         return (new_context, new_wait_for_human_call_id)
     else:
         logger.info("No new wait_for_human calls found")

examples/demos/procurement_agent/project/workflow.py

@@ -1,6 +1,6 @@
 import json
 import asyncio
-from typing import Any, Dict, List
+from typing import Any, Dict, List, override
 from datetime import timedelta
 
 from agents import Runner
@@ -82,6 +82,7 @@ def __init__(self):
         )
 
     @workflow.signal(name=SignalName.RECEIVE_EVENT)
+    @override
     async def on_task_event_send(self, params: SendEventParams) -> None:
 
         if self._state is None:
@@ -95,22 +96,30 @@ async def on_task_event_send(self, params: SendEventParams) -> None:
         self._trace_id = params.task.id
         self._parent_span_id = params.task.id
 
+        if params.event.content is None:
+            workflow.logger.warning("Received event with no content")
+            return
+
         await adk.messages.create(task_id=params.task.id, content=params.event.content)
 
         if self._state.turn_number > 1:
-        # put human message in human queue
-            await self.human_queue.put(params.event.content.content)
-            return await adk.messages.create(task_id=params.task.id, content=TextContent(
+            # put human message in human queue
+            # Extract text content from the event
+            if hasattr(params.event.content, 'content'):
+                await self.human_queue.put(params.event.content.content)
+
+            await adk.messages.create(task_id=params.task.id, content=TextContent(
                 author="agent",
                 content="",
             ))
         else:
-            return await adk.messages.create(task_id=params.task.id, content=TextContent(
+            await adk.messages.create(task_id=params.task.id, content=TextContent(
                 author="agent",
                 content="",
             ))
 
     @workflow.run
+    @override
     async def on_task_create(self, params: CreateTaskParams) -> str:
         logger.info(f"Received task create params: {params}")
 
@@ -223,10 +232,10 @@ async def on_task_create(self, params: CreateTaskParams) -> str:
                     hooks = TemporalStreamingHooks(task_id=params.task.id)
 
                     # Execute agent with graceful degradation pattern (from temporal-community demos)
-                    result = await Runner.run(procurement_agent, self._state.input_list, hooks=hooks)
+                    result = await Runner.run(procurement_agent, self._state.input_list, hooks=hooks)  # type: ignore[arg-type]
 
                     # Update state with result
-                    self._state.input_list = result.to_input_list()
+                    self._state.input_list = result.to_input_list()  # type: ignore[assignment]
                     logger.info(f"Successfully processed event at turn {self._state.turn_number}")
 
                     # Extract learnings from NEW wait_for_human calls only (using going backwards approach)
@@ -242,7 +251,7 @@ async def on_task_create(self, params: CreateTaskParams) -> str:
 
                             # Create extraction agent and run with only the NEW context
                             extract_agent = new_extract_learnings_agent()
-                            extraction_result = await Runner.run(extract_agent, new_context, hooks=hooks)
+                            extraction_result = await Runner.run(extract_agent, new_context, hooks=hooks)  # type: ignore[arg-type]
 
                             logger.info(f"About to extract learning: {extraction_result.final_output}")
                             # Append the learning and track the call_id