File size: 526 Bytes
550e28f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import os
from langchain_groq import ChatGroq
def get_llm():
"""
Returns the language model instance (LLM) using ChatGroq API.
The LLM used is Llama 3.3 with a versatile 70 billion parameter model.
"""
api_key = os.getenv("CHAT_GROQ_API_KEY")
if not api_key:
raise ValueError("Please set the CHAT_GROQ_API_KEY environment variable.")
llm = ChatGroq(
model="llama-3.3-70b-versatile",
temperature=0,
max_tokens=1024,
api_key=api_key
)
return llm
|