bupa1018 commited on
Commit
15a869d
·
1 Parent(s): 880dc1b

Update llm.py

Browse files
Files changed (1) hide show
  1. llm.py +2 -0
llm.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  def get_groq_llm(model_name, temperature, api_key):
2
  llm = ChatGroq(model=model_name, temperature=temperature, api_key=api_key)
3
  return llm
 
1
+ from langchain_groq import ChatGroq
2
+
3
  def get_groq_llm(model_name, temperature, api_key):
4
  llm = ChatGroq(model=model_name, temperature=temperature, api_key=api_key)
5
  return llm