llm

type AnthropicAPIAccess string

type AnthropicConfig struct {
	Platform         string `mapstructure:"platform" json:"platform" yaml:"platform"`                                                // Canonical platform name for Anthropic-compatible APIs (e.g., anthropic, copilot)
	BaseURL          string `mapstructure:"base_url" json:"base_url" yaml:"base_url"`                                                // Custom API base URL (overrides platform defaults)
	AdaptiveThinking bool   `mapstructure:"adaptive_thinking" json:"adaptive_thinking,omitempty" yaml:"adaptive_thinking,omitempty"` // Forces Anthropic adaptive-thinking request plumbing for the configured custom model ID when true
}

type BashConfig struct {
	Timeout time.Duration `mapstructure:"timeout" json:"timeout" yaml:"timeout"` // Timeout is the maximum allowed timeout for a bash tool call
}

type Config struct {
	IsSubAgent              bool     `mapstructure:"is_sub_agent" json:"is_sub_agent" yaml:"is_sub_agent"` // IsSubAgent is true if the LLM is a sub-agent
	Provider                string   `mapstructure:"provider" json:"provider" yaml:"provider"`             // Provider is the LLM provider (anthropic, openai)
	Model                   string   `mapstructure:"model" json:"model" yaml:"model"`                      // Model is the main driver
	WeakModel               string   `mapstructure:"weak_model" json:"weak_model" yaml:"weak_model"`       // WeakModel is the less capable but faster model to use
	MaxTokens               int      `mapstructure:"max_tokens" json:"max_tokens" yaml:"max_tokens"`
	WeakModelMaxTokens      int      `mapstructure:"weak_model_max_tokens" json:"weak_model_max_tokens" yaml:"weak_model_max_tokens"`    // WeakModelMaxTokens is the maximum tokens for the weak model
	ThinkingBudgetTokens    int      `mapstructure:"thinking_budget_tokens" json:"thinking_budget_tokens" yaml:"thinking_budget_tokens"` // ThinkingBudgetTokens is sent as Anthropic manual budget_tokens on non-adaptive Claude models; adaptive Claude models ignore it
	ReasoningEffort         string   `mapstructure:"reasoning_effort" json:"reasoning_effort" yaml:"reasoning_effort"`                   // ReasoningEffort controls supported provider effort settings (e.g. OpenAI reasoning models, Anthropic adaptive thinking models where "none" disables adaptive thinking)
	AllowedReasoningEfforts []string ``                                                                                                  // AllowedReasoningEfforts restricts selectable reasoning efforts for new conversations (empty means unrestricted)
	/* 126-byte string literal not displayed */
	AllowedCommands      []string           `mapstructure:"allowed_commands" json:"allowed_commands" yaml:"allowed_commands"`             // AllowedCommands is a list of allowed command patterns for the bash tool
	AllowedDomainsFile   string             `mapstructure:"allowed_domains_file" json:"allowed_domains_file" yaml:"allowed_domains_file"` // AllowedDomainsFile is the path to the file containing allowed domains for web_fetch tool
	AllowedTools         []string           `mapstructure:"allowed_tools" json:"allowed_tools" yaml:"allowed_tools"`                      // AllowedTools is a list of allowed tools for the main agent (empty means use defaults)
	WorkingDirectory     string             `mapstructure:"working_directory" json:"working_directory" yaml:"working_directory"`
	ToolMode             ToolMode           `mapstructure:"tool_mode" json:"tool_mode" yaml:"tool_mode"`                                    // ToolMode controls file-interaction behavior (e.g. full or patch)
	AnthropicAPIAccess   AnthropicAPIAccess `mapstructure:"anthropic_api_access" json:"anthropic_api_access" yaml:"anthropic_api_access"`   // AnthropicAPIAccess controls how to authenticate with Anthropic API
	AnthropicAccount     string             `mapstructure:"anthropic_account" json:"anthropic_account" yaml:"anthropic_account"`            // AnthropicAccount specifies which Anthropic subscription account to use
	Aliases              map[string]string  `mapstructure:"aliases" json:"aliases,omitempty" yaml:"aliases,omitempty"`                      // Aliases maps short model names to full model names
	ModelAliasesResolved bool               `mapstructure:"-" json:"-" yaml:"-"`                                                            // ModelAliasesResolved prevents effective model names from being resolved as aliases again
	Retry                RetryConfig        `mapstructure:"retry" json:"retry" yaml:"retry"`                                                // Retry configuration for API calls
	Sysprompt            string             `mapstructure:"sysprompt" json:"sysprompt,omitempty" yaml:"sysprompt,omitempty"`                // Sysprompt is the path to a custom system prompt template file
	SyspromptArgs        map[string]string  `mapstructure:"sysprompt_args" json:"sysprompt_args,omitempty" yaml:"sysprompt_args,omitempty"` // SyspromptArgs are custom template arguments for system prompt rendering
	Bash                 *BashConfig        `mapstructure:"bash" json:"bash,omitempty" yaml:"bash,omitempty"`                               // Bash contains bash tool configuration

	// Profile system configuration
	Profile  string                   `mapstructure:"profile" json:"profile,omitempty" yaml:"profile,omitempty"`    // Active profile name
	Profiles map[string]ProfileConfig `mapstructure:"profiles" json:"profiles,omitempty" yaml:"profiles,omitempty"` // Named configuration profiles

	// Provider-specific configurations
	OpenAI    *OpenAIConfig    `mapstructure:"openai" json:"openai,omitempty" yaml:"openai,omitempty"`          // OpenAI-specific configuration including compatible providers
	Anthropic *AnthropicConfig `mapstructure:"anthropic" json:"anthropic,omitempty" yaml:"anthropic,omitempty"` // Anthropic-specific configuration including compatible providers

	// Skills configuration
	Skills *SkillsConfig `mapstructure:"skills" json:"skills,omitempty" yaml:"skills,omitempty"` // Skills configuration for agentic skills system

	// Context configuration
	Context *ContextConfig `mapstructure:"context" json:"context,omitempty" yaml:"context,omitempty"` // Context configuration for context file discovery

	// Runtime feature toggle configuration
	Extensions              any                     `mapstructure:"-" json:"-" yaml:"-"`                                                                         // Extensions is the active extension runtime for lifecycle events
	EnableFSSearchTools     bool                    `mapstructure:"enable_fs_search_tools" json:"enable_fs_search_tools" yaml:"enable_fs_search_tools"`          // EnableFSSearchTools enables glob_tool and grep_tool and updates prompt/tool guidance accordingly
	ConversationSummaryMode ConversationSummaryMode `mapstructure:"conversation_summary_mode" json:"conversation_summary_mode" yaml:"conversation_summary_mode"` // ConversationSummaryMode controls whether persisted conversation summaries come from the LLM or first user message
	RecipeName              string                  `mapstructure:"recipe_name" json:"recipe_name" yaml:"recipe_name"`                                           // RecipeName is the active recipe/fragment name for extension context metadata
	CompactRatio            float64                 `mapstructure:"compact_ratio" json:"compact_ratio" yaml:"compact_ratio"`                                     // CompactRatio is the context utilization threshold for automatic compaction (>0.0-1.0)
}

