Persona-postgenerator / llm_helper.py
Deaksh's picture
Update llm_helper.py
04f20c1 verified
raw
history blame
302 Bytes
from langchain_groq import ChatGroq
import os
from dotenv import load_dotenv
load_dotenv()
llm = ChatGroq(api_key=os.getenv("GROQ_API_KEY"), model_name="llama3-70b-8192")
if __name__ == "__main__":
response = llm.invoke("Two most important ingredient in samosa are ")
print(response.content)