store

Documentation

Index

• type AuthSettings
• type GrafanaVectorAPISettings
• type ReadVectorStore
  • func NewReadVectorStore(s Settings, secrets map[string]string) (ReadVectorStore, context.CancelFunc, error)
• type SearchResult
• type Settings
• type VectorStore
  • func NewVectorStore(s Settings) (VectorStore, error)
• type VectorStoreAuthType
• type VectorStoreType
• type WriteVectorStore

Types

type AuthSettings

type AuthSettings struct {
	BasicAuthUser string `json:"basicAuthUser"`
}

type GrafanaVectorAPISettings

type GrafanaVectorAPISettings struct {
	URL           string `json:"url"`
	AuthType      string `json:"authType"`
	BasicAuthUser string `json:"basicAuthUser"`
}

type ReadVectorStore

type ReadVectorStore interface {
	CollectionExists(ctx context.Context, collection string) (bool, error)
	Search(ctx context.Context, collection string, vector []float32, topK uint64, filter map[string]interface{}) ([]SearchResult, error)
	Health(ctx context.Context) error
}

func NewReadVectorStore

func NewReadVectorStore(s Settings, secrets map[string]string) (ReadVectorStore, context.CancelFunc, error)

type SearchResult

type SearchResult struct {
	Payload map[string]any `json:"payload"`
	Score   float64        `json:"score"`
}

type Settings

type Settings struct {
	Type VectorStoreType `json:"type"`

	GrafanaVectorAPI GrafanaVectorAPISettings `json:"grafanaVectorAPI"`

	Qdrant qdrantSettings `json:"qdrant"`
}

type VectorStore

type VectorStore interface {
	ReadVectorStore
	WriteVectorStore
}

func NewVectorStore

func NewVectorStore(s Settings) (VectorStore, error)

type VectorStoreAuthType

type VectorStoreAuthType string

const (
	VectorStoreAuthTypeBasicAuth VectorStoreAuthType = "basic-auth"
)

type VectorStoreType

type VectorStoreType string

const (
	VectorStoreTypeQdrant           VectorStoreType = "qdrant"
	VectorStoreTypeGrafanaVectorAPI VectorStoreType = "grafana/vectorapi"
)

type WriteVectorStore

type WriteVectorStore interface {
	Collections(ctx context.Context) ([]string, error)
	CreateCollection(ctx context.Context, collection string, size uint64) error
	PointExists(ctx context.Context, collection string, id uint64) (bool, error)
	UpsertColumnar(ctx context.Context, collection string, ids []uint64, embeddings [][]float32, payloadJSONs []string) error
}