some fixes with msg history

app.py

@@ -80,11 +80,17 @@ def assistant(state: AgentState):
         {textual_description_of_tool}
         """
     )
-    user_prompt = HumanMessage(content=f"Question: {question.get('question', question)}")
-    messages = [system_prompt, user_prompt] + state.get("messages", [])
+    # Always include conversation history
+    messages = [system_prompt] + state.get("messages", [])
+    # Add the user question only if not already present
+    if not any(isinstance(m, HumanMessage) and m.content.startswith("Question:") for m in messages):
+        user_prompt = HumanMessage(content=f"Question: {question.get('question', question)}")
+        messages.append(user_prompt)
     # If tool_outputs exist, add them as context
     if state.get("tool_outputs"):
-        messages.append(HumanMessage(content=f"Tool results: {state['tool_outputs']}"))
+        tool_msg = HumanMessage(content=f"Tool results: {state['tool_outputs']}")
+        messages.append(tool_msg)
+        state.setdefault("messages", []).append(tool_msg)
     print(f"Messages sent to LLM: {messages}")
     response = llm_with_tools.invoke(messages)
     print(f"Raw LLM response: {response}")
@@ -94,6 +100,7 @@ def assistant(state: AgentState):
         print(f"Tool calls requested: {tool_calls}")
         state["tool_calls"] = tool_calls
         state["answer"] = ""  # Not final yet
+        state.setdefault("messages", []).append(AIMessage(content="Calling tool: " + str(tool_calls)))
     else:
         state["answer"] = response.content.strip()
         print(f"Model response: {state['answer']}")
@@ -121,6 +128,8 @@ def tool_node(state: AgentState):
             outputs.append(result)
     state["tool_outputs"] = outputs
     state["tool_calls"] = None  # Clear tool calls
+    # Append tool output to conversation history
+    state.setdefault("messages", []).append(HumanMessage(content=f"Tool results: {outputs}"))
     return state
 
 #building the graph