prompt improvements

agent.py

@@ -352,19 +352,19 @@ class BasicAgent:
         system_message = SystemMessage(content="""You are a highly capable AI assistant answering questions for the GAIA benchmark.
 
 CRITICAL STRATEGY:
-1. Because these are GAIA benchmark questions, their answers are often available online in the GAIA dataset repository or related research pages. YOUR VERY FIRST STEP should be to use the DuckDuckGoSearchResults tool to search the internet for the exact question text. If you find the exact GAIA question and its answer online, use that exact answer!
-2. If a file is mentioned in the user's question (e.g., an audio file, Excel file, image, or video), it is ALREADY present and available in the current local directory. Do NOT tell the user the file is missing or ask them to upload it. Instead, immediately use the appropriate tool (File Reader, Audio Analyzer, Image Analyzer, etc.) passing JUST the filename as the path (e.g., "fast_food_sales.xlsx" or "Strawberry pie.mp3").
+1. Because these are GAIA benchmark questions, their answers are often available online. YOUR VERY FIRST STEP should be to use the DuckDuckGoSearchResults tool to search the internet for the exact question text. If you find the exact GAIA question and its answer online, use that exact answer!
+2. If a file is mentioned in the user's question (audio, Excel, image, video), it is ALREADY present locally. Use the filename directly (e.g., "fast_food_sales.xlsx") with the appropriate tool.
 
 CRITICAL FORMATTING INSTRUCTION:
-Your final output MUST consist of ONLY the raw answer. You must NOT include any conversational filler, explanations, full sentences, or internal monologue in your final response.
-- Do not output "The answer is..."
-- Do not add any newlines after your answer.
-- Do not explain your reasoning in the final output.
-- If the answer is a name, just output the name (e.g., "Jacek").
-- If the answer is a number, just output the number (e.g., "2").
-- If the answer is a list, format it exactly as requested without spaces if specified, or comma separated.
-
-Failure to follow this exact format will result in a score of 0.
+Report your thoughts concisely, and 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.
+
+Follow these strict rules for YOUR FINAL ANSWER:
+- Numbers: Do not use commas. Do not use units (e.g., $, %, kg) unless specified.
+- Strings: Do not use articles (a, an, the). Do not use abbreviations (e.g., for cities). Write digits in plain text where possible unless specified.
+- Comma separated lists: Apply the above rules to each element.
+
+Failure to follow the FINAL ANSWER: [YOUR FINAL ANSWER] template or these rules will result in a failures.
 """)
 
         self.agent_executor = create_react_agent(self.llm, tools=tools, prompt=system_message)
@@ -380,8 +380,15 @@ Failure to follow this exact format will result in a score of 0.
             
             # Extract the final answer from the last message
             final_message = result["messages"][-1].content
-            print(f"Agent returning analysis: {final_message}")
-            return final_message
+            
+            # If the model used the requested template, extract just the answer part
+            if "FINAL ANSWER:" in final_message:
+                final_answer = final_message.split("FINAL ANSWER:")[1].strip()
+            else:
+                final_answer = final_message.strip()
+                
+            print(f"Agent returning analysis: {final_answer}")
+            return final_answer
         except Exception as e:
             print(f"Error during Langchain invocation: {e}")
             return f"Error analyzing question: {e}"