github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-01 09:06:55 +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
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.configs.config import Config
from pilot.openapi.knowledge.knowledge_service import KnowledgeService from pilot.openapi.knowledge.knowledge_service import KnowledgeService
from pilot.openapi.knowledge.request.knowledge_request import KnowledgeSpaceRequest from pilot.openapi.knowledge.request.knowledge_request import KnowledgeSpaceRequest
from pilot.scene.base_chat import BaseChat from pilot.scene.base_chat import BaseChat
from pilot.scene.base import ChatScene from pilot.scene.base import ChatScene
from pilot.scene.chat_factory import ChatFactory from pilot.scene.chat_factory import ChatFactory
@@ -127,8 +128,13 @@ def plugins_select_info():
def knowledge_list(): def knowledge_list():
"""return knowledge space list"""
params: dict = {}
request = KnowledgeSpaceRequest() 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]) @router.post("/v1/chat/mode/params/list", response_model=Result[dict])