Documentation
¶
Index ¶
- func Bm25Retrieve(ctx context.Context, client *elasticsearch.Client, ...) ([]*schema.Document, error)
- func BuildRetriever(ctx context.Context, conf *config.Config) (r compose.Runnable[string, []*schema.Document], err error)
- func EsHit2Document(ctx context.Context, hit types.Hit) (doc *schema.Document, err error)
- func NewQdrantRetriever(ctx context.Context, config *QdrantRetrieverConfig) (retriever.Retriever, error)
- func RRFFusion(inputs [][]*schema.Document) []*schema.Document
- type QdrantRetriever
- type QdrantRetrieverConfig
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func Bm25Retrieve ¶
func Bm25Retrieve(ctx context.Context, client *elasticsearch.Client, indexName, knowledgeName, query string, excludeIDs []string, topK int) ([]*schema.Document, error)
Bm25Retrieve 使用 ES BM25 对 content 字段做全文检索。 仅在使用 ES 向量存储时调用,复用同一索引中的文档。
func BuildRetriever ¶
func EsHit2Document ¶
func NewQdrantRetriever ¶
func NewQdrantRetriever(ctx context.Context, config *QdrantRetrieverConfig) (retriever.Retriever, error)
NewQdrantRetriever 创建 Qdrant retriever(自定义实现,支持命名向量)
Types ¶
type QdrantRetriever ¶
type QdrantRetriever struct {
// contains filtered or unexported fields
}
QdrantRetriever Qdrant retriever 实现(支持命名向量和过滤)
func (*QdrantRetriever) GetType ¶
func (r *QdrantRetriever) GetType() string
GetType 返回 retriever 类型
type QdrantRetrieverConfig ¶
type QdrantRetrieverConfig struct {
Client *qdrant.Client // Required: Qdrant client
Collection string // Required: Collection name
Embedding embedding.Embedder // Required: Query embedding component
VectorField string // Optional: Vector field name (for named vectors)
ScoreThreshold *float64 // Optional: Score threshold
TopK int // Optional: Number of results (default: 10)
}
QdrantRetrieverConfig Qdrant retriever 配置
Click to show internal directories.
Click to hide internal directories.