vectordb

type Match struct {
	Vector
	Score float32
	Doc   []byte
}

type Upsert struct {
	Vector
	Doc []byte
}

type Vector struct {
	ID                   string            `protobuf:"bytes,1,opt,name=ID,proto3" json:"ID,omitempty"`
	Values               []float32         `protobuf:"fixed32,2,rep,packed,name=Values,proto3" json:"Values,omitempty"`
	MD5                  []byte            `protobuf:"bytes,3,opt,name=MD5,proto3" json:"MD5,omitempty"`
	Meta                 map[string]string `` /* 149-byte string literal not displayed */
	XXX_NoUnkeyedLiteral struct{}          `json:"-"`
	XXX_unrecognized     []byte            `json:"-"`
	XXX_sizecache        int32             `json:"-"`
}

type VectorDB interface {
	// ListRepos list repos
	ListRepos(ctx context.Context) ([]string, error)

	// DeleteRepo delete repo
	DeleteRepo(ctx context.Context, repos string) error

	// Upsert 插入or更新
	Upsert(ctx context.Context, repo string, v *Upsert) error

	// Delete 删除
	Delete(ctx context.Context, repo string, id ...string) error

	// Range 遍历所有
	Range(ctx context.Context, repo string, f func(*Vector)) error

	// Search 搜索向量相似度最高的topk
	Search(ctx context.Context, repo string, v []float32, topK int) ([]*Match, error)
}