chat

type Document struct {
	// Name is the display name of the document (e.g. "report.pdf").
	Name string `json:"name"`

	// MimeType is the post-processing MIME type of the document. For images
	// this is always "image/jpeg" or "image/png" regardless of the original
	// format. For text files it is the exact MIME (e.g. "text/plain",
	// "text/markdown", "text/html"). For binary documents it is the original
	// MIME (e.g. "application/pdf").
	MimeType string `json:"mime_type"`

	// Size is the byte length of the document content (InlineData or InlineText).
	// Optional; zero means unknown.
	Size int64 `json:"size,omitempty"`

	// Source holds the actual document content.
	Source DocumentSource `json:"source"`
}

type DocumentSource struct {
	// InlineText holds the raw text for text/* MIME types (TXT, MD, HTML, CSV, …).
	// Used for StrategyTXT attachments.
	InlineText string `json:"inline_text,omitempty"`

	// InlineData holds binary content (images, PDFs, Office docs, …) that is
	// base64-encoded when sent to the provider. Used for StrategyB64 attachments.
	InlineData []byte `json:"inline_data,omitempty"`
}

type FinishReason string

type ImageResizeResult struct {
	// Data is the (possibly re-encoded) image bytes.
	Data []byte
	// MimeType is the MIME type of the output image.
	MimeType string
	// OriginalWidth and OriginalHeight are the dimensions of the input image.
	OriginalWidth, OriginalHeight int
	// Width and Height are the dimensions of the output image.
	Width, Height int
	// Resized indicates whether the image was actually modified.
	Resized bool
}

type ImageURLDetail string

type Message struct {
	Role         MessageRole   `json:"role"`
	Content      string        `json:"content"`
	MultiContent []MessagePart `json:"multi_content,omitempty"`

	// This property is used for the "reasoning" feature supported by deepseek-reasoner
	// which is not in the official documentation.
	// the doc from deepseek:
	// - https://api-docs.deepseek.com/api/create-chat-completion#responses
	ReasoningContent string `json:"reasoning_content,omitempty"`

	// ThinkingSignature is used for Anthropic's extended thinking feature
	ThinkingSignature string `json:"thinking_signature,omitempty"`

	ThoughtSignature []byte `json:"thought_signature,omitempty"`

	FunctionCall *tools.FunctionCall `json:"function_call,omitempty"`

	// For Role=assistant prompts this may be set to the tool calls generated by the model, such as function calls.
	ToolCalls []tools.ToolCall `json:"tool_calls,omitempty"`

	// ToolDefinitions contains the definitions of tools referenced in ToolCalls.
	// This is used to provide tool metadata (name, description, category) when loading historical sessions.
	ToolDefinitions []tools.Tool `json:"tool_definitions,omitempty"`

	// For Role=tool prompts this should be set to the ID given in the assistant's prior request to call a tool.
	ToolCallID string `json:"tool_call_id,omitempty"`

	// IsError indicates the tool call failed (only for Role=tool messages).
	IsError bool `json:"is_error,omitempty"`

	CreatedAt string `json:"created_at,omitempty"`

	// Usage tracks token usage for this message (only set for assistant messages)
	Usage *Usage `json:"usage,omitempty"`

	// Model is the model that generated this message (only set for assistant messages)
	Model string `json:"model,omitempty"`

	// Cost is the cost of this message in dollars (only set for assistant messages)
	Cost float64 `json:"cost,omitempty"`

	// FinishReason indicates why the model stopped generating for this message.
	// "stop" = natural end, "tool_calls" = tool invocation, "length" = token limit.
	// Only set for assistant messages.
	FinishReason FinishReason `json:"finish_reason,omitempty"`

	// CacheControl indicates whether this message is a cached message (only used by anthropic)
	CacheControl bool `json:"cache_control,omitempty"`
}

type MessageDelta struct {
	Role              string              `json:"role,omitempty"`
	Content           string              `json:"content,omitempty"`
	ReasoningContent  string              `json:"reasoning_content,omitempty"`
	ThinkingSignature string              `json:"thinking_signature,omitempty"`
	ThoughtSignature  []byte              `json:"thought_signature,omitempty"`
	FunctionCall      *tools.FunctionCall `json:"function_call,omitempty"`
	ToolCalls         []tools.ToolCall    `json:"tool_calls,omitempty"`
}

type MessageFile struct {
	Path     string `json:"path,omitempty"`      // Local file path (used for upload)
	FileID   string `json:"file_id,omitempty"`   // Provider-specific file ID (after upload)
	MimeType string `json:"mime_type,omitempty"` // MIME type of the file
}

type MessageImageURL struct {
	URL    string         `json:"url,omitempty"`
	Detail ImageURLDetail `json:"detail,omitempty"`
}

type MessagePart struct {
	Type MessagePartType `json:"type,omitempty"`
	Text string          `json:"text,omitempty"`
	// Note: superseded by Document+MessagePartTypeDocument. Will be removed in a future release.
	ImageURL *MessageImageURL `json:"image_url,omitempty"`
	// Note: superseded by Document+MessagePartTypeDocument. Will be removed in a future release.
	File *MessageFile `json:"file,omitempty"`
	// Document is set when Type is MessagePartTypeDocument.
	Document *Document `json:"document,omitempty"`
}

type MessagePartType string

type MessageRole string

type MessageStream interface {
	// Recv gets the next completion chunk
	Recv() (MessageStreamResponse, error)
	// Close closes the stream
	Close()
}

type MessageStreamChoice struct {
	Index        int          `json:"index"`
	Delta        MessageDelta `json:"delta"`
	FinishReason FinishReason `json:"finish_reason,omitempty"`
}

type MessageStreamResponse struct {
	ID      string                `json:"id"`
	Object  string                `json:"object"`
	Created int64                 `json:"created"`
	Model   string                `json:"model"`
	Choices []MessageStreamChoice `json:"choices"`
	Usage   *Usage                `json:"usage,omitempty"`
}

type Usage struct {
	InputTokens       int64 `json:"input_tokens"`
	OutputTokens      int64 `json:"output_tokens"`
	CachedInputTokens int64 `json:"cached_input_tokens"`
	CacheWriteTokens  int64 `json:"cached_write_tokens"`
	ReasoningTokens   int64 `json:"reasoning_tokens,omitempty"`
}