Spaces:
Sleeping
Sleeping
import gradio as gr | |
import random | |
from smolagents import GradioUI, CodeAgent, HfApiModel, LiteLLMModel | |
# Import our custom tools from their modules | |
from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool | |
from retriever import load_guest_dataset | |
#model = HfApiModel( | |
# max_tokens=2096, | |
# temperature=0, | |
# model_id='mistralai/Mistral-7B-Instruct-v0.3', | |
# custom_role_conversions=None, | |
#) | |
hf_api = HfApi() | |
secrets = hf_api.get_secrets() | |
model = LiteLLMModel( | |
model_id="gemini/gemini-2.0-flash-lite", | |
api_key=secrets.get("GEMINI_API_TOKEN") | |
) | |
# 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() | |
# Load the guest dataset and initialize the guest info tool | |
guest_info_tool = load_guest_dataset() | |
# Create Alfred with all the tools | |
alfred = CodeAgent( | |
tools=[guest_info_tool, weather_info_tool, hub_stats_tool, search_tool], | |
model=model, | |
add_base_tools=True, # Add any additional base tools | |
planning_interval=3 # Enable planning every 3 steps | |
) | |
if __name__ == "__main__": | |
GradioUI(alfred).launch() |