multi-round chat

multi-agent-supervisor/main-multi-agent.py

@@ -10,59 +10,62 @@ if __name__ == "__main__":
     # Create the supervisor
     supervisor = create_sysadmin_supervisor()
 
-    # Example run - demonstrating both invoke and streaming with debug output
-    query = {
-        "messages": [
-            {
-                "role": "user",
-                "content": "Nginx returns 502 Bad Gateway on my server. What can I do?",
-            }
-        ]
-    }
+    # Interactive conversation loop
+    messages = []
+    print("Welcome to the multi-agent sysadmin assistant!")
+    print("Type your sysadmin question below. Type 'exit' to quit.")
+    while True:
+        user_input = input("\n📝 User: ")
+        if user_input.strip().lower() == 'exit':
+            print("Goodbye!")
+            break
+        messages.append({"role": "user", "content": user_input})
+        query = {"messages": messages}
 
-    print("🚀 Starting multi-agent sysadmin analysis...")
-    print(f"📝 User Query: {query['messages'][0]['content']}")
-    print("=" * 80)
+        print("\n=== Using invoke() method ===")
+        result = supervisor.invoke(query)
 
-    # Show explanation of the supervisor pattern
-    explain_supervisor_pattern()
+        print("\n📊 FINAL RESULT:")
+        print("-" * 40)
+        print(result["messages"][-1].content)
+        print("-" * 40)
+        print(f"\n📈 Total messages exchanged: {len(result['messages'])}")
 
-    print("\n=== Using invoke() method ===")
-    result = supervisor.invoke(query)
+        # Add the assistant's reply to the conversation history
+        messages.append({"role": "assistant", "content": result["messages"][-1].content})
 
-    print("\n📊 FINAL RESULT:")
-    print("-" * 40)
-    print(result["messages"][-1].content)
-    print("-" * 40)
+        # Ask if the user wants to continue
+        cont = input("\nWould you like to continue the conversation? (y/n): ")
+        if cont.strip().lower() not in ('y', 'yes'):
+            print("Session ended.")
+            break
 
-    print(f"\n📈 Total messages exchanged: {len(result['messages'])}")
+        print("\n=== Using stream() method for detailed step-by-step analysis ===")
+        step_count = 0
+        max_steps = 20  # Prevent infinite loops
 
-    print("\n=== Using stream() method for detailed step-by-step analysis ===")
-    step_count = 0
-    max_steps = 20  # Prevent infinite loops
+        try:
+            chunks_processed = []
+            for chunk in supervisor.stream(query):
+                step_count += 1
+                chunks_processed.append(chunk)
+                print_step_info(step_count, chunk)
 
-    try:
-        chunks_processed = []
-        for chunk in supervisor.stream(query):
-            step_count += 1
-            chunks_processed.append(chunk)
-            print_step_info(step_count, chunk)
+                # Safety check to prevent infinite loops
+                if step_count >= max_steps:
+                    print(f"\n⚠️ Reached maximum steps ({max_steps}), stopping stream...")
+                    break
 
-            # Safety check to prevent infinite loops
-            if step_count >= max_steps:
-                print(f"\n⚠️ Reached maximum steps ({max_steps}), stopping stream...")
-                break
+            print(f"\n✅ Streaming completed successfully with {step_count} steps")
+            print(f"📊 Total chunks processed: {len(chunks_processed)}")
 
-        print(f"\n✅ Streaming completed successfully with {step_count} steps")
-        print(f"📊 Total chunks processed: {len(chunks_processed)}")
+            # Check if the last chunk contains a complete final response
+            if chunks_processed:
+                last_chunk = chunks_processed[-1]
+                print(f"🔍 Last chunk keys: {list(last_chunk.keys()) if isinstance(last_chunk, dict) else type(last_chunk)}")
 
-        # Check if the last chunk contains a complete final response
-        if chunks_processed:
-            last_chunk = chunks_processed[-1]
-            print(f"🔍 Last chunk keys: {list(last_chunk.keys()) if isinstance(last_chunk, dict) else type(last_chunk)}")
-        
-    except Exception as e:
-        print(f"\n❌ Streaming error after {step_count} steps: {e}")
-        print("💡 The invoke() method worked fine, so the supervisor itself is functional.")
-        import traceback
-        traceback.print_exc()
+        except Exception as e:
+            print(f"\n❌ Streaming error after {step_count} steps: {e}")
+            print("💡 The invoke() method worked fine, so the supervisor itself is functional.")
+            import traceback
+            traceback.print_exc()