Adding changes to support temporal auto reload + fixing where you cant ctrl+c

Author: jasonyang101

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

@@ -22,7 +22,7 @@ async def main():
     worker = AgentexWorker(
         task_queue=task_queue_name,
     )
-
+    
     await worker.run(
         activities=get_all_activities(),
         workflow=At010AgentChatWorkflow,

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

@@ -70,7 +70,6 @@ async def on_task_event_send(self, params: SendEventParams) -> None:
 
         if not params.event.content:
             return
-
         if params.event.content.type != "text":
             raise ValueError(f"Expected text message, got {params.event.content.type}")
 

pyproject.toml

@@ -19,6 +19,7 @@ dependencies = [
     "rich>=13.9.2,<14",
     "fastapi>=0.115.0,<0.116",
     "uvicorn>=0.31.1",
+    "watchfiles>=0.24.0,<1.0",
     "python-on-whales>=0.73.0,<0.74",
     "pyyaml>=6.0.2,<7",
     "jsonschema>=4.23.0,<5",

src/agentex/lib/cli/handlers/agent_handlers.py

@@ -129,8 +129,32 @@ def build_agent(
 def run_agent(manifest_path: str):
     """Run an agent locally from the given manifest"""
     import asyncio
-
+    import signal
+    import sys
+
+    # Flag to track if we're shutting down
+    shutting_down = False
+
+    def signal_handler(signum, frame):
+        """Handle signals by raising KeyboardInterrupt"""
+        nonlocal shutting_down
+        if shutting_down:
+            # If we're already shutting down and get another signal, force exit
+            print(f"\nForce exit on signal {signum}")
+            sys.exit(1)
+        
+        shutting_down = True
+        print(f"\nReceived signal {signum}, shutting down...")
+        raise KeyboardInterrupt()
+    
+    # Set up signal handling for the main thread
+    signal.signal(signal.SIGINT, signal_handler)
+    signal.signal(signal.SIGTERM, signal_handler)
+    
     try:
         asyncio.run(_run_agent(manifest_path))
+    except KeyboardInterrupt:
+        print("Shutdown completed.")
+        sys.exit(0)
     except RunError as e:
         raise RuntimeError(str(e)) from e