vdb

type Document struct {
	ID       string         `json:"id"`
	Title    string         `json:"title"`
	Content  string         `json:"content"`
	Metadata map[string]any `json:"metadata"`
}

type SearchOption func(*SearchOptions)

type SearchOptions struct {
	Topk      int     // 返回结果数量限制
	Threshold float64 // 相似度阈值
}

type VDB interface {
	Store(context.Context, []*Document) error

	Search(context.Context, []float64, ...SearchOption) ([]Document, error)
	SearchQuery(context.Context, string, ...SearchOption) ([]Document, error)

	Delete(context.Context, string) error
	Close() error
}