Merge branch 'main' of https://huggingface.co/spaces/Dkapsis/gaia_final_assignment

agents.py

@@ -123,7 +123,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=[],
     )

multi_agent.py

@@ -12,7 +12,6 @@ import agents
 def orchestrate(message, file_path):
     final_prompt = prompts.get_manager_prompt(message, file_path)
     initial_answer = agents.create_manager_agent(message).run(message)
-
     final_answer = agents.create_final_answer_agent(message).run(prompts.get_final_answer_prompt(message, initial_answer))
 
     return final_answer

prompts.py

@@ -33,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 a string and no other formats are acceptable.
-
         Question: {message}
         """
 
@@ -65,7 +63,7 @@ def get_final_answer_prompt(message: str, initial_answer: str):
         
         **Final answer:**
 
-        The final answer must always be a string and no other formats are acceptable.
+        The final answer must always be in text format and no other formats are acceptable.
         """
 
     return prompt

requirements.txt

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