response

type Chat struct {
	Response
	model.NonStreamedChat
	model.StreamedChat
	// contains filtered or unexported fields
}

type Classify struct {
	Response
	ID              string                 `json:"id"`
	Classifications []model.Classification `json:"classifications"`
}

type Detail struct {
	TypeURL *string `json:"typeUrl,omitempty"`
	Value   *string `json:"value,omitempty"`
}

type Detokenize struct {
	Response
	Text string `json:"text"`
}

type Embed struct {
	Response
	ID         string      `json:"id"`
	Embeddings [][]float64 `json:"embeddings"`
	Texts      []string    `json:"texts"`
}

type Generate struct {
	Response
	ID          string             `json:"id"`
	Prompt      *string            `json:"prompt"`
	Generations []model.Generation `json:"generations"`
}

type Headers struct {
	EndpointMonthlyCallLimit   int `json:"-"`
	RatelimitLimit             int `json:"-"`
	TrialEndpointCallRemaining int `json:"-"`
	RatelimitRemaining         int `json:"-"`
	RatelimitReset             int `json:"-"`
	TrialEndpointCallLimit     int `json:"-"`
}

type ListModels struct {
	Response
	Models   []Model `json:"models"`
	NextPage string  `json:"next_page"`
}

type Model struct {
	Response
	Name             string   `json:"name"`
	Endpoints        []string `json:"endpoints"`
	Finetuned        bool     `json:"finetuned"`
	ContextLength    int      `json:"context_length"`
	TokenizerURL     string   `json:"tokenizer_url"`
	DefaultEndpoints []string `json:"default_endpoints"`
}

type Rerank struct {
	Response
	Results []model.RerankResult `json:"results"`
}

type Response struct {
	Meta    model.Meta `json:"meta,omitempty"`
	Headers Headers    `json:"-"`
	Code    int        `json:"-"`
	RawBody *string    `json:"-"`
}

type Summarize struct {
	Response
	ID      string `json:"id"`
	Summary string `json:"summary"`
}

type Tokenize struct {
	Response
	Tokens       []int64  `json:"tokens"`
	TokenStrings []string `json:"token_strings"`
}