Update pipeline.py

pipeline.py

@@ -20,7 +20,7 @@ except LookupError:
     english_words = set(words.words())
 
 # LangChain / Groq / LLM imports
-# from langchain_groq import ChatGroq
+from langchain_groq import ChatGroq
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_community.vectorstores import FAISS
 from langchain.chains import RetrievalQA, LLMChain
@@ -201,22 +201,19 @@ try:
         logger.warning("No Groq API key found for fallback LLM")
         
     # Initialize the Groq client as fallback
-    groq_fallback_llm = GroqClient(
+    groq_fallback_llm = ChatGroq(
         model=GROQ_MODELS["default"],
         temperature=0.7,
         groq_api_key=fallback_groq_api_key,
         max_tokens=2048
     )
     
-    # Verify connection by checking models (optional)
-    try:
-        groq_fallback_llm.list_models()
-        logger.info("Fallback Groq LLM initialized successfully")
-    except Exception as e:
-        logger.error(f"Fallback Groq LLM connection test failed: {e}")
-        
-    
-        
+    # # Verify connection by checking models (optional)
+    # try:
+    #     groq_fallback_llm.list_models()
+    #     logger.info("Fallback Groq LLM initialized successfully")
+    # except Exception as e:
+    #     logger.error(f"Fallback Groq LLM connection test failed: {e}")
 except Exception as e:
         logger.error(f"Failed to initialize fallback Groq LLM: {e}")