nickmuchi commited on
Commit
d2773f5
1 Parent(s): 698009d

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +1 -1
functions.py CHANGED
@@ -204,7 +204,7 @@ def embed_text(query,title,embedding_model,_docsearch):
204
  chain_type_kwargs = {"prompt": prompt}
205
 
206
  chain = VectorDBQA.from_chain_type(
207
- llm=streaming_llm,
208
  chain_type="stuff",
209
  vectorstore=_docsearch,
210
  chain_type_kwargs=chain_type_kwargs,
 
204
  chain_type_kwargs = {"prompt": prompt}
205
 
206
  chain = VectorDBQA.from_chain_type(
207
+ llm=chat_llm,
208
  chain_type="stuff",
209
  vectorstore=_docsearch,
210
  chain_type_kwargs=chain_type_kwargs,