Fix get relations

docker-compose.yaml

@@ -5,7 +5,7 @@ services:
     container_name: hybrid-agi-db
     ports:
       - 6379:6379
-      - 3000:3000
+      # - 3000:3000
   hybrid-agi-db-browser:
     image: redislabs/redisinsight:1.14.0
     container_name: hybrid-agi-db-browser

hybridagi/hybridstores/fact_memory/fact_memory.py

@@ -78,7 +78,7 @@ def get_rel_map(
             return rel_map
 
         query = f"""
-            MATCH (n1:{self._node_label})
+            MATCH (n1:{self.indexed_label})
             WHERE n1.name IN $subjs
             WITH n1
             MATCH p=(n1)-[e*1..{depth}]->(z)

tests/hybridstores/fact_memory/test_fact_memory.py

@@ -22,4 +22,22 @@ def test_delete_triplet():
 
     memory.add_triplet("myself", "is a", "robot")
     memory.delete_triplet("myself", "is a", "robot")
-    assert len(memory.get_triplets("myself")) == 0
+    assert len(memory.get_triplets("myself")) == 0
+
+def test_get_rel_map():
+    emb = FakeEmbeddings(dim=250)
+    memory = FactMemory(
+        index_name="test",
+        embeddings=emb,
+        wipe_on_start=True,
+    )
+
+    memory.add_triplet("myself", "is a", "robot")
+    memory.add_triplet("robot", "has", "sensors")
+    memory.add_triplet("robot", "can", "move")
+
+    rel_map = memory.get_rel_map(subjs=["myself", "robot"], depth=2, limit=30)
+
+    assert len(rel_map) == 2
+    assert len(rel_map["myself"]) == 3
+    assert len(rel_map["robot"]) == 2