llm_helper.py

import os
from langchain_groq import ChatGroq

llm = ChatGroq(
    api_key=os.environ["GROQ_API_KEY"],  # This will raise an error if unset
    model_name="llama-3.1-8b-instant"
)

if __name__ == "__main__":
    response = llm.invoke("Two most important ingredient in samosa are")
    print(response.content)