guangliang.yin commited on
Commit
2573026
·
1 Parent(s): f82d274

改为简单链

Browse files
Files changed (1) hide show
  1. app.py +13 -10
app.py CHANGED
@@ -17,6 +17,7 @@ from project.prompt.answer_by_private_prompt import (
17
  )
18
  from langchain.chains.combine_documents.refine import RefineDocumentsChain
19
  from langchain.chains.llm import LLMChain
 
20
 
21
  chain: Optional[Callable] = None
22
 
@@ -68,15 +69,17 @@ def web_loader(file, openai_key, puzhiai_key, zilliz_uri, user, password):
68
 
69
  llm = ZhipuAILLM(model="glm-3-turbo", temperature=0.1, zhipuai_api_key=puzhiai_key)
70
 
71
- initial_chain = LLMChain(llm=llm, prompt=DEFAULT_TEXT_QA_PROMPT)
72
- refine_chain = LLMChain(llm=llm, prompt=DEFAULT_REFINE_PROMPT)
73
- combine_documents_chain = RefineDocumentsChain(
74
- initial_llm_chain=initial_chain,
75
- refine_llm_chain=refine_chain,
76
- document_variable_name="context_str",
77
- initial_response_name="existing_answer",
78
- document_prompt=EXAMPLE_PROMPT,
79
- )
 
 
80
 
81
  chain = RetrievalQAWithSourcesChain(combine_documents_chain=combine_documents_chain,
82
  retriever=docsearch.as_retriever())
@@ -100,7 +103,7 @@ if __name__ == "__main__":
100
  """
101
  <h1><center>Langchain And Zilliz App</center></h1>
102
 
103
- v.2.27.15.2
104
 
105
  """
106
  )
 
17
  )
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
 
 
69
 
70
  llm = ZhipuAILLM(model="glm-3-turbo", temperature=0.1, zhipuai_api_key=puzhiai_key)
71
 
72
+ #initial_chain = LLMChain(llm=llm, prompt=DEFAULT_TEXT_QA_PROMPT)
73
+ #refine_chain = LLMChain(llm=llm, prompt=DEFAULT_REFINE_PROMPT)
74
+ #combine_documents_chain = RefineDocumentsChain(
75
+ # initial_llm_chain=initial_chain,
76
+ # refine_llm_chain=refine_chain,
77
+ # document_variable_name="context_str",
78
+ # initial_response_name="existing_answer",
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
  """
104
  <h1><center>Langchain And Zilliz App</center></h1>
105
 
106
+ v.2.27.15.21
107
 
108
  """
109
  )