testing models

agents.py

@@ -6,7 +6,7 @@ import prompts
 
 MANAGER_MODEL = "deepseek-ai/DeepSeek-R1"
 AGENT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
-FINAL_ANSWER_MODEL = "deepseek-ai/DeepSeek-R1" # OpenAIServerModel
+FINAL_ANSWER_MODEL = "gpt-4o" # OpenAIServerModel
 WEB_SEARCH_MODEL        = "Qwen/Qwen2.5-Coder-32B-Instruct"
 IMAGE_ANALYSIS_MODEL    = "HuggingFaceM4/idefics2-8b"
 AUDIO_ANALYSIS_MODEL    = "Qwen/Qwen2.5-Coder-32B-Instruct"
@@ -105,7 +105,7 @@ def create_final_answer_agent(message):
     return CodeAgent(
         name="final_answer_agent",
         description="Given a question and an initial answer, return the final refined answer following strict formatting rules.",
-        model=InferenceClientModel(FINAL_ANSWER_MODEL),
+        model=OpenAIServerModel(FINAL_ANSWER_MODEL),
         max_steps=3,
         tools=[],
     )

prompts.py

@@ -3,7 +3,6 @@
 def get_web_search_prompt(message, file_path=None):
     prompt = f"""
     As an expert web search assistant, you search the web to answer the question. Your task is to search the web and provide accurate answers to the question: {message}
-    The final answer must always be in text format and no other formats are acceptable.
         """
 
     return prompt
@@ -11,7 +10,6 @@ def get_web_search_prompt(message, file_path=None):
 def get_image_analysis_prompt(message, file_path=None):
     prompt = f"""
     As an expert image analysis assistant, you analyze the image to answer the question. Given a question and image file, analyze the image and answer the question: {message}
-    The final answer must always be in text format and no other formats are acceptable.
         """
 
     return prompt
@@ -19,7 +17,6 @@ def get_image_analysis_prompt(message, file_path=None):
 def get_audio_analysis_prompt(message, file_path=None):
     prompt = f"""
     As an expert audio analysis assistant, you analyze the audio to answer the question. Given a question and audio file, analyze the audio and answer the question: {message}
-    The final answer must always be in text format and no other formats are acceptable.
         """
 
     return prompt
@@ -36,8 +33,6 @@ def get_manager_prompt(message, file_path=None):
 
         In case you cannot answer the question and there is not a good coworker, delegate to the Code Generation Agent.\n.
 
-        The final answer must always be in text format and no other formats are acceptable.
-
         Question: {message}
         """
 

requirements.txt

@@ -3,4 +3,5 @@ requests
 smolagents
 pandas
 duckduckgo-search
-markdownify
+markdownify
+smolagents[openai]