ai

type Api = string

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

type AssistantEventStream interface {
	Events() <-chan AssistantEvent
	Close() error
	Wait() error
}

type AssistantMessage struct {
	Content      []ContentBlock
	Api          string
	Provider     string
	Model        string
	Usage        Usage
	StopReason   StopReason
	ErrorMessage string
	Timestamp    time.Time
}

type CacheRetention = string

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

type CompleteOptions struct {
	StreamOptions
}

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

type Context struct {
	System   string
	Messages []Message
	Tools    []toolspec.Definition
	Metadata map[string]string
}

type EventDone struct{}

type EventError struct {
	Err   error
	Error *AssistantMessage
}

type EventStart struct {
	Time    time.Time
	Partial *AssistantMessage
}

type EventStop struct {
	Reason  StopReason
	Message *AssistantMessage
}

type EventTextDelta struct {
	ContentIndex int
	Text         string
	Partial      *AssistantMessage
}

type EventTextEnd struct {
	ContentIndex int
	Content      string
	Partial      *AssistantMessage
}

type EventTextStart struct {
	ContentIndex int
	Partial      *AssistantMessage
}

type EventThinkingDelta struct {
	ContentIndex int
	Thinking     string
	Partial      *AssistantMessage
}

type EventThinkingEnd struct {
	ContentIndex int
	Content      string
	Partial      *AssistantMessage
}

type EventThinkingStart struct {
	ContentIndex int
	Partial      *AssistantMessage
}

type EventToolCallDelta struct {
	ContentIndex int
	ID           string
	Name         string
	Arguments    string
	Partial      *AssistantMessage
}

type EventToolCallEnd struct {
	ContentIndex int
	ToolCall     ToolCall
	Partial      *AssistantMessage
}

type EventToolCallStart struct {
	ContentIndex int
	Partial      *AssistantMessage
}

type EventUsage struct {
	Usage Usage
}

type ImageContent struct {
	Data     string // base64 encoded
	MimeType string // e.g. "image/jpeg", "image/png"
}

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

type Model struct {
	ID            string
	Name          string
	API           string
	Provider      string
	BaseURL       string
	Reasoning     bool
	Input         []string // e.g. ["text", "image"]
	Cost          ModelCost
	ContextWindow int
	MaxTokens     int
	Headers       map[string]string
}

type ModelCost struct {
	Input      float64
	Output     float64
	CacheRead  float64
	CacheWrite float64
}

type ModelLister interface {
	ListModels(ctx context.Context) ([]Model, error)
}

type Provider = string

type ProviderAdapter interface {
	API() string
	Stream(model Model, ctx Context, opts StreamOptions) (AssistantEventStream, error)
	StreamSimple(model Model, ctx Context, opts SimpleStreamOptions) (AssistantEventStream, error)
}

type ProviderGetter interface {
	Get(api string) (ProviderAdapter, bool)
}

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

type SimpleStreamOptions struct {
	StreamOptions
	Reasoning       ThinkingLevel
	ThinkingBudgets *ThinkingBudgets
}

type StopReason string

type StreamOptions struct {
	Temperature     *float64
	MaxTokens       *int
	APIKey          string
	BaseURL         string
	Transport       Transport
	CacheRetention  CacheRetention
	SessionID       string
	Headers         map[string]string
	Metadata        map[string]any
	MaxRetryDelayMS *int
	Timeout         time.Duration
}

type TextContent struct {
	Text          string
	TextSignature string // legacy id string or TextSignatureV1 JSON
}

type TextSignatureV1 struct {
	V     int    `json:"v"`
	ID    string `json:"id"`
	Phase string `json:"phase,omitempty"` // "commentary" | "final_answer"
}

type ThinkingBudgets struct {
	Minimal *int
	Low     *int
	Medium  *int
	High    *int
}

type ThinkingContent struct {
	Thinking  string
	Signature string
	Redacted  bool
}

type ThinkingLevel = string

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

type ToolResultMessage struct {
	ToolCallID string
	ToolName   string
	Content    []ContentBlock // TextContent | ImageContent
	Details    any
	IsError    bool
	Timestamp  time.Time
}

type Transport = string

type Usage struct {
	InputTokens  int
	OutputTokens int
	CacheRead    int
	CacheWrite   int
	TotalTokens  int
	Cost         UsageCost
}

type UsageCost struct {
	Input      float64
	Output     float64
	CacheRead  float64
	CacheWrite float64
	Total      float64
}

type UserMessage struct {
	Content   any
	Timestamp time.Time
}