Spaces:
Running
Running
File size: 521 Bytes
f3793b5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | """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()
|