Bismack/LearningBuddy

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.check_answers import SimpleTool as CheckAnswers10from tools.motivator import SimpleTool as Motivator11from tools.final_answer import FinalAnswerTool as FinalAnswer12 13from managed_agents.concept_quiz_generator.app import agent_concept_quiz_generator14 15 16model = LiteLLMModel(17max_tokens=8192,18model_id='gemini/gemini-2.0-flash',19api_key=os.environ.get("GOOGLE_API_KEY"),20)21 22web_search = WebSearch()23check_answers = CheckAnswers()24motivator = Motivator()25final_answer = FinalAnswer()26 27 28with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:29    prompt_templates = yaml.safe_load(stream)30 31agent = CodeAgent(32    model=model,33    tools=[web_search, check_answers, motivator],34    managed_agents=[agent_concept_quiz_generator],35    max_steps=10,36    name='CodeAgent',37    verbosity_level=2,38    grammar=None,39    planning_interval=4,40    description=None,41    executor_type='local',42    executor_kwargs={},43    max_print_outputs_length=None,44    prompt_templates=prompt_templates45)46if __name__ == "__main__":47    GradioUI(agent).launch()48