ent

type DocumentScore struct {
	Document string
	Score    float64
}

type RankResult struct {
	Query          string
	DocumentScores []DocumentScore
}

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