schema

type Content struct {
	Id     string         `json:"id,omitempty"`
	Type   string         `json:"type" writer:",width:4"`
	Text   string         `json:"text,omitempty" writer:",width:60,wrap"`
	Source *contentSource `json:"source,omitempty"`
	Url    *contentImage  `json:"image_url,omitempty"`

	// Tool Result
	ToolId string `json:"tool_use_id,omitempty"`
	Result string `json:"content,omitempty"`
	// contains filtered or unexported fields
}

type Embedding struct {
	Embedding []float64 `json:"embedding"`
	Index     int       `json:"index"`
}

type Embeddings struct {
	Id    string      `json:"id" writer:",width:32"`
	Data  []Embedding `json:"data" writer:",wrap"`
	Model string      `json:"model"`
	Usage struct {
		PromptTokerns int `json:"prompt_tokens"`
		TotalTokens   int `json:"total_tokens"`
	} `json:"usage" writer:",wrap"`
}

type Message struct {
	// user, system or assistant
	Role string `json:"role,omitempty"`

	// Message Id
	Id string `json:"id,omitempty"`

	// Model
	Model string `json:"model,omitempty"`

	// Content can be a string, array of strings, content
	// object or an array of content objects
	Content any `json:"content,omitempty"`

	// Any tool calls
	ToolCalls []ToolCall `json:"tool_calls,omitempty"`

	// Tool Call Id
	ToolCallId string `json:"tool_call_id,omitempty"`

	// Time the message was created, in unix seconds
	Created int64 `json:"created,omitempty"`
}

type MessageChoice struct {
	Message      *Message      `json:"message,omitempty"`
	Delta        *MessageDelta `json:"delta,omitempty"`
	Index        int           `json:"index"`
	FinishReason string        `json:"finish_reason,omitempty"`
}

type MessageChunk struct {
	Id                string          `json:"id,omitempty"`
	Model             string          `json:"model,omitempty"`
	Created           int64           `json:"created,omitempty"`
	SystemFingerprint string          `json:"system_fingerprint,omitempty"`
	TokenUsage        *TokenUsage     `json:"usage,omitempty"`
	Choices           []MessageChoice `json:"choices,omitempty"`
}

type MessageDelta struct {
	Role    string `json:"role,omitempty"`
	Content string `json:"content,omitempty"`
}

type Model struct {
	Id      string `json:"id" writer:",width:30"`
	Created int64  `json:"created,omitempty"`
	Owner   string `json:"owned_by,omitempty"`
}

type TokenUsage struct {
	PromptTokens     int `json:"prompt_tokens,omitempty"`
	CompletionTokens int `json:"completion_tokens,omitempty"`
	TotalTokens      int `json:"total_tokens,omitempty"`
}

type Tool struct {
	Name        string          `json:"name"`
	Description string          `json:"description"`
	Type        string          `json:"type,omitempty"`
	Parameters  *toolParameters `json:"parameters,omitempty"`   // Used by OpenAI, Mistral
	InputSchema *toolParameters `json:"input_schema,omitempty"` // Used by anthropic
}

type ToolCall struct {
	Id       string       `json:"id,omitempty"`
	Type     string       `json:"type,omitempty"`
	Function ToolFunction `json:"function,omitempty"`
}

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