Update multiagents.py

multiagents.py

@@ -1,6 +1,6 @@
+
 # a multi agent proposal to solve HF agent course final assignment
 import os
-import textwrap
 import dotenv
 from smolagents import CodeAgent
 from smolagents import OpenAIServerModel
@@ -28,15 +28,15 @@ vllm_model = OpenAIServerModel(
 )
 
 openai_41nano_model = OpenAIServerModel(
-    model_id="llama-3.3-70b-versatile",
-    api_base="https://api.groq.com/openai/v1",
-    api_key=os.environ["GROQ_API_KEY"],
+    model_id="gpt-4.1-nano",
+    api_base="https://api.openai.com/v1",
+    api_key=os.environ["OPENAI_API_KEY"],
 )
 
 openai_41mini_model = OpenAIServerModel(
-    model_id="llama-3.3-70b-versatile",
-    api_base="https://api.groq.com/openai/v1",
-    api_key=os.environ["GROQ_API_KEY"],
+    model_id="gpt-3.5-turbo",
+    api_base="https://api.openai.com/v1",
+    api_key=os.environ["OPENAI_API_KEY"],
 )
 
 
@@ -81,43 +81,44 @@ audiovideo_agent = CodeAgent(
     max_steps=7,
 )
 
-# Try initializing the manager agent with a system_prompt parameter instead of handling it in the MultiAgent class
+
+
 manager_agent = CodeAgent(
     model=openai_41mini_model,
-    tools=[PythonInterpreterTool()],
-    managed_agents=[web_agent, audiovideo_agent],
-    additional_authorized_imports=["pandas", "numpy", "bs4"],
+    tools=[ PythonInterpreterTool()],
+    managed_agents=[web_agent, audiovideo_agent],    
+    additional_authorized_imports=["pandas", "numpy","bs4"],
     planning_interval=5,
     verbosity_level=2,
     final_answer_checks=[check_final_answer],
     max_steps=15,
     name="manager_agent",
     description="A manager agent that coordinates the work of other agents to answer questions.",
-    # Add system prompt here if the CodeAgent supports it
-    system_prompt="You are the top agent of a multi-agent system that can answer questions by coordinating the work of other agents. You will receive a question and you will decide which agent to use to answer it. You can use: web_agent to search the web and fetch page content, audiovideo_agent for video/audio file information. You can also use your own knowledge. Think step by step and don't skip steps."
 )
 
 class MultiAgent:
     def __init__(self):
-        print("MultiAgent initialized.")
+        print("BasicAgent initialized.")
 
     def __call__(self, question: str) -> str:
-        mylog(self.__class__.__name__, question)
+        mylog(self.__class__.__name__, question)        
 
         try:
-            if not isinstance(question, str):
-                question = str(question)
+            prefix = """You are the top agent of a multi-agent system that can answer questions by coordinating the work of other agents.
+            You will receive a question and you will decide which agent to use to answer it.
+            You can use the web_agent to search the web for information and for fetching the content of a web page, or the audiovideo_agent to extract information from video or audio files.
+            You can also use your own knowledge to answer the question.
+            You need to respect the output format that is given to you.
+            Finding the correct answer to the question need reasoning and plannig, read the question carrefully, think step by step and do not skip any steps.
+            """
 
-            # Simplify the prompt - let the agent handle the system message internally
-            # Just add the output format requirements to the question
-            enhanced_question = f"""{question.strip()}
+            question = prefix + "\nTHE QUESTION:\n" + question + '\n' + myprompts.output_format
 
-Please provide your answer following this format:
-{myprompts.output_format.strip() if hasattr(myprompts, 'output_format') else 'Provide a clear, concise answer.'}"""
+            fixed_answer = ""
 
-            # Pass directly to the manager agent without complex system message formatting
-            return manager_agent.run(enhanced_question)
+            fixed_answer = manager_agent.run(question)
             
+            return fixed_answer
         except Exception as e:
             error = f"An error occurred while processing the question: {e}"
             print(error)
@@ -132,4 +133,5 @@ What was the actual enrollment of the Malko competition in 2023?
 """
     agent = MultiAgent()
     answer = agent(question)
-    print(f"Answer: {answer}")
+    print(f"Answer: {answer}")
+