Update app.py

app.py

@@ -40,8 +40,8 @@ class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
 
 def assistant(state: AgentState):
-    if len(state["messages"]) > 3:
-        return {"messages": chat_with_tools.invoke([sys_msg] + state["messages"][-2:])}
+    if len(state["messages"]) > 7:
+        return {"messages": chat_with_tools.invoke([sys_msg] + state["messages"][-6:])}
     return {"messages": chat_with_tools.invoke([sys_msg] + state["messages"])}
 
 ## The graph