vikingowl
d37cc2dad3
Extends the Content discriminated union with a fifth variant for inline image payloads. Image carries the raw bytes (captured at user-input time so the message snapshot is self-contained and survives source-file deletion), the IANA media type for the provider's image part, and the original path for logging. HasImages() lets providers decide whether to fall back to a text-only representation; providers that don't know about ContentImage will simply skip those blocks via TextContent().
98 lines
2.5 KiB
Go
98 lines
2.5 KiB
Go
package message
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
)
|
|
|
|
// ContentType discriminates the content block union.
|
|
type ContentType int
|
|
|
|
const (
|
|
ContentText ContentType = iota + 1
|
|
ContentToolCall
|
|
ContentToolResult
|
|
ContentThinking
|
|
ContentImage
|
|
)
|
|
|
|
func (ct ContentType) String() string {
|
|
switch ct {
|
|
case ContentText:
|
|
return "text"
|
|
case ContentToolCall:
|
|
return "tool_call"
|
|
case ContentToolResult:
|
|
return "tool_result"
|
|
case ContentThinking:
|
|
return "thinking"
|
|
case ContentImage:
|
|
return "image"
|
|
default:
|
|
return fmt.Sprintf("unknown(%d)", ct)
|
|
}
|
|
}
|
|
|
|
// Content is a discriminated union. Exactly one payload field is set per Type.
|
|
type Content struct {
|
|
Type ContentType
|
|
Text string // ContentText
|
|
ToolCall *ToolCall // ContentToolCall
|
|
ToolResult *ToolResult // ContentToolResult
|
|
Thinking *Thinking // ContentThinking
|
|
Image *Image // ContentImage
|
|
}
|
|
|
|
// ToolCall represents the model's request to invoke a tool.
|
|
type ToolCall struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Arguments json.RawMessage `json:"arguments"`
|
|
}
|
|
|
|
// ToolResult is the output of executing a tool, correlated by ToolCallID.
|
|
type ToolResult struct {
|
|
ToolCallID string `json:"tool_call_id"`
|
|
Content string `json:"content"`
|
|
IsError bool `json:"is_error"`
|
|
}
|
|
|
|
// Thinking represents a reasoning/thinking trace.
|
|
// Signature must round-trip unchanged (Anthropic requirement).
|
|
type Thinking struct {
|
|
Text string `json:"text,omitempty"`
|
|
Signature string `json:"signature,omitempty"`
|
|
Redacted bool `json:"redacted,omitempty"`
|
|
}
|
|
|
|
// Image carries inline image bytes for vision-capable models. Data is the
|
|
// raw image bytes captured at user-input time so the message snapshot is
|
|
// self-contained (file deletion or rename after the turn does not break
|
|
// translation). MediaType is the IANA media type (e.g. "image/png").
|
|
// Path is retained for human-readable display and logging only.
|
|
type Image struct {
|
|
Data []byte `json:"data"`
|
|
MediaType string `json:"media_type"`
|
|
Path string `json:"path,omitempty"`
|
|
}
|
|
|
|
func NewTextContent(text string) Content {
|
|
return Content{Type: ContentText, Text: text}
|
|
}
|
|
|
|
func NewToolCallContent(tc ToolCall) Content {
|
|
return Content{Type: ContentToolCall, ToolCall: &tc}
|
|
}
|
|
|
|
func NewToolResultContent(tr ToolResult) Content {
|
|
return Content{Type: ContentToolResult, ToolResult: &tr}
|
|
}
|
|
|
|
func NewThinkingContent(th Thinking) Content {
|
|
return Content{Type: ContentThinking, Thinking: &th}
|
|
}
|
|
|
|
func NewImageContent(img Image) Content {
|
|
return Content{Type: ContentImage, Image: &img}
|
|
}
|