agent/graph.py

"""On-demand audit graph: per-message extract → judge (Gemma)."""

from __future__ import annotations

from langgraph.graph import END, START, StateGraph

from agent.nodes.audit import audit_all_messages
from agent.state import AuditState


def build_audit_graph():
    graph = StateGraph(AuditState)
    graph.add_node("audit_all_messages", audit_all_messages)
    graph.add_edge(START, "audit_all_messages")
    graph.add_edge("audit_all_messages", END)
    return graph.compile()


audit_graph = build_audit_graph()