debug

type EdgeInfo struct {
	From      string                 `json:"from"`
	To        string                 `json:"to"`
	Condition string                 `json:"condition,omitempty"`
	Metadata  map[string]interface{} `json:"metadata"`
}

type ExecutionStep struct {
	ID            string                 `json:"id"`
	ThreadID      string                 `json:"thread_id"`
	NodeID        string                 `json:"node_id"`
	StepType      string                 `json:"step_type"` // "enter", "exit", "error"
	Timestamp     time.Time              `json:"timestamp"`
	Duration      time.Duration          `json:"duration,omitempty"`
	Input         interface{}            `json:"input,omitempty"`
	Output        interface{}            `json:"output,omitempty"`
	Error         string                 `json:"error,omitempty"`
	Metadata      map[string]interface{} `json:"metadata"`
	StateSnapshot *core.BaseState        `json:"state_snapshot,omitempty"`
}

type GraphTopology struct {
	Nodes []NodeInfo `json:"nodes"`
	Edges []EdgeInfo `json:"edges"`
}

type GraphVisualizer struct {
	// contains filtered or unexported fields
}

type NodeInfo struct {
	ID          string                 `json:"id"`
	Name        string                 `json:"name"`
	Type        string                 `json:"type"`
	Metadata    map[string]interface{} `json:"metadata"`
	IsStartNode bool                   `json:"is_start_node"`
	IsEndNode   bool                   `json:"is_end_node"`
	Position    *Position              `json:"position,omitempty"`
}

type Position struct {
	X int `json:"x"`
	Y int `json:"y"`
}

type VisualizationSubscriber interface {
	OnStepExecuted(step *ExecutionStep)
	OnGraphCompleted(threadID string, steps []ExecutionStep)
	OnError(err error)
}

type VisualizerConfig struct {
	EnableRealTimeUpdates bool   `json:"enable_real_time_updates"`
	MaxHistorySize        int    `json:"max_history_size"`
	OutputFormat          string `json:"output_format"` // "mermaid", "dot", "json"
	IncludeMetadata       bool   `json:"include_metadata"`
	IncludeTimestamps     bool   `json:"include_timestamps"`
}

type WebSocketSubscriber struct {
	SendMessage func(message interface{}) error
	// contains filtered or unexported fields
}