Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
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 =
|
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(),
|