vdb

type Document struct {
	ID     string    `json:"id"`
	Query  string    `json:"query"`
	Answer string    `json:"answer"`
	Score  []float64 `json:"score"`
	Scores []float32 `json:"scores"`
	Vector []float64 `json:"vector"`
}

type SearchOption func(*SearchOptions)

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

type VDB interface {
	Store(context.Context, string, []float64) error

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

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