Update app.py

app.py

@@ -16,8 +16,9 @@ from langchain_community.tools import TavilySearchResults
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-SYSTEM_MESSAGE = """You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template:[YOUR FINAL ANSWER]. 
+SYSTEM_MESSAGE = """You are a general AI assistant. I will ask you a question. Make 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. 
+Please do not include any thinking process inside your final response. Only respond in the form of "Final Answer: [YOUR FINAL ANSWER]" 
 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.
@@ -37,7 +38,7 @@ class BasicAgent:
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
         print(f"Agent returning fixed answer: {answer}")
-        return answer
+        return answer[14:]
 
 @tool    
 def search_tavily(state):