model

type AnthropicConfig struct {
	APIKey      string
	BaseURL     string
	Model       string
	MaxTokens   int
	MaxRetries  int
	System      string
	Temperature *float64
	HTTPClient  *http.Client
}

type AnthropicProvider struct {
	APIKey      string
	BaseURL     string
	ModelName   string
	MaxTokens   int
	MaxRetries  int
	System      string
	Temperature *float64
	CacheTTL    time.Duration
	// contains filtered or unexported fields
}

type FallbackOptions struct {
	// PrimaryModel is used as display/source model when Request.Model is empty.
	PrimaryModel string
	// OnSwitch is called before each fallback switch attempt.
	OnSwitch func(FallbackSwitchEvent)
}

type FallbackSwitchEvent struct {
	FromModel string
	ToModel   string
	LastError error
	Stream    bool
}

type Message struct {
	Role        string
	Content     string
	ToolCalls   []ToolCall
	Attachments []MessageAttachment // Optional: image/audio/file attachments
}

type MessageAttachment struct {
	Type       string // "image", "audio", or "file"
	Data       string // base64-encoded data
	MimeType   string // e.g., "image/jpeg", "audio/wav"
	SourceType string // "base64" or "url"
}

type MiddlewareState interface {
	SetModelInput(any)
	SetModelOutput(any)
	SetValue(string, any)
}

type Model interface {
	Complete(ctx context.Context, req Request) (*Response, error)
	CompleteStream(ctx context.Context, req Request, cb StreamHandler) error
}

type OpenAIConfig struct {
	APIKey       string
	BaseURL      string // Optional: for Azure or proxies
	Model        string // e.g., "gpt-4o", "gpt-4-turbo"
	MaxTokens    int
	MaxRetries   int
	System       string
	Temperature  *float64
	HTTPClient   *http.Client
	UseResponses bool // true = /responses API, false = /chat/completions
}

type OpenAIProvider struct {
	APIKey      string
	BaseURL     string // Optional: for Azure or proxies
	ModelName   string
	MaxTokens   int
	MaxRetries  int
	System      string
	Temperature *float64
	CacheTTL    time.Duration
	// contains filtered or unexported fields
}

type Provider interface {
	Model(ctx context.Context) (Model, error)
}

type ProviderFunc func(context.Context) (Model, error)

type Request struct {
	Messages          []Message
	Tools             []ToolDefinition
	System            string
	Model             string
	SessionID         string
	MaxTokens         int
	Temperature       *float64
	EnablePromptCache bool // Enable prompt caching for system and recent messages
}

type Response struct {
	Message    Message
	Usage      Usage
	StopReason string
}

type StreamHandler func(StreamResult) error

type StreamOnlyModel struct {
	Inner Model
}

type StreamOnlyProvider struct {
	Inner Provider
}

type StreamResult struct {
	Delta    string
	ToolCall *ToolCall
	Final    bool
	Response *Response
}

type ToolCall struct {
	ID        string
	Name      string
	Arguments map[string]any
	Result    string // Result stores the execution result for this specific tool call
}

type ToolDefinition struct {
	Name        string
	Description string
	Parameters  map[string]any
}

type Usage struct {
	InputTokens         int
	OutputTokens        int
	TotalTokens         int
	CacheReadTokens     int
	CacheCreationTokens int
}