File size: 364 Bytes
e23acaf
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import os
from langchain_groq import ChatGroq
from src.config.settings import settings
from src.utils.logger import get_logger

logger = get_logger(__name__)


def get_llm():

    logger.info(f"Loading Groq model: {settings.llm_model}")

    return ChatGroq(
        model=settings.llm_model,
        api_key=os.getenv("GROQ_API_KEY"),
        temperature=0
    )