vdb

Documentation

Index

• type Document
• type SearchOption
  • func WithThreshold(threshold float64) SearchOption
  • func WithTopk(topk int) SearchOption
• type SearchOptions
• type VDB

Types

type Document

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

Document 文档

type SearchOption

type SearchOption func(*SearchOptions)

SearchOption 搜索选项

func WithThreshold

func WithThreshold(threshold float64) SearchOption

WithThreshold 设置相似度阈值

func WithTopk

func WithTopk(topk int) SearchOption

WithTopk 设置返回结果数量

type SearchOptions

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

SearchOptions 搜索配置

type VDB

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
}

VDB ...