middleware

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

type HumanInTheLoop struct {
	PassThrough
	// contains filtered or unexported fields
}

type HumanInTheLoopConfig struct {
	// BeforeToolValidator, when set, lets a human approve/reject a tool call.
	// If it returns false, the tool observation is set to a rejection message.
	BeforeToolValidator func(action string, input string) (bool, string)
	// FallbackObservation is used when a tool call is rejected.
	FallbackObservation string
}

type Middleware interface {
	// BeforeModel runs before the model is invoked.
	BeforeModel(ctx context.Context, state *State, req *ModelRequest, next Next)
	// AfterModel runs after the model returns.
	AfterModel(ctx context.Context, state *State, req *ModelRequest, resp *ModelResponse, next Next)
	// BeforeTool runs before a tool is called.
	BeforeTool(ctx context.Context, state *State, req *ToolRequest, next Next)
	// AfterTool runs after a tool returns.
	AfterTool(ctx context.Context, state *State, req *ToolRequest, resp *ToolResponse, next Next)
}

type MiddlewareConfig struct {
	BeforeModel func(ctx context.Context, state *State, req *ModelRequest, next Next)
	AfterModel  func(ctx context.Context, state *State, req *ModelRequest, resp *ModelResponse, next Next)
	BeforeTool  func(ctx context.Context, state *State, req *ToolRequest, next Next)
	AfterTool   func(ctx context.Context, state *State, req *ToolRequest, resp *ToolResponse, next Next)
}

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

type ModelCallLimit struct {
	PassThrough
	// contains filtered or unexported fields
}

type ModelRequest struct {
	// Inputs are the inputs passed to the model, including the scratchpad.
	Inputs map[string]any
	// Options are the chain call options.
	Options []any
	// Step is the index of the current iteration.
	Step int
	// Skip, when set by a before-hook, prevents the underlying plan from
	// running. The after-hooks still execute so they can record the outcome.
	Skip bool
}

type ModelResponse struct {
	// Actions are the actions the model decided to take.
	Actions []schema.AgentAction
	// Finish is set when the model decided to finish.
	Finish *schema.AgentFinish
	// Err is set when the model invocation failed.
	Err error
}

type ModelRetry struct {
	PassThrough
	// contains filtered or unexported fields
}

type Next func(ctx context.Context)

type PII struct {
	PassThrough
	// contains filtered or unexported fields
}

type PassThrough struct{}

type RetryConfig struct {
	// MaxAttempts is the maximum number of model attempts before giving up.
	MaxAttempts int
	// ShouldRetry reports whether a given error should be retried.
	ShouldRetry func(error) bool
	// OnRetry is invoked before each retry with the attempt index and error.
	OnRetry func(attempt int, err error)
}

type State struct {
	// Inputs are the original agent inputs.
	Inputs map[string]string
	// Steps are the intermediate steps accumulated so far.
	Steps []schema.AgentStep
	// Output is the final output, set when the agent finishes.
	Output map[string]any
}

type Summarization struct {
	PassThrough
	// MaxSteps is the maximum number of steps to retain before summarizing.
	MaxSteps int
	// SummarizeFunc, when set, is called to compact the steps.
	SummarizeFunc func(steps []schema.AgentStep) []schema.AgentStep
}

type TodoList struct {
	PassThrough
	// contains filtered or unexported fields
}

type TodoListConfig struct {
	// TodoList is the list of tasks the agent should complete.
	TodoList []string
	// InjectSystemMessage, when set, is a function that adds the todo list to
	// the model inputs.
	InjectSystemMessage func(inputs map[string]any, todos []string)
}

type ToolCallLimit struct {
	PassThrough
	// contains filtered or unexported fields
}

type ToolError struct {
	PassThrough
	// FormatObservation, when set, transforms the raw error into an observation.
	FormatObservation func(err error) string
}

type ToolRequest struct {
	// Action is the action describing the tool to call.
	Action schema.AgentAction
	// Step is the index of the current iteration.
	Step int
	// Skip, when set by a before-hook, prevents the underlying tool from
	// running. The after-hooks still execute so they can record the outcome.
	Skip bool
}

type ToolResponse struct {
	// Observation is the tool result.
	Observation string
	// Err is set when the tool call failed.
	Err error
}

type ToolRetry struct {
	PassThrough
	// contains filtered or unexported fields
}