added database

Author: vasanthasaikalluri

backend/src/graph_query.py

@@ -181,7 +181,7 @@ def get_completed_documents(driver):
     return documents
 
 
-def get_graph_results(uri, username, password,document_names):
+def get_graph_results(uri, username, password,database,document_names):
     """
     Retrieves graph data by executing a specified Cypher query using credentials and parameters provided.
     Processes the results to extract nodes and relationships and packages them in a structured output.
@@ -198,7 +198,7 @@ def get_graph_results(uri, username, password,document_names):
     """
     try:
         logging.info(f"Starting graph query process")
-        driver = get_graphDB_driver(uri, username, password)  
+        driver = get_graphDB_driver(uri, username, password,database)  
         document_names= list(map(str.strip, json.loads(document_names)))
         query = GRAPH_QUERY.format(graph_chunk_limit=GRAPH_CHUNK_LIMIT)
         records, summary , keys = execute_query(driver, query.strip(), document_names)