user agent and logging for neo4j (#344)

* user agent and logging for neo4j

* remove changes

Author: aashipandya

backend/example.env

@@ -19,3 +19,4 @@ NUMBER_OF_CHUNKS_TO_COMBINE = ""
 GEMINI_ENABLED = True|False
 # Enable Google Cloud logs (default is True)
 GCP_LOG_METRICS_ENABLED = True|False
+NEO4J_USER_AGENT = ""

backend/src/graph_query.py

@@ -1,6 +1,10 @@
 import logging
 from neo4j import time 
 from neo4j import GraphDatabase
+import os
+from neo4j.debug import watch
+
+watch("neo4j")
 
 QUERY_MAP = {
     "document"          : " + [docs] ",
@@ -50,7 +54,7 @@ def get_graphDB_driver(uri, username, password):
     """
     try:
         logging.info(f"Attempting to connect to the Neo4j database at {uri}")
-        driver = GraphDatabase.driver(uri, auth=(username, password))
+        driver = GraphDatabase.driver(uri, auth=(username, password), user_agent=os.environ.get('NEO4J_USER_AGENT'))
         logging.info("Connection successful")
         return driver
     except Exception as e:

backend/src/shared/common_fn.py

@@ -10,6 +10,9 @@
 import re
 import os
 from pathlib import Path
+from neo4j.debug import watch
+
+watch("neo4j")
 
 def check_url_source(source_type, yt_url:str=None, queries_list:List[str]=None):
     try:
@@ -59,7 +62,7 @@ def get_chunk_and_graphDocument(graph_document_list, chunkId_chunkDoc_list):
   return lst_chunk_chunkId_document  
 
 def create_graph_database_connection(uri, userName, password, database):
-  graph = Neo4jGraph(url=uri, database=database, username=userName, password=password)
+  graph = Neo4jGraph(url=uri, database=database, username=userName, password=password, driver_config={'user_agent':os.environ.get('NEO4J_USER_AGENT')})
   return graph