Update do_graphlookup.py

taner45 · Jan 16, 2025 · ee16519 · ee16519
1 parent ddb662e
commit ee16519
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/apps/graph_rag_demo/do_graphlookup.py b/apps/graph_rag_demo/do_graphlookup.py
@@ -9,7 +9,7 @@ def graph_lookup(node_name,max_depth):
     graph_lookup_docs = []
     try:
         uri = os.getenv("ATLAS_CONNECTION_STRING")
-        client = MongoClient(uri)
+        client = MongoClient(uri, appname="devrel.showcase.apps.graph_rag_demo")
         database = client["langchain_db"]
         collection = database["nodes_relationships"]
         pipeline = [
@@ -38,4 +38,4 @@ def graph_lookup(node_name,max_depth):
     finally:
         client.close()
     pprint(graph_lookup_docs)
-    return graph_lookup_docs
+    return graph_lookup_docs