retriever

package
v0.1.5 Latest Latest
Warning

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

Go to latest
Published: Aug 13, 2026 License: Apache-2.0 Imports: 14 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func CreateFileRetrievalTool

func CreateFileRetrievalTool(uploadsBase string) tool.BaseTool

CreateFileRetrievalTool 创建文件检索工具(供 Agent 调用)

func CreateRetrievalTool

func CreateRetrievalTool(kbConfigs []KnowledgeBaseConfig) tool.BaseTool

CreateRetrievalTool 创建检索工具(供 Agent 调用)

func FormatKnowledgeResults

func FormatKnowledgeResults(docs []*schema.Document) string

FormatKnowledgeResults 将检索结果格式化为字符串

Types

type FileParserRetriever

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

FileParserRetriever 文件解析检索器

func NewFileParserRetriever

func NewFileParserRetriever(baseDir string) *FileParserRetriever

NewFileParserRetriever 创建文件解析检索器

func (*FileParserRetriever) Retrieve

func (fpr *FileParserRetriever) Retrieve(ctx context.Context, query string) ([]*schema.Document, error)

Retrieve 解析指定路径的文件内容 query 格式: /path/to/file.md 或 ["file1.md", "file2.md"] 或 *

type KnowledgeBaseConfig

type KnowledgeBaseConfig struct {
	ID           string `json:"id"`
	Name         string `json:"name"`
	RetrievalURL string `json:"retrieval_url"`
	Token        string `json:"token"`
	TopK         int    `json:"top_k"`
}

KnowledgeBaseConfig 知识库配置

type KnowledgeRetriever

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

KnowledgeRetriever 知识检索器

func NewKnowledgeRetriever

func NewKnowledgeRetriever(configs []KnowledgeBaseConfig) *KnowledgeRetriever

NewKnowledgeRetriever 创建知识检索器

func (*KnowledgeRetriever) Retrieve

func (kr *KnowledgeRetriever) Retrieve(ctx context.Context, query string) ([]*schema.Document, error)

Retrieve 根据 query 从知识库检索相关内容

Jump to

Keyboard shortcuts

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