Update agent/nodes.py

agent/nodes.py

@@ -921,11 +921,14 @@ def AnswerRefinementNode(state: AgentState) -> AgentState:
         Original Question: {question}
         Initial Answer: {initial_answer}
         Extracted Context/Data: {extracted_data if extracted_data else "No specific data was extracted, the answer was generated based on general knowledge or initial processing."}
-
-        Your task is to critically review the Initial Answer in the context of the Original Question and Extracted Context/Data.
-        Refine the Initial Answer to ensure it is accurate, directly answers the question, and strictly follows the FINAL ANSWER formatting rules.
-        If the Initial Answer seems correct and appropriately formatted, you can simply re-state it.
-        If the Initial Answer is "unknown" or an error message, try to re-evaluate the question using the available context to provide a valid answer if possible.
+        
+        Your task is to review and improve the Initial Answer ONLY if there are clear issues.
+        
+        IMPORTANT: If the Initial Answer appears reasonable and addresses the question, KEEP IT AS IS. 
+        Only change the answer if you can clearly identify one of these problems:
+        1. Formatting issues (wrong format for the expected answer type)
+        2. Clear factual errors that you can correct with the provided context
+        3. The answer doesn't actually address the question asked
 
         Strict FINAL ANSWER formatting rules:
         - A number OR
@@ -943,9 +946,12 @@ def AnswerRefinementNode(state: AgentState) -> AgentState:
         3. For comma-separated lists:
            - Apply the above rules to each element
            - Separate elements with commas only (no spaces unless part of the element)
-
-        Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
-        """
+        
+        If you cannot make a clear improvement, return the Initial Answer unchanged.
+        
+        DO NOT change a reasonable answer to "unknown" unless the Initial Answer is clearly wrong or irrelevant.
+        
+        {SYSTEM_PROMPT.strip()}"""
 
         response = client.chat.completions.create(
             model="gpt-4-turbo",