native

type Annotations struct {
	ReadOnly    bool
	Destructive bool
	Idempotent  bool
	OpenWorld   bool
	// Kind is the coarse class the permission policy and the UI read.
	Kind agentwire.ToolKind
}

type ApprovalRequest struct {
	CallID string
	Spec   ToolSpec
	Input  json.RawMessage
}

type Approver interface {
	Approve(ctx context.Context, r ApprovalRequest) (agentwire.Decision, error)
}

type ApproverFunc func(ctx context.Context, r ApprovalRequest) (agentwire.Decision, error)

type Compactor interface {
	Compact(ctx context.Context, m Model, msgs []Message) ([]Message, error)
}

type CompactorFunc func(ctx context.Context, m Model, msgs []Message) ([]Message, error)

type Config struct {
	// NewModel builds the Model of one session from the credentials and the
	// model name. A provider module supplies the constructor.
	NewModel func(c agentwire.Credentials, model string) (Model, error)
	// Credentials resolves the provider credentials of one session. It is
	// called per session; StartRequest.Credentials wins when set. The
	// library never reads env or secrets for it.
	Credentials func() (agentwire.Credentials, error)
	// ToolSets are the consumer's own tools, next to the session's MCP
	// servers and the built-in file tools.
	ToolSets []ToolSet
	// ToolSetFuncs open one ToolSet per session. Use them for tools that
	// cannot be shared, such as an in-process MCP transport pair.
	ToolSetFuncs []func(ctx context.Context) (ToolSet, error)
	// BuiltinTools roots the built-in read, write, edit, glob, grep, and
	// bash tools at StartRequest.WorkingDir.
	BuiltinTools bool
	// MaxFileBytes caps one read or write of the built-in file tools.
	// Zero uses a 10 MiB default.
	MaxFileBytes int64
	// Store persists the message log. Nil keeps no log and resume never
	// finds anything.
	Store Store
	// Compactor folds the message list when it outgrows the budget. Nil
	// disables compaction.
	Compactor Compactor
	// ContextTokens is the compaction budget.
	ContextTokens int
	// MaxTurns bounds one prompt. Zero uses DefaultMaxTurns. A cap ends the
	// turn with the max_turn_requests reason.
	MaxTurns int
	// MaxOutputTokens bounds one model call. Zero uses the provider
	// default.
	MaxOutputTokens int
	// BeforeCall runs before every model call. A non-nil error fails the
	// turn; a daily token cap goes here.
	BeforeCall func(ctx context.Context, sessionID string) error
	// AfterCall runs after every model call with its usage.
	AfterCall func(ctx context.Context, sessionID string, u Usage)
	// Approver replaces the default policy approver.
	Approver Approver
}

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

type FilePart struct {
	MediaType string `json:"media_type,omitempty"`
	Data      []byte `json:"data,omitempty"`
	URL       string `json:"url,omitempty"`
}

type Finish struct {
	Reason FinishReason
	Raw    string
	Usage  Usage
}

type FinishReason string

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

type Message struct {
	Role  Role   `json:"role"`
	Parts []Part `json:"parts"`
}

type Model interface {
	// Name is the model identifier the driver reports on init.
	Name() string
	// Stream runs one model call and yields the response as stream parts.
	// The sequence must yield exactly one StreamFinish.
	Stream(ctx context.Context, req Request) Seq[StreamPart]
}

type Part struct {
	Text       *TextPart      `json:"text,omitempty"`
	File       *FilePart      `json:"file,omitempty"`
	Reasoning  *ReasoningPart `json:"reasoning,omitempty"`
	ToolCall   *ToolCall      `json:"tool_call,omitempty"`
	ToolResult *ToolResult    `json:"tool_result,omitempty"`
}

type PolicyApprover struct {
	Policy agentwire.PermissionPolicy
	// Emit surfaces one permission request to the consumer. It must return
	// the channel the decision arrives on. Nil means the policy alone
	// decides and everything else is denied.
	Emit func(ctx context.Context, r ApprovalRequest) (wait func(ctx context.Context) (agentwire.Decision, error), cancel func())
}

type ReasoningPart struct {
	Text         string          `json:"text"`
	ProviderMeta json.RawMessage `json:"provider_meta,omitempty"`
}

type Request struct {
	// System is the system prompt.
	System string
	// Messages is the full conversation, tool results included.
	Messages []Message
	// Tools is the callable surface for this call.
	Tools []ToolSpec
	// ToolChoice steers tool use. The zero value is Auto.
	ToolChoice ToolChoice
	// MaxOutputTokens bounds one response. Zero means the provider default.
	MaxOutputTokens int
	// Effort is low | medium | high | max. The provider maps it onto its own
	// thinking budget or reasoning effort.
	Effort string
	// ProviderOptions carries provider-specific extensions. The provider
	// modules document their keys.
	ProviderOptions map[string]json.RawMessage
}

type Role string

type Seq[T any] = iter.Seq2[T, error]

type SkillCatalog struct {
	Skills []SkillMeta
}

type SkillMeta struct {
	Name string
	// Description is the frontmatter description.
	Description string
	// Dir is the absolute skill directory.
	Dir string
	// AllowedTools lists tools the skill's frontmatter pre-approves.
	AllowedTools []string
}

type Store interface {
	// Load returns the stored messages. ErrNotFound means the store has no
	// log for the session and the caller starts fresh.
	Load(ctx context.Context, sessionID string) ([]Message, error)
	Append(ctx context.Context, sessionID string, msgs ...Message) error
	Replace(ctx context.Context, sessionID string, msgs []Message) error
}

type Stream = Seq[StreamPart]

type StreamKind int

type StreamPart struct {
	Kind StreamKind
	// ID is the tool-call id for the tool parts.
	ID string
	// Delta is the text of a text or reasoning delta, or one JSON fragment
	// of a tool input.
	Delta string
	// Call is the complete call on StreamToolCallDone.
	Call *ToolCall
	// Finish terminates the stream.
	Finish *Finish
}

type TextPart struct {
	Text string `json:"text"`
}

type Tool interface {
	Spec() ToolSpec
	Call(ctx context.Context, input json.RawMessage) (ToolResult, error)
}

type ToolCall struct {
	ID    string          `json:"id"`
	Name  string          `json:"name"`
	Input json.RawMessage `json:"input,omitempty"`
}

type ToolChoice struct {
	Mode ToolChoiceMode
	// Name names the tool when Mode is ToolNamed.
	Name string
}

type ToolChoiceMode int

type ToolResult struct {
	CallID  string `json:"call_id"`
	Name    string `json:"name,omitempty"`
	Content []Part `json:"content"`
	IsError bool   `json:"is_error,omitempty"`
}

type ToolSet interface {
	Name() string
	Tools(ctx context.Context) ([]Tool, error)
	Close() error
}

type ToolSpec struct {
	Name        string
	Title       string
	Description string
	// InputSchema is a JSON Schema object.
	InputSchema json.RawMessage
	Annotations Annotations
}

type Usage struct {
	Input      int64
	Output     int64
	CacheRead  int64
	CacheWrite int64
	Reasoning  int64
}