github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-10-11 20:03:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:add knowledge space argument

Browse Source
This commit is contained in:
aries_ckt
2023-08-07 16:36:41 +08:00
parent cc57ed22ca
commit aa3c3205a4
9 changed files with 146 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
pilot/server/knowledge/api.py
View File

@@ -20,6 +20,7 @@ from pilot.server.knowledge.request.request import (
DocumentSyncRequest,
ChunkQueryRequest,
DocumentQueryRequest,
SpaceArgumentRequest,
)
from pilot.server.knowledge.request.request import KnowledgeSpaceRequest
@@ -54,13 +55,33 @@ def space_list(request: KnowledgeSpaceRequest):
@router.post("/knowledge/space/delete")
def space_delete(request: KnowledgeSpaceRequest):
print(f"/space/list params:")
print(f"/space/delete params:")
try:
return Result.succ(knowledge_space_service.delete_space(request.name))
except Exception as e:
return Result.faild(code="E000X", msg=f"space list error {e}")
@router.post("/knowledge/{space_name}/arguments")
def arguments(space_name: str):
print(f"/knowledge/space/arguments params:")
try:
return Result.succ(knowledge_space_service.arguments(space_name))
except Exception as e:
return Result.faild(code="E000X", msg=f"space list error {e}")
@router.post("/knowledge/{space_name}/argument/save")
def arguments_save(space_name: str, argument_request: SpaceArgumentRequest):
print(f"/knowledge/space/argument/save params:")
try:
return Result.succ(
knowledge_space_service.argument_save(space_name, argument_request)
)
except Exception as e:
return Result.faild(code="E000X", msg=f"space list error {e}")
@router.post("/knowledge/{space_name}/document/add")
def document_add(space_name: str, request: KnowledgeDocumentRequest):
print(f"/document/add params: {space_name}, {request}")