From 04dcd905022b55bc5354828d6e5c283ccec5d6b1 Mon Sep 17 00:00:00 2001 From: aries_ckt <916701291@qq.com> Date: Tue, 31 Oct 2023 16:14:26 +0800 Subject: [PATCH] feat:document summary --- pilot/server/knowledge/service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pilot/server/knowledge/service.py b/pilot/server/knowledge/service.py index cd18c539f..4db3d6c51 100644 --- a/pilot/server/knowledge/service.py +++ b/pilot/server/knowledge/service.py @@ -429,7 +429,7 @@ class KnowledgeService: from llama_index import PromptHelper from llama_index.prompts.default_prompt_selectors import DEFAULT_TREE_SUMMARIZE_PROMPT_SEL texts = [doc.page_content for doc in chunk_docs] - prompt_helper = PromptHelper(context_window=3900) + prompt_helper = PromptHelper() texts = prompt_helper.repack(prompt=DEFAULT_TREE_SUMMARIZE_PROMPT_SEL, text_chunks=texts) logger.info( f"async_document_summary, doc:{doc.doc_name}, chunk_size:{len(texts)}, begin generate summary"