demoPOC commited on
Commit
0f3983b
·
verified ·
1 Parent(s): cf31c18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,7 +35,7 @@ from langchain.docstore.document import Document
35
  from langchain.vectorstores import Chroma
36
  from langchain.text_splitter import RecursiveCharacterTextSplitter
37
 
38
- from langchain.chains import VectorDBQA, ConversationChain
39
 
40
  from langchain.document_loaders import UnstructuredFileLoader, TextLoader
41
  from langchain import PromptTemplate
@@ -242,7 +242,7 @@ def getRAGChain(customerName, customerDistrict, custDetailsPresent, vectordb,llm
242
  # "memory": memory
243
  # }
244
  # )
245
- chain = ConversationChain(
246
  llm=getLLMModel(llmID),
247
  chain_type='stuff',
248
  retriever=getRetriever(vectordb),
 
35
  from langchain.vectorstores import Chroma
36
  from langchain.text_splitter import RecursiveCharacterTextSplitter
37
 
38
+ from langchain.chains import VectorDBQA, ConversationChain, ConversationalRetrievalChain
39
 
40
  from langchain.document_loaders import UnstructuredFileLoader, TextLoader
41
  from langchain import PromptTemplate
 
242
  # "memory": memory
243
  # }
244
  # )
245
+ chain = ConversationalRetrievalChain.from_llm(
246
  llm=getLLMModel(llmID),
247
  chain_type='stuff',
248
  retriever=getRetriever(vectordb),