feat: Kuzu Graph DB integration and adds bi-lingual (english, chinese) support to graph testing suite

.gitignore

@@ -1,5 +1,5 @@
 # Python-related files
-__pycache__/
+__pycache__
 *.py[cod]
 *.egg-info/
 .eggs/
@@ -67,9 +67,10 @@ lightrag-dev/
 gui/
 
 # unit-test files
-test_*
+# test_*
 
 # Cline files
 memory-bank
 memory-bank/
 .clinerules
+throwaway

.prettierignore

@@ -0,0 +1 @@
+README.md

lightrag/base.py

@@ -560,7 +560,9 @@ async def get_edges_by_chunk_ids(self, chunk_ids: list[str]) -> list[dict]:
         return all_edges
 
     @abstractmethod
-    async def upsert_node(self, node_id: str, node_data: dict[str, str]) -> None:
+    async def upsert_node(
+        self, node_id: str, node_data: dict[str, str | int | float]
+    ) -> None:
         """Insert a new node or update an existing node in the graph.
 
         Importance notes for in-memory storage:
@@ -575,7 +577,10 @@ async def upsert_node(self, node_id: str, node_data: dict[str, str]) -> None:
 
     @abstractmethod
     async def upsert_edge(
-        self, source_node_id: str, target_node_id: str, edge_data: dict[str, str]
+        self,
+        source_node_id: str,
+        target_node_id: str,
+        edge_data: dict[str, str | int | float],
     ) -> None:
         """Insert a new edge or update an existing edge in the graph.
 

lightrag/kg/__init__.py

@@ -15,6 +15,10 @@
             "PGGraphStorage",
             "MongoGraphStorage",
             "MemgraphStorage",
+            "KuzuDBStorage",
+            # "AGEStorage",
+            # "TiDBGraphStorage",
+            # "GremlinStorage",
         ],
         "required_methods": ["upsert_node", "upsert_edge"],
     },
@@ -53,6 +57,8 @@
     "Neo4JStorage": ["NEO4J_URI", "NEO4J_USERNAME", "NEO4J_PASSWORD"],
     "MongoGraphStorage": [],
     "MemgraphStorage": ["MEMGRAPH_URI"],
+    "KuzuDBStorage": [],
+    # "TiDBGraphStorage": ["TIDB_USER", "TIDB_PASSWORD", "TIDB_DATABASE"],
     "AGEStorage": [
         "AGE_POSTGRES_DB",
         "AGE_POSTGRES_USER",
@@ -101,6 +107,7 @@
     "FaissVectorDBStorage": ".kg.faiss_impl",
     "QdrantVectorDBStorage": ".kg.qdrant_impl",
     "MemgraphStorage": ".kg.memgraph_impl",
+    "KuzuDBStorage": ".kg.kuzu_impl",
 }