provider

type CacheControl struct {
	Type string `json:"type"`
	TTL  string `json:"ttl,omitempty"`
}

type ChatClient interface {
	Chat(ctx context.Context, req ChatRequest) (<-chan StreamEvent, error)
	Model() string
	ContextWindow() int
	Pricing() (float64, float64, float64)
}

type ChatRequest struct {
	Model           string               `json:"model"`
	Messages        []message.Message    `json:"messages"`
	Tools           []message.ToolDef    `json:"tools,omitempty"`
	Stream          bool                 `json:"stream"`
	MaxTokens       int                  `json:"max_tokens,omitempty"`
	StreamOptions   *StreamOptions       `json:"stream_options,omitempty"`
	Thinking        *ThinkingConfig      `json:"thinking,omitempty"`
	ReasoningEffort string               `json:"reasoning_effort,omitempty"`
	Reasoning       *ReasoningConfig     `json:"reasoning,omitempty"`
	Models          []string             `json:"models,omitempty"`
	Route           string               `json:"route,omitempty"`
	ProviderPrefs   *ProviderPreferences `json:"provider,omitempty"`
	CacheControl    *CacheControl        `json:"cache_control,omitempty"`
	SessionID       string               `json:"session_id,omitempty"`
}

type Client struct {
	Profile ModelProfile
	// contains filtered or unexported fields
}

type CompletionTokensDetails struct {
	ReasoningTokens int `json:"reasoning_tokens,omitempty"`
}

type ModelProfile struct {
	ContextWindow      int
	MaxOutputTokens    int
	ThinkingType       string
	ReasoningEffort    string
	ReasoningMaxTokens int
	OpenRouter         bool
	InputPrice         float64
	CachedInputPrice   float64
	OutputPrice        float64
	FallbackModels     []string
	Route              string
	ProviderPrefs      *ProviderPreferences
	PromptCache        bool
	PromptCacheTTL     string
}

type PromptTokensDetails struct {
	CachedTokens     int `json:"cached_tokens"`
	CacheWriteTokens int `json:"cache_write_tokens,omitempty"`
}

type ProviderError struct {
	StatusCode int
	Body       string
}

type ProviderPreferences struct {
	Order             []string `json:"order,omitempty"`
	AllowFallbacks    *bool    `json:"allow_fallbacks,omitempty"`
	RequireParameters *bool    `json:"require_parameters,omitempty"`
	DataCollection    string   `json:"data_collection,omitempty"`
	Only              []string `json:"only,omitempty"`
	Ignore            []string `json:"ignore,omitempty"`
	Quantizations     []string `json:"quantizations,omitempty"`
	Sort              string   `json:"sort,omitempty"`
}

type ReasoningConfig struct {
	Effort    string `json:"effort,omitempty"`
	MaxTokens int    `json:"max_tokens,omitempty"`
	Exclude   *bool  `json:"exclude,omitempty"`
	Enabled   *bool  `json:"enabled,omitempty"`
}

type StreamEvent struct {
	Type             string                    `json:"type"`
	Delta            string                    `json:"delta,omitempty"`
	ReasoningDelta   string                    `json:"reasoningDelta,omitempty"`
	ReasoningDetails []message.ReasoningDetail `json:"reasoningDetails,omitempty"`
	ToolCall         *StreamToolCall           `json:"toolCall,omitempty"`
	Usage            *Usage                    `json:"usage,omitempty"`
	FinishReason     string                    `json:"finishReason,omitempty"`
	Error            error                     `json:"-"`
}

type StreamOptions struct {
	IncludeUsage bool `json:"include_usage"`
}

type StreamToolCall struct {
	ID        string
	Name      string
	Arguments string
}

type ThinkingConfig struct {
	Type string `json:"type"`
}

type Usage struct {
	PromptTokens            int                      `json:"prompt_tokens"`
	CompletionTokens        int                      `json:"completion_tokens"`
	TotalTokens             int                      `json:"total_tokens"`
	PromptCacheHitTokens    int                      `json:"prompt_cache_hit_tokens"`
	PromptTokensDetails     *PromptTokensDetails     `json:"prompt_tokens_details"`
	CompletionTokensDetails *CompletionTokensDetails `json:"completion_tokens_details,omitempty"`
	Cost                    float64                  `json:"cost,omitempty"`
}