cohere

type CohereClient struct {
	// contains filtered or unexported fields
}

type EmbeddingRequest struct {
	// ID of the model to use.
	Model string `json:"model"`

	Texts []string `json:"texts"`

	InputType string `json:"input_type,omitempty"`

	EmbeddingTypes []string `json:"embedding_types"`

	// Allowed values: NONE,START, END (default)
	// Passing START will discard the start of the input.
	// END will discard the end of the input. In both cases, input is discarded until the remaining input is
	// exactly the maximum input token length for the model.
	// If NONE is selected, when the input exceeds the maximum input token length an error will be returned.
	Truncate string `json:"truncate,omitempty"`
}

type EmbeddingResponse struct {
	Id         string     `json:"id"`
	Embeddings Embeddings `json:"embeddings"`
}

type Embeddings struct {
	Float [][]float32 `json:"float"`
	Int8  [][]int8    `json:"int8"`
}

type Meta struct {
	APIVersion struct {
		Version        string `json:"version"`
		IsExperimental bool   `json:"is_experimental"`
	} `json:"api_version"`
	BilledUnits struct {
		SearchUnits int `json:"search_units"`
	} `json:"billed_units"`
}

type RerankResponse struct {
	Results []RerankResult `json:"results"`
	Id      string         `json:"id"`
	Meta    Meta           `json:"meta"`
}

type RerankResult struct {
	Index          int     `json:"index"`
	RelevanceScore float32 `json:"relevance_score"`
}