ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter6/agent-with-event-trigger/example_with_mcp.py

@@ -0,0 +1,103 @@
+"""
+Example demonstrating the Event-Triggered Agent with MCP tools
+"""
+
+import os
+import asyncio
+from dotenv import load_dotenv
+from agent import EventTriggeredAgent, SystemHintConfig, resolve_provider_and_key
+from event_types import Event, EventType
+
+# Load environment variables
+load_dotenv()
+
+
+async def main():
+    """Main example function"""
+    print("=" * 80)
+    print("Event-Triggered Agent with MCP Tools Example")
+    print("=" * 80)
+    print()
+    
+    # Get API credentials
+    provider = os.getenv("LLM_PROVIDER", "kimi")
+    provider, api_key = resolve_provider_and_key(provider)
+    
+    if not api_key:
+        print("❌ Please set the provider API key in your .env file (DASHSCOPE_API_KEY for dashscope/qwen/bailian)")
+        return
+    
+    # Create agent configuration
+    config = SystemHintConfig(
+        enable_timestamps=True,
+        enable_tool_counter=True,
+        enable_todo_list=True,
+        enable_detailed_errors=True,
+        enable_system_state=True,
+        save_trajectory=True,
+        trajectory_file="example_trajectory.json",
+        use_mcp_servers=True  # Enable MCP servers
+    )
+    
+    # Initialize agent
+    print("Initializing agent...")
+    agent = EventTriggeredAgent(
+        api_key=api_key,
+        provider=provider,
+        config=config,
+        verbose=True
+    )
+    
+    # Load MCP tools
+    print("\nLoading MCP tools...")
+    await agent.load_mcp_tools()
+    
+    print("\n" + "=" * 80)
+    print("Testing Event Processing")
+    print("=" * 80)
+    print()
+    
+    # Create a test event
+    event = Event(
+        event_type=EventType.WEB_MESSAGE,
+        content="Search the web for 'Python async programming best practices' and summarize the top 3 results.",
+        metadata={
+            "source": "web_interface",
+            "user_id": "demo_user",
+            "session_id": "test_session_001"
+        }
+    )
+    
+    # Handle the event
+    try:
+        result = agent.handle_event(event, max_iterations=15)
+        
+        print("\n" + "=" * 80)
+        print("Result Summary")
+        print("=" * 80)
+        print(f"Success: {result['success']}")
+        print(f"Iterations: {result['iterations']}")
+        print(f"Tool Calls: {len(result['tool_calls'])}")
+        
+        if result.get('final_answer'):
+            print(f"\nFinal Answer:\n{result['final_answer']}")
+        
+        if result.get('trajectory_file'):
+            print(f"\nTrajectory saved to: {result['trajectory_file']}")
+            
+    except Exception as e:
+        print(f"\n❌ Error processing event: {e}")
+        import traceback
+        traceback.print_exc()
+    finally:
+        # Cleanup MCP connections
+        print("\nCleaning up MCP connections...")
+        await agent.mcp_manager.disconnect_all()
+        print("✅ Cleanup complete")
+
+
+if __name__ == "__main__":
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        print("\n\n⚠️  Interrupted by user")