Update agent.py

agent.py

@@ -20,7 +20,7 @@ def calculator_tool_func(query: str):
         result = str(eval(query, {"__builtins__": {}}))
         return result
     except Exception:
-        return "Could not calculate the input expression."
+        return "I couldn't compute that expression."
 
 calculator_tool = Tool(
     name="Calculator",
@@ -72,16 +72,16 @@ def run_ninu(query: str):
     conversation = []
 
     system_prompt = """
-You are a general AI assistant. I will ask you a question. Report your thoughts, 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. 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.
+You are a smart, helpful, and curious assistant. Think step by step and use any available tools if needed. Explain your reasoning, be honest about what you know and don't know, and speak naturally in English.
 """
     conversation.append(HumanMessage(content=system_prompt.strip()))
     conversation.append(HumanMessage(content=query))
 
     response = ninu.invoke({"messages": conversation})
     for message in reversed(response["messages"]):
-        if isinstance(message.content, str) and "FINAL ANSWER:" in message.content:
-            return message.content.split("FINAL ANSWER:")[-1].strip()
-    return "FINAL ANSWER: No valid answer found."
+        if isinstance(message.content, str):
+            return message.content.strip()
+    return "No response received."
 
 # --- Example test ---
 if __name__ == "__main__":