openai

type APIMessage struct {
	Role             string            `json:"role"`
	Content          any               `json:"content"` // string 或 []schema.ContentPart
	ReasoningContent string            `json:"reasoning_content,omitempty"`
	Name             string            `json:"name,omitempty"`
	ToolCalls        []schema.ToolCall `json:"tool_calls,omitempty"`
	ToolCallID       string            `json:"tool_call_id,omitempty"`
}

type ChatModel struct {
	// contains filtered or unexported fields
}

type ChatModelConfig struct {
	BaseURL string
	APIKey  string
	Model   string

	MaxCompletionTokens uint64             `json:"max_completion_tokens,omitempty"`
	ResponseFormat      ResponseFormatType `json:"response_format,omitempty"`
	Stop                string             `json:"stop,omitempty"`
	Temperature         float64            `json:"temperature,omitempty"`
	TopP                float64            `json:"top_p,omitempty"`
	N                   uint8              `json:"n,omitempty"`
	PresencePenalty     float64            `json:"presence_penalty,omitempty"`
	FrequencyPenalty    float64            `json:"frequency_penalty,omitempty"`

	Tools    []schema.Tool `json:"-"`
	Thinking Thinking      `json:"thinking,omitempty"`

	PromptCacheKey   string `json:"prompt_cache_key,omitempty"`
	SafetyIdentifier string `json:"safety_identifier,omitempty"`

	Debug      bool // 启用请求日志（输出到 stderr）
	TimeOut    time.Duration
	HTTPClient *http.Client `json:"-"`
}

type ChatModelResponse struct {
	ID      string       `json:"id"`
	Object  string       `json:"object"`
	Created int64        `json:"created"`
	Model   string       `json:"model"`
	Choices []Choice     `json:"choices"`
	Usage   schema.Usage `json:"usage"`
}

type Choice struct {
	Index        int            `json:"index"`
	Message      schema.Message `json:"message"`
	FinishReason string         `json:"finish_reason"`
}

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

type Delta struct {
	Role             string            `json:"role,omitempty"`
	Content          string            `json:"content"`
	ReasoningContent string            `json:"reasoning_content,omitempty"`
	ToolCalls        []schema.ToolCall `json:"tool_calls,omitempty"`
}

type RequestBody struct {
	Model               string             `json:"model"`
	Messages            []APIMessage       `json:"messages"`
	MaxCompletionTokens uint64             `json:"max_completion_tokens,omitempty"`
	ResponseFormat      ResponseFormatType `json:"response_format,omitempty"`
	Stream              bool               `json:"stream,omitempty"`
	StreamOptions       *StreamOptions     `json:"stream_options,omitempty"`
	Tools               []Tool             `json:"tools,omitempty"`
	PromptCacheKey      string             `json:"prompt_cache_key,omitempty"`
	SafetyIdentifier    string             `json:"safety_identifier,omitempty"`
	Thinking            Thinking           `json:"thinking,omitempty"`
	Temperature         float64            `json:"temperature,omitempty"`
	TopP                float64            `json:"top_p,omitempty"`
	N                   uint8              `json:"n,omitempty"`
	PresencePenalty     float64            `json:"presence_penalty,omitempty"`
	FrequencyPenalty    float64            `json:"frequency_penalty,omitempty"`
}

type ResponseFormatType string

type StreamChoice struct {
	Index        int    `json:"index"`
	Delta        Delta  `json:"delta"`
	FinishReason string `json:"finish_reason"`
}

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

type StreamResponse struct {
	Choices []StreamChoice `json:"choices"`
	Usage   *schema.Usage  `json:"usage,omitempty"`
}

type Thinking struct {
	Type ThinkingType     `json:"type,omitempty"`
	Keep ThinkingKeepEnum `json:"keep,omitempty"`
}

type ThinkingKeepEnum string

type ThinkingType string

type Tool struct {
	Type     ToolType    `json:"type"`
	Function schema.Tool `json:"function"`
}

type ToolType string