lm.py

import os
from dotenv import load_dotenv
from langchain_groq import ChatGroq

load_dotenv()
key1 = os.getenv('GROQ_API_KEY_1')
key2 = os.getenv('GROQ_API_KEY_2')



def get_query_llm():
    return ChatGroq(model="mistral-saba-24b",api_key=key1)

def get_answer_llm():
    return ChatGroq(model="mistral-saba-24b",api_key=key2)