type ConsoleMessageHandler struct {
	Silent bool
}

type ContextConfig struct {
	// Patterns is a list of filenames to search for in each directory.
	// Default is ["AGENTS.md"]. Files are searched in order; first match wins per directory.
	Patterns []string `mapstructure:"patterns" json:"patterns" yaml:"patterns"`
}

type ConversationAnthropicSnapshot struct {
	Platform         string `json:"platform,omitempty" yaml:"platform,omitempty"`
	AdaptiveThinking bool   `json:"adaptive_thinking,omitempty" yaml:"adaptive_thinking,omitempty"`
}

type ConversationConfigSnapshot struct {
	Version                 int                            `json:"version" yaml:"version"`
	Profile                 string                         `json:"profile,omitempty" yaml:"profile,omitempty"`
	Provider                string                         `json:"provider" yaml:"provider"`
	Model                   string                         `json:"model" yaml:"model"`
	WeakModel               string                         `json:"weak_model,omitempty" yaml:"weak_model,omitempty"`
	MaxTokens               int                            `json:"max_tokens,omitempty" yaml:"max_tokens,omitempty"`
	WeakModelMaxTokens      int                            `json:"weak_model_max_tokens,omitempty" yaml:"weak_model_max_tokens,omitempty"`
	ThinkingBudgetTokens    int                            `json:"thinking_budget_tokens,omitempty" yaml:"thinking_budget_tokens,omitempty"`
	ReasoningEffort         string                         `json:"reasoning_effort" yaml:"reasoning_effort"`
	ConversationSummaryMode ConversationSummaryMode        `json:"conversation_summary_mode,omitempty" yaml:"conversation_summary_mode,omitempty"`
	CompactRatio            float64                        `json:"compact_ratio,omitempty" yaml:"compact_ratio,omitempty"`
	OpenAI                  *ConversationOpenAISnapshot    `json:"openai,omitempty" yaml:"openai,omitempty"`
	Anthropic               *ConversationAnthropicSnapshot `json:"anthropic,omitempty" yaml:"anthropic,omitempty"`
}

