briqt
6 months ago
3 changed files with 3 additions and 0 deletions
+ 1
- 0
api/core/rag/retrieval/dataset_retrieval.py
View File
| source = { | source = { | ||||
| "dataset_id": item.metadata.get("dataset_id"), | "dataset_id": item.metadata.get("dataset_id"), | ||||
| "dataset_name": item.metadata.get("dataset_name"), | "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"), | "document_name": item.metadata.get("title"), | ||||
| "data_source_type": "external", | "data_source_type": "external", | ||||
| "retriever_from": invoke_from.to_source(), | "retriever_from": invoke_from.to_source(), |
+ 1
- 0
api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py
View File
| "position": position, | "position": position, | ||||
| "dataset_id": item.metadata.get("dataset_id"), | "dataset_id": item.metadata.get("dataset_id"), | ||||
| "dataset_name": item.metadata.get("dataset_name"), | "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"), | "document_name": item.metadata.get("title"), | ||||
| "data_source_type": "external", | "data_source_type": "external", | ||||
| "retriever_from": self.retriever_from, | "retriever_from": self.retriever_from, |
+ 1
- 0
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
View File
| "_source": "knowledge", | "_source": "knowledge", | ||||
| "dataset_id": item.metadata.get("dataset_id"), | "dataset_id": item.metadata.get("dataset_id"), | ||||
| "dataset_name": item.metadata.get("dataset_name"), | "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"), | "document_name": item.metadata.get("title"), | ||||
| "data_source_type": "external", | "data_source_type": "external", | ||||
| "retriever_from": "workflow", | "retriever_from": "workflow", |
Loading…