Update base_agent.py

base_agent.py

@@ -57,4 +57,30 @@ class BaseAgent:
             return response.choices[0].message.content
         except Exception as e:
             print(f"Error calling LLM: {str(e)}")
-            return f"Error: {str(e)}"
+            return f"Error: {str(e)}"
+    
+    @staticmethod
+    def extract_final_answer(text: str) -> str:
+        """
+        Extract the final answer from potentially verbose LLM output.
+        
+        Args:
+            text: The full text from the LLM
+            
+        Returns:
+            The extracted answer, removing any explanation
+        """
+        # Look for explicit markers
+        markers = ["Final Answer:", "Answer:", "Therefore, the answer is"]
+        for marker in markers:
+            if marker in text:
+                parts = text.split(marker, 1)
+                if len(parts) > 1:
+                    return parts[1].strip()
+        
+        # If no markers, return the last non-empty line
+        lines = [line.strip() for line in text.splitlines() if line.strip()]
+        if lines:
+            return lines[-1]
+        
+        return text.strip()