types

type BasicAuthCredentialsProvider struct {
	Username string
	Password string
}

type CollectionQueryBuilder struct {
	QueryTexts      []string
	QueryEmbeddings []*Embedding
	Where           map[string]interface{}
	WhereDocument   map[string]interface{}
	NResults        int32
	Include         []QueryEnum
	Offset          int32
	Limit           int32
	Ids             []string
}

type CollectionQueryOption func(*CollectionQueryBuilder) error

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

type CredentialsProvider interface {
	Authenticate(apiClient *openapi.Configuration) error
}

type DistanceFunction string

type Embedding struct {
	ArrayOfFloat32 *[]float32
	ArrayOfInt32   *[]int32
}

type EmbeddingFunction interface {
	// EmbedDocuments returns a vector for each text.
	EmbedDocuments(ctx context.Context, texts []string) ([]*Embedding, error)
	// EmbedQuery embeds a single text.
	EmbedQuery(ctx context.Context, text string) (*Embedding, error)
	EmbedRecords(ctx context.Context, records []*Record, force bool) error
}

type IDGenerator interface {
	Generate(document string) string
}

type InvalidEmbeddingValueError struct {
	Value interface{}
}

type InvalidMetadataValueError struct {
	Key   string
	Value interface{}
}

type Option func(*Record) error

type QueryEnum string

type Record struct {
	ID        string
	Embedding Embedding
	Metadata  map[string]interface{}
	Document  string
	URI       string
	// contains filtered or unexported fields
}

type RecordSet struct {
	Records           []*Record
	IDGenerator       IDGenerator
	EmbeddingFunction EmbeddingFunction
}

type RecordSetOption func(*RecordSet) error

type SHA256Generator struct{}

type TokenAuthCredentialsProvider struct {
	Token  string
	Header TokenTransportHeader
}

type TokenTransportHeader string

type ULIDGenerator struct{}

type UUIDGenerator struct{}