Refactor vector_factory.py to use pgvector as the default vector store

langgenius · Apr 27, 2024 · b378adf · b378adf
1 parent 93e225c
commit b378adf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/api/core/rag/datasource/vdb/vector_factory.py b/api/core/rag/datasource/vdb/vector_factory.py
@@ -146,9 +146,10 @@ def _init_vector(self) -> BaseVector:
             else:
                 dataset_id = self._dataset.id
                 collection_name = Dataset.gen_collection_name_by_id(dataset_id)
-                index_struct_dict = {"type": "relyt", "vector_store": {"class_prefix": collection_name}}
+                index_struct_dict = {
+                    "type": "pgvector",
+                    "vector_store": {"class_prefix": collection_name}}
                 self._dataset.index_struct = json.dumps(index_struct_dict)
-            dim = len(self._embeddings.embed_query("hello relyt"))
             return PGVector(
                 collection_name=collection_name,
                 config=PGVectorConfig(
@@ -158,7 +159,6 @@ def _init_vector(self) -> BaseVector:
                     password=config.get("PGVECTOR_PASSWORD"),
                     database=config.get("PGVECTOR_DATABASE"),
                 ),
-                dimension=dim,
             )
         else:
             raise ValueError(f"Vector store {config.get('VECTOR_STORE')} is not supported.")