add vector field for other vectordb (#7051)

2024-08-07 17:14:03 +08:00
parent aad02113c6
commit 80c94f02e9
10 changed files with 23 additions and 14 deletions
--- a/api/core/rag/datasource/vdb/qdrant/qdrant_vector.py
+++ b/api/core/rag/datasource/vdb/qdrant/qdrant_vector.py
@@ -399,7 +399,6 @@ class QdrantVector(BaseVector):
                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
@@ -418,6 +417,7 @@ class QdrantVector(BaseVector):
    ) -> Document:
        return Document(
            page_content=scored_point.payload.get(content_payload_key),
+            vector=scored_point.vector,
            metadata=scored_point.payload.get(metadata_payload_key) or {},
        )