Update base_agent.py

base_agent.py

@@ -59,28 +59,22 @@ class BaseAgent:
             print(f"Error calling LLM: {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()
+    def ask_james(question) -> str:
         
-        # 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]
+        system_prompt = """
+        You are a general AI Assistant.
+        You must provide only the exact answer requested by the user, in the correct type and minimal format, based purely on the provided information.
+
+        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.
+        If you cannot find the answer based on the provided information, reply only with "Unknown".
         
-        return text.strip()
+        You must not explain your answer, restate the question, or add any commentary.
+
+        Output must be a single line, without punctuation or extra whitespace.
+        """
+        final_answer = self.llm_call(system_prompt, question, temperature=0.0)
+
+        return final_answer