mycode

type AfterToolHook func(context.Context, ToolCall, Result) (Result, bool)

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

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

type BeforeToolHook func(context.Context, ToolCall) (Result, bool)

type Block struct {
	Type      string         `json:"type"`
	Text      string         `json:"text,omitempty"`
	Data      string         `json:"data,omitempty"`
	MIMEType  string         `json:"mime_type,omitempty"`
	Name      string         `json:"name,omitempty"`
	ID        string         `json:"id,omitempty"`
	Input     map[string]any `json:"input,omitempty"`
	ToolUseID string         `json:"tool_use_id,omitempty"`
	Output    string         `json:"output,omitempty"`
	Metadata  map[string]any `json:"metadata,omitempty"`
	IsError   *bool          `json:"is_error,omitempty"`
	Content   []Block        `json:"content,omitempty"`
	Meta      map[string]any `json:"meta,omitempty"`
}

type Call[T any] struct {
	ID    string
	Input T
	CWD   string
	Emit  func(string)
}

type Config struct {
	// Provider
	Model    string
	Provider string // optional; inferred from Model when empty
	APIKey   string
	APIBase  string

	// Runtime
	CWD              string
	System           string
	Tools            []Tool
	Hooks            Hooks
	MaxTurns         int
	Temperature      *float64
	ReasoningEffort  string
	CompactThreshold float64 // fraction of the context window that triggers compaction; 0 (default) disables it

	// Model capabilities. Nil resolves from the bundled catalog (falling back
	// to 16384 output tokens / 128000 context); set to use exactly these values.
	Metadata *ModelMetadata

	// Session persistence (optional)
	Store     *SessionStore // nil keeps the run in memory
	SessionID string        // empty generates a random id
	Messages  []Message     // explicit history; nil auto-resumes from Store; rejected when SessionID already exists
}

type Event struct {
	Type string
	Data map[string]any
}

type Hooks struct {
	BeforeTool []BeforeToolHook
	AfterTool  []AfterToolHook
}

type Message struct {
	Role    string         `json:"role"`
	Content []Block        `json:"content,omitempty"`
	Meta    map[string]any `json:"meta,omitempty"`
}

type ModelMetadata struct {
	ContextWindow      int  `json:"context_window"`
	MaxOutputTokens    int  `json:"max_output_tokens"`
	SupportsReasoning  bool `json:"supports_reasoning"`
	SupportsImageInput bool `json:"supports_image_input"`
	SupportsPDFInput   bool `json:"supports_pdf_input"`
}

type ModelOverride struct {
	ContextWindow      int
	MaxOutputTokens    int
	SupportsReasoning  *bool
	SupportsImageInput *bool
	SupportsPDFInput   *bool
}

type Option func(*Tool)

type Provider struct {
	ID                      string
	Label                   string
	DefaultBaseURL          string
	EnvAPIKeyNames          []string
	DefaultModels           []string
	SupportsReasoningEffort bool
	AutoDiscoverable        bool
}

type Result struct {
	Output   string         `json:"output"`
	Content  []Block        `json:"content,omitempty"`
	Metadata map[string]any `json:"metadata,omitempty"`
	IsError  bool           `json:"is_error"`
}

type SessionData struct {
	Session  SessionSummary `json:"session"`
	Messages []Message      `json:"messages"`
}

type SessionMeta struct {
	CWD       string `json:"cwd"`
	Title     string `json:"title"`
	CreatedAt string `json:"created_at"`
	UpdatedAt string `json:"updated_at"`
}

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

type SessionSummary struct {
	ID string `json:"id"`
	SessionMeta
}

type Tool struct {
	Name          string
	Description   string
	InputSchema   map[string]any
	StreamsOutput bool
	Runner        func(context.Context, ToolCall) Result
}

type ToolCall struct {
	ID    string
	Name  string
	Input map[string]any
	CWD   string
	Emit  func(string)
	// contains filtered or unexported fields
}