diff --git a/server/knowledge_base/utils.py b/server/knowledge_base/utils.py index ab4c2f9..e34c326 100644 --- a/server/knowledge_base/utils.py +++ b/server/knowledge_base/utils.py @@ -16,7 +16,7 @@ from configs.model_config import ( TEXT_SPLITTER ) import importlib -from text_splitter import zh_title_enhance +from text_splitter import zh_title_enhance as func_zh_title_enhance import langchain.document_loaders from langchain.docstore.document import Document from langchain.text_splitter import TextSplitter @@ -308,7 +308,7 @@ class KnowledgeFile: print(f"文档切分示例:{docs[0]}") if zh_title_enhance: - docs = zh_title_enhance(docs) + docs = func_zh_title_enhance(docs) self.splited_docs = docs return self.splited_docs