type ConversationOpenAISnapshot struct {
	Platform      string                  `json:"platform,omitempty" yaml:"platform,omitempty"`
	APIMode       OpenAIAPIMode           `json:"api_mode,omitempty" yaml:"api_mode,omitempty"`
	TextVerbosity OpenAITextVerbosity     `json:"text_verbosity,omitempty" yaml:"text_verbosity,omitempty"`
	ServiceTier   OpenAIServiceTier       `json:"service_tier,omitempty" yaml:"service_tier,omitempty"`
	ManualCache   bool                    `json:"manual_cache,omitempty" yaml:"manual_cache,omitempty"`
	Models        *CustomModels           `json:"models,omitempty" yaml:"models,omitempty"`
	Pricing       map[string]ModelPricing `json:"pricing,omitempty" yaml:"pricing,omitempty"`
}

type ConversationSummaryMode string

type CustomModels struct {
	Reasoning    []string `mapstructure:"reasoning" json:"reasoning" yaml:"reasoning"`             // Models that support reasoning (o1, o3, etc.)
	NonReasoning []string `mapstructure:"non_reasoning" json:"non_reasoning" yaml:"non_reasoning"` // Models that don't support reasoning (gpt-4, etc.)
}

type CustomPricing map[string]ModelPricing

type DeltaEntry struct {
	Kind           string `json:"kind"`
	Delta          string `json:"delta,omitempty"`
	Content        string `json:"content,omitempty"`
	ConversationID string `json:"conversation_id"`
	Role           string `json:"role"`
}

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

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type MessageEvent struct {
	Type    string
	Content string
	Done    bool
}

type MessageHandler interface {
	HandleText(text string)
	HandleToolUse(toolCallID string, toolName string, input string)
	HandleToolResult(toolCallID string, toolName string, result tooltypes.ToolResult)
	HandleThinking(thinking string)
	HandleDone()
}

type MessageOpt struct {
	// Initiator identifies whether a Copilot-backed request originated from the user or agent internals.
	// Empty means "user".
	Initiator string
	// PromptCache indicates if prompt caching should be used
	PromptCache bool
	// UseWeakModel allows temporarily overriding the model for this message
	UseWeakModel bool
	// NoToolUse indicates that no tool use should be performed
	NoToolUse bool
	// NoSaveConversation indicates that the following conversation should not be saved
	NoSaveConversation bool
	// Images contains image paths or URLs to include with the message
	Images []string
	// MaxTurns limits the number of turns within a single SendMessage call
	// A value of 0 means no limit, and negative values are treated as 0
	MaxTurns int
	// CompactRatio is the ratio of context window at which to trigger auto-compact (>0.0-1.0).
	// A zero value uses the thread/configured default; it does not disable auto-compaction.
	CompactRatio float64
	// DisableUsageLog disables LLM usage logging for this message
	DisableUsageLog bool
}

