Update app.py
Browse files
app.py
CHANGED
@@ -12,6 +12,7 @@ from langchain_community.document_loaders import PyPDFLoader
|
|
12 |
from langchain_openai import ChatOpenAI
|
13 |
from langchain.callbacks import get_openai_callback
|
14 |
from io import BytesIO
|
|
|
15 |
|
16 |
openai.api_key = os.getenv("OPENAI_API_KEY")
|
17 |
|
@@ -39,7 +40,7 @@ def summarize_document(docs, llm):
|
|
39 |
|
40 |
{context}
|
41 |
"""
|
42 |
-
chain =
|
43 |
with get_openai_callback() as cb:
|
44 |
summary = chain.invoke({"context": docs})
|
45 |
return summary, cb
|
|
|
12 |
from langchain_openai import ChatOpenAI
|
13 |
from langchain.callbacks import get_openai_callback
|
14 |
from io import BytesIO
|
15 |
+
from langchain.chains.combine_documents import create_stuff_documents_chain
|
16 |
|
17 |
openai.api_key = os.getenv("OPENAI_API_KEY")
|
18 |
|
|
|
40 |
|
41 |
{context}
|
42 |
"""
|
43 |
+
chain = create_stuff_documents_chain(llm, prompt)
|
44 |
with get_openai_callback() as cb:
|
45 |
summary = chain.invoke({"context": docs})
|
46 |
return summary, cb
|