Remove previous query_or_respond node/function and conditional edge

app.py

@@ -71,14 +71,6 @@ def retrieve(query: str):
     )
     return serialized, retrieved_docs
 
-
-def query_or_respond(state: MessagesState):
-    """Generate tool call for retrieval or respond."""
-    llm_with_tools = llm.bind_tools([retrieve])
-    response = llm_with_tools.invoke(state["messages"], config={"callbacks": [langfuse_handler]})
-    # MessagesState appends messages to state instead of overwriting
-    return {"messages": [response]} 
-
 def trigger_ai_message_with_tool_call(state: MessagesState) -> AIMessage:
     """
     Takes the last user message from the state and returns an AIMessage
@@ -118,17 +110,9 @@ graph_builder = StateGraph(MessagesState)
 tools = ToolNode([retrieve])
 memory = MemorySaver()
 
-graph_builder.add_node(query_or_respond)
 graph_builder.add_node(tools)
 graph_builder.add_node(generate)
 
-graph_builder.set_entry_point("query_or_respond")
-
-graph_builder.add_conditional_edges(
-    "query_or_respond",
-    tools_condition,
-    {END: END, "tools": "tools"},
-)
 
 graph_builder.add_edge("tools", "generate")
 graph_builder.add_edge("generate", END)