Update pipeline.py

pipeline.py

@@ -27,13 +27,13 @@ from langchain.chains import RetrievalQA, LLMChain
 from langchain.prompts import PromptTemplate
 from langchain.docstore.document import Document
 from langchain_core.caches import BaseCache
-from langchain_core.callbacks import CallbackManager
+# from langchain_core.callbacks import CallbackManager
 # from langchain.callbacks.base import BaseCallbacks  # Updated import
 # from langchain.callbacks.manager import CallbackManager
 # from langchain.callbacks import StdOutCallbackHandler
 
 # Custom chain imports
-from groq_client import GroqClient
+# from groq_client import GroqClient
 from classification_chain import get_classification_chain
 from refusal_chain import get_refusal_chain
 from tailor_chain import get_tailor_chain
@@ -208,13 +208,15 @@ class NoCache(BaseCache):
     def clear(self):
         pass
 
+# Rebuild the ChatGroq model after defining NoCache
+ChatGroq.model_rebuild()
 # Initialize ChatGroq with cache
 try:
     fallback_groq_api_key = os.environ.get("GROQ_API_KEY_FALLBACK", os.environ.get("GROQ_API_KEY"))
     if not fallback_groq_api_key:
         logger.warning("No Groq API key found for fallback LLM")
     groq_fallback_llm = ChatGroq(
-        model=GROQ_MODELS["default"],
+        model="default",  # Replace with your actual model name if different
         temperature=0.7,
         groq_api_key=fallback_groq_api_key,
         max_tokens=2048,
@@ -223,7 +225,6 @@ try:
 except Exception as e:
     logger.error(f"Failed to initialize fallback Groq LLM: {e}")
     raise RuntimeError("ChatGroq initialization failed.") from e
-
 # -------------------------------------------------------
 # Rate-limit & Cache
 # -------------------------------------------------------