src/llms/groqllm.py

from langchain_groq import ChatGroq
import os 
from dotenv import load_dotenv

class GroqLLM:
    def __init__(self):
        load_dotenv()

    def get_llm(self):
        try:
            print(os.getenv("GROQ_API_KEY"))
            os.environ["GROQ_API_KEY"] = self.groq_api_key = os.getenv("GROQ_API_KEY")
            llm=ChatGroq(api_key=self.groq_api_key, model="openai/gpt-oss-120b", streaming=False)
            return llm
        except Exception as e:
            raise ValueError(f"Error occurred with exception: {e}")
        
    def get_moon(self):
        try:
            print(os.getenv("GROQ_API_KEY"))
            os.environ["GROQ_API_KEY"] = self.groq_api_key = os.getenv("GROQ_API_KEY")
            llm=ChatGroq(api_key=self.groq_api_key, model="moonshotai/kimi-k2-instruct", streaming=False)
            return llm
        except Exception as e:
            raise ValueError(f"Error occurred with exception: {e}")