FYP / lm.py
Saim-11's picture
Upload 6 files
aa7cb11 verified
raw
history blame contribute delete
340 Bytes
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)