update

agent.py

@@ -32,9 +32,20 @@ def get_graph(llm):
 
     def call_model(state: State):
         print("\n-------------------- Agent has been called -----------------------------------\n")
-        prompt = prompt_template.invoke(state["messages"])
-        print("\nThe Prompt is: ", prompt, "\n")
-        response = llm.invoke(prompt)
+        messages = state["messages"]
+        messages.append(HumanMessage(content="Write a plan how to solve this qustion?"))
+
+
+        prompt_plan = prompt_template.invoke(messages)
+        plan = llm.invoke(prompt_plan).content
+        messages.append(AIMessage(content=plan))
+        messages.append(HumanMessage(content="Now give me the answer to the question."))
+        prompt_answer = prompt_template.invoke(messages)
+        response = llm.invoke(prompt_answer)
+
+
+        print("\nThe Prompt is: ", prompt_answer, "\n")
+
         print("Agent has made a decision: ",response.content)
         return {"messages": [response], "aggregate": ["Agent"]}