vectorstore.go

package rag

import "context"

// VectorStore defines the interface for vector storage and search.
// QdrantClient satisfies this interface.
type VectorStore interface {
	// CreateCollection creates a new vector collection with the given
	// name and vector dimensionality.
	CreateCollection(ctx context.Context, name string, vectorSize uint64) error

	// CollectionExists checks whether a collection with the given name exists.
	CollectionExists(ctx context.Context, name string) (bool, error)

	// DeleteCollection deletes the collection with the given name.
	DeleteCollection(ctx context.Context, name string) error

	// ListCollections returns all collection names in the store.
	ListCollections(ctx context.Context) ([]string, error)

	// CollectionInfo returns metadata about a collection. Implementations
	// should populate at least PointCount and VectorSize in the returned
	// CollectionInfo struct.
	CollectionInfo(ctx context.Context, name string) (*CollectionInfo, error)

	// UpsertPoints inserts or updates points in the named collection.
	UpsertPoints(ctx context.Context, collection string, points []Point) error

	// Search performs a vector similarity search, returning up to limit results.
	// One or more optional filter maps restrict results by payload field values.
	Search(ctx context.Context, collection string, vector []float32, limit uint64, filter ...map[string]string) ([]SearchResult, error)
}

// Vector represents an RFC-compatible vector payload for storage.
// It is equivalent to Point, but uses the Values field name from the spec.
type Vector struct {
	ID      string
	Values  []float32
	Payload map[string]any
}

// CollectionInfo holds backend-agnostic metadata about a collection.
type CollectionInfo struct {
	Name       string
	PointCount uint64
	VectorSize uint64
	Status     string // e.g. "green", "yellow", "red", "unknown"
	Count      uint64
	Vectors    uint64
	Index      string
}