Update pipeline.py

pipeline.py

@@ -173,6 +173,20 @@ def do_web_search(query: str) -> str:
     response = manager_agent.run(search_query)
     return response
 
+# Modify the classification logic to recognize box breathing
+def classify_query(query: str) -> str:
+    wellness_keywords = ["box breathing", "meditation", "yoga", "mindfulness", "breathing exercises"]
+    
+    if any(keyword in query.lower() for keyword in wellness_keywords):
+        return "Wellness"
+    
+    # If not recognized as wellness, use the classification chain
+    class_result = classification_chain.invoke({"query": query})
+    classification = class_result.get("text", "").strip()
+    
+    if classification == "OutOfScope":
+        return "OutOfScope"
+    return classification
 
 # # Orchestrator: run_with_chain
 # def run_with_chain(query: str) -> str: