latest

routers/database/tools/get_nodes_and_edges.py

@@ -89,8 +89,8 @@ async def get_all_nodes_and_edges():
 
 
 @router.get("/get-connected-nodes-and-edges")
-async def get_connected_nodes_and_edges(unique_id: str = Query(...), db_name: str = Query(...)):
-    logging.info(f"Getting connected nodes and edges for {unique_id} from database {db_name}")
+async def get_connected_nodes_and_edges(uuid_string: str = Query(...), db_name: str = Query(...)):
+    logging.info(f"Getting connected nodes and edges for {uuid_string} from database {db_name}")
     neo_driver = driver.get_driver(db_name=db_name)
     if neo_driver is None:
         return {"status": "error", "message": "Failed to connect to the database"}
@@ -98,11 +98,11 @@ async def get_connected_nodes_and_edges(unique_id: str = Query(...), db_name: st
     try:
         with neo_driver.session(database=db_name) as neo_session:
             query = """
-            MATCH (n {unique_id: $unique_id})
+            MATCH (n {uuid_string: $uuid_string})
             OPTIONAL MATCH (n)-[r]-(connected)
             RETURN n, collect(connected) as connected_nodes, collect(r) as relationships
             """
-            result = neo_session.run(query, unique_id=unique_id)
+            result = neo_session.run(query, uuid_string=uuid_string)
             record = result.single()
             if record:
                 main_node = record['n']