Upload app.py

app.py

@@ -29,11 +29,17 @@ class BasicAgent:
         #print(f"Agent returning fixed answer: {answer}")
 
 
+        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."
+        question = f"{system_prompt} \
+        The question: \
+        {question} \
+        "
+
         # Wrap the question in a HumanMessage from langchain_core
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
-        return answer[14:]
+        return answer