adapter

package
v1.11.1 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Apr 17, 2026 License: MIT Imports: 6 Imported by: 0

Documentation

Overview

Package adapter 提供 RAG 层到 agent/workflow 层的适配器

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type HybridRetrieverAdapter

type HybridRetrieverAdapter struct {
	// contains filtered or unexported fields
}

HybridRetrieverAdapter 将 HybridRetriever 适配为 workflow/steps 需要的接口

func NewHybridRetrieverAdapter

func NewHybridRetrieverAdapter(retriever *rag.HybridRetriever) *HybridRetrieverAdapter

NewHybridRetrieverAdapter 创建适配器

func (*HybridRetrieverAdapter) Retrieve

func (a *HybridRetrieverAdapter) Retrieve(ctx context.Context, query string, queryEmbedding []float64) ([]types.RetrievalRecord, error)

Retrieve 实现 workflow/steps.HybridRetriever 接口

func (*HybridRetrieverAdapter) Retriever

func (a *HybridRetrieverAdapter) Retriever() *rag.HybridRetriever

Retriever 返回底层的 HybridRetriever 实例

type RetrievalToolAdapter

type RetrievalToolAdapter struct {
	// contains filtered or unexported fields
}

RetrievalToolAdapter 将 RAG Pipeline 适配为 agent 层可用的检索接口

func NewRetrievalToolAdapter

func NewRetrievalToolAdapter(pipeline *retrieval.Pipeline, embedder rag.EmbeddingProvider, collection string) *RetrievalToolAdapter

NewRetrievalToolAdapter 从 Pipeline 创建适配器

func (*RetrievalToolAdapter) Collection

func (a *RetrievalToolAdapter) Collection() string

Collection 返回集合名称

func (*RetrievalToolAdapter) Embedder

Embedder 返回 embedding provider

func (*RetrievalToolAdapter) Pipeline

func (a *RetrievalToolAdapter) Pipeline() *retrieval.Pipeline

Pipeline 返回底层的 Pipeline 实例

func (*RetrievalToolAdapter) Retrieve

func (a *RetrievalToolAdapter) Retrieve(ctx context.Context, query string, topK int) ([]types.RetrievalRecord, error)

Retrieve 实现 agent 层需要的检索接口,返回 types.RetrievalRecord

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL