optimize db connection and knowledge QA

This commit is contained in:
xuyuan23
2023-12-01 11:38:02 +08:00
parent c4d23d81c5
commit 62ede7d7fb
2 changed files with 5 additions and 5 deletions

View File

@@ -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}")

View File

@@ -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)