ai

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

type AgentOption func(*Agent)

type Capabilities struct {
	PromptCaching     bool // explicit caching API (Anthropic, Gemini)
	StructuredOutput  bool // JSON-schema-constrained output
	ParallelToolCalls bool
	MaxContextTokens  int
}

type ChatEvent struct {
	Type     ChatEventType
	Text     string        // EventText, EventThinking
	ToolUse  *ToolUseBlock // EventToolUse
	Response *ChatResponse // EventDone
	Err      error         // EventError
}

type ChatEventType int

type ChatRequest struct {
	Model           string
	System          string
	Messages        []ProviderMessage
	Tools           []ToolDef
	ToolChoice      ToolChoice
	MaxOutputTokens int
	Temperature     float64
	JSONSchema      *JSONSchema // structured output, optional
	CachePrefix     bool        // hint: cache system + leading messages if supported
}

type ChatResponse struct {
	Content    []ContentBlock // mix of TextBlock, ThinkingBlock, ToolUseBlock
	StopReason StopReason
	Usage      TokenUsage
}

type Client interface {
	// ChatStream sends a conversation to the LLM and streams the response.
	// systemPrompt is prepended as a system instruction.
	// onToken is called for each streamed chunk. Chunks may be:
	//   - plain text (regular model output)
	//   - "\x00THOUGHT:<text>" — model thinking/reasoning
	//   - "\x00TOOL_START:<name>(<args>)" — tool execution starting
	//   - "\x00TOOL_DONE:<name>|<status>|<duration>" — tool execution finished
	// Returns the full assembled response text, or an error.
	ChatStream(ctx context.Context, systemPrompt string, messages []Message, onToken func(string)) (string, error)
}

type ContentBlock interface {
	// contains filtered or unexported methods
}

type GeminiProvider struct {
	APIKey     string
	Model      string
	BaseURL    string       // override for testing; empty uses the real Gemini API
	HTTPClient *http.Client // optional; defaults to a client with no timeout (context-based cancellation)

	// ModelConfig holds per-model tuning (maxOutputTokens, temperature,
	// thinkingBudget). Set by the caller from config.GetModelConfig().
	ModelConfig struct {
		MaxOutputTokens int
		Temperature     float64
		ThinkingBudget  int
	}
}

type JSONSchema struct {
	Name   string
	Schema json.RawMessage
}

type Message struct {
	Role    string // "user" or "assistant"
	Content string
}

type ModelInfo interface {
	// ProviderName returns the provider name (e.g. "gemini", "anthropic").
	ProviderName() string
	// ModelName returns the model identifier (e.g. "gemini-2.5-pro").
	ModelName() string
}

type ModelSwitcher interface {
	// SwitchModel changes the active model to the given ID and applies the
	// provided tuning parameters. Returns an error if the model is invalid.
	SwitchModel(modelID string, maxOutputTokens int, temperature float64, thinkingBudget int) error
}

type Provider interface {
	Name() string
	ModelID() string
	Capabilities() Capabilities
	Chat(ctx context.Context, req ChatRequest) (*ChatResponse, error)
	StreamChat(ctx context.Context, req ChatRequest) (<-chan ChatEvent, error)
}

type ProviderMessage struct {
	Role    Role
	Content []ContentBlock
}

type Role string

type StopReason string

type TextBlock struct{ Text string }

type ThinkingBlock struct {
	Text      string
	Signature string
}

type TokenUsage struct {
	InputTokens  int
	OutputTokens int
	CacheHits    int // tokens served from cache (if supported)
}

type ToolChoice string

type ToolConfigurer interface {
	// SetHeadRef configures the git ref used for file reading tools.
	SetHeadRef(ref string)
	// SetBaseRef configures the git ref for reading base-branch files (before changes).
	SetBaseRef(ref string)
	// SetRawDiffs provides the raw unified diffs for the git_diff tool.
	SetRawDiffs(diffs map[string]string)
	// SetReviewGetter provides a function that returns the latest PR review summary.
	SetReviewGetter(fn func() string)
}

type ToolDef struct {
	Name        string
	Description string
	Parameters  ToolParams
	ReadOnly    bool // true if this tool only reads state (safe for parallel execution)
}

type ToolExecutor struct {
	HeadRef      string            // e.g. "origin/feature-branch" — the PR head ref for git show
	BaseRef      string            // e.g. "origin/main" — the PR base ref for reading pre-change files
	RawDiffs     map[string]string // filePath -> raw unified diff (set by UI after PR load, used by review flow)
	ReviewGetter func() string     // returns the latest PR review summary, or "" if none
	// contains filtered or unexported fields
}

type ToolParam struct {
	Type        string
	Description string
	Enum        []string   // optional enum values
	Items       *ToolParam // for array types
}

type ToolParams struct {
	Type       string // always "object" for function parameters
	Properties map[string]ToolParam
	Required   []string
}

type ToolResultBlock struct {
	ToolUseID string
	Name      string // tool name (needed by Gemini's functionResponse)
	Content   string
	IsError   bool
}

type ToolUseBlock struct {
	ID        string
	Name      string
	Args      json.RawMessage
	Signature string // opaque thought signature (Gemini); must be echoed back
}

type UsageTracker struct {
	InputTokens  int
	OutputTokens int
	CacheHits    int
	Calls        int // number of API calls
	// contains filtered or unexported fields
}