Updated system prompt

app.py

@@ -2,6 +2,7 @@ import os
 import gradio as gr
 import requests
 import inspect
+import yaml
 import pandas as pd
 from smolagents import OpenAIServerModel, ToolCallingAgent, HfApiModel, DuckDuckGoSearchTool
 
@@ -22,7 +23,19 @@ class BasicAgent:
             api_base="https://api.openai.com/v1",
             api_key=os.environ["OPENAI_API_KEY"],
         )
-        agent = ToolCallingAgent(tools=[DuckDuckGoSearchTool()], model=model)
+        
+        with open("prompts.yaml", "r") as stream:
+            prompt_templates = yaml.safe_load(stream)
+
+        agent = ToolCallingAgent(
+            name="Jeff's GAIA L1 Agent",
+            verbosity_level=1,
+            tools=[DuckDuckGoSearchTool()], 
+            add_base_tools=True,
+            max_steps=5,
+            model=model,
+            prompt_templates=prompt_templates)
+        
         self.agent = agent
         print(f"Agent initialized with model ID: {os.environ['MODEL_ID']}")
         print(f"Agent initialized with tools: {self.agent.tools}")