fix when empty file_name

README.md

@@ -1,5 +1,5 @@
 ---
-title: Template Final Assignment
+title: SmolAgents submission
 emoji: 🕵🏻‍♂️
 colorFrom: indigo
 colorTo: indigo

agents/react_agent.py

@@ -110,7 +110,6 @@ class ReactAgent:
             parsed_content = file_info["parser"](content)
             user_question = question["question"]
             if file_info["type"] == "image":
-                parsed_content.save("image.png")
                 return self.agent.run(user_question, images=[parsed_content])
 
             user_question = (
@@ -120,7 +119,7 @@ class ReactAgent:
             )
             return self.agent.run(user_question)
 
-        return self.agent.run(question)
+        return self.agent.run(question["question"])
 
     def steer_system_prompt(self):
         prev_system_prompt = self.agent.system_prompt
@@ -129,3 +128,8 @@ class ReactAgent:
         gaia_answer_rules = """It is crucial that you wrap your final answer in the ```code``` block by using the `final_answer` tool or your mom will die."""
         system_prompt = prompt_prefix + gaia_answer_rules + "\n\nNow Begin!"
         self.agent.system_prompt = system_prompt
+
+if __name__ == "__main__":
+    question = {'task_id': '8e867cd7-cff9-4e6c-867a-ff5ddc2550be', 'question': 'How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia.', 'Level': '1', 'file_name': ''}
+    agent = ReactAgent()
+    response = agent(question)