Update pipeline.py

pipeline.py

@@ -173,7 +173,52 @@ def do_web_search(query: str) -> str:
     response = manager_agent.run(search_query)
     return response
 
-# Orchestrator: run_with_chain
+
+# # Orchestrator: run_with_chain
+# def run_with_chain(query: str) -> str:
+#     print("DEBUG: Starting run_with_chain...")
+    
+#     # Moderate the query for harmful content
+#     moderated_query = moderate_text(query)
+#     if moderated_query == "OutOfScope":
+#         return "Sorry, this query contains harmful or inappropriate content."
+
+#     # Classify the query
+#     class_result = classification_chain.invoke({"query": moderated_query})
+#     classification = class_result.get("text", "").strip()
+#     print("DEBUG: Classification =>", classification)
+
+#     if classification == "OutOfScope":
+#         refusal_text = refusal_chain.run({"topic": "this topic"})
+#         final_refusal = tailor_chain.run({"response": refusal_text})
+#         return final_refusal.strip()
+
+#     if classification == "Wellness":
+#         rag_result = wellness_rag_chain({"query": moderated_query})
+#         csv_answer = rag_result["result"].strip()
+#         if not csv_answer:
+#             web_answer = do_web_search(moderated_query)
+#         else:
+#             lower_ans = csv_answer.lower()
+#             if any(phrase in lower_ans for phrase in ["i do not know", "not sure", "no context", "cannot answer"]):
+#                 web_answer = do_web_search(moderated_query)
+#             else:
+#                 web_answer = ""
+#         final_merged = cleaner_chain.merge(kb=csv_answer, web=web_answer)
+#         final_answer = tailor_chain.run({"response": final_merged})
+#         return final_answer.strip()
+
+#     if classification == "Brand":
+#         rag_result = brand_rag_chain({"query": moderated_query})
+#         csv_answer = rag_result["result"].strip()
+#         final_merged = cleaner_chain.merge(kb=csv_answer, web="")
+#         final_answer = tailor_chain.run({"response": final_merged})
+#         return final_answer.strip()
+
+#     refusal_text = refusal_chain.run({"topic": "this topic"})
+#     final_refusal = tailor_chain.run({"response": refusal_text})
+#     return final_refusal.strip()
+
 def run_with_chain(query: str) -> str:
     print("DEBUG: Starting run_with_chain...")
     
@@ -182,9 +227,8 @@ def run_with_chain(query: str) -> str:
     if moderated_query == "OutOfScope":
         return "Sorry, this query contains harmful or inappropriate content."
 
-    # Classify the query
-    class_result = classification_chain.invoke({"query": moderated_query})
-    classification = class_result.get("text", "").strip()
+    # Classify the query manually, ensuring box breathing is recognized
+    classification = classify_query(moderated_query)
     print("DEBUG: Classification =>", classification)
 
     if classification == "OutOfScope":
@@ -217,3 +261,4 @@ def run_with_chain(query: str) -> str:
     refusal_text = refusal_chain.run({"topic": "this topic"})
     final_refusal = tailor_chain.run({"response": refusal_text})
     return final_refusal.strip()
+