agents/agents.py

from langchain_groq import ChatGroq
import os
from dotenv import load_dotenv

# Load environment variables
load_dotenv()
os.getenv("GROQ_API_KEY") 

def get_agent_groq(temperature=0.7):

    # Load the Groq model
    chat = ChatGroq(
        temperature=temperature,
        model="llama3-70b-8192", #define model available on Groq
        api_key=os.getenv("GROQ_API_KEY")
    )
    return chat