fix: 修复总结标题时传参错误

dataelement · May 9, 2024 · 47b2220 · 47b2220
1 parent 38f2fa0
commit 47b2220
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/backend/bisheng/api/services/knowledge_imp.py b/src/backend/bisheng/api/services/knowledge_imp.py
@@ -407,7 +407,7 @@ def read_chunk_text(input_file, file_name, size, chunk_overlap, separator):
             logger.info(f'need_extract_title')
             for one in documents:
                 # 配置了相关llm的话，就对文档做总结
-                title = extract_title(llm, documents)
+                title = extract_title(llm, one.page_content)
                 one.metadata['title'] = title
 
         text_splitter = ElemCharacterTextSplitter(separators=separator,