store

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

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

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
}

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

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

	GrafanaVectorAPI GrafanaVectorAPISettings `json:"grafanaVectorAPI"`

	Qdrant qdrantSettings `json:"qdrant"`
}

type VectorStore interface {
	ReadVectorStore
	WriteVectorStore
}

type VectorStoreAuthType string

type VectorStoreType string

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
}