embeddings

type EmbeddingData struct {
	Object    string             `json:"object"`
	Index     int                `json:"index"`
	Embedding EmbeddingDataUnion `json:"embedding"`
}

type EmbeddingDataUnion struct {
	OfFloat  []float64 `json:",omitempty"`
	OfBase64 *string   `json:",omitempty"`
}

type InputUnion struct {
	OfString *string  `json:",omitempty"`
	OfList   []string `json:",omitempty"`
}

type Request struct {
	Model          string         `json:"model"`
	Input          InputUnion     `json:"input"`
	Dimensions     *int           `json:"dimensions"`
	EncodingFormat *string        `json:"encoding_format"` // "float" or "base64"
	ExtraFields    map[string]any `json:",omitempty"`
}

type Response struct {
	Object string          `json:"object"`
	Model  string          `json:"model"`
	Usage  *Usage          `json:"usage"`
	Data   []EmbeddingData `json:"data"`
}

type Usage struct {
	PromptTokens int64 `json:"prompt_tokens"`
	TotalTokens  int64 `json:"total_tokens"`
}