dify-plugin-daemon/pkg/entities/model_entities/text_embedding.go

package model_entities

import "github.com/shopspring/decimal"

type EmbeddingUsage struct {
	Tokens      *int            `json:"tokens" validate:"required"`
	TotalTokens *int            `json:"total_tokens" validate:"required"`
	UnitPrice   decimal.Decimal `json:"unit_price" validate:"required"`
	PriceUnit   decimal.Decimal `json:"price_unit" validate:"required"`
	TotalPrice  decimal.Decimal `json:"total_price" validate:"required"`
	Currency    *string         `json:"currency" validate:"required"`
	Latency     *float64        `json:"latency" validate:"required"`
}

type TextEmbeddingResult struct {
	Model      string         `json:"model" validate:"required"`
	Embeddings [][]float64    `json:"embeddings" validate:"required,dive"`
	Usage      EmbeddingUsage `json:"usage" validate:"required"`
}

type GetTextEmbeddingNumTokensResponse struct {
	NumTokens []int `json:"num_tokens" validate:"required"`
}