Put retriever and tools together

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import os
 from smolagents import CodeAgent, HfApiModel, LiteLLMModel
 from retriever import load_guest_dataset
-from tools import *
+from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
 
 is_running_on_space = os.environ.get('SPACE_ID') is not None
 if is_running_on_space:
@@ -13,11 +13,24 @@ else:
     api_base="http://127.0.0.1:11434",
     num_ctx=8192,)
 
-# guest_info_tool = load_guest_dataset()
+guest_info_tool = load_guest_dataset()
 
-# Create Alfred, our gala agent, with the guest info tool
-alfred = CodeAgent(tools=[hub_stats_tool, search_tool, weather_info_tool], model=model)
+# Initialize the web search tool
+search_tool = DuckDuckGoSearchTool()
+
+# Initialize the weather tool
+weather_info_tool = WeatherInfoTool()
 
+# Initialize the Hub stats tool
+hub_stats_tool = HubStatsTool()
+
+# Create Alfred, our gala agent, with the guest info tool
+alfred = CodeAgent(
+  tools=[guest_info_tool, hub_stats_tool, weather_info_tool, search_tool],
+  model=model,
+  add_base_tools=True,
+  planning_interval=3
+)
 
 def query_guest(guest_name):
   return alfred.run(f"Tell me about our guest named '{guest_name}'.")
@@ -25,6 +38,8 @@ def query_guest(guest_name):
 def ai_builder(builder):
   return alfred.run(f"What is {builder} most downloaded model on huggingface hub?")
 
+def query(query):
+  return alfred.run(query)
 
-demo = gr.Interface(fn=ai_builder, inputs="text", outputs="text")
+demo = gr.Interface(fn=query, inputs="text", outputs="text")
 demo.launch()

tools.py

@@ -1,10 +1,7 @@
 from huggingface_hub import list_models
-from smolagents import DuckDuckGoSearchTool, Tool
+from smolagents import Tool, DuckDuckGoSearchTool
 import random
 
-# Initialize the DuckDuckGo search tool
-search_tool = DuckDuckGoSearchTool()
-
 class WeatherInfoTool(Tool):
     name = "weather_info"
     description = "Fetches dummy weather information for a given location."
@@ -27,9 +24,6 @@ class WeatherInfoTool(Tool):
         data = random.choice(weather_conditions)
         return f"Weather in {location}: {data['condition']}, {data['temp_c']}°C"
 
-# Initialize the tool
-weather_info_tool = WeatherInfoTool()
-
 class HubStatsTool(Tool):
     name = "hub_stats"
     description = "Fetches the most downloaded model from a specific author on the Hugging Face Hub."
@@ -53,6 +47,3 @@ class HubStatsTool(Tool):
                 return f"No models found for author {author}."
         except Exception as e:
             return f"Error fetching models for {author}: {str(e)}"
-
-# Initialize the tool
-hub_stats_tool = HubStatsTool()