Dataset / langchain_folder /llm_helper.py
vansh9878's picture
files added
825e978
raw
history blame contribute delete
291 Bytes
from dotenv import load_dotenv
from langchain_groq import ChatGroq
import os
load_dotenv()
llm = ChatGroq(groq_api_key = os.getenv('groq_api'), model_name = "llama-3.1-8b-instant")
#llm = ChatGroq(groq_api_key = os.getenv('groq_deepseek_api'), model_name = "DeepSeek R1 Distill Llama 70B")