model

type Capabilities struct {
	Provider     string                       `json:"provider"`
	Model        string                       `json:"model"`
	Features     map[string]domain.Capability `json:"features"`
	BillingRoute string                       `json:"billing_route"`
}

type Config struct {
	Endpoint string
	APIKey   string `json:"-"`
	Model    string
	Client   HTTPClient `json:"-"`
	Profile  string
}

type Event struct {
	Type         string          `json:"type"`
	ID           string          `json:"id"`
	Text         string          `json:"text,omitempty"`
	ToolCall     *ToolCall       `json:"tool_call,omitempty"`
	Usage        *Usage          `json:"usage,omitempty"`
	Continuation json.RawMessage `json:"-"`
	FinishReason string          `json:"finish_reason,omitempty"`
}

type HTTPClient interface {
	Do(*http.Request) (*http.Response, error)
}

type Message struct {
	Role       string          `json:"role"`
	Text       string          `json:"text"`
	ToolCalls  []ToolCall      `json:"tool_calls,omitempty"`
	ToolCallID string          `json:"tool_call_id,omitempty"`
	Opaque     json.RawMessage `json:"-"`
}

type Model struct {
	ID           string `json:"id"`
	Name         string `json:"name,omitempty"`
	ContextLimit int    `json:"context_limit"`
	OutputLimit  int    `json:"output_limit"`
}

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

type ModelProvider interface {
	Probe(context.Context) (Capabilities, error)
	ListModels(context.Context) ([]Model, error)
	StreamTurn(context.Context, TurnRequest, func(Event) error) error
	EstimateUsage(TurnRequest) Usage
}

type Tool struct {
	Name        string          `json:"name"`
	Description string          `json:"description"`
	Schema      json.RawMessage `json:"schema"`
}

type ToolCall struct {
	ID        string          `json:"id"`
	Name      string          `json:"name"`
	Arguments json.RawMessage `json:"arguments"`
	Invalid   string          `json:"invalid,omitempty"`
}

type ToolSchemas map[string]*jsonschema.Schema

type Turn struct {
	OperationID     string          `json:"operation_id"`
	Model           string          `json:"model"`
	System          string          `json:"system"`
	Messages        []Message       `json:"messages"`
	Tools           []Tool          `json:"tools"`
	MaxOutputTokens int             `json:"max_output_tokens"`
	Continuation    json.RawMessage `json:"continuation,omitempty"`
	TimeoutMS       int64           `json:"timeout_ms"`
	Session         string          `json:"-"`
}

type TurnRequest struct {
	OperationID     string
	Model           string
	System          string
	Messages        []Message
	Tools           []Tool
	MaxOutputTokens int
	Continuation    json.RawMessage
	Session         string
}

type TurnResult struct {
	ProviderID   string          `json:"provider_id"`
	Text         string          `json:"text"`
	ToolCalls    []ToolCall      `json:"tool_calls"`
	Usage        Usage           `json:"usage"`
	Continuation json.RawMessage `json:"continuation,omitempty"`
	FinishReason string          `json:"finish_reason"`
}

type Usage struct {
	InputTokens         int64  `json:"input_tokens"`
	OutputTokens        int64  `json:"output_tokens"`
	CacheTokens         int64  `json:"cache_tokens"`
	CacheCreationTokens int64  `json:"cache_creation_tokens"`
	Known               bool   `json:"known"`
	Source              string `json:"source"`
}