feat: Integrate Neo4j and fix CI issues

This commit resolves the issues from the previous PR eosphoros-ai#2951 by successfully integrating the Neo4j database.

Key changes include:
- Fixed code style issues, passing `make fmt` and `make fmt-check`.
- Added `examples/neo4j_example.py` to demonstrate the usage of the new Neo4j connector.
- Added `tests/intetration_tests/datasource/test_conn_neo4j.py` and `tests/intetration_tests/graph_store/test_neo4j_store.py` to provide integration tests for the new Neo4j functionality.
- Resolved an issue with Neo4j startup in high-version Java environments.
- Addressed all feedback from the code review, including fixing broken tests and moving an import statement.

Note: I was unable to run the integration tests locally due to a lack of Docker permissions. The tests have been written and should be run in the CI/CD pipeline.

Author: google-labs-jules[bot]

examples/neo4j_example.py

@@ -0,0 +1,37 @@
+import asyncio
+from typing import List
+
+from dbgpt.storage.graph_store.graph import Edge, Graph, Vertex
+from dbgpt_ext.storage.graph_store.neo4j_store import Neo4jStore
+
+
+async def main():
+    # Create a Neo4jStore instance
+    neo4j_store = Neo4jStore()
+
+    # Create a graph
+    graph = Graph("test_graph")
+    v1 = Vertex("1", "node1")
+    v2 = Vertex("2", "node2")
+    graph.add_vertex(v1)
+    graph.add_vertex(v2)
+    edge = Edge("1", "2", "edge1")
+    graph.add_edge(edge)
+
+    # Add the graph to the store
+    await neo4j_store.add_graph(graph)
+
+    # Get the graph from the store
+    retrieved_graph = await neo4j_store.get_graph("test_graph")
+    print(retrieved_graph)
+
+    # Get all graph names
+    graph_names = await neo4j_store.get_all_graph_names()
+    print(graph_names)
+
+    # Delete the graph from the store
+    await neo4j_store.delete_graph("test_graph")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

packages/dbgpt-ext/src/dbgpt_ext/datasource/conn_neo4j.py

@@ -119,11 +119,11 @@ def from_uri_db(
 
     def create_graph(self, graph_name: str) -> bool:
         """Create a graph in Neo4j.
-        
+
         Note: In Neo4j, databases are created at the system level.
         For knowledge graphs, we don't need to explicitly create a graph structure.
         This method is here for compatibility with TuGraph adapter.
-        
+
         Returns:
             bool: True (graph always exists in Neo4j)
         """
@@ -133,10 +133,10 @@ def create_graph(self, graph_name: str) -> bool:
 
     def is_exist(self, graph_name: str) -> bool:
         """Check if a database exists in Neo4j.
-        
+
         Args:
             graph_name: Database name to check
-            
+
         Returns:
             bool: True if database exists
         """
@@ -149,7 +149,7 @@ def is_exist(self, graph_name: str) -> bool:
 
     def delete_graph(self, graph_name: str) -> None:
         """Delete all data from the current database.
-        
+
         Note: This deletes all nodes and relationships, not the database itself.
         """
         with self._driver.session(database=self._database) as session:
@@ -171,11 +171,11 @@ def get_system_info(self) -> Dict:
                         system_info["neo4j_version"] = versions[0]
                         # For compatibility with TuGraph adapter
                         system_info["lgraph_version"] = versions[0]
-        except Exception as e:
+        except Exception:
             # Fallback for older Neo4j versions
             system_info["neo4j_version"] = "unknown"
             system_info["lgraph_version"] = "5.0.0"  # Default compatible version
-        
+
         return system_info
 
     def get_table_names(self) -> Iterator[str]:
@@ -187,7 +187,9 @@ def get_table_names(self) -> Iterator[str]:
 
             # Get relationship types
             result = session.run("CALL db.relationshipTypes()")
-            rel_types = [record["relationshipType"] + "_relationship" for record in result]
+            rel_types = [
+                record["relationshipType"] + "_relationship" for record in result
+            ]
 
             return iter(node_labels + rel_types)
 
@@ -340,4 +342,3 @@ def get_database_names(self) -> List[str]:
     def is_graph_type(cls) -> bool:
         """Return whether the connector is a graph database connector."""
         return True
-

packages/dbgpt-ext/src/dbgpt_ext/storage/graph_store/neo4j_store.py

@@ -3,7 +3,6 @@
 import logging
 import os
 from dataclasses import dataclass, field
-from typing import List
 
 from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
 from dbgpt.storage.graph_store.base import GraphStoreBase, GraphStoreConfig
@@ -182,10 +181,10 @@ def get_config(self) -> Neo4jStoreConfig:
 
     def is_exist(self, name: str) -> bool:
         """Check if graph (database) exists in Neo4j.
-        
+
         Args:
             name: Database name to check
-            
+
         Returns:
             bool: True if database exists
         """