Update rag_chain.py
Browse files- 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 =
|
| 7 |
-
|
| 8 |
-
|
| 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,
|