Witcher2603/AlfredAgent

1import yaml2import os3from smolagents import GradioUI, CodeAgent, InferenceClientModel4 5# Get current directory path6CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))7 8from tools.suggest_menu import SimpleTool as SuggestMenu9from tools.web_search import DuckDuckGoSearchTool as WebSearch10from tools.final_answer import FinalAnswerTool as FinalAnswer11 12 13 14model = InferenceClientModel(15model_id='Qwen/Qwen2.5-Coder-32B-Instruct',16)17 18suggest_menu = SuggestMenu()19web_search = WebSearch()20final_answer = FinalAnswer()21 22 23with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:24    prompt_templates = yaml.safe_load(stream)25 26agent = CodeAgent(27    model=model,28    tools=[suggest_menu, web_search],29    managed_agents=[],30    max_steps=20,31    verbosity_level=1,32    grammar=None,33    planning_interval=None,34    name=None,35    description=None,36    executor_type='local',37    executor_kwargs={},38    max_print_outputs_length=None,39    prompt_templates=prompt_templates40)41if __name__ == "__main__":42    GradioUI(agent).launch()43