ADDED SANITIZE FUNCTION

pipeline.py

@@ -228,9 +228,24 @@ def merge_responses(kb_answer: str, web_answer: str) -> str:
         return f"Knowledge Base Answer: {kb_answer.strip()}\n\nWeb Search Result: {web_answer.strip()}"
     except Exception as e:
         return f"Error merging responses: {str(e)}"
+def sanitize_message(message: Any) -> str:
+    """Sanitize message input to ensure it's a valid string."""
+    if hasattr(message, 'content'):
+        return str(message.content)
+    if isinstance(message, (list, dict)):
+        return str(message)
+    return str(message)
 
+# Modify your run_pipeline function to include the sanitization
 def run_pipeline(query: str) -> str:
     try:
+        # Sanitize input
+        query = sanitize_message(query)
+        
+        # Rest of your pipeline code...
+        moderation_result = moderate_text(query)
+        if not moderation_result.is_safe:
+            return "Sorry, this query contains harmful or inappropriate content."
         # Validate and moderate input
         moderation_result = moderate_text(query)
         if not moderation_result.is_safe: