Update Distributed Agent Runtime Cross-platform Sample (#5164)

fix: improve service shutdown handling in for cross-platform compatibility (#5124)

Co-authored-by: Eric Zhu <[email protected]>

python/samples/core_grpc_worker_runtime/run_host.py

@@ -1,12 +1,24 @@
 import asyncio
+import os
 
 from autogen_ext.runtimes.grpc import GrpcWorkerAgentRuntimeHost
 
 
 async def main() -> None:
     service = GrpcWorkerAgentRuntimeHost(address="localhost:50051")
     service.start()
-    await service.stop_when_signal()
+
+    try:
+        # Wait for the service to stop
+        if os.system() == "Windows":
+            # On Windows, the signal is not available, so we wait for a new event
+            await asyncio.Event().wait()
+        else:
+            await service.stop_when_signal()
+    except KeyboardInterrupt:
+        print("Stopping service...")
+    finally:
+        await service.stop()
 
 
 if __name__ == "__main__":