Update agent.py

agent.py

@@ -47,8 +47,13 @@ class BasicAgent():
         return answer
 
     def assistant(state: AgentState):
+        prompt = """When providing the final answer, be as concise as possible, provide only the required final answer. For example:
+        QUESTION: What was the actual enrollment count of the clinical trial on H. pylori in acne vulgaris patients from Jan-May 2018 as listed on the NIH website?
+        FINAL ANSWER: 90
+        
+        """
         return {
-            "messages": [chat_with_tools.invoke(state["messages"])],
+            "messages": [chat_with_tools.invoke(prompt + state["messages"])],
         }