anthropic

type APIMessage struct {
	Role    string         `json:"role"`    // "user" 或 "assistant"
	Content []ContentBlock `json:"content"` // 内容块数组
}

type APIRequest struct {
	Model       string       `json:"model"`
	Messages    []APIMessage `json:"messages"`
	MaxTokens   int          `json:"max_tokens"`
	System      string       `json:"system,omitempty"` // 顶层系统提示
	Temperature float64      `json:"temperature,omitempty"`
	TopP        float64      `json:"top_p,omitempty"`
	TopK        int          `json:"top_k,omitempty"`
	Stream      bool         `json:"stream,omitempty"`
	Tools       []APITool    `json:"tools,omitempty"`
	Thinking    *Thinking    `json:"thinking,omitempty"`
}

type APIResponse struct {
	ID         string         `json:"id"`
	Type       string         `json:"type"`
	Role       string         `json:"role"`
	Content    []ContentBlock `json:"content"`
	Model      string         `json:"model"`
	StopReason string         `json:"stop_reason"`
	Usage      Usage          `json:"usage"`
}

type APITool struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	InputSchema map[string]any `json:"input_schema"`
}

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

type ChatModelConfig struct {
	BaseURL string
	APIKey  string
	Model   string

	// 模型参数
	MaxTokens   int     `json:"max_tokens"`
	Temperature float64 `json:"temperature,omitempty"`
	TopP        float64 `json:"top_p,omitempty"`
	TopK        int     `json:"top_k,omitempty"`

	// 工具
	Tools []schema.Tool `json:"-"`

	// Thinking（扩展思考）
	Thinking *Thinking `json:"thinking,omitempty"`

	// 流式
	Stream bool `json:"stream,omitempty"`

	// HTTP
	TimeOut    time.Duration
	HTTPClient *http.Client `json:"-"`
}

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

type ContentBlock struct {
	Type string `json:"type"` // "text", "tool_use", "tool_result", "thinking", "image", "document", "audio", "video"

	// type=text
	Text string `json:"text,omitempty"`

	// type=tool_use
	ID    string         `json:"id,omitempty"`
	Name  string         `json:"name,omitempty"`
	Input map[string]any `json:"input,omitempty"`

	// type=tool_result
	ToolUseID string         `json:"tool_use_id,omitempty"`
	Content   string         `json:"content,omitempty"` // tool_result 的文本内容
	IsError   bool           `json:"is_error,omitempty"`
	CntBlocks []ContentBlock `json:"content_blocks,omitempty"` // tool_result 的多模态内容

	// type=thinking
	Thinking string `json:"thinking,omitempty"`

	// type=image, document, audio, video
	Source *ContentSource `json:"source,omitempty"`
}

type ContentSource struct {
	Type      string `json:"type"`                 // "base64" 或 "url"
	MediaType string `json:"media_type,omitempty"` // "image/png", "application/pdf" 等
	Data      string `json:"data,omitempty"`       // base64 数据（type=base64 时）
	URL       string `json:"url,omitempty"`        // URL（type=url 时）
}

type StreamDelta struct {
	Type        string `json:"type"` // "text_delta", "thinking_delta", "input_json_delta"
	Text        string `json:"text,omitempty"`
	Thinking    string `json:"thinking,omitempty"`
	PartialJSON string `json:"partial_json,omitempty"`
}

type StreamEvent struct {
	Type         string        `json:"type"`
	ContentBlock *ContentBlock `json:"content_block,omitempty"`
	Delta        *StreamDelta  `json:"delta,omitempty"`
	Usage        *StreamUsage  `json:"usage,omitempty"`
}

type StreamUsage struct {
	InputTokens  int `json:"input_tokens"`
	OutputTokens int `json:"output_tokens"`
}

type Thinking struct {
	Type         ThinkingType `json:"type"`                    // "enabled"
	BudgetTokens int          `json:"budget_tokens,omitempty"` // 思考预算
}

type ThinkingType string

type Usage struct {
	InputTokens              int `json:"input_tokens"`
	OutputTokens             int `json:"output_tokens"`
	CacheCreationInputTokens int `json:"cache_creation_input_tokens,omitempty"`
	CacheReadInputTokens     int `json:"cache_read_input_tokens,omitempty"`
}