Tuchuanhuhuhu commited on
Commit
ad48b38
·
1 Parent(s): 523e0f2

移除多余的callback

Browse files
Files changed (1) hide show
  1. modules/models/base_model.py +1 -2
modules/models/base_model.py CHANGED
@@ -269,8 +269,7 @@ class BaseLLMModel:
269
  from langchain.callbacks import StdOutCallbackHandler
270
  prompt_template = "Write a concise summary of the following:\n\n{text}\n\nCONCISE SUMMARY IN " + language + ":"
271
  PROMPT = PromptTemplate(template=prompt_template, input_variables=["text"])
272
- handler = StdOutCallbackHandler()
273
- llm = ChatOpenAI(callbacks=[handler])
274
  chain = load_summarize_chain(llm, chain_type="map_reduce", return_intermediate_steps=True, map_prompt=PROMPT, combine_prompt=PROMPT)
275
  summary = chain({"input_documents": list(index.docstore.__dict__["_dict"].values())}, return_only_outputs=True)["output_text"]
276
  print(i18n("总结") + f": {summary}")
 
269
  from langchain.callbacks import StdOutCallbackHandler
270
  prompt_template = "Write a concise summary of the following:\n\n{text}\n\nCONCISE SUMMARY IN " + language + ":"
271
  PROMPT = PromptTemplate(template=prompt_template, input_variables=["text"])
272
+ llm = ChatOpenAI()
 
273
  chain = load_summarize_chain(llm, chain_type="map_reduce", return_intermediate_steps=True, map_prompt=PROMPT, combine_prompt=PROMPT)
274
  summary = chain({"input_documents": list(index.docstore.__dict__["_dict"].values())}, return_only_outputs=True)["output_text"]
275
  print(i18n("总结") + f": {summary}")