aman1762 commited on
Commit
62f1a72
·
verified ·
1 Parent(s): d3dba58

Update rag_chain.py

Browse files
Files changed (1) hide show
  1. rag_chain.py +5 -4
rag_chain.py CHANGED
@@ -3,10 +3,11 @@ from langchain_groq import ChatGroq
3
 
4
 
5
  def build_rag_chain(vectorstore, groq_api_key):
6
- llm = Groq(
7
- api_key=groq_api_key,
8
- model_name="llama3-8b-8192"
9
- )
 
10
 
11
  return RetrievalQA.from_chain_type(
12
  llm=llm,
 
3
 
4
 
5
  def build_rag_chain(vectorstore, groq_api_key):
6
+ llm = ChatGroq(
7
+ api_key=groq_api_key,
8
+ model="llama3-8b-8192"
9
+ )
10
+
11
 
12
  return RetrievalQA.from_chain_type(
13
  llm=llm,