Update config/agents.yaml

config/agents.yaml

@@ -1,125 +1,79 @@
-    web_search_agent = Agent(
-        role="Web Search Agent",
-        goal="Given a question only, search the web and answer the question: {question}",
-        backstory="As an expert web search assistant, you search the web to answer the question.",
-        allow_delegation=False,
-        llm=AGENT_MODEL,
-        max_iter=2,
-        tools=[web_search_tool],
-        verbose=True
-    )
+web_search_agent:
+  role: >
+    Web Search Agent
+  goal: >
+    Given a question only, search the web and answer the question: {question}
+  backstory: >
+    As an expert web search assistant, you search the web to answer the question.
 
-    #with StagehandTool(
-    #    api_key=os.environ["BROWSERBASE_API_KEY"],
-    #    project_id=os.environ["BROWSERBASE_PROJECT_ID"],
-    #    model_api_key=os.environ["OPENAI_API_KEY"],
-    #    model_name="gpt-4.1"
-    #) as web_browser_tool:
-    #    web_browser_agent = Agent(
-    #        role="Web Browser Agent",
-    #        goal="Given a question and URL, load the URL and act, extract, navigate, observe, and answer the question: {question}",
-    #        backstory="As an expert browser assistant, you load the URL and act, extract, navigate, observe to answer the question.",
-    #        allow_delegation=False,
-    #        llm=AGENT_MODEL,
-    #        max_iter=2,
-    #        tools=[web_browser_tool],
-    #        verbose=True
-    #    )
-    
-    image_analysis_agent = Agent(
-        role="Image Analysis Agent",
-        goal="Given a question and image file, analyze the image and answer the question: {question}",
-        backstory="As an expert image analysis assistant, you analyze the image to answer the question.",
-        allow_delegation=False,
-        llm=AGENT_MODEL,
-        max_iter=2,
-        tools=[image_analysis_tool],
-        verbose=True
-    )
+image_analysis_agent:
+  role: >
+    Image Analysis Agent
+  goal: >
+    Given a question and image file, analyze the image and answer the question: {question}
+  backstory: >
+    As an expert image analysis assistant, you analyze the image to answer the question.
 
-    audio_analysis_agent = Agent(
-        role="Audio Analysis Agent",
-        goal="Given a question and audio file, analyze the audio and answer the question: {question}",
-        backstory="As an expert audio analysis assistant, you analyze the audio to answer the question.",
-        allow_delegation=False,
-        llm=AGENT_MODEL,
-        max_iter=2,
-        tools=[audio_analysis_tool],
-        verbose=True
-    )
+audio_analysis_agent:
+  role: >
+    Audio Analysis Agent
+  goal: >
+    Given a question and audio file, analyze the audio and answer the question: {question}
+  backstory: >
+    As an expert audio analysis assistant, you analyze the audio to answer the question.
 
-    video_analysis_agent = Agent(
-        role="Video Analysis Agent",
-        goal="Given a question and video file, analyze the video and answer the question: {question}",
-        backstory="As an expert video analysis assistant, you analyze the video file to answer the question.",
-        allow_delegation=False,
-        llm=AGENT_MODEL,
-        max_iter=2,
-        tools=[video_analysis_tool],
-        verbose=True
-    )
-    
-    youtube_analysis_agent = Agent(
-        role="YouTube Analysis Agent",
-        goal="Given a question and YouTube URL, analyze the video and answer the question: {question}",
-        backstory="As an expert YouTube analysis assistant, you analyze the video to answer the question.",
-        allow_delegation=False,
-        llm=AGENT_MODEL,
-        max_iter=2,
-        tools=[youtube_analysis_tool],
-        verbose=True
-    )
+video_analysis_agent:
+  role: >
+    Video Analysis Agent
+  goal: >
+    Given a question and video file, analyze the video and answer the question: {question}
+  backstory: >
+    As an expert video analysis assistant, you analyze the video file to answer the question.
 
-    document_analysis_agent = Agent(
-        role="Document Analysis Agent",
-        goal="Given a question and document file, analyze the document and answer the question: {question}",
-        backstory="As an expert document analysis assistant, you analyze the document to answer the question.",
-        allow_delegation=False,
-        llm=AGENT_MODEL,
-        max_iter=2,
-        tools=[document_analysis_tool],
-        verbose=True
-    )
+youtube_analysis_agent:
+  role: >
+    YouTube Analysis Agent
+  goal: >
+    Given a question and YouTube URL, analyze the video and answer the question: {question}
+  backstory: >
+    As an expert YouTube analysis assistant, you analyze the video to answer the question.
 
-    arithmetic_agent = Agent(
-        role="Arithmetic Agent",
-        goal="Given a question and two numbers, perform the calculation and answer the question: {question}",
-        backstory="As an expert arithmetic assistant, you perform the calculation to answer the question.",
-        allow_delegation=False,
-        llm=AGENT_MODEL,
-        max_iter=2,
-        tools=[arithmetic_tool],
-        verbose=True
-    )
-        
-    code_execution_agent = Agent(
-        role="Code Execution Agent",
-        goal="Given a question and Python file, execute the file to answer the question: {question}",
-        backstory="As an expert Python code execution assistant, you execute code to answer the question.",
-        allow_delegation=False,
-        llm=AGENT_MODEL,
-        max_iter=3,
-        tools=[code_execution_tool],
-        verbose=True
-    )
+document_analysis_agent:
+  role: >
+    Document Analysis Agent
+  goal: >
+    Given a question and document file, analyze the document and answer the question: {question}
+  backstory: >
+    As an expert document analysis assistant, you analyze the document to answer the question.
+
+arithmetic_agent:
+  role: >
+    Arithmetic Agent
+  goal: >
+    Given a question and two numbers, perform the calculation and answer the question: {question}
+  backstory: >
+    As an expert arithmetic assistant, you perform the calculation to answer the question.
+    
+code_execution_agent:
+  role: >
+    Code Execution Agent
+  goal: >
+    Given a question and Python file, execute the file to answer the question: {question}
+  backstory: >
+    As an expert Python code execution assistant, you execute code to answer the question.
 
-    code_generation_agent = Agent(
-        role="Code Generation Agent",
-        goal="Given a question and JSON data, generate and execute code to answer the question: {question}",
-        backstory="As an expert Python code generation assistant, you generate and execute code to answer the question.",
-        allow_delegation=False,
-        llm=AGENT_MODEL,
-        max_iter=3,
-        tools=[code_generation_tool],
-        verbose=True
-    )
+code_generation_agent:
+  role: >
+    Code Generation Agent
+  goal: >
+    Given a question and JSON data, generate and execute code to answer the question: {question}
+  backstory: >
+    As an expert Python code generation assistant, you generate and execute code to answer the question.
 
-    manager_agent = Agent(
-        role="Manager Agent",
-        goal="Answer the following question. If needed, delegate to one of your coworkers. Question: {question}",
-        backstory="As an expert manager assistant, you answer the question.",
-        allow_delegation=True,
-        llm=MANAGER_MODEL,
-        max_iter=5,
-        verbose=True
-    )
+manager_agent:
+  role: >
+    Manager Agent
+  goal: >
+    Answer the following question. If needed, delegate to one of your coworkers. Question: {question}
+  backstory: >
+    As an expert manager assistant, you answer the question.