hooks

type ErrorEvent struct {
	RunID    string         `json:"run_id"`
	AgentID  string         `json:"agent_id"`
	Error    error          `json:"error"`
	Phase    string         `json:"phase"`
	Metadata map[string]any `json:"metadata,omitempty"`
}

type Hook interface {
	// Name 返回钩子名称
	Name() string

	// Enabled 是否启用
	Enabled() bool
}

type LLMEndEvent struct {
	RunID            string         `json:"run_id"`
	RequestID        string         `json:"request_id"`
	Model            string         `json:"model"`
	Response         any            `json:"response"`
	PromptTokens     int            `json:"prompt_tokens"`
	CompletionTokens int            `json:"completion_tokens"`
	Duration         int64          `json:"duration_ms"`
	Error            error          `json:"error,omitempty"`
	Metadata         map[string]any `json:"metadata,omitempty"`
}

type LLMHook interface {
	Hook
	// OnLLMStart LLM 调用开始
	OnLLMStart(ctx context.Context, event *LLMStartEvent) error
	// OnLLMEnd LLM 调用完成
	OnLLMEnd(ctx context.Context, event *LLMEndEvent) error
	// OnLLMStream LLM 流式输出
	OnLLMStream(ctx context.Context, event *LLMStreamEvent) error
}

type LLMStartEvent struct {
	RunID       string         `json:"run_id"`
	RequestID   string         `json:"request_id"`
	Model       string         `json:"model"`
	Provider    string         `json:"provider"`
	Messages    []any          `json:"messages"`
	Temperature float64        `json:"temperature,omitempty"`
	Metadata    map[string]any `json:"metadata,omitempty"`
}

type LLMStreamEvent struct {
	RunID      string `json:"run_id"`
	RequestID  string `json:"request_id"`
	Model      string `json:"model"`
	Content    string `json:"content"`
	ChunkIndex int    `json:"chunk_index"`
}

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

type RetrieverEndEvent struct {
	RunID     string         `json:"run_id"`
	QueryID   string         `json:"query_id"`
	Query     string         `json:"query"`
	Documents []any          `json:"documents"`
	DocCount  int            `json:"doc_count"`
	Duration  int64          `json:"duration_ms"`
	Error     error          `json:"error,omitempty"`
	Metadata  map[string]any `json:"metadata,omitempty"`
}

type RetrieverHook interface {
	Hook
	// OnRetrieverStart 检索开始
	OnRetrieverStart(ctx context.Context, event *RetrieverStartEvent) error
	// OnRetrieverEnd 检索完成
	OnRetrieverEnd(ctx context.Context, event *RetrieverEndEvent) error
}

type RetrieverStartEvent struct {
	RunID    string         `json:"run_id"`
	QueryID  string         `json:"query_id"`
	Query    string         `json:"query"`
	TopK     int            `json:"top_k"`
	Metadata map[string]any `json:"metadata,omitempty"`
}

type RunEndEvent struct {
	RunID    string         `json:"run_id"`
	AgentID  string         `json:"agent_id"`
	Output   any            `json:"output"`
	Duration int64          `json:"duration_ms"`
	Metadata map[string]any `json:"metadata,omitempty"`
}

type RunHook interface {
	Hook
	// OnStart Agent 开始执行
	OnStart(ctx context.Context, event *RunStartEvent) error
	// OnEnd Agent 执行完成
	OnEnd(ctx context.Context, event *RunEndEvent) error
	// OnError 发生错误
	OnError(ctx context.Context, event *ErrorEvent) error
}

type RunStartEvent struct {
	RunID    string         `json:"run_id"`
	AgentID  string         `json:"agent_id"`
	Input    any            `json:"input"`
	Metadata map[string]any `json:"metadata,omitempty"`
}

type RunStreamEndEvent struct {
	// RunID 运行 ID
	RunID string `json:"run_id"`

	// AgentID Agent ID
	AgentID string `json:"agent_id"`

	// ChunkCount 流式输出的 chunk 数量
	ChunkCount int `json:"chunk_count"`

	// Duration 执行耗时（毫秒）
	Duration int64 `json:"duration_ms"`

	// Error 错误（如果有）
	Error error `json:"error,omitempty"`

	// Metadata 元数据
	Metadata map[string]any `json:"metadata,omitempty"`
}

type RunStreamStartEvent struct {
	// RunID 运行 ID
	RunID string `json:"run_id"`

	// AgentID Agent ID
	AgentID string `json:"agent_id"`

	// Input 输入数据
	Input any `json:"input"`

	// IsStream 标记是否为流式模式
	IsStream bool `json:"is_stream"`

	// Metadata 元数据
	Metadata map[string]any `json:"metadata,omitempty"`
}

type StreamHook interface {
	// OnStreamStart 流式执行开始
	OnStreamStart(ctx context.Context, event *RunStreamStartEvent) error

	// OnStreamEnd 流式执行结束
	OnStreamEnd(ctx context.Context, event *RunStreamEndEvent) error
}

type Timing uint32

type TimingChecker interface {
	// Timings 返回关心的时机（位掩码）
	// 返回 TimingAll 表示关心所有时机
	// 返回 TimingNone 表示不关心任何时机（禁用）
	Timings() Timing
}

type ToolEndEvent struct {
	RunID    string         `json:"run_id"`
	ToolName string         `json:"tool_name"`
	ToolID   string         `json:"tool_id"`
	Output   any            `json:"output"`
	Duration int64          `json:"duration_ms"`
	Error    error          `json:"error,omitempty"`
	Metadata map[string]any `json:"metadata,omitempty"`
}

type ToolHook interface {
	Hook
	// OnToolStart 工具调用开始
	OnToolStart(ctx context.Context, event *ToolStartEvent) error
	// OnToolEnd 工具调用完成
	OnToolEnd(ctx context.Context, event *ToolEndEvent) error
}

type ToolStartEvent struct {
	RunID    string         `json:"run_id"`
	ToolName string         `json:"tool_name"`
	ToolID   string         `json:"tool_id"`
	Input    map[string]any `json:"input"`
	Metadata map[string]any `json:"metadata,omitempty"`
}