schema

type ContentPart struct {
	Type       string      `json:"type"`                  // 内容类型，见 ContentType* 常量
	Text       string      `json:"text,omitempty"`        // type="text"
	ImageURL   *ImageURL   `json:"image_url,omitempty"`   // type="image_url"
	InputAudio *InputAudio `json:"input_audio,omitempty"` // type="input_audio"
	VideoURL   *MediaURL   `json:"video_url,omitempty"`   // type="video_url"
	FileURL    *MediaURL   `json:"file_url,omitempty"`    // type="file_url"
	InlineData *InlineData `json:"inline_data,omitempty"` // type="inline_data"
}

type Document struct {
	ID       string         `json:"id"`                  // 唯一标识
	Content  string         `json:"content"`             // 文本内容
	MetaData map[string]any `json:"meta_data,omitempty"` // 开放元数据
}

type FunctionCall struct {
	Name      string `json:"name,omitempty"`
	Arguments string `json:"arguments,omitempty"`
}

type ImageURL struct {
	URL    string `json:"url"`              // http(s)://... 或 data:image/png;base64,...
	Detail string `json:"detail,omitempty"` // "low", "high", "auto"
}

type IndexItem interface {
	GetID() string
	GetContent() string
	GetEmbedding() []float32
}

type InlineData struct {
	MediaType string `json:"media_type"` // MIME 类型："image/png", "audio/mp3", "video/mp4", "application/pdf"
	Data      string `json:"data"`       // base64 编码的数据
}

type InputAudio struct {
	Data   string `json:"data"`   // base64 编码的音频数据
	Format string `json:"format"` // 音频格式："wav", "mp3"
}

type MediaURL struct {
	URL string `json:"url"` // http(s)://... 或 data:xxx;base64,...
}

type Message struct {
	Role             RoleType `json:"role"`
	Content          string   `json:"content,omitempty"`
	ReasoningContent string   `json:"reasoning_content,omitempty"`
	// 消息发送者的名称（可选）
	Name string `json:"name,omitempty"`
	// 当设置为 true 时，表示这条消息是未完成的，模型需要继续生成这条消息的剩余内容。（可选）
	Partial   bool       `json:"partial,omitempty"`
	ToolCalls []ToolCall `json:"tool_calls,omitempty"`

	// 多模态内容：当非空时，序列化由 adapter 处理，Content 字段被忽略
	// JSON 序列化标记为 "-"，各 provider adapter 自行转换为对应格式
	ContentParts []ContentPart `json:"-"`

	// 输出侧多模态：模型生成的非文本内容
	OutputImages []OutputImage `json:"output_images,omitempty"`
	OutputAudio  *OutputAudio  `json:"output_audio,omitempty"`

	// tool 消息专用：关联到哪个 ToolCall
	ToolCallID string `json:"tool_call_id,omitempty"`

	// tool 消息专用：标记工具执行是否出错（Anthropic API 原生支持 is_error）
	IsError bool `json:"is_error,omitempty"`

	Usage *Usage `json:"-"`
}

type OutputAudio struct {
	Data   string `json:"data"`   // base64 编码的音频数据
	Format string `json:"format"` // 音频格式："mp3", "wav", "opus"
}

type OutputImage struct {
	URL           string `json:"url,omitempty"`            // 图片 URL
	Base64        string `json:"base64,omitempty"`         // base64 编码的图片数据
	RevisedPrompt string `json:"revised_prompt,omitempty"` // 模型修订后的 prompt（如 DALL·E）
}

type RoleType string

type StreamReader struct {
	Usage Usage
	// contains filtered or unexported fields
}

type Tool struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	Parameters  any    `json:"parameters"` // JSON Schema 对象
}

type ToolCall struct {
	ID       string       `json:"id"`
	Type     string       `json:"type"`
	Index    int          `json:"index,omitempty"`
	Function FunctionCall `json:"function"`
}

type ToolResult struct {
	CallID       string        `json:"call_id"`
	Content      string        `json:"content"`
	IsError      bool          `json:"is_error"`
	ContentParts []ContentPart `json:"content_parts,omitempty"` // 多模态工具结果
}

type ToolResultContent struct {
	Content      string        // 文本内容
	ContentParts []ContentPart // 多模态内容片段
}

type Usage struct {
	PromptTokens        uint64 `json:"prompt_tokens"`
	CompletionTokens    uint64 `json:"completion_tokens"`
	TotalTokens         uint64 `json:"total_tokens"`
	CachedTokens        uint64 `json:"cached_tokens"`
	PromptTokensDetails struct {
		CachedTokens uint64 `json:"cached_tokens"`
	} `json:"prompt_tokens_details"`
}