Deaksh commited on
Commit
0ac93ae
·
verified ·
1 Parent(s): b4b16b3

Update llm_helper.py

Browse files
Files changed (1) hide show
  1. llm_helper.py +4 -4
llm_helper.py CHANGED
@@ -1,10 +1,10 @@
1
  from langchain_groq import ChatGroq
2
  import os
3
- from dotenv import load_dotenv
4
-
5
- load_dotenv()
6
- llm = ChatGroq(groq_api_key=os.getenv("GROQ_API_KEY"), model_name="llama-3.3-70b-versatile")
7
 
 
 
 
 
8
 
9
  if __name__ == "__main__":
10
  response = llm.invoke("Two most important ingredient in samosa are ")
 
1
  from langchain_groq import ChatGroq
2
  import os
 
 
 
 
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 ")