From fa6999bf00bc85a21a9f3a5c2fecc7b57f827b71 Mon Sep 17 00:00:00 2001
From: John Stegeman <john.stegeman@gmail.com>
Date: Fri, 25 Apr 2025 07:48:34 -0400
Subject: [PATCH] Add database name to environment file and post-processing
 codl

---
 customer-graph/.env.example              | 1 +
 customer-graph/ingest_post_processing.py | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/customer-graph/.env.example b/customer-graph/.env.example
index d916980..390d984 100644
--- a/customer-graph/.env.example
+++ b/customer-graph/.env.example
@@ -2,6 +2,7 @@
 NEO4J_URI=<YOUR_NEO4J_URI>
 NEO4J_USERNAME=<YOUR_NEO4J_USERNAME>
 NEO4J_PASSWORD=<YOUR_NEO4J_PASSWORD>
+NEO4J_DATABASE=<NEO4J_DATABASE>
 
 #OpenAI
 OPENAI_API_KEY=<YOUR_OPENAI_API_KEY>
\ No newline at end of file
diff --git a/customer-graph/ingest_post_processing.py b/customer-graph/ingest_post_processing.py
index e8c574b..cede694 100644
--- a/customer-graph/ingest_post_processing.py
+++ b/customer-graph/ingest_post_processing.py
@@ -7,6 +7,7 @@
 NEO4J_URI=os.getenv("NEO4J_URI")
 NEO4J_USERNAME=os.getenv("NEO4J_USERNAME")
 NEO4J_PASSWORD=os.getenv("NEO4J_PASSWORD")
+NEO4J_DATABASE=os.getenv("NEO4J_DATABASE")
 
 
 
@@ -30,7 +31,7 @@
 
 # create text embeddings for products
 print("Creating Product Text Embeddings")
-with driver.session(database="neo4j") as session:
+with driver.session(database=NEO4J_DATABASE) as session:
     session.run('''
     MATCH (n:Product) WHERE size(n.description) <> 0
     WITH collect(n) AS nodes, toInteger(rand()*$numberOfBatches) AS partition