From 2cfb2765ed30df310d5fc0e173edebf80892b5c5 Mon Sep 17 00:00:00 2001
From: Hamza Sheikh <164283922+SheikhMuhammadHamza-HS@users.noreply.github.com>
Date: Mon, 14 Jul 2025 16:35:02 +0500
Subject: [PATCH] Update routing.py

---
 examples/agent_patterns/routing.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/examples/agent_patterns/routing.py b/examples/agent_patterns/routing.py
index 3dcaefa98..55ad5f0e4 100644
--- a/examples/agent_patterns/routing.py
+++ b/examples/agent_patterns/routing.py
@@ -63,8 +63,11 @@ async def main():
 
         user_msg = input("Enter a message: ")
         inputs.append({"content": user_msg, "role": "user"})
+        if user_msg.lower().strip() == "exit":
+            print("")
+            break
         agent = result.current_agent
 
 
-if __name__ == "__main__":
-    asyncio.run(main())
+
+asyncio.run(main())