Update pipeline.py

pipeline.py

@@ -55,7 +55,10 @@ def moderate_text(query: str) -> str:
     Classifies the query as harmful or not using Mistral Moderation via Pydantic AI.
     Returns "OutOfScope" if harmful, otherwise returns the original query.
     """
-    response = mistral_agent.call("classify", {"inputs": [query]})
+    # Use the agent to interact with the moderation model, here assuming predict() method
+    response = mistral_agent.predict({"inputs": [query]})  # Adjust if necessary
+    
+    # Assuming response contains a 'results' field similar to the previous example
     categories = response['results'][0]['categories']
 
     # Check if harmful content is flagged in moderation categories