Add conditional edges for chat flow control

- Introduced a `should_continue` function to determine whether to proceed to the next node based on the response state, enhancing the decision-making process in the chat flow.
- Replaced direct edges with conditional edges for `intent_update`, `property_search`, and `exit_check`, allowing for more dynamic control of the conversation based on user responses.
- These changes aim to improve the responsiveness and adaptability of the AI chat system, ensuring a smoother user experience by managing the flow of conversation more effectively.

ai_chat.py

@@ -454,11 +454,26 @@ graph.add_node("property_search", RunnableLambda(extract_and_search_properties))
 graph.add_node("exit_check", RunnableLambda(detect_end_chat))
 graph.add_node("chat", RunnableLambda(chat_with_session_memory))
 graph.set_entry_point("persona_update")
+
+# Add conditional edges - if a node returns a response, go to END
+def should_continue(state):
+    """Check if we should continue to the next node or end"""
+    return state.get("response") is None
+
 graph.add_edge("persona_update", "classify_intent")
 graph.add_edge("classify_intent", "intent_update")
-graph.add_edge("intent_update", "property_search")
-graph.add_edge("property_search", "exit_check")
-graph.add_edge("exit_check", "chat")
+graph.add_conditional_edges("intent_update", should_continue, {
+    True: "property_search",
+    False: END
+})
+graph.add_conditional_edges("property_search", should_continue, {
+    True: "exit_check",
+    False: END
+})
+graph.add_conditional_edges("exit_check", should_continue, {
+    True: "chat",
+    False: END
+})
 graph.add_edge("chat", END)
 chat_graph = graph.compile()