update config modify (#79)

dataelement · Oct 16, 2023 · 24976a8 · 24976a8
2 parents 3c90a82 + b28740e
commit 24976a8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/backend/bisheng/api/v1/knowledge.py b/src/backend/bisheng/api/v1/knowledge.py
@@ -328,7 +328,7 @@ def _read_chunk_text(input_file, file_name, size, chunk_overlap, separator):
         file_type = file_name.split('.')[-1]
         if file_type not in filetype_load_map:
             raise Exception('Unsupport file type')
-        loader = filetype_load_map[file_type]
+        loader = filetype_load_map[file_type](file_path=input_file)
         text_splitter = CharacterTextSplitter(separator=separator,
                                               chunk_size=size,
                                               chunk_overlap=chunk_overlap,

diff --git a/src/backend/bisheng/chat/manager.py b/src/backend/bisheng/chat/manager.py
@@ -485,7 +485,7 @@ async def process_source_document(self, source_document: List[Document], chat_id
 
         from bisheng.settings import settings
         # 使用大模型进行关键词抽取，模型配置临时方案
-        keyword_conf = settings.default_llm
+        keyword_conf = settings.default_llm or {}
         host_base_url = keyword_conf.get('host_base_url')
         model = keyword_conf.get('model')