aman1762 commited on
Commit
d3dba58
·
verified ·
1 Parent(s): 7012e3b

Update rag_chain.py

Browse files
Files changed (1) hide show
  1. rag_chain.py +2 -1
rag_chain.py CHANGED
@@ -1,5 +1,6 @@
1
  from langchain.chains import RetrievalQA
2
- from langchain.llms import Groq
 
3
 
4
  def build_rag_chain(vectorstore, groq_api_key):
5
  llm = Groq(
 
1
  from langchain.chains import RetrievalQA
2
+ from langchain_groq import ChatGroq
3
+
4
 
5
  def build_rag_chain(vectorstore, groq_api_key):
6
  llm = Groq(