farmax commited on
Commit
4147ba2
·
verified ·
1 Parent(s): c1afe21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ from langchain_community.vectorstores import Chroma
9
  from langchain_community.document_loaders import UnstructuredPDFLoader, PyPDFLoader
10
  from langchain.text_splitter import CharacterTextSplitter
11
  from langchain.chains import RetrievalQAWithSourcesChain
12
- # from langchain.chains import load_qa_with_sources_from_chain_type
13
  from langchain.schema import Document
14
  from langchain.memory import ConversationBufferMemory
15
  from langchain.callbacks.manager import CallbackManager
@@ -77,7 +77,7 @@ def initialize_LLM(llm_option, llm_temperature, max_tokens, top_k, vector_db, pr
77
  print(f"Using default LLM {default_llm} for {language}")
78
  llm_name = default_llm
79
 
80
- qa_chain = RetrievalQAWithSourcesChain(
81
  llm=llm_name,
82
  chain_type="stuff",
83
  retriever=vector_db.as_retriever(),
 
9
  from langchain_community.document_loaders import UnstructuredPDFLoader, PyPDFLoader
10
  from langchain.text_splitter import CharacterTextSplitter
11
  from langchain.chains import RetrievalQAWithSourcesChain
12
+ from langchain.chains import load_qa_with_sources_from_chain_type
13
  from langchain.schema import Document
14
  from langchain.memory import ConversationBufferMemory
15
  from langchain.callbacks.manager import CallbackManager
 
77
  print(f"Using default LLM {default_llm} for {language}")
78
  llm_name = default_llm
79
 
80
+ qa_chain = load_qa_with_sources_from_chain_type(
81
  llm=llm_name,
82
  chain_type="stuff",
83
  retriever=vector_db.as_retriever(),