Update recommender/graph_retrieval.py

recommender/graph_retrieval.py

@@ -7,6 +7,7 @@ from nodes.intent import CreditCardState
 from neo4j import GraphDatabase
 from neo4j.exceptions import ServiceUnavailable, TransientError
 import time
+import re
 
 #neo4j credentials
 NEO4J_URI = os.environ.get("NEO4J_URI")
@@ -20,9 +21,16 @@ def get_driver():
         _driver = GraphDatabase.driver(NEO4J_URI, auth=(NEO4J_USER, NEO4J_PASS))
     return _driver
 
+def normalize_card_name(name):
+    name = name.lower()
+    name = name.replace("+", "plus")  
+    name = re.sub(r"[^a-z0-9 ]", "", name) 
+    name = re.sub(r"\s+", " ", name).strip()  
+    return name
+
 class Neo4jConnectionError(Exception):
     pass
-    
+
 # --- Eligibility Filter ---
 def eligibility_filter(cards, user_income, user_cibil, user_age,min_joining_fee, max_joining_fee,
                             min_annual_fee, max_annual_fee): 
@@ -44,9 +52,6 @@ def eligibility_filter(cards, user_income, user_cibil, user_age,min_joining_fee,
                 eligible_cards.append(card_name)
     return eligible_cards
     
-class Neo4jConnectionError(Exception):
-    pass
-    
 class Neo4jRetrievalTool(BaseTool):
     name: str = "neo4j_card_retriever"
     description: str = "Runs Cypher and builds FAISS on filtered cards."