Deaksh commited on
Commit
c0ea645
·
verified ·
1 Parent(s): 04f20c1

Update llm_helper.py

Browse files
Files changed (1) hide show
  1. llm_helper.py +6 -5
llm_helper.py CHANGED
@@ -1,10 +1,11 @@
1
- from langchain_groq import ChatGroq
2
  import os
3
- from dotenv import load_dotenv
4
 
5
- load_dotenv()
6
- llm = ChatGroq(api_key=os.getenv("GROQ_API_KEY"), model_name="llama3-70b-8192")
 
 
7
 
8
  if __name__ == "__main__":
9
- response = llm.invoke("Two most important ingredient in samosa are ")
10
  print(response.content)
 
 
1
  import os
2
+ from langchain_groq import ChatGroq
3
 
4
+ llm = ChatGroq(
5
+ api_key=os.environ["GROQ_API_KEY"], # This will raise an error if unset
6
+ model_name="llama3-70b-8192"
7
+ )
8
 
9
  if __name__ == "__main__":
10
+ response = llm.invoke("Two most important ingredient in samosa are")
11
  print(response.content)