Update tailor_chain.py

tailor_chain.py

@@ -1,15 +1,23 @@
 from langchain.chains import LLMChain
+from prompts import tailor_prompt
+import os
+from langchain_groq import ChatGroq
+
 from prompts import tailor_prompt
 
 def get_tailor_chain() -> LLMChain:
     """
-    Creates the tailor chain to simplify and personalize the assistant's responses.
+    Builds the chain that tailors the final response to DailyWellnessAI's style.
     """
-    tailor_chain = LLMChain(
-        llm=your_llm,  # Update this with your actual LLM model
+    chat_groq_model = ChatGroq(
+        model="Gemma2-9b-It",
+        groq_api_key=os.environ["GROQ_API_KEY"]
+    )
+    chain = LLMChain(
+        llm=chat_groq_model,
         prompt=tailor_prompt
     )
-    return tailor_chain
+    return chain
 
 def tailor_with_history(response: str, chat_history: list) -> str:
     """