Update chatbot_new.py

chatbot_new.py

@@ -17,8 +17,8 @@ META_PATH = "hbl_vector_store/hbl_metadata.pkl"
 EMBED_MODEL = "./bge-m3"
 OLLAMA_URL = "http://localhost:11434/api/chat"
 LLM_MODEL = "qwen3.5:4b"
-TOP_K = 5
-MIN_SCORE = 0.45  # re-validate against the new embeddings, this was tuned for the old ones
+TOP_K = 4
+MIN_SCORE = 0.58  # re-validate against the new embeddings, this was tuned for the old ones
 HISTORY_TURNS = 3  # number of past user/assistant exchanges kept for classifier context
 
 
@@ -105,7 +105,6 @@ HBL_TERMS = [
 ]
 
 OTHER_BANK_TERMS = [
-    "haider bank",
     "meezan bank",
     "meezan",
     "ubl",
@@ -139,14 +138,14 @@ def detect_bank_entity(query):
     hbl_hit = any(_has_term(q, t) for t in HBL_TERMS)
     other_hit = any(_has_term(q, t) for t in OTHER_BANK_TERMS)
 
-    if hbl_hit:
-        # HBL mentioned -> HBL-related even if another bank is named too
-        # (comparison / switching-bank questions). The KB only holds HBL
-        # docs, so retrieval keeps the answer scoped correctly regardless.
-        return "HBL"
-
+    if hbl_hit and other_hit:
+        return "OTHER_BANK"
+    
     if other_hit:
         return "OTHER_BANK"
+    
+    if hbl_hit:
+        return "HBL"
 
     return "UNKNOWN"
 
@@ -272,14 +271,14 @@ def retrieve(query):
 SYSTEM_PROMPT = """
 You are Habib Bank Limited Assistant.
 
-Strict rules:
-- Answer only using provided HBL documents.
+Answer only from the provided HBL documents.
+
+Rules:
 - Do not use outside knowledge.
-- Do not guess.
-- If information is unavailable say:
+- Do not guess or invent information.
+- If the answer is not present in the documents, say:
 "I could not find this information in the available HBL documents."
-
-Never answer questions outside HBL domain.
+- Do not discuss topics unrelated to HBL.
 """