talk

type ImageContent struct {
	MediaType string
	Data      string
}

type LLMClient interface {
	Chat(ctx context.Context, req *Request, fn func(Response) error) error
}

type Message struct {
	Role       Role
	Content    string
	Images     []ImageContent // Inline images for a vision request; only meaningful on a user message.
	Thinking   string
	ToolCalls  []ToolCall
	ToolCallID string // Correlates a tool result with its originating ToolCall.
}

type Request struct {
	Model         string
	Messages      []Message
	Tools         []tool.ToolDef
	Stream        bool
	Think         *bool
	Options       map[string]any
	MaxIterations int // Maximum tool-call loop iterations. Defaults to 20 if 0.
	MaxTokens     int // Maximum estimated token budget for messages. 0 means no limit.
}

type RequestOption func(*Request)

type Response struct {
	Content   string
	Thinking  string
	Done      bool
	ToolCalls []ToolCall
	Usage     *Usage // non-nil on the final chunk (Done=true) when the provider reports it
}

type Role string

type StatusError struct {
	StatusCode int
	Body       string
	Provider   string
}

type ToolCall struct {
	ID        string // Provider-assigned correlation ID (e.g. OpenAI tool_call_id).
	Name      string
	Arguments map[string]any
}

type Usage struct {
	InputTokens  int
	OutputTokens int

	// Cache stats (zero when the provider does not support caching).
	CacheCreationInputTokens int
	CacheReadInputTokens     int
}