Update pipeline.py

pipeline.py

@@ -1,5 +1,3 @@
-# pipeline.py
-
 import os
 import getpass
 import pandas as pd
@@ -139,12 +137,12 @@ def run_with_chain_context(inputs: Dict[str, Any]) -> Dict[str, str]:
     chat_history = inputs.get("chat_history", [])
 
     # 1) Classification
-    class_result = classification_chain.invoke({"query": user_query})
+    class_result = classification_chain.invoke({"query": user_query, "chat_history": chat_history})
     classification = class_result.get("text", "").strip()
 
     if classification == "OutOfScope":
-        refusal_text = refusal_chain.run({})
-        final_refusal = tailor_chain.run({"response": refusal_text})
+        refusal_text = refusal_chain.run({"chat_history": chat_history})
+        final_refusal = tailor_chain.run({"response": refusal_text, "chat_history": chat_history})
         return {"answer": final_refusal.strip()}
 
     if classification == "Wellness":
@@ -154,8 +152,8 @@ def run_with_chain_context(inputs: Dict[str, Any]) -> Dict[str, str]:
         })
         csv_answer = rag_result["result"].strip()
         web_answer = do_web_search(user_query) if not csv_answer else ""
-        final_merged = cleaner_chain.merge(kb=csv_answer, web=web_answer)
-        final_answer = tailor_chain.run({"response": final_merged}).strip()
+        final_merged = cleaner_chain.merge(kb=csv_answer, web=web_answer, chat_history=chat_history)
+        final_answer = tailor_chain.run({"response": final_merged, "chat_history": chat_history}).strip()
         return {"answer": final_answer}
 
     if classification == "Brand":
@@ -164,13 +162,14 @@ def run_with_chain_context(inputs: Dict[str, Any]) -> Dict[str, str]:
             "chat_history": chat_history  # Pass history here
         })
         csv_answer = rag_result["result"].strip()
-        final_merged = cleaner_chain.merge(kb=csv_answer, web="")
-        final_answer = tailor_chain.run({"response": final_merged}).strip()
+        final_merged = cleaner_chain.merge(kb=csv_answer, web="", chat_history=chat_history)
+        final_answer = tailor_chain.run({"response": final_merged, "chat_history": chat_history}).strip()
         return {"answer": final_answer}
 
-    refusal_text = refusal_chain.run({})
-    final_refusal = tailor_chain.run({"response": refusal_text}).strip()
+    refusal_text = refusal_chain.run({"chat_history": chat_history})
+    final_refusal = tailor_chain.run({"response": refusal_text, "chat_history": chat_history}).strip()
     return {"answer": final_refusal}
+
 ###############################################################################
 # 7) Build a "Runnable" wrapper so .with_listeners() works
 ###############################################################################