AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 16:05:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixes 14217: database retrieve api and chat-messages api response doc_metadata (#14219)

Browse Source
This commit is contained in:
yuhaowin 2025-02-27 14:56:46 +08:00 committed by GitHub
parent 5f692dfce2
commit 1e3197a1ea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/core/rag/retrieval/dataset_retrieval.py
View File

@ -203,6 +203,7 @@ class DatasetRetrieval:
"segment_id": segment.id, "segment_id": segment.id,
"retriever_from": invoke_from.to_source(), "retriever_from": invoke_from.to_source(),
"score": record.score or 0.0, "score": record.score or 0.0,
"doc_metadata": document.doc_metadata,
} }
if invoke_from.to_source() == "dev": if invoke_from.to_source() == "dev":

1
api/core/tools/utils/dataset_retriever/dataset_multi_retriever_tool.py
View File

@ -123,6 +123,7 @@ class DatasetMultiRetrieverTool(DatasetRetrieverBaseTool):
"segment_id": segment.id, "segment_id": segment.id,
"retriever_from": self.retriever_from, "retriever_from": self.retriever_from,
"score": document_score_list.get(segment.index_node_id, None), "score": document_score_list.get(segment.index_node_id, None),
"doc_metadata": document.doc_metadata,
} }
if self.retriever_from == "dev": if self.retriever_from == "dev":

1
api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py
View File

@ -172,6 +172,7 @@ class DatasetRetrieverTool(DatasetRetrieverBaseTool):
"segment_id": segment.id, "segment_id": segment.id,
"retriever_from": self.retriever_from, "retriever_from": self.retriever_from,
"score": record.score or 0.0, "score": record.score or 0.0,
"doc_metadata": document.doc_metadat, # type: ignore
} }
if self.retriever_from == "dev": if self.retriever_from == "dev":

1
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
View File

@ -240,6 +240,7 @@ class KnowledgeRetrievalNode(BaseNode[KnowledgeRetrievalNodeData]):
"segment_word_count": segment.word_count, "segment_word_count": segment.word_count,
"segment_position": segment.position, "segment_position": segment.position,
"segment_index_node_hash": segment.index_node_hash, "segment_index_node_hash": segment.index_node_hash,
"doc_metadata": document.doc_metadata,
}, },
"title": document.name, "title": document.name,
} }

1
api/core/workflow/nodes/llm/node.py
View File

@ -459,6 +459,7 @@ class LLMNode(BaseNode[LLMNodeData]):
"index_node_hash": metadata.get("segment_index_node_hash"), "index_node_hash": metadata.get("segment_index_node_hash"),
"content": context_dict.get("content"), "content": context_dict.get("content"),
"page": metadata.get("page"), "page": metadata.get("page"),
"doc_metadata": metadata.get("doc_metadata"),
} }
return source return source

1
api/fields/hit_testing_fields.py
View File

@ -7,6 +7,7 @@ document_fields = {
"data_source_type": fields.String, "data_source_type": fields.String,
"name": fields.String, "name": fields.String,
"doc_type": fields.String, "doc_type": fields.String,
"doc_metadata": fields.Raw,
} }
segment_fields = { segment_fields = {