Spaces:
Sleeping
Sleeping
Tuchuanhuhuhu
commited on
Commit
·
ad48b38
1
Parent(s):
523e0f2
移除多余的callback
Browse files
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 |
-
|
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}")
|