| |
| |
|
|
|
|
| export type NodeName =
|
| | 'gatekeeper'
|
| | 'planner'
|
| | 'tool_executor'
|
| | 'auditor'
|
| | 'synthesizer'
|
|
|
| export const NODE_ORDER: NodeName[] = [
|
| 'gatekeeper',
|
| 'planner',
|
| 'tool_executor',
|
| 'auditor',
|
| 'synthesizer',
|
| ]
|
|
|
| export const NODE_LABELS: Record<NodeName, string> = {
|
| gatekeeper: 'Gatekeeper',
|
| planner: 'Planner',
|
| tool_executor: 'Tool Executor',
|
| auditor: 'Auditor',
|
| synthesizer: 'Synthesizer',
|
| }
|
|
|
| export const TOOL_LABELS: Record<string, string> = {
|
| librarian_rag_tool: 'Librarian',
|
| analyst_sql_tool: 'SQL Analyst',
|
| analyst_trend_tool: 'Trend Analyst',
|
| scout_web_search_tool: 'Scout',
|
| FINISH: 'Finish',
|
| }
|
|
|
| interface BaseEvent {
|
| seq: number
|
| ts: number
|
| }
|
|
|
| export interface PlanStep {
|
| tool_name: string
|
| tool_input: string
|
| }
|
|
|
| export interface RunStartEvent extends BaseEvent {
|
| type: 'run_start'
|
| run_id: string
|
| conversation_id: string
|
| query: string
|
| }
|
|
|
| export interface NodeStartEvent extends BaseEvent {
|
| type: 'node_start'
|
| node: NodeName
|
| attempt: number
|
| }
|
|
|
| export interface NodeEndEvent extends BaseEvent {
|
| type: 'node_end'
|
| node: NodeName
|
| attempt: number
|
| duration_ms: number
|
| }
|
|
|
| export interface PlanEvent extends BaseEvent {
|
| type: 'plan'
|
| attempt: number
|
| steps: PlanStep[]
|
| }
|
|
|
| export interface ToolStartEvent extends BaseEvent {
|
| type: 'tool_start'
|
| tool_name: string
|
| tool_input: string
|
| }
|
|
|
| export interface ToolEndEvent extends BaseEvent {
|
| type: 'tool_end'
|
| tool_name: string
|
| duration_ms: number
|
| ok: boolean
|
| preview: string | null
|
| result_count: number | null
|
| error: string | null
|
| }
|
|
|
| export interface AuditEvent extends BaseEvent {
|
| type: 'audit'
|
| attempt: number
|
| confidence_score: number
|
| is_relevant: boolean
|
| reasoning: string
|
| action: 'accept' | 'replan'
|
| }
|
|
|
| export interface ClarificationEvent extends BaseEvent {
|
| type: 'clarification'
|
| question: string
|
| }
|
|
|
| export interface TokenEvent extends BaseEvent {
|
| type: 'token'
|
| text: string
|
| }
|
|
|
| export interface RunEndEvent extends BaseEvent {
|
| type: 'run_end'
|
| run_id: string
|
| duration_ms: number
|
| node_timings_ms: Record<string, number>
|
| llm_calls: number
|
| output_tokens: number
|
| replans: number
|
| }
|
|
|
| export interface ErrorEvent extends BaseEvent {
|
| type: 'error'
|
| code: string
|
| message: string
|
| retryable: boolean
|
| }
|
|
|
| export interface HeartbeatEvent extends BaseEvent {
|
| type: 'heartbeat'
|
| }
|
|
|
| export type AgentEvent =
|
| | RunStartEvent
|
| | NodeStartEvent
|
| | NodeEndEvent
|
| | PlanEvent
|
| | ToolStartEvent
|
| | ToolEndEvent
|
| | AuditEvent
|
| | ClarificationEvent
|
| | TokenEvent
|
| | RunEndEvent
|
| | ErrorEvent
|
| | HeartbeatEvent
|
|
|
