model

type APIError struct {
	StatusCode int
	Type       string
	Message    string
	Body       string
}

type Completer interface {
	Complete(ctx context.Context, req Request) (Response, error)
}

type FunctionCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"` // JSON 字符串
}

type FunctionSchema struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	Parameters  map[string]any `json:"parameters"` // JSON Schema（type=object + properties + required）
}

type Message struct {
	Role       string     `json:"role"`
	Content    string     `json:"content,omitempty"`
	ToolCalls  []ToolCall `json:"tool_calls,omitempty"`   // assistant 消息携带模型决定要调的工具
	ToolCallID string     `json:"tool_call_id,omitempty"` // tool 消息：对应哪一次工具调用
	Name       string     `json:"name,omitempty"`
}

type OpenAICompatibleProvider struct {
	BaseURL    string
	APIKey     string
	HTTPClient *http.Client
}

type Request struct {
	Model       string           `json:"-"`
	Messages    []Message        `json:"messages"`
	Temperature float64          `json:"temperature,omitempty"`
	Tools       []ToolDefinition `json:"tools,omitempty"`
	// ToolChoice 可为字符串（"auto"/"none"/"required"）或对象
	// （{"type":"function","function":{"name":"X"}} 强制调用某个函数）。
	ToolChoice any `json:"tool_choice,omitempty"`
}

type Response struct {
	Content      string
	ToolCalls    []ToolCall
	FinishReason string
	Usage        Usage
	Raw          []byte // 调试用；不必常驻
}

type ToolCall struct {
	ID       string       `json:"id"`
	Type     string       `json:"type"` // 目前固定 "function"
	Function FunctionCall `json:"function"`
}

type ToolDefinition struct {
	Type     string         `json:"type"` // "function"
	Function FunctionSchema `json:"function"`
}

type Usage struct {
	PromptTokens       int
	CompletionTokens   int
	TotalTokens        int
	CachedPromptTokens int
}