rag
| 函数名 | 函数描述/介绍 |
|---|---|
| rag.AddDocument | _addDocument 向指定集合添加文档 |
| rag.BuildCollectionFromFile | |
| rag.BuildCollectionFromRaw | |
| rag.BuildCollectionFromReader | |
| rag.BuildIndexKnowledgeFromFile | |
| rag.BuildKnowledgeFromEntityRepos | |
| rag.BuildSearchIndexKnowledge | BuildSearchIndexKnowledge builds a search index for the given text content. It generates 5-10 search questions that users might ask to find this cont... |
| rag.BuildSearchIndexKnowledgeFromFile | BuildSearchIndexKnowledgeFromFile builds a search index from a file. It reads the file content and calls BuildSearchIndexKnowledge. Parameters: ... |
| rag.DeleteAllKnowledgeBase | _deleteAllKnowledgeBase 删除所有知识库及其关联的 RAG 内容 清空所有: RAG 向量库、RAG 集合综述库、知识库条目库、知识库集合、问题索引库等 |
| rag.DeleteCollection | _deleteCollection 删除指定的 RAG 集合 |
| rag.DeleteDocument | _deleteDocument 从指定集合删除文档 |
| rag.DeleteKnowledgeBase | _deleteKnowledgeBase 删除指定的知识库及其关联的 RAG 内容 包括: RAG 向量库、RAG 集合综述库、知识库条目库、知识库集合、问题索引库等 |
| rag.DeleteRAG | _deleteRAG 删除指定的 RAG 系统 |
| rag.Embedding | _embedding 生成文本的嵌入向量 使用默认的嵌入服务生成文本的向量表示（优先使用在线服务，回退到本地服务） |
| rag.EnableMockMode | |
| rag.Export | |
| rag.Get | |
| rag.GetCollection | |
| rag.GetCollectionInfo | _getCollectionInfo 获取指定集合的详细信息 |
| rag.HasCollection | _hasCollection 检查指定集合是否存在 |
| rag.Import | |
| rag.ListCollection | _listCollection 获取所有 RAG 集合列表 |
| rag.ListRAG | _listRAG 列出所有 RAG 系统列表 |
| rag.LocalEmbedding | _localEmbedding 使用本地嵌入服务生成文本的向量表示 本地服务需要安装本地模型（如 Qwen3-Embedding-0.6B-Q4_K_M） |
| rag.NewRagDatabase | |
| rag.NewTempRagDatabase | _newTempRagDatabase 创建临时 RAG 数据库 |
| rag.OnlineEmbedding | _onlineEmbedding 使用在线嵌入服务生成文本的向量表示 使用 AIBalance 免费在线服务，无需安装本地模型 |
| rag.Query | |
| rag.QueryDocuments | _queryDocuments 在指定集合中查询文档 |
| rag.aiService | |
| rag.aiServiceType | |
| rag.buildFilter | |
| rag.buildQuery | |
| rag.chunkSize | |
| rag.ctx | |
| rag.db | |
| rag.docMetadata | WithDocumentMetadataKeyValue sets document metadata key-value pairs |
| rag.docRawMetadata | WithDocumentRawMetadata sets raw document metadata |
| rag.documentHandler | |
| rag.embeddingHandle | _embeddingHandle 创建自定义嵌入处理器 |
| rag.entryLength | |
| rag.extraPrompt | |
| rag.getEntityFilter | |
| rag.importName | WithRAGCollectionName sets the specific collection name to query |
| rag.importOverwrite | |
| rag.importRebuildGraph | |
| rag.khopLimit | |
| rag.khopk | WithKHopK 设置k-hop的跳数，k>=2时返回k-hop路径，k=0返回所有路径 |
| rag.khopkMax | WithKHopKMax 设置最大路径长度，最小值为2 |
| rag.khopkMin | WithKHopKMin 设置最小路径长度，最小值为2 |
| rag.log | |
| rag.noEntityRepository | _noEntityRepository 禁用实体仓库 |
| rag.noHNSWGraph | |
| rag.noKnowledgeBase | _noKnowledgeBase 禁用知识库 |
| rag.noMetadata | |
| rag.noOriginInput | |
| rag.onlyPQCode | |
| rag.pathDepth | |
| rag.progressHandler | |
| rag.queryCollection | WithRAGCollectionName sets the specific collection name to query |
| rag.queryConcurrent | WithRAGConcurrent sets the concurrency level for query operations |
| rag.queryCtx | WithRAGCtx sets the context for RAG query operations |
| rag.queryEnhance | WithRAGEnhance sets the enhancement strategies to apply |
| rag.queryLimit | WithRAGLimit sets the maximum number of results to return |
| rag.queryScoreLimit | WithRAGCollectionScoreLimit sets the score limit for collection filtering |
| rag.queryStatus | WithRAGQueryStatus sets the query status callback function |
| rag.queryType | WithRAGDocumentType sets the document type filter |
| rag.ragCosineDistance | |
| rag.ragDescription | |
| rag.ragEmbeddingModel | |
| rag.ragForceNew | WithForceNew sets whether to force creation of new collection |
| rag.ragHNSWParameters | WithHNSWParameters sets HNSW parameters |
| rag.ragImportFile | |
| rag.ragModelDimension | WithModelDimension sets the model dimension |
| rag.statusCard |
函数定义#
AddDocument#
详细描述#
_addDocument 向指定集合添加文档
Example:
err = rag.AddDocument("my_collection", "doc1", "content", {"key": "value"})
定义#
AddDocument(knowledgeBaseName string, documentName string, document string, metadata map[string]any, opts ...rag.RAGSystemConfigOption) error
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| knowledgeBaseName | string | |
| documentName | string | |
| document | string | |
| metadata | map[string]any | |
| opts | ...rag.RAGSystemConfigOption |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | error |
BuildCollectionFromFile#
详细描述#
定义#
BuildCollectionFromFile(kbName string, path string, option ...any) (<-chan *schema.KnowledgeBaseEntry, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| kbName | string | |
| path | string | |
| option | ...any |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | <-chan *schema.KnowledgeBaseEntry | |
| r2 | error |
BuildCollectionFromRaw#
详细描述#
定义#
BuildCollectionFromRaw(kbName string, content []byte, option ...any) (<-chan *schema.KnowledgeBaseEntry, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| kbName | string | |
| content | []byte | |
| option | ...any |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | <-chan *schema.KnowledgeBaseEntry | |
| r2 | error |
BuildCollectionFromReader#
详细描述#
定义#
BuildCollectionFromReader(kbName string, reader io.Reader, option ...any) (<-chan *schema.KnowledgeBaseEntry, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| kbName | string | |
| reader | io.Reader | |
| option | ...any |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | <-chan *schema.KnowledgeBaseEntry | |
| r2 | error |
BuildIndexKnowledgeFromFile#
详细描述#
定义#
BuildIndexKnowledgeFromFile(kbName string, path string, option ...any) error
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| kbName | string | |
| path | string | |
| option | ...any |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | error |
BuildKnowledgeFromEntityRepos#
详细描述#
定义#
BuildKnowledgeFromEntityRepos(name string, option ...any) (<-chan *schema.KnowledgeBaseEntry, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| name | string | |
| option | ...any |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | <-chan *schema.KnowledgeBaseEntry | |
| r2 | error |
BuildSearchIndexKnowledge#
详细描述#
BuildSearchIndexKnowledge builds a search index for the given text content.
It generates 5-10 search questions that users might ask to find this content,
and stores the original content as the knowledge entry.
Parameters:
kbName: the knowledge base name
text: the content to index (e.g., tool description, usage, parameters)
options: optional configuration (rag options, AI options, etc.)
The function will:
Use AI to generate 5-10 search questions based on the text
Store the original text as the knowledge entry
Set docMetadata with question_index and search_target for each question
Example:
text = `工具名：端口扫描器目标：扫描目标主机的开放端口用法：指定目标IP和端口范围，工具会返回开放的端口列表`result = rag.BuildSearchIndexKnowledge("my-tools", text)~println("Generated questions:", result.Questions)定义#
BuildSearchIndexKnowledge(kbName string, text string, option ...any) (*aiforge.SearchIndexResult, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| kbName | string | |
| text | string | |
| option | ...any |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | *aiforge.SearchIndexResult | |
| r2 | error |
BuildSearchIndexKnowledgeFromFile#
详细描述#
BuildSearchIndexKnowledgeFromFile builds a search index from a file.
It reads the file content and calls BuildSearchIndexKnowledge.
Parameters:
kbName: the knowledge base name
filename: the path to the file containing the content to index
options: optional configuration (rag options, AI options, etc.)
Example:
result = rag.BuildSearchIndexKnowledgeFromFile("my-tools", "/path/to/tool-description.txt")~println("Generated questions:", result.Questions)定义#
BuildSearchIndexKnowledgeFromFile(kbName string, filename string, option ...any) (*aiforge.SearchIndexResult, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| kbName | string | |
| filename | string | |
| option | ...any |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | *aiforge.SearchIndexResult | |
| r2 | error |
DeleteAllKnowledgeBase#
详细描述#
_deleteAllKnowledgeBase 删除所有知识库及其关联的 RAG 内容
清空所有: RAG 向量库、RAG 集合综述库、知识库条目库、知识库集合、问题索引库等
Example:
err = rag.DeleteAllKnowledgeBase()
定义#
DeleteAllKnowledgeBase() error
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | error |
DeleteCollection#
详细描述#
_deleteCollection 删除指定的 RAG 集合
Example:
err = rag.DeleteCollection("my_collection")
定义#
DeleteCollection(name string) error
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| name | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | error |
DeleteDocument#
详细描述#
_deleteDocument 从指定集合删除文档
Example:
err = rag.DeleteDocument("my_collection", "doc1")
定义#
DeleteDocument(knowledgeBaseName string, documentName string, opts ...rag.RAGSystemConfigOption) error
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| knowledgeBaseName | string | |
| documentName | string | |
| opts | ...rag.RAGSystemConfigOption |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | error |
DeleteKnowledgeBase#
详细描述#
_deleteKnowledgeBase 删除指定的知识库及其关联的 RAG 内容
包括: RAG 向量库、RAG 集合综述库、知识库条目库、知识库集合、问题索引库等
Example:
err = rag.DeleteKnowledgeBase("my_knowledge_base")
定义#
DeleteKnowledgeBase(name string) error
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| name | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | error |
DeleteRAG#
详细描述#
_deleteRAG 删除指定的 RAG 系统
Example:
err = rag.DeleteRAG("my_rag")
定义#
DeleteRAG(name string) error
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| name | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | error |
Embedding#
详细描述#
_embedding 生成文本的嵌入向量
使用默认的嵌入服务生成文本的向量表示（优先使用在线服务，回退到本地服务）
Example:
result, err = rag.Embedding("你好") if err != nil { // handle error } // result is []float32
定义#
Embedding(text string) ([]float32, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| text | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | []float32 | |
| r2 | error |
EnableMockMode#
详细描述#
定义#
EnableMockMode()
Export#
详细描述#
定义#
Export(collectionName string, fileName string, opts ...RAGSystemConfigOption) error
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| collectionName | string | |
| fileName | string | |
| opts | ...RAGSystemConfigOption |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | error |
Get#
详细描述#
定义#
Get(name string, opts ...RAGSystemConfigOption) (*RAGSystem, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| name | string | |
| opts | ...RAGSystemConfigOption |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | *RAGSystem | |
| r2 | error |
GetCollection#
详细描述#
定义#
GetCollection(name string, opts ...RAGSystemConfigOption) (*RAGSystem, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| name | string | |
| opts | ...RAGSystemConfigOption |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | *RAGSystem | |
| r2 | error |
GetCollectionInfo#
详细描述#
_getCollectionInfo 获取指定集合的详细信息
Example:
info, err = rag.GetCollectionInfo("my_collection")
定义#
GetCollectionInfo(name string) (*vectorstore.CollectionInfo, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| name | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | *vectorstore.CollectionInfo | |
| r2 | error |
HasCollection#
详细描述#
_hasCollection 检查指定集合是否存在
Example:
exists = rag.HasCollection("my_collection")
定义#
HasCollection(name string) bool
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| name | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | bool |
Import#
详细描述#
定义#
Import(inputPath string, optFuncs ...RAGSystemConfigOption) error
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| inputPath | string | |
| optFuncs | ...RAGSystemConfigOption |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | error |
ListCollection#
详细描述#
_listCollection 获取所有 RAG 集合列表
Example:
collections = rag.ListCollection()
定义#
ListCollection() []string
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | []string |
ListRAG#
详细描述#
_listRAG 列出所有 RAG 系统列表
Example:
ragSystems = rag.ListRAG()
定义#
ListRAG() []string
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | []string |
LocalEmbedding#
详细描述#
_localEmbedding 使用本地嵌入服务生成文本的向量表示
本地服务需要安装本地模型（如 Qwen3-Embedding-0.6B-Q4_K_M）
Example:
result, err = rag.LocalEmbedding("你好") if err != nil { // handle error - 本地服务不可用 } // result is []float32, dimension: 1024
定义#
LocalEmbedding(text string) ([]float32, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| text | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | []float32 | |
| r2 | error |
NewRagDatabase#
详细描述#
定义#
NewRagDatabase(path string) (*gorm.DB, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| path | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | *gorm.DB | |
| r2 | error |
NewTempRagDatabase#
详细描述#
_newTempRagDatabase 创建临时 RAG 数据库
Example:
db, err = rag.NewTempRagDatabase()
定义#
NewTempRagDatabase() (*gorm.DB, error)
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | *gorm.DB | |
| r2 | error |
OnlineEmbedding#
详细描述#
_onlineEmbedding 使用在线嵌入服务生成文本的向量表示
使用 AIBalance 免费在线服务，无需安装本地模型
Example:
result, err = rag.OnlineEmbedding("你好") if err != nil { // handle error - 在线服务不可用 } // result is []float32, dimension: 1024
定义#
OnlineEmbedding(text string) ([]float32, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| text | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | []float32 | |
| r2 | error |
Query#
详细描述#
定义#
Query(query string, opts ...RAGSystemConfigOption) (<-chan *RAGSearchResult, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| query | string | |
| opts | ...RAGSystemConfigOption |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | <-chan *RAGSearchResult | |
| r2 | error |
QueryDocuments#
详细描述#
_queryDocuments 在指定集合中查询文档
Example:
results, err = rag.QueryDocuments("my_collection", "query", 10)
定义#
QueryDocuments(knowledgeBaseName string, query string, limit int, opts ...rag.RAGSystemConfigOption) ([]*rag.SearchResult, error)
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| knowledgeBaseName | string | |
| query | string | |
| limit | int | |
| opts | ...rag.RAGSystemConfigOption |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | []*rag.SearchResult | |
| r2 | error |
aiService#
详细描述#
定义#
aiService(aiService aicommon.AICallbackType) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| aiService | aicommon.AICallbackType |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
aiServiceType#
详细描述#
定义#
aiServiceType(aiServiceName string, aiServiceConfig ...aispec.AIConfigOption) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| aiServiceName | string | |
| aiServiceConfig | ...aispec.AIConfigOption |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
buildFilter#
详细描述#
定义#
buildFilter(filter *ypb.EntityFilter) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| filter | *ypb.EntityFilter |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
buildQuery#
详细描述#
定义#
buildQuery(query string) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| query | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
chunkSize#
详细描述#
定义#
chunkSize(size int64) Option
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| size | int64 |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | Option |
ctx#
详细描述#
定义#
ctx(ctx context.Context) AnalysisOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| ctx | context.Context |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | AnalysisOption |
db#
详细描述#
定义#
db(db *gorm.DB) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| db | *gorm.DB |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
docMetadata#
详细描述#
WithDocumentMetadataKeyValue sets document metadata key-value pairs
定义#
docMetadata(key string, value any) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| key | string | |
| value | any |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
docRawMetadata#
详细描述#
WithDocumentRawMetadata sets raw document metadata
定义#
docRawMetadata(metadata map[string]any) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| metadata | map[string]any |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
documentHandler#
详细描述#
定义#
documentHandler(documentHandler func(doc schema.VectorStoreDocument) (schema.VectorStoreDocument, error)) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| documentHandler | func(doc schema.VectorStoreDocument) (schema.VectorStoreDocument, error) |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
embeddingHandle#
详细描述#
_embeddingHandle 创建自定义嵌入处理器
Example:
embeddingOpt = rag.embeddingHandle((text) => { return [0.1, 0.2, 0.3] // 返回嵌入向量 })
定义#
embeddingHandle(handle func(text string) any) rag.RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| handle | func(text string) any |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | rag.RAGSystemConfigOption |
entryLength#
详细描述#
定义#
entryLength(length int) RefineOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| length | int |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RefineOption |
extraPrompt#
详细描述#
定义#
extraPrompt(prompt string) AnalysisOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| prompt | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | AnalysisOption |
getEntityFilter#
详细描述#
定义#
getEntityFilter(reposName string, entityTypes []string, names []string, HiddenIndex []string, keywords []string) *ypb.EntityFilter
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| reposName | string | |
| entityTypes | []string | |
| names | []string | |
| HiddenIndex | []string | |
| keywords | []string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | *ypb.EntityFilter |
importName#
详细描述#
WithRAGCollectionName sets the specific collection name to query
定义#
importName(collectionName string) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| collectionName | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
importOverwrite#
详细描述#
定义#
importOverwrite(overwriteExisting bool) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| overwriteExisting | bool |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
importRebuildGraph#
详细描述#
定义#
importRebuildGraph(rebuildHNSWIndex bool) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| rebuildHNSWIndex | bool |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
khopLimit#
详细描述#
定义#
khopLimit(k int) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| k | int |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
khopk#
详细描述#
WithKHopK 设置k-hop的跳数，k>=2时返回k-hop路径，k=0返回所有路径
定义#
khopk(k int) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| k | int |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
khopkMax#
详细描述#
WithKHopKMax 设置最大路径长度，最小值为2
定义#
khopkMax(kMax int) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| kMax | int |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
khopkMin#
详细描述#
WithKHopKMin 设置最小路径长度，最小值为2
定义#
khopkMin(kMin int) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| kMin | int |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
log#
详细描述#
定义#
log(handler func(format string, args ...any)) AnalysisOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| handler | func(format string, args ...any) |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | AnalysisOption |
noEntityRepository#
详细描述#
_noEntityRepository 禁用实体仓库
Example:
rag.noEntityRepository()
定义#
noEntityRepository() rag.RAGSystemConfigOption
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | rag.RAGSystemConfigOption |
noHNSWGraph#
详细描述#
定义#
noHNSWGraph(noHNSWGraph bool) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| noHNSWGraph | bool |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
noKnowledgeBase#
详细描述#
_noKnowledgeBase 禁用知识库
Example:
rag.noKnowledgeBase()
定义#
noKnowledgeBase() rag.RAGSystemConfigOption
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | rag.RAGSystemConfigOption |
noMetadata#
详细描述#
定义#
noMetadata(noMetadata bool) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| noMetadata | bool |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
noOriginInput#
详细描述#
定义#
noOriginInput(noOriginInput bool) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| noOriginInput | bool |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
onlyPQCode#
详细描述#
定义#
onlyPQCode(onlyPQCode bool) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| onlyPQCode | bool |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
pathDepth#
详细描述#
定义#
pathDepth(deep int) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| deep | int |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
progressHandler#
详细描述#
定义#
progressHandler(progressHandler func(percent float64, message string, messageType string)) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| progressHandler | func(percent float64, message string, messageType string) |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
queryCollection#
详细描述#
WithRAGCollectionName sets the specific collection name to query
定义#
queryCollection(collectionName string) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| collectionName | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
queryConcurrent#
详细描述#
WithRAGConcurrent sets the concurrency level for query operations
定义#
queryConcurrent(concurrent int) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| concurrent | int |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
queryCtx#
详细描述#
WithRAGCtx sets the context for RAG query operations
定义#
queryCtx(ctx context.Context) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| ctx | context.Context |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
queryEnhance#
详细描述#
WithRAGEnhance sets the enhancement strategies to apply
定义#
queryEnhance(enhance ...string) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| enhance | ...string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
queryLimit#
详细描述#
WithRAGLimit sets the maximum number of results to return
定义#
queryLimit(limit int) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| limit | int |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
queryScoreLimit#
详细描述#
WithRAGCollectionScoreLimit sets the score limit for collection filtering
定义#
queryScoreLimit(scoreLimit float64) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| scoreLimit | float64 |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
queryStatus#
详细描述#
WithRAGQueryStatus sets the query status callback function
定义#
queryStatus(callback func(label string, i any, tags ...string)) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| callback | func(label string, i any, tags ...string) |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
queryType#
详细描述#
WithRAGDocumentType sets the document type filter
定义#
queryType(documentType ...string) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| documentType | ...string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
ragCosineDistance#
详细描述#
定义#
ragCosineDistance()
ragDescription#
详细描述#
定义#
ragDescription(description string) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| description | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
ragEmbeddingModel#
详细描述#
定义#
ragEmbeddingModel(model string) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| model | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
ragForceNew#
详细描述#
WithForceNew sets whether to force creation of new collection
定义#
ragForceNew(force bool) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| force | bool |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
ragHNSWParameters#
详细描述#
WithHNSWParameters sets HNSW parameters
定义#
ragHNSWParameters(m int, ml float64, efSearch int, efConstruct int) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| m | int | |
| ml | float64 | |
| efSearch | int | |
| efConstruct | int |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
ragImportFile#
详细描述#
定义#
ragImportFile(importFile string) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| importFile | string |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
ragModelDimension#
详细描述#
WithModelDimension sets the model dimension
定义#
ragModelDimension(dimension int) RAGSystemConfigOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| dimension | int |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | RAGSystemConfigOption |
statusCard#
详细描述#
定义#
statusCard(handler func(id string, data any, tags ...string)) AnalysisOption
参数#
| 参数名 | 参数类型 | 参数解释 |
|---|---|---|
| handler | func(id string, data any, tags ...string) |
返回值#
| 返回值(顺序) | 返回值类型 | 返回值解释 |
|---|---|---|
| r1 | AnalysisOption |