MahatirTusher commited on
Commit
e4a5b6a
·
verified ·
1 Parent(s): 1cd1a63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -315,7 +315,10 @@ def create_qa_chain(vectorstore, llm):
315
  llm=llm,
316
  retriever=retriever,
317
  chain_type="stuff",
318
- chain_type_kwargs={"prompt": qa_prompt}
 
 
 
319
  )
320
  return qa_chain
321
 
 
315
  llm=llm,
316
  retriever=retriever,
317
  chain_type="stuff",
318
+ chain_type_kwargs={
319
+ "prompt": qa_prompt,
320
+ "document_variable_name": "context" # Match the variable name in the prompt
321
+ }
322
  )
323
  return qa_chain
324