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

2025-02-27 14:56:46 +08:00 · 2025-02-27 14:56:46 +08:00 · 1e3197a1ea
commit 1e3197a1ea
parent 5f692dfce2
6 changed files with 6 additions and 0 deletions
--- a/api/core/rag/retrieval/dataset_retrieval.py
+++ b/api/core/rag/retrieval/dataset_retrieval.py
@ -203,6 +203,7 @@ class DatasetRetrieval:
                                "segment_id": segment.id,
                                "retriever_from": invoke_from.to_source(),
                                "score": record.score or 0.0,
+                                "doc_metadata": document.doc_metadata,
                            }

                            if invoke_from.to_source() == "dev":