src/langgraphagenticai/LLMS/groqllm.py

import streamlit as st
import os
from langchain_groq import ChatGroq

class GroqLLM:
    def __init__(self,user_controls_input):
        self.user_controls_input = user_controls_input

    def get_llm_model(self):
        try:
            groq_api_key = self.user_controls_input["GROQ_API_KEY"]
            selected_groq_model = self.user_controls_input["selected_groq_model"]
            if groq_api_key=='' and os.environ["GROQ_API_KEY"]=='':
                st.error("Please Enter the Groq API key")
            llm = ChatGroq(api_key=groq_api_key,model=selected_groq_model)
        except Exception as e:
            raise ValueError(f"Error occured with Exception: {e}")
        
        return llm