llm

type AssetID string

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

type ChatRequest struct {
	Model       string
	Messages    []Message
	Tools       []ToolDefinition
	Assets      map[AssetID]ImageAsset
	Temperature float64

	// OnStreamProgress is called during streaming when the provider receives
	// non-content progress updates (e.g. reasoning_content chunks from
	// DeepSeek-style models). charCount is the cumulative number of
	// characters received so far; active is true while reasoning is still
	// streaming.
	OnStreamProgress func(charCount int, active bool)
}

type Client interface {
	CreateMessage(ctx context.Context, req ChatRequest) (Message, error)
	StreamMessage(ctx context.Context, req ChatRequest, onDelta func(string)) (Message, error)
}

type ErrorCode string

type FunctionDefinition struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	Parameters  map[string]any `json:"parameters"`
}

type ImageAsset struct {
	MediaType string
	Data      []byte
}

type ImagePartRef struct {
	AssetID AssetID `json:"asset_id"`
}

type InternalRequest struct {
	Model    string
	Messages []Message
	Tools    []ToolDefinition
	Assets   map[AssetID]ImageAsset
	Stream   bool
}

type Message struct {
	ID        string      `json:"id,omitempty"`
	Role      Role        `json:"role"`
	Parts     []Part      `json:"content,omitempty"`
	CreatedAt string      `json:"created_at,omitempty"`
	Meta      MessageMeta `json:"meta,omitempty"`
	Usage     *Usage      `json:"usage,omitempty"`

	// Legacy compatibility fields used by existing runner/tui/provider paths.
	Content    string     `json:"-"`
	Name       string     `json:"-"`
	ToolCallID string     `json:"-"`
	ToolCalls  []ToolCall `json:"-"`
}

type MessageMeta map[string]any

type ModelCapabilities struct {
	SupportsVision   bool
	SupportsToolUse  bool
	SupportsThinking bool
}

type Part struct {
	Type       PartType        `json:"type"`
	Text       *TextPart       `json:"text,omitempty"`
	Image      *ImagePartRef   `json:"image,omitempty"`
	ToolUse    *ToolUsePart    `json:"tool_use,omitempty"`
	ToolResult *ToolResultPart `json:"tool_result,omitempty"`
	Thinking   *ThinkingPart   `json:"thinking,omitempty"`
}

type PartType string

type ProviderClient interface {
	Chat(ctx context.Context, req InternalRequest) (<-chan StreamEvent, error)
}

type ProviderError struct {
	Code      ErrorCode `json:"code"`
	Provider  string    `json:"provider,omitempty"`
	Message   string    `json:"message,omitempty"`
	Status    int       `json:"status,omitempty"`
	Retryable bool      `json:"retryable,omitempty"`
}

type Role string

type StreamEvent struct {
	Type    StreamEventType
	Text    string
	ToolUse *ToolUsePart
	Err     *ProviderError
}

type StreamEventType string

type TextPart struct {
	Value string `json:"value"`
}

type ThinkingPart struct {
	Value string `json:"value"`
}

type ToolCall struct {
	ID       string           `json:"id"`
	Type     string           `json:"type"`
	Function ToolFunctionCall `json:"function"`
}

type ToolDefinition struct {
	Type     string             `json:"type"`
	Function FunctionDefinition `json:"function"`
}

type ToolFunctionCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type ToolResultPart struct {
	ToolUseID string `json:"tool_use_id"`
	Content   string `json:"content"`
	IsError   bool   `json:"is_error,omitempty"`
}

type ToolUsePart struct {
	ID        string `json:"id"`
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type Usage struct {
	InputTokens   int `json:"input_tokens,omitempty"`
	OutputTokens  int `json:"output_tokens,omitempty"`
	ContextTokens int `json:"context_tokens,omitempty"`
	TotalTokens   int `json:"total_tokens,omitempty"`
}