agent

type Agent interface {
	// Execute executes the agent with the given input and returns the output.
	// Internally the agent may perform a multi-step operation based on config,
	// instructions and available tools.
	Execute(context.Context, Session, Input, ...AgentExecutionContextOpt) (Output, error)
}

type AgentExecutionContext struct {
	// OnToolCall is called when the agent is about to call a tool.
	// This is used for introspection only and not to mutate the agent's behavior.
	OnToolCall func(context.Context, Session, Input, string, string) error
}

type AgentExecutionContextOpt func(*AgentExecutionContext)

type AgentToolCallIntrospectionFn func(context.Context, string, string) (string, error)

type AgentUIConfig struct {
	Width                int
	Height               int
	InitialSystemMessage string
	TextInputPlaceholder string
	TitleText            string
	MaxHistory           int

	// Only for informational purposes.
	ModelName   string
	ModelVendor string
	ModelFast   bool
}

type AnswerFormat string

type Input struct {
	Query string
}

type McpClientToolBuilderConfig struct {
	// Common config
	ClientName    string
	ClientVersion string

	// SSE client config
	SseURL  string
	Headers map[string]string

	// Stdout client config
	SkipDefaultTools           bool
	SQLQueryToolEnabled        bool
	SQLQueryToolDBPath         string
	PackageRegistryToolEnabled bool

	// Enable debug mode for the MCP client.
	Debug bool
}

type Memory interface {
	AddInteraction(ctx context.Context, interaction *schema.Message) error
	GetInteractions(ctx context.Context) ([]*schema.Message, error)
	Clear(ctx context.Context) error
}

type Model struct {
	Vendor string
	Name   string
	Fast   bool
	Client model.ToolCallingChatModel
}

type Output struct {
	Answer string
	Format AnswerFormat
}

type ReactQueryAgentConfig struct {
	MaxSteps     int
	SystemPrompt string
}

type Session interface {
	ID() string
	Memory() Memory
}

type ToolBuilder interface {
	Build(context.Context) ([]tool.BaseTool, error)
}