guangliang.yin commited on
Commit
80b6833
1 Parent(s): 2573026

改为简单链-2

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -18,6 +18,7 @@ from project.prompt.answer_by_private_prompt import (
18
  from langchain.chains.combine_documents.refine import RefineDocumentsChain
19
  from langchain.chains.llm import LLMChain
20
  from langchain.chains.combine_documents import create_stuff_documents_chain
 
21
 
22
  chain: Optional[Callable] = None
23
 
@@ -79,7 +80,11 @@ def web_loader(file, openai_key, puzhiai_key, zilliz_uri, user, password):
79
  # document_prompt=EXAMPLE_PROMPT,
80
  #)
81
 
82
- combine_documents_chain = create_stuff_documents_chain(llm, prompt=QUESTION_PROMPT)
 
 
 
 
83
 
84
  chain = RetrievalQAWithSourcesChain(combine_documents_chain=combine_documents_chain,
85
  retriever=docsearch.as_retriever())
@@ -103,7 +108,7 @@ if __name__ == "__main__":
103
  """
104
  <h1><center>Langchain And Zilliz App</center></h1>
105
 
106
- v.2.27.15.21
107
 
108
  """
109
  )
 
18
  from langchain.chains.combine_documents.refine import RefineDocumentsChain
19
  from langchain.chains.llm import LLMChain
20
  from langchain.chains.combine_documents import create_stuff_documents_chain
21
+ from langchain.chains import StuffDocumentsChain
22
 
23
  chain: Optional[Callable] = None
24
 
 
80
  # document_prompt=EXAMPLE_PROMPT,
81
  #)
82
 
83
+ combine_documents_chain = StuffDocumentsChain(
84
+ llm_chain=llm,
85
+ document_prompt=QUESTION_PROMPT,
86
+ document_variable_name="context"
87
+ )
88
 
89
  chain = RetrievalQAWithSourcesChain(combine_documents_chain=combine_documents_chain,
90
  retriever=docsearch.as_retriever())
 
108
  """
109
  <h1><center>Langchain And Zilliz App</center></h1>
110
 
111
+ v.2.27.15.27
112
 
113
  """
114
  )