Update pipeline.py

pipeline.py

@@ -11,6 +11,12 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, ManagedAgent, LiteLLMMod
 import subprocess  # Import subprocess to run shell commands
 from langchain.llms.base import LLM  # Import LLM
 
+# Import the functions from respective chain files
+from classification_chain import get_classification_chain
+from refusal_chain import get_refusal_chain
+from tailor_chain import get_tailor_chain
+from cleaner_chain import get_cleaner_chain
+
 # Mistral Client Setup
 from mistralai import Mistral  # Import the Mistral client
 from pydantic_ai import Agent  # Import Pydantic AI's Agent