Update agent.py

agent.py

@@ -43,21 +43,24 @@ class BasicAgent():
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        response = self.agent.invoke({"messages":question})
-        answer = response['messages'][-1].content
-        print(f"Agent returning answer: {answer}")
-        return answer
-
-    def assistant(self,state: AgentState):
-        prompt = """Please reason step by step. 
-        
+        agent_prompt = f"""        
         When providing the final answer, be as concise as possible, provide only the 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
+
+        Now:
+        QUESTION: {question}
+        FINAL ANSWER: """
         
-        """
+        messages=[HumanMessage(content=agent_prompt)]
+        response = self.agent.invoke({"messages":messages})
+        answer = response['messages'][-1].content
+        print(f"Agent returning answer: {answer}")
+        return answer
+
+    def assistant(self, state: AgentState):
         return {
-            "messages": [self.chat_with_tools.invoke([prompt] + state["messages"])],
+            "messages": [self.chat_with_tools.invoke(state["messages"])],
         }