Persona-postgenerator / llm_helper.py
Deaksh's picture
Update llm_helper.py
c0ea645 verified
raw
history blame contribute delete
304 Bytes
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="llama3-70b-8192"
)
if __name__ == "__main__":
response = llm.invoke("Two most important ingredient in samosa are")
print(response.content)