src/main.py

from app.graph.workflow import build_graph

graph = build_graph()

def run_agent(prompt):
    result = graph.invoke({
        "prompt": prompt,
        "iteration": 0
    })
    return result