Create tailor_chain.py

tailor_chain.py

@@ -0,0 +1,20 @@
+# tailor_chain.py
+import os
+from langchain.chains import LLMChain
+from langchain_groq import ChatGroq
+
+from prompts import tailor_prompt
+
+def get_tailor_chain() -> LLMChain:
+    """
+    Builds the chain that tailors the final response to DailyWellnessAI's style.
+    """
+    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 chain