Add MainAgent class

agent.py

@@ -0,0 +1,51 @@
+from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, VisitWebpageTool, AmazonBedrockServerModel
+
+from tools import AudioTranscriber, txt_reader, pdf_reader, excel_reader, math_calculator
+
+MAIN_PROMPT = """
+You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template:
+
+[YOUR FINAL ANSWER]
+
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. 
+If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 
+If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. 
+If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+
+Each question will contain the question, the task ID, and if available, a file name.
+"""
+
+
+class MainAgent:
+    """Main agent that orchestrates file handling and web search tasks."""
+
+    def __init__(self):
+        model = AmazonBedrockServerModel(model_id="anthropic.claude-3-5-sonnet-20241022-v2:0")
+        file_handler_agent = ToolCallingAgent(
+            name="FileHandlerAgent",
+            description="Handles file-related tasks such as reading Excel, text, PDF files, etc.",
+            tools=[
+                AudioTranscriber(),
+                txt_reader,
+                pdf_reader,
+                excel_reader,
+            ],
+            max_steps=3,
+            model=model,
+        )
+        self.agent = ToolCallingAgent(
+            name="web_search_agent",
+            description="An agent that can search the web, visit webpages, and perform calculations.",
+            managed_agents=[file_handler_agent],
+            tools=[DuckDuckGoSearchTool(), VisitWebpageTool(), math_calculator],
+            max_steps=5,
+            model=model,
+        )
+
+    def run(self, question: str, task_id: str, file_name: str | None) -> str:
+        """Run the agent with the provided question, task ID, and optional file name."""
+        query = f"""q: {question}
+task_id: {task_id}
+{'file_name: ' + file_name if file_name else ''}
+"""
+        return self.agent.run(query)