Skip to content

Commit 8c470f8

Browse files
authored
fix: wrong parameters when recreate vectorstore (#5043)
1 parent f9e3895 commit 8c470f8

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

libs/chatchat-server/chatchat/server/knowledge_base/kb_doc_api.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -400,8 +400,8 @@ def download_doc(
400400
def recreate_vector_store(
401401
knowledge_base_name: str = Body(..., examples=["samples"]),
402402
allow_empty_kb: bool = Body(True),
403-
vs_type: str = Body(Settings.kb_settings.DEFAULT_VS_TYPE),
404-
embed_model: str = Body(get_default_embedding()),
403+
vs_type: str = Body(Settings.kb_settings.DEFAULT_VS_TYPE, description="为空知识库指定向量库类型。已有知识库默认使用原向量库类型。"),
404+
embed_model: str = Body(get_default_embedding(), description="为空知识库指定Embedding模型。已有知识库默认使用原Embedding模型。"),
405405
chunk_size: int = Body(Settings.kb_settings.CHUNK_SIZE, description="知识库中单段文本最大长度"),
406406
chunk_overlap: int = Body(Settings.kb_settings.OVERLAP_SIZE, description="知识库中相邻文本重合长度"),
407407
zh_title_enhance: bool = Body(Settings.kb_settings.ZH_TITLE_ENHANCE, description="是否开启中文标题加强"),
@@ -416,7 +416,9 @@ def recreate_vector_store(
416416

417417
def output():
418418
try:
419-
kb = KBServiceFactory.get_service(knowledge_base_name, vs_type, embed_model)
419+
kb = KBServiceFactory.get_service_by_name(knowledge_base_name)
420+
if kb is None:
421+
kb = KBServiceFactory.get_service(knowledge_base_name, vs_type, embed_model)
420422
if not kb.exists() and not allow_empty_kb:
421423
yield {"code": 404, "msg": f"未找到知识库 ‘{knowledge_base_name}’"}
422424
else:

0 commit comments

Comments
 (0)