feat: add ensure payload index exists in qdrant

Author: CaralHsi

src/memos/graph_dbs/neo4j_community.py

@@ -270,24 +270,14 @@ def _create_basic_property_indexes(self) -> None:
         except Exception as e:
             logger.warning(f"Failed to create basic property indexes: {e}")
 
-        # Step 2: Qdrant payload indexes
+        # Step 2: VectorDB indexes
         try:
-            # Qdrant supports `create_payload_index`, which is idempotent
-            self.vec_db.client.create_payload_index(
-                collection_name=self.vec_db.config.collection_name,
-                field_name="user_name",
-                field_schema="keyword",
-            )
-            logger.debug("Qdrant payload index on 'user_name' ensured.")
-
-            self.vec_db.client.create_payload_index(
-                collection_name=self.vec_db.config.collection_name,
-                field_name="memory_type",
-                field_schema="keyword",
-            )
-            logger.debug("Qdrant payload index on 'memory_type' ensured.")
+            if hasattr(self.vec_db, "ensure_payload_indexes"):
+                self.vec_db.ensure_payload_indexes(["user_name", "memory_type"])
+            else:
+                logger.debug("VecDB does not support payload index creation; skipping.")
         except Exception as e:
-            logger.warning(f"Failed to create Qdrant payload indexes: {e}")
+            logger.warning(f"Failed to create VecDB payload indexes: {e}")
 
     def _parse_node(self, node_data: dict[str, Any]) -> dict[str, Any]:
         """Parse Neo4j node and optionally fetch embedding from vector DB."""

src/memos/vec_dbs/base.py

@@ -107,3 +107,11 @@ def upsert(self, data: list[VecDBItem | dict[str, Any]]) -> None:
     @abstractmethod
     def delete(self, ids: list[str]) -> None:
         """Delete items from the vector database."""
+
+    @abstractmethod
+    def ensure_payload_indexes(self, fields: list[str]) -> None:
+        """
+        Create payload indexes for specified fields in the collection.
+        Args:
+            fields (list[str]): List of field names to index (as keyword).
+        """

src/memos/vec_dbs/qdrant.py

@@ -278,6 +278,25 @@ def update(self, id: str, data: VecDBItem | dict[str, Any]) -> None:
                 collection_name=self.config.collection_name, payload=data.payload, points=[id]
             )
 
+    def ensure_payload_indexes(self, fields: list[str]) -> None:
+        """
+        Create payload indexes for specified fields in the collection.
+        This is idempotent: it will skip if index already exists.
+
+        Args:
+            fields (list[str]): List of field names to index (as keyword).
+        """
+        for field in fields:
+            try:
+                self.client.create_payload_index(
+                    collection_name=self.config.collection_name,
+                    field_name=field,
+                    field_schema="keyword",  # Could be extended in future
+                )
+                logger.debug(f"Qdrant payload index on '{field}' ensured.")
+            except Exception as e:
+                logger.warning(f"Failed to create payload index on '{field}': {e}")
+
     def upsert(self, data: list[VecDBItem | dict[str, Any]]) -> None:
         """
         Add or update data in the vector database.