github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-30 15:01:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:return knowledge dict

Browse Source 
format knowledge result
This commit is contained in:
aries_ckt 2023-06-28 10:35:55 +08:00
parent e6509e0183
commit 14387d9661
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pilot/openapi/api_v1/api_v1.py
View File

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