Update pipeline.py

pipeline.py

@@ -172,6 +172,7 @@ def classify_query(query: str) -> str:
         
         # Use chain for everything else
         class_result = classification_chain.invoke({"query": query_input.query})
+        print(class_result)
         classification = class_result.get("text", "").strip()
         
         return classification if classification != "" else "OutOfScope"
@@ -395,7 +396,7 @@ def run_pipeline(query: str) -> str:
             return "Sorry, this query contains harmful or inappropriate content."
         
         # Classify
-        classification = classificatin_chain.invoke({"query":"moderation_result.original_text"})
+        classification = classify_query(moderation_result.original_text)
         
         # If out-of-scope, refuse
         if classification == "OutOfScope":