From 62ede7d7fba4e025910b586136ab066a67bc7807 Mon Sep 17 00:00:00 2001 From: xuyuan23 <643854343@qq.com> Date: Fri, 1 Dec 2023 11:38:02 +0800 Subject: [PATCH] optimize db connection and knowledge QA --- pilot/base_modules/meta_data/meta_data.py | 2 +- pilot/openapi/api_v1/api_v1.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/pilot/base_modules/meta_data/meta_data.py b/pilot/base_modules/meta_data/meta_data.py index 5b79a38a5..84b83b7f3 100644 --- a/pilot/base_modules/meta_data/meta_data.py +++ b/pilot/base_modules/meta_data/meta_data.py @@ -58,7 +58,7 @@ if CFG.LOCAL_DB_TYPE == "mysql": + CFG.LOCAL_DB_HOST + ":" + str(CFG.LOCAL_DB_PORT) - + f"/{db_name}" + + f"/{db_name}", pool_pre_ping=True, pool_recycle=300 ) else: engine = create_engine(f"sqlite:///{db_path}") diff --git a/pilot/openapi/api_v1/api_v1.py b/pilot/openapi/api_v1/api_v1.py index 03bf0de48..5deb6fc41 100644 --- a/pilot/openapi/api_v1/api_v1.py +++ b/pilot/openapi/api_v1/api_v1.py @@ -120,9 +120,9 @@ def knowledge_list_info(): return params -def knowledge_list(): +def knowledge_list(user_id: str = None): """return knowledge space list""" - request = KnowledgeSpaceRequest() + request = KnowledgeSpaceRequest(user_id=user_id) spaces = knowledge_service.get_knowledge_space(request) space_list = [] for space in spaces: @@ -287,9 +287,9 @@ async def params_list(chat_mode: str = ChatScene.ChatNormal.value(), user_token: elif ChatScene.ChatExecution.value() == chat_mode: result = plugins_select_info() elif ChatScene.ChatKnowledge.value() == chat_mode: - result = knowledge_list() + result = knowledge_list(user_id=user_token.user_id) elif ChatScene.ChatKnowledge.ExtractRefineSummary.value() == chat_mode: - result = knowledge_list() + result = knowledge_list(user_id=user_token.user_id) else: return Result.succ(None) return Result.succ(result)