Create graph_runner.py

graph_runner.py

@@ -0,0 +1,22 @@
+from langgraph.graph import StateGraph, START, END
+from state import AgentState
+from coder import coder
+from reviewer import reviewer
+from saver import saver
+
+def run_agent(prompt: str) -> str:
+    graph = StateGraph(AgentState)
+
+    graph.add_node("coder", coder)
+    graph.add_node("reviewer", reviewer)
+    graph.add_node("saver", saver)
+
+    graph.add_edge(START, "coder")
+    graph.add_edge("coder", "reviewer")
+    graph.add_edge("reviewer", "saver")
+    graph.add_edge("saver", END)
+
+    app = graph.compile()
+
+    result = app.invoke({"request": prompt})
+    return result["code"]