Update config/agents.yaml

config/agents.yaml

@@ -1 +1,125 @@
-x
+    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
+    )
+
+    #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
+    )
+
+    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
+    )
+
+    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
+    )
+
+    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
+    )
+
+    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
+    )
+
+    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
+    )
+
+    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
+    )