request

type Classify struct {
	Inputs   []string          `json:"inputs"`
	Examples []model.Example   `json:"examples,omitempty"`
	Model    *model.EmbedModel `json:"model,omitempty"`
	Preset   *string           `json:"preset,omitempty"`
	Truncate *model.Truncate   `json:"truncate,omitempty"`
}

type DetectLanguage struct {
	Texts []string `json:"texts"`
}

type Detokenize struct {
	Tokens []int64      `json:"tokens"`
	Model  *model.Model `json:"model,omitempty"`
}

type Embed struct {
	Texts    []string          `json:"texts"`
	Model    *model.EmbedModel `json:"model,omitempty"`
	Truncate *model.Truncate   `json:"truncate,omitempty"`
}

type Generate struct {
	Prompt            string                   `json:"prompt"`
	Model             *model.Model             `json:"model,omitempty"`
	NumGenerations    *int                     `json:"num_generations,omitempty"`
	MaxTokens         *int                     `json:"max_tokens,omitempty"`
	Preset            *string                  `json:"preset,omitempty"`
	Temperature       *float64                 `json:"temperature,omitempty"`
	K                 *int                     `json:"k,omitempty"`
	P                 *float64                 `json:"p,omitempty"`
	FrequencyPenalty  *float64                 `json:"frequency_penalty,omitempty"`
	PresencePenalty   *float64                 `json:"presence_penalty,omitempty"`
	EndSequences      []string                 `json:"end_sequences,omitempty"`
	StopSequences     []string                 `json:"stop_sequences,omitempty"`
	ReturnLikelihoods *model.ReturnLikelihoods `json:"return_likelihoods,omitempty"`
	LogitBias         model.LogitBias          `json:"logit_bias,omitempty"`
	Truncate          *model.Truncate          `json:"truncate,omitempty"`
	Stream            bool                     `json:"stream,omitempty"`
}

type Rerank struct {
	Query           string             `json:"query"`
	Model           *model.RerankModel `json:"model,omitempty"`
	Documents       []string           `json:"documents"`
	TopN            *int               `json:"top_n,omitempty"`
	ReturnDocuments bool               `json:"return_documents"`
	MaxChunksPerDoc *int               `json:"max_chunks_per_doc,omitempty"`
}

type Summarize struct {
	Text              string                `json:"text"`
	Length            *model.TextLength     `json:"length,omitempty"`
	Format            *model.TextFormat     `json:"format,omitempty"`
	Model             *model.Model          `json:"model,omitempty"`
	Extractiveness    *model.Extractiveness `json:"extractiveness,omitempty"`
	Temperature       *float64              `json:"temperature,omitempty"`
	AdditionalCommand *string               `json:"additional_command,omitempty"`
}

type Tokenize struct {
	Text  string       `json:"text"`
	Model *model.Model `json:"model,omitempty"`
}