From 14387d9661998272ed9f1849edcdb7ec408fd114 Mon Sep 17 00:00:00 2001 From: aries_ckt <916701291@qq.com> Date: Wed, 28 Jun 2023 10:35:55 +0800 Subject: [PATCH] fix:return knowledge dict format knowledge result --- pilot/openapi/api_v1/api_v1.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/pilot/openapi/api_v1/api_v1.py b/pilot/openapi/api_v1/api_v1.py index 9a3d3d88e..ba5ef2f04 100644 --- a/pilot/openapi/api_v1/api_v1.py +++ b/pilot/openapi/api_v1/api_v1.py @@ -21,6 +21,7 @@ from pilot.server.api_v1.api_view_model import ( from pilot.configs.config import Config from pilot.openapi.knowledge.knowledge_service import KnowledgeService from pilot.openapi.knowledge.request.knowledge_request import KnowledgeSpaceRequest + from pilot.scene.base_chat import BaseChat from pilot.scene.base import ChatScene from pilot.scene.chat_factory import ChatFactory @@ -127,8 +128,13 @@ def plugins_select_info(): def knowledge_list(): + """return knowledge space list""" + params: dict = {} request = KnowledgeSpaceRequest() - return knowledge_service.get_knowledge_space(request) + spaces = knowledge_service.get_knowledge_space(request) + for space in spaces: + params.update({space.name: space.name}) + return params @router.post("/v1/chat/mode/params/list", response_model=Result[dict])