Update cleaner_chain.py

cleaner_chain.py

@@ -6,16 +6,11 @@ from prompts import cleaner_prompt
 
 class CleanerChain(LLMChain):
     def merge(self, kb: str, web: str) -> str:
-        return self.run({"kb_answer": kb, "web_answer": web})
+        # Use invoke() instead of run() to comply with new LangChain practices
+        return self.invoke({"kb_answer": kb, "web_answer": web})
 
 def get_cleaner_chain() -> CleanerChain:
-    """
-    Builds the 'CleanerChain' that merges CSV answer + web result.
-    """
-    chat_groq_model = ChatGroq(
-        model="Gemma2-9b-It",
-        groq_api_key=os.environ["GROQ_API_KEY"]
-    )
+    chat_groq_model = ChatGroq(model="Gemma2-9b-It", groq_api_key=os.environ["GROQ_API_KEY"])
     chain = CleanerChain(
         llm=chat_groq_model,
         prompt=cleaner_prompt