| |
| import os |
| from langchain.chains import LLMChain |
| from langchain_groq import ChatGroq |
| 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}) |
|
|
| 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"] |
| ) |
| chain = CleanerChain( |
| llm=chat_groq_model, |
| prompt=cleaner_prompt |
| ) |
| return chain |
|
|
