revert page column (#8217)

2024-09-10 18:21:22 +08:00
parent d4dc54447a
commit bb3002b173
3 changed files with 1 additions and 6 deletions
--- a/api/core/rag/datasource/vdb/vector_factory.py
+++ b/api/core/rag/datasource/vdb/vector_factory.py
@@ -27,7 +27,7 @@ class AbstractVectorFactory(ABC):
 class Vector:
    def __init__(self, dataset: Dataset, attributes: list = None):
        if attributes is None:
-            attributes = ["doc_id", "dataset_id", "document_id", "doc_hash", "page"]
+            attributes = ["doc_id", "dataset_id", "document_id", "doc_hash"]
        self._dataset = dataset
        self._embeddings = self._get_embeddings()
        self._attributes = attributes