tool update

app.py

@@ -51,6 +51,8 @@ class BasicAgent:
         # context = self.wiki_ret(question)
         context = state.get("context", "")
         question = state.get("messages", "")
+        print("question agent: ", question)
+        print("context: ", context)
         prompt = f"""
         You are a general AI assistant. I will ask you a question. 
         YOUR FINAL ANSWER should be a number OR 
@@ -71,6 +73,7 @@ class BasicAgent:
         # Just make up a task_id.
         # Call the LLM
         messages = [HumanMessage(content=prompt)]
+        print("messages: ", messages)
         response = (llm_with_tools.invoke(messages)).content
         print("response: ", response)
         # cleaned_text = re.sub(r"<think>.*?</think>", "", response.content, flags=re.DOTALL)