chukbert commited on
Commit
a59fe19
·
verified ·
1 Parent(s): 15bf530

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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 = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=None)
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