maybe fixed nodes?

graph.py

@@ -4,7 +4,7 @@ import os
 from langchain_core.messages import HumanMessage, SystemMessage
 
 from langchain_openai import ChatOpenAI
-from langgraph.graph import START, MessagesState, StateGraph
+from langgraph.graph import START, MessagesState, StateGraph, END
 from langgraph.prebuilt import ToolNode, tools_condition
 
 from tools import TOOLS
@@ -41,13 +41,10 @@ def build_graph():
     builder.add_edge(START, "assistant")
     builder.add_conditional_edges(
         "assistant",
-        tools_condition,
-        {
-            "tools": "tools",
-            "final": "END"
-        }
+        tools_condition
     )
     builder.add_edge("tools", "assistant")
+    builder.add_edge("assistant", END)
 
 
     # Compile graph