Refactor MainAgent ot use just one agent

agent.py

@@ -1,7 +1,17 @@
-from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, VisitWebpageTool, AmazonBedrockServerModel
+import os
 
-from tools import AudioTranscriber, txt_reader, pdf_reader, excel_reader, math_calculator
-from utils import BEDROCK_MODEL_ID
+from smolagents import OpenAIServerModel, ToolCallingAgent
+
+from tools import (
+    AudioTranscriber,
+    txt_reader,
+    pdf_reader,
+    excel_reader,
+    math_calculator,
+    search_engine,
+    image_transcriber,
+)
+from utils import OPENAI_MODEL_ID
 
 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:
@@ -21,24 +31,19 @@ class MainAgent:
     """Main agent that orchestrates file handling and web search tasks."""
 
     def __init__(self):
-        model = AmazonBedrockServerModel(model_id=BEDROCK_MODEL_ID)
-        file_handler_agent = ToolCallingAgent(
-            name="FileHandlerAgent",
-            description="Handles file-related tasks such as reading Excel, text, PDF files, etc.",
+        model = OpenAIServerModel(model_id=OPENAI_MODEL_ID, api_key=os.getenv("OPENAI_API_KEY"))
+        self.agent = ToolCallingAgent(
+            name="main_agent",
+            description="An agent that can search the web, visit webpages, perform calculations, and handle files.",
             tools=[
+                search_engine,
+                math_calculator,
                 AudioTranscriber(),
                 txt_reader,
                 pdf_reader,
                 excel_reader,
+                image_transcriber,
             ],
-            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,
         )

requirements.txt

@@ -31,6 +31,7 @@ jmespath==1.0.1 ; python_version >= '3.7'
 jsonpatch==1.33 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4, 3.5, 3.6'
 jsonpointer==3.0.0 ; python_version >= '3.7'
 langchain-core==0.3.61 ; python_version >= '3.9'
+langchain-openai==0.3.18
 langgraph==0.4.7
 langgraph-checkpoint==2.0.26 ; python_version >= '3.9'
 langgraph-prebuilt==0.2.1 ; python_version >= '3.9'
@@ -60,6 +61,7 @@ python-dotenv==1.1.0
 python-multipart==0.0.20 ; python_version >= '3.8'
 pytz==2025.2
 pyyaml==6.0.2 ; python_version >= '3.8'
+regex==2024.11.6 ; python_version >= '3.8'
 requests==2.32.3
 requests-toolbelt==1.0.0 ; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
 rich==14.0.0 ; python_full_version >= '3.8.0'
@@ -74,6 +76,7 @@ sniffio==1.3.1 ; python_version >= '3.7'
 soupsieve==2.7 ; python_version >= '3.8'
 starlette==0.46.2 ; sys_platform != 'emscripten'
 tenacity==9.1.2 ; python_version >= '3.9'
+tiktoken==0.9.0 ; python_version >= '3.9'
 tomlkit==0.13.2 ; python_version >= '3.8'
 tqdm==4.67.1 ; python_version >= '3.7'
 typer==0.16.0 ; sys_platform != 'emscripten'

utils.py

@@ -6,6 +6,7 @@ import requests
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 BEDROCK_MODEL_ID = "anthropic.claude-3-5-sonnet-20240620-v1:0"
+OPENAI_MODEL_ID = "o4-mini"
 
 bedrock_runtime = boto3.client("bedrock-runtime", region_name=os.getenv("AWS_REGION"))