浏览代码

Fix external knowledge Issues: (#17685) (#17843)

tags/1.3.0
briqt 6 个月前
父节点
当前提交
91cfa90503
没有帐户链接到提交者的电子邮件

+ 1
- 0
api/core/rag/retrieval/dataset_retrieval.py 查看文件

@@ -206,6 +206,7 @@ class DatasetRetrieval:
source = {
"dataset_id": item.metadata.get("dataset_id"),
"dataset_name": item.metadata.get("dataset_name"),
"document_id": item.metadata.get("document_id") or item.metadata.get("title"),
"document_name": item.metadata.get("title"),
"data_source_type": "external",
"retriever_from": invoke_from.to_source(),

+ 1
- 0
api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py 查看文件

@@ -86,6 +86,7 @@ class DatasetRetrieverTool(DatasetRetrieverBaseTool):
"position": position,
"dataset_id": item.metadata.get("dataset_id"),
"dataset_name": item.metadata.get("dataset_name"),
"document_id": item.metadata.get("document_id") or item.metadata.get("title"),
"document_name": item.metadata.get("title"),
"data_source_type": "external",
"retriever_from": self.retriever_from,

+ 1
- 0
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py 查看文件

@@ -259,6 +259,7 @@ class KnowledgeRetrievalNode(LLMNode):
"_source": "knowledge",
"dataset_id": item.metadata.get("dataset_id"),
"dataset_name": item.metadata.get("dataset_name"),
"document_id": item.metadata.get("document_id") or item.metadata.get("title"),
"document_name": item.metadata.get("title"),
"data_source_type": "external",
"retriever_from": "workflow",

正在加载...
取消
保存