clients

Documentation

Index

• func New(apiKey string, timeout time.Duration, logger logrus.FieldLogger) *client
• type Data
• type Document
• type RankInput
• type RankResponse
• type Usage

Functions

func New

func New(apiKey string, timeout time.Duration, logger logrus.FieldLogger) *client

Types

type Data

type Data struct {
	Index          int      `json:"index"`
	RelevanceScore float64  `json:"relevance_score"`
	Document       Document `json:"document"`
}

type Document

type Document struct {
	Text string `json:"text"`
}

type RankInput

type RankInput struct {
	Documents       []string `json:"documents"`
	Query           string   `json:"query"`
	Model           string   `json:"model"`
	ReturnDocuments bool     `json:"return_documents"`
	Truncation      bool     `json:"truncation"`
}

type RankResponse

type RankResponse struct {
	ID    string `json:"id"`
	Data  []Data `json:"data"`
	Usage Usage  `json:"usage"`
}

type Usage

type Usage struct {
	TotalTokens int `json:"total_tokens"`
}