Update pipeline.py

pipeline.py

@@ -402,7 +402,7 @@ def run_pipeline(query: str) -> str:
         
         # If out-of-scope, refuse
         if classification == "OutOfScope":
-            refusal_text = refusal_chain.run({"topic": topic})
+            refusal_text = refusal_chain.invoke({"topic": topic,"query":query}})
             return tailor_chain.run({"response": refusal_text}).strip()
         
         # Otherwise, do a RAG query and also do a web search (cached)
@@ -431,7 +431,7 @@ def run_pipeline(query: str) -> str:
             return tailor_chain.run({"response": final_merged}).strip()
         
         # If it doesn't fall under known categories, return refusal by default.
-        refusal_text = refusal_chain.run({"topic": topic})
+        refusal_text = refusal_chain.invoke({"topic": topic,"query":query})
         return tailor_chain.run({"response": refusal_text}).strip()
         
     except ValidationError as e: