ship

app.py

@@ -96,9 +96,13 @@ def should_continue(state):
 # ------- Runnables ------- #
 
 # retrieve graph
-graph_builder = StateGraph(State).add_sequence([retrieve_adjusted, generate])
-graph_builder.add_edge(START, "retrieve_adjusted")
-graph = graph_builder.compile() 
+graph_builder = StateGraph(State)
+graph_builder.add_node("retrieve", retrieve_adjusted)
+graph_builder.add_node("generate", generate)
+graph_builder.add_edge(START, "retrieve")
+graph_builder.add_edge("retrieve", "generate")
+graph_builder.add_edge("generate", END)
+graph = graph_builder.compile()
 
 
 @tool