Implemented multi-agent system.

app.py

@@ -7,7 +7,7 @@ from retriever import load_guest_dataset
 from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
 
 # Initialize the Hugging Face model
-model = HfApiModel()
+model = HfApiModel(max_tokens=8096)
 
 # Initialize the web search tool
 search_tool = DuckDuckGoSearchTool()
@@ -26,12 +26,46 @@ collection = chroma_client.create_collection(name="guests-collection", embedding
 guest_info_tool = load_guest_dataset(collection)
 
 # Create Alfred with all the tools
-alfred = CodeAgent(
-    tools=[guest_info_tool, weather_info_tool, hub_stats_tool, search_tool], 
+search_agent = CodeAgent(
+    name="search_agent",
+    description="Browses the web to find information",
+    tools=[search_tool],
     model=model,
-    add_base_tools=False,  # Add any additional base tools
-    planning_interval=3   # Enable planning every 3 steps
+    add_base_tools=False,
+    planning_interval=3,
+    verbosity_level=2,
+    max_steps=5
+)
+hub_stats_agent = CodeAgent(
+    name="hub_stats_agent",
+    description="Browses the HF statistics",
+    tools=[hub_stats_tool],
+    model=model,
+    add_base_tools=False,
+    planning_interval=3,
+    verbosity_level=2,
+    max_steps=2
+)
+weather_agent = CodeAgent(
+    name="weather_agent",
+    description="Weather forecasting",
+    tools=[weather_info_tool],
+    model=model,
+    add_base_tools=False,
+    planning_interval=3,
+    verbosity_level=2,
+    max_steps=2
+)
+
+alfred_manager = CodeAgent(
+    tools=[guest_info_tool],
+    managed_agents=[search_agent, hub_stats_agent, weather_agent],
+    model=model,
+    add_base_tools=False,
+    planning_interval=5,
+    verbosity_level=2,
+    max_steps=15
 )
 
 if __name__ == "__main__":
-    GradioUI(alfred).launch()
+    GradioUI(alfred_manager).launch()