type ModelPricing struct {
	// Input token cost per token.
	Input float64 `mapstructure:"input" json:"input" yaml:"input"`
	// Cached input token cost per token.
	CachedInput float64 `mapstructure:"cached_input" json:"cached_input" yaml:"cached_input"`
	// Cache write input token cost per token.
	CacheWriteInput float64 `mapstructure:"cache_write_input" json:"cache_write_input" yaml:"cache_write_input"`
	// Output token cost per token.
	Output float64 `mapstructure:"output" json:"output" yaml:"output"`
	// Long-context input token cost per token.
	LongContextInput float64 `mapstructure:"long_context_input" json:"long_context_input,omitempty" yaml:"long_context_input,omitempty"`
	// Long-context cached input token cost per token.
	LongContextCachedInput float64 `` /* 126-byte string literal not displayed */
	// Long-context cache write input token cost per token.
	LongContextCacheWriteInput float64 `` /* 141-byte string literal not displayed */
	// Long-context output token cost per token.
	LongContextOutput float64 `mapstructure:"long_context_output" json:"long_context_output,omitempty" yaml:"long_context_output,omitempty"`
	// Prompt token threshold for long-context pricing.
	LongContextThreshold int `mapstructure:"long_context_threshold" json:"long_context_threshold,omitempty" yaml:"long_context_threshold,omitempty"`
	// Maximum context window size.
	ContextWindow int `mapstructure:"context_window" json:"context_window" yaml:"context_window"`
}

type OpenAIAPIMode string

type OpenAIConfig struct {
	Platform      string                  `mapstructure:"platform" json:"platform" yaml:"platform"`                                       // Canonical platform name for OpenAI-compatible APIs (e.g., openai, codex, fireworks)
	BaseURL       string                  `mapstructure:"base_url" json:"base_url" yaml:"base_url"`                                       // Custom API base URL (overrides platform defaults)
	APIKeyEnvVar  string                  `mapstructure:"api_key_env_var" json:"api_key_env_var" yaml:"api_key_env_var"`                  // Environment variable name for API key (overrides platform default)
	APIMode       OpenAIAPIMode           `mapstructure:"api_mode" json:"api_mode" yaml:"api_mode"`                                       // Preferred API mode selection (chat_completions or responses)
	TextVerbosity OpenAITextVerbosity     `mapstructure:"text_verbosity" json:"text_verbosity" yaml:"text_verbosity"`                     // Optional Responses API text verbosity (low, medium, or high); omitted values use the upstream default
	ServiceTier   OpenAIServiceTier       `mapstructure:"service_tier" json:"service_tier" yaml:"service_tier"`                           // Optional service tier hint (e.g. auto, default, fast, flex, priority, scale)
	EnableSearch  *bool                   `mapstructure:"enable_search" json:"enable_search,omitempty" yaml:"enable_search,omitempty"`    // Enable native OpenAI Responses web_search tool when supported (defaults to true)
	WebSocketMode *bool                   `mapstructure:"websocket_mode" json:"websocket_mode,omitempty" yaml:"websocket_mode,omitempty"` // Use Responses API WebSocket transport when supported (defaults to true)
	ManualCache   bool                    `mapstructure:"manual_cache" json:"manual_cache" yaml:"manual_cache"`                           // Enables manual cache affinity headers for Chat Completions when prompt caching is requested
	Models        *CustomModels           `mapstructure:"models" json:"models,omitempty" yaml:"models,omitempty"`                         // Custom model configuration
	Pricing       map[string]ModelPricing `mapstructure:"pricing" json:"pricing,omitempty" yaml:"pricing,omitempty"`                      // Custom pricing configuration
}

type OpenAIServiceTier string

type OpenAITextVerbosity string

type ProfileConfig map[string]any

type RetryConfig struct {
	Attempts     int    `mapstructure:"attempts" json:"attempts" yaml:"attempts"`                // Maximum number of retry attempts (default: 3)
	InitialDelay int    `mapstructure:"initial_delay" json:"initial_delay" yaml:"initial_delay"` // Initial delay in milliseconds (default: 1000) - OpenAI only
	MaxDelay     int    `mapstructure:"max_delay" json:"max_delay" yaml:"max_delay"`             // Maximum delay in milliseconds (default: 10000) - OpenAI only
	BackoffType  string `mapstructure:"backoff_type" json:"backoff_type" yaml:"backoff_type"`    // Backoff strategy: "fixed", "exponential" (default: "exponential") - OpenAI only
}

