From ce0aeaf9b4934144976df8ccb264120eddb68193 Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Tue, 15 Apr 2025 16:49:36 +0800
Subject: [PATCH] Fix mis-using directed edge in PostgreSQL AGE graph storage

---
 lightrag/kg/postgres_impl.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py
index 872bc6bb..3e66e83e 100644
--- a/lightrag/kg/postgres_impl.py
+++ b/lightrag/kg/postgres_impl.py
@@ -1262,7 +1262,7 @@ class PGGraphStorage(BaseGraphStorage):
         tgt_label = target_node_id.strip('"')
 
         query = """SELECT * FROM cypher('%s', $$
-                     MATCH (a:base {entity_id: "%s"})-[r]->(b:base {entity_id: "%s"})
+                     MATCH (a:base {entity_id: "%s"})-[r]-(b:base {entity_id: "%s"})
                      RETURN properties(r) as edge_properties
                      LIMIT 1
                    $$) AS (edge_properties agtype)""" % (
@@ -1374,7 +1374,7 @@ class PGGraphStorage(BaseGraphStorage):
                      MATCH (source:base {entity_id: "%s"})
                      WITH source
                      MATCH (target:base {entity_id: "%s"})
-                     MERGE (source)-[r:DIRECTED]->(target)
+                     MERGE (source)-[r:DIRECTED]-(target)
                      SET r += %s
                      RETURN r
                    $$) AS (r agtype)""" % (
@@ -1447,7 +1447,7 @@ class PGGraphStorage(BaseGraphStorage):
             tgt_label = target.strip('"')
 
             query = """SELECT * FROM cypher('%s', $$
-                         MATCH (a:base {entity_id: "%s"})-[r]->(b:base {entity_id: "%s"})
+                         MATCH (a:base {entity_id: "%s"})-[r]-(b:base {entity_id: "%s"})
                          DELETE r
                        $$) AS (r agtype)""" % (self.graph_name, src_label, tgt_label)
 
@@ -1723,7 +1723,7 @@ class PGGraphStorage(BaseGraphStorage):
         if node_label == "*":
             query = f"""SELECT * FROM cypher('{self.graph_name}', $$
                     MATCH (n:base)
-                    OPTIONAL MATCH (n)-[r]->(target:base)
+                    OPTIONAL MATCH (n)-[r]-(target:base)
                     RETURN collect(distinct n) AS n, collect(distinct r) AS r
                     LIMIT {max_nodes}
                     $$) AS (n agtype, r agtype)"""