Bläddra i källkod
Fixes 14217: database retrieve api and chat-messages api response doc_metadata (#14219)
tags/1.0.0
yuhaowin
8 månader sedan
6 ändrade filer med 6 tillägg och 0 borttagningar
+ 1
- 0
api/core/rag/retrieval/dataset_retrieval.py
Visa fil
| "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
- 0
api/core/tools/utils/dataset_retriever/dataset_multi_retriever_tool.py
Visa fil
| "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
- 0
api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py
Visa fil
| "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
- 0
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
Visa fil
| "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
- 0
api/core/workflow/nodes/llm/node.py
Visa fil
| "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
- 0
api/fields/hit_testing_fields.py
Visa fil
| "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 = { |
Laddar…