Fix CI/CD by waiting for the neo4j index to be created

Author: YoanSallami

coverage-badge.svg

@@ -27,7 +27,6 @@ def __init__(
         embedding_model=None,
         metric="cosine",
         wipe_on_start=False,
-        **kwargs,
     ):
         super().__init__(
             index_name=index_name,
@@ -77,13 +76,12 @@ def __init__(
         asyncio.get_event_loop().run_until_complete(
             self.query(query, params=params),
         )
-
         for entity_model in self.entity_models:
             node_label = self.sanitize_label(entity_model.get_schema().get("title"))
             index_name = to_snake_case(node_label)
             query = "\n".join(
                 [
-                    f"CREATE VECTOR INDEX `{index_name}` IF NOT EXISTS",
+                    f"CREATE VECTOR INDEX $indexName IF NOT EXISTS",
                     f"FOR (n:{node_label}) ON n.embedding",
                     "OPTIONS {indexConfig : {"
                     " `vector.dimensions`: $dimension,"
@@ -92,12 +90,17 @@ def __init__(
                 ]
             )
             params = {
+                "indexName": index_name,
                 "dimension": self.embedding_dim,
                 "similarityFunction": self.metric,
             }
             asyncio.get_event_loop().run_until_complete(
                 self.query(query, params=params),
             )
+        if self.entity_models:
+            asyncio.get_event_loop().run_until_complete(
+                self.query("CALL db.awaitIndexes(300)"),
+            )
 
     def sanitize_label(self, label):
         """Sanitize labels for Neo4j compatibility"""

synalinks/src/knowledge_bases/database_adapters/neo4j_adapter.py

@@ -47,7 +47,7 @@ async def test_adapter(self, mock_embedding):
         adapter = Neo4JAdapter(
             index_name="neo4j://localhost:7687",
             embedding_model=embedding_model,
-            data_models=[Document, Chunk],
+            entity_models=[Document, Chunk],
             relation_models=[IsPartOf],
             wipe_on_start=True,
         )
@@ -64,7 +64,7 @@ async def test_adapter_update_entity(self, mock_embedding):
         adapter = Neo4JAdapter(
             index_name="neo4j://localhost:7687",
             embedding_model=embedding_model,
-            data_models=[Document, Chunk],
+            entity_models=[Document, Chunk],
             relation_models=[IsPartOf],
             wipe_on_start=True,
         )
@@ -88,7 +88,7 @@ async def test_adapter_update_relation(self, mock_embedding):
         adapter = Neo4JAdapter(
             index_name="neo4j://localhost:7687",
             embedding_model=embedding_model,
-            data_models=[Document, Chunk],
+            entity_models=[Document, Chunk],
             relation_models=[IsPartOf],
             wipe_on_start=True,
         )
@@ -120,7 +120,7 @@ async def test_adapter_similarity_search(self, mock_embedding):
         adapter = Neo4JAdapter(
             index_name="neo4j://localhost:7687",
             embedding_model=embedding_model,
-            data_models=[Document, Chunk],
+            entity_models=[Document, Chunk],
             relation_models=[IsPartOf],
             wipe_on_start=True,
         )

synalinks/src/knowledge_bases/database_adapters/neo4j_adapter_test.py

@@ -3,7 +3,7 @@
 from synalinks.src.api_export import synalinks_export
 
 # Unique source of truth for the version number.
-__version__ = "0.3.002"
+__version__ = "0.3.003"
 
 
 @synalinks_export("synalinks.version")