agents

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

type AgentConfig struct {
	// Model is a string identifier or a ChatModel instance.
	Model any
	// Tools available to the agent.
	Tools []graph.Tool
	// SystemPrompt is prepended to the message list.
	SystemPrompt string
	// Middleware applied to the agent.
	Middleware []Middleware
	// Checkpointer enables persistence/interrupts.
	Checkpointer graph.Checkpointer
	// Name of the agent graph.
	Name string
	// ResponseFormat configures structured output.
	ResponseFormat *structured.ResponseFormat
}

type AgentOption func(*AgentConfig)

type AgentState = graph.State

type Base struct{}

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

type ChatModel interface {
	// BindTools returns a copy of the model bound to the given tools.
	BindTools(tools []graph.Tool) (ChatModel, error)
	// GenerateMessages produces an assistant message from the input messages.
	GenerateMessages(ctx context.Context, messages []*graph.Message) (*graph.Message, error)
	// StreamMessages streams an assistant message incrementally.
	StreamMessages(ctx context.Context, messages []*graph.Message) (<-chan *graph.Message, error)
	// SupportsStructuredOutput reports provider-native structured output support.
	SupportsStructuredOutput(schema map[string]any) bool
	// GenerateStructured generates structured output constrained by schema.
	GenerateStructured(ctx context.Context, messages []*graph.Message, schema map[string]any) (any, error)
}

type JumpTo string

type Middleware interface {
	// Name returns a unique identifier for the middleware.
	Name() string
	// BeforeModel runs before the model is invoked. It may return a command to
	// short-circuit or redirect.
	BeforeModel(ctx context.Context, req *ModelRequest) (*ModelResponse, error)
	// AfterModel runs after the model returns.
	AfterModel(ctx context.Context, req *ModelRequest, resp *ModelResponse) error
	// WrapModelCall wraps the model invocation, allowing full control.
	WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelHandler) (*ModelResponse, error)
	// BeforeTool runs before a tool is called.
	BeforeTool(ctx context.Context, req *ToolCallRequest) (*ToolCallResponse, error)
	// AfterTool runs after a tool returns.
	AfterTool(ctx context.Context, req *ToolCallRequest, resp *ToolCallResponse) error
}

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

type MiddlewareHooks struct {
	Name          string
	BeforeModel   func(ctx context.Context, req *ModelRequest) (*ModelResponse, error)
	AfterModel    func(ctx context.Context, req *ModelRequest, resp *ModelResponse) error
	WrapModelCall func(ctx context.Context, req *ModelRequest, handler ModelHandler) (*ModelResponse, error)
	BeforeTool    func(ctx context.Context, req *ToolCallRequest) (*ToolCallResponse, error)
	AfterTool     func(ctx context.Context, req *ToolCallRequest, resp *ToolCallResponse) error
}

type ModelHandler func(ctx context.Context, req *ModelRequest) (*ModelResponse, error)

type ModelOption func(*modelConfig)

type ModelProfile struct {
	Provider string
	Name     string
}

type ModelRequest struct {
	// Messages are the input messages.
	Messages []*graph.Message
	// Tools are the tools offered to the model.
	Tools []graph.Tool
	// Context holds runtime context.
	Context any
	// Step is the current iteration index.
	Step int
}

type ModelResponse struct {
	// Messages are the messages produced by the model.
	Messages []*graph.Message
	// StructuredResponse is set when the model produced structured output.
	StructuredResponse any
	// Commands are control-flow commands emitted by middleware.
	Commands []graph.Command
}

type SubAgent struct {
	Tools        []graph.Tool
	Checkpoint   graph.Checkpointer
	SystemPrompt string
	// contains filtered or unexported fields
}

type ToolCallRequest struct {
	// ToolCall is the tool invocation.
	ToolCall graph.ToolCall
	// Step is the current iteration index.
	Step int
}

type ToolCallResponse struct {
	// Message is the tool message produced.
	Message *graph.Message
}