1  from langchain_groq import ChatGroq
2  from langchain_core.language_models.chat_models import BaseChatModel
3  
4  from src.setup.config import groq_config
5  
6  
7  def initiate_llm(temperature: int = 0, model_name: str = groq_config.preferred_model) -> BaseChatModel:
8      return ChatGroq(model=model_name, temperature=temperature)
9