provider

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

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
}

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