Shangong/HF_tutorial-AlfredAgent

1import yaml2import os3from smolagents import GradioUI, CodeAgent, LiteLLMModel4 5# Get current directory path6CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))7 8from tools.web_search import DuckDuckGoSearchTool as WebSearch9from tools.visit_webpage import VisitWebpageTool as VisitWebpage10from tools.suggest_menu import SimpleTool as SuggestMenu11from tools.catering_service_tool import SimpleTool as CateringServiceTool12from tools.superhero_party_theme_generator import SuperheroPartyThemeTool as SuperheroPartyThemeGenerator13from tools.final_answer import FinalAnswerTool as FinalAnswer14 15 16 17model = LiteLLMModel(18max_tokens=8192,19model_id='gemini/gemini-2.0-flash',20api_base=None,21)22 23web_search = WebSearch()24visit_webpage = VisitWebpage()25suggest_menu = SuggestMenu()26catering_service_tool = CateringServiceTool()27superhero_party_theme_generator = SuperheroPartyThemeGenerator()28final_answer = FinalAnswer()29 30 31with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:32    prompt_templates = yaml.safe_load(stream)33 34agent = CodeAgent(35    model=model,36    tools=[web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator],37    managed_agents=[],38    max_steps=10,39    verbosity_level=2,40    grammar=None,41    planning_interval=None,42    name=None,43    description=None,44    executor_type='local',45    executor_kwargs={},46    max_print_outputs_length=None,47    prompt_templates=prompt_templates48)49if __name__ == "__main__":50    GradioUI(agent).launch()51