llm

type BedrockHTTPError struct {
	Code int
	Msg  string
}

type CacheControl struct {
	Type string `json:"type"` // currently always "ephemeral"
}

type Client interface {
	// StreamMessage runs a streaming request with default options.
	StreamMessage(
		ctx context.Context,
		system []SystemBlock,
		messages []MessageParam,
		tools []ToolParam,
		onDelta func(string),
		onThinkingDelta func(string),
	) (*Message, time.Duration, error)

	// StreamMessageWith runs a streaming request honoring per-call
	// overrides from opts (currently just EffortOverride).
	StreamMessageWith(
		ctx context.Context,
		system []SystemBlock,
		messages []MessageParam,
		tools []ToolParam,
		onDelta func(string),
		onThinkingDelta func(string),
		opts StreamOpts,
	) (*Message, time.Duration, error)

	// Provider identifies which upstream this client talks to.
	Provider() ProviderID

	// Model returns the bare model name (no provider prefix).
	Model() string

	// Credential returns the credential this client was built with.
	Credential() config.Credential

	// MaxTokens returns the per-call output token cap configured on this
	// client. Zero means "use the default" (32768).
	MaxTokens() int64

	// Effort returns the reasoning effort configured at construction time.
	Effort() string
}

type Config struct {
	Credential config.Credential
	Model      string // bare model name (no provider prefix)
	Effort     string // "", "low", "medium", "high", "max", "adaptive"
	MaxTokens  int64  // 0 = use DefaultMaxTokens
	PluginCfg  PluginConfig
	HTTPClient *http.Client // optional override; nil = use NewPluginHTTPClient(PluginCfg)

	// BaseURL overrides the adapter's default API endpoint. Empty means
	// use the provider's default. Set from a credential's endpoint override
	// (e.g. the Codex backend) or by tests redirecting to httptest servers.
	BaseURL string

	StreamIdle    time.Duration // 0 = read from env or use DefaultStreamIdleTimeout
	ThinkingStall time.Duration // 0 = read from env or use DefaultThinkingStallTimeout
}

type ContentBlock struct {
	Type         ContentBlockType `json:"type"`
	Text         string           `json:"text,omitempty"`        // BlockText, BlockThinking
	ID           string           `json:"id,omitempty"`          // BlockToolUse
	Name         string           `json:"name,omitempty"`        // BlockToolUse
	Input        map[string]any   `json:"input,omitempty"`       // BlockToolUse — already-parsed; never a raw JSON string
	ToolUseID    string           `json:"tool_use_id,omitempty"` // BlockToolResult
	Output       string           `json:"output,omitempty"`      // BlockToolResult
	IsError      bool             `json:"is_error,omitempty"`    // BlockToolResult
	MediaType    string           `json:"media_type,omitempty"`  // BlockImage (e.g. "image/png")
	Data         string           `json:"data,omitempty"`        // BlockImage (base64-encoded payload)
	Signature    string           `json:"signature,omitempty"`   // BlockThinking — Anthropic signature or OpenAI reasoning-item ID
	CacheControl *CacheControl    `json:"cache_control,omitempty"`
}

type ContentBlockType string

type Message struct {
	StopReason  StopReason
	TextContent string         // concatenated text blocks (convenience for extractTextFromMessage)
	Content     []ContentBlock // full ordered content for replay
	ToolCalls   []ToolCall     // convenience extraction; duplicates Content's tool_use blocks
	Usage       Usage
	Raw         any // raw provider response, retained for LogLLMCall debugging
}

type MessageParam struct {
	Role    Role           `json:"role"`
	Content []ContentBlock `json:"content"`
}

type PluginConfig struct {
	// Headers maps HTTP header name → value. A nil pointer value means
	// "strip this header from every outgoing API request". A non-nil
	// pointer means "set (or override) this header to the given string".
	Headers map[string]*string `json:"headers"`

	// SystemPrefix is prepended as the first system-prompt text block on
	// every StreamMessage call. Empty means no-op.
	SystemPrefix string `json:"system_prefix"`
}

type PluginSource func(provider, model string, cred config.Credential) PluginConfig

type ProviderID string

type Role string

type StopReason string

type StreamOpts struct {
	// EffortOverride, when non-nil, replaces Client.Effort() for this call
	// only. Empty string disables reasoning entirely. Used by the retry
	// loops to force a non-thinking response on the final attempt after
	// repeated thinking stalls.
	EffortOverride *string
}

type SystemBlock struct {
	Text         string
	CacheControl *CacheControl
}

type ThinkingStallError struct {
	Elapsed time.Duration
	Summary string
}

type ToolCall struct {
	ID    string
	Name  string
	Input map[string]any
}

type ToolParam struct {
	Name        string
	Description string
	InputSchema map[string]any // raw JSON Schema object
}

type Usage struct {
	InputTokens         int64
	OutputTokens        int64
	CacheCreationTokens int64
	CacheReadTokens     int64
	ReasoningTokens     int64          // openai o-series, gpt-5-thinking
	CostUSD             float64        // openrouter when usage.include=true
	ProviderExtra       map[string]any // raw provider blob for future fields
}