File size: 490 Bytes
fb8e56f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# 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
|