Feat/delete single dataset retrival (#6570)

2024-07-24 12:50:11 +08:00
parent 0fb741f269
commit e4bb943fe5
22 changed files with 651 additions and 115 deletions
--- a/api/core/rag/datasource/vdb/qdrant/qdrant_vector.py
+++ b/api/core/rag/datasource/vdb/qdrant/qdrant_vector.py
@@ -396,9 +396,11 @@ class QdrantVector(BaseVector):
        documents = []
        for result in results:
            if result:
-                documents.append(self._document_from_scored_point(
+                document = self._document_from_scored_point(
                    result, Field.CONTENT_KEY.value, Field.METADATA_KEY.value
-                ))
+                )
+                document.metadata['vector'] = result.vector
+                documents.append(document)

        return documents