Update research_agent.py

research_agent.py

@@ -39,14 +39,14 @@ class ResearchAgent(BaseAgent):
         Provide a concise, accurate answer based only on the provided information.
         You must provide only the exact answer requested by the user, in the correct type and minimal format, based purely on the provided information.
 
-        - If the question asks for a number, output only the number (as an integer).
-        - If the question asks for a name, output only the name (as plain text).
-        - If the question asks for a list, output only the list items (one per line, no explanations).
-        - If you cannot find the answer based on the provided information, reply only with "Unknown".
+        Finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
 
-        You must not explain your answer, restate the question, or add any commentary.
+        YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. 
+        If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 
+        If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. 
+        If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
 
-        Output must be a single line, without punctuation or extra whitespace.
+        If you cannot find the answer based on the provided information, reply only with "Unknown".
         """
         
         user_prompt = f"""
@@ -69,14 +69,12 @@ class ResearchAgent(BaseAgent):
 
             You must provide only the exact answer requested by the user, in the correct type and minimal format, based purely on the provided information.
 
-            - If the question asks for a number, output only the number (as an integer).
-            - If the question asks for a name, output only the name (as plain text).
-            - If the question asks for a list, output only the list items (one per line, no explanations).
-            - If you cannot find the answer based on the provided information, reply only with "Unknown".
-
-            You must not explain your answer, restate the question, or add any commentary.
-
-            Output must be a single line, without punctuation or extra whitespace.
+            Finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
+    
+            YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. 
+            If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 
+            If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. 
+            If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
             """
             final_answer = self.llm_call(system_prompt, question, temperature=0.0)