type SkillsConfig struct {
	// Enabled controls whether skills are active. When the SkillsConfig is nil
	// (not specified in config), skills default to enabled. Set to false to disable.
	Enabled bool `mapstructure:"enabled" json:"enabled" yaml:"enabled"`
	// Allowed is an allowlist of skill names. When empty, all discovered skills are available.
	// When specified, only the listed skills will be enabled.
	Allowed []string `mapstructure:"allowed" json:"allowed" yaml:"allowed"`
}

type StreamingMessageHandler interface {
	MessageHandler
	HandleTextDelta(delta string)     // Called for each text chunk as it streams
	HandleThinkingStart()             // Called when a thinking block starts
	HandleThinkingDelta(delta string) // Called for each thinking chunk as it streams
	HandleThinkingBlockEnd()          // Called when a thinking block ends (for visual separation)
	HandleContentBlockEnd()           // Called when any content block ends
}

type StringCollectorHandler struct {
	Silent bool
	// contains filtered or unexported fields
}

type Thread interface {
	// SetState sets the state for the thread
	SetState(s tooltypes.State)
	// GetState returns the current state of the thread
	GetState() tooltypes.State
	// AddUserMessage adds a user message with optional images to the thread
	AddUserMessage(ctx context.Context, message string, imagePaths ...string)
	// SendMessage sends a message to the LLM and processes the response
	SendMessage(ctx context.Context, message string, handler MessageHandler, opt MessageOpt) (finalOutput string, err error)
	// GetUsage returns the current token usage for the thread
	GetUsage() Usage
	// GetConversationID returns the current conversation ID
	GetConversationID() string
	// SetConversationID sets the conversation ID
	SetConversationID(id string)
	// SaveConversation saves the current thread to the conversation store
	SaveConversation(ctx context.Context, summarise bool) error
	// IsPersisted returns whether this thread is being persisted
	IsPersisted() bool
	// EnablePersistence enables conversation persistence for this thread
	EnablePersistence(ctx context.Context, enabled bool)
	// Provider returns the provider of the thread
	Provider() string
	// GetMessages returns the messages from the thread
	GetMessages() ([]Message, error)
	// GetConfig returns the configuration of the thread
	GetConfig() Config
	// AggregateSubagentUsage aggregates usage from a subagent into this thread's usage
	// This aggregates token counts and costs but NOT context window (which should remain isolated)
	AggregateSubagentUsage(usage Usage)
	// SetMetadataValue stores provider-neutral conversation metadata for persistence.
	SetMetadataValue(key string, value any)
	// GetMetadata returns provider-neutral conversation metadata.
	GetMetadata() map[string]any
}

type ToolMode string

type Usage struct {
	InputTokens              int     `json:"inputTokens"`              // Regular input tokens count
	OutputTokens             int     `json:"outputTokens"`             // Output tokens generated
	CacheCreationInputTokens int     `json:"cacheCreationInputTokens"` // Tokens used for creating cache entries
	CacheReadInputTokens     int     `json:"cacheReadInputTokens"`     // Tokens used for reading from cache
	InputCost                float64 `json:"inputCost"`                // Cost for input tokens in USD
	OutputCost               float64 `json:"outputCost"`               // Cost for output tokens in USD
	CacheCreationCost        float64 `json:"cacheCreationCost"`        // Cost for cache creation in USD
	CacheReadCost            float64 `json:"cacheReadCost"`            // Cost for cache read in USD
	CurrentContextWindow     int     `json:"currentContextWindow"`     // Current context window size
	MaxContextWindow         int     `json:"maxContextWindow"`         // Max context window size
}

type UsageMessageHandler interface {
	HandleUsage(usage Usage)
}

type UserMessageHandler interface {
	HandleUserMessage(content string, images []string)
}