provider

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"`
}

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

type ModelProfile struct {
	ContextWindow    int
	MaxOutputTokens  int
	ThinkingType     string
	ReasoningEffort  string
	InputPrice       float64
	CachedInputPrice float64
	OutputPrice      float64
}

type PromptTokensDetails struct {
	CachedTokens int `json:"cached_tokens"`
}

type ProviderError struct {
	StatusCode int
	Body       string
}

type StreamEvent struct {
	Type           string          `json:"type"`
	Delta          string          `json:"delta,omitempty"`
	ReasoningDelta string          `json:"reasoningDelta,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"`
}