xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-07-17 18:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: kb get api format

Browse Source
This commit is contained in:
Junyan Qin
2025-07-11 20:57:09 +08:00
parent 14c161b733
commit bd9331ce62
1 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/api/http/controller/groups/knowledge/base.py
+14 -10
View File
@@ -24,7 +24,9 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
elif quart.request.method == 'POST': elif quart.request.method == 'POST':
json_data = await quart.request.json json_data = await quart.request.json
knowledge_base_uuid = await self.ap.knowledge_base_service.create_knowledge_base( knowledge_base_uuid = await self.ap.knowledge_base_service.create_knowledge_base(
json_data.get('name'), json_data.get('description'), json_data.get('embedding_model_uuid') json_data.get('name'),
json_data.get('description'),
json_data.get('embedding_model_uuid'),
) )
return self.success(data={'uuid': knowledge_base_uuid}) return self.success(data={'uuid': knowledge_base_uuid})
@@ -35,20 +37,22 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
) )
async def handle_specific_knowledge_base(knowledge_base_uuid: str) -> str: async def handle_specific_knowledge_base(knowledge_base_uuid: str) -> str:
if quart.request.method == 'GET': if quart.request.method == 'GET':
knowledge_base = await self.ap.knowledge_base_service.get_knowledge_base_by_id(int(knowledge_base_uuid)) knowledge_base = await self.ap.knowledge_base_service.get_knowledge_base_by_id(knowledge_base_uuid)
if knowledge_base is None: if knowledge_base is None:
return self.http_status(404, -1, 'knowledge base not found') return self.http_status(404, -1, 'knowledge base not found')
return self.success( return self.success(
data={ data={
'name': knowledge_base.name, 'base': {
'description': knowledge_base.description, 'name': knowledge_base.name,
'uuid': knowledge_base.id, 'description': knowledge_base.description,
}, 'uuid': knowledge_base.id,
},
}
) )
elif quart.request.method == 'DELETE': elif quart.request.method == 'DELETE':
await self.ap.knowledge_base_service.delete_kb_by_id(int(knowledge_base_uuid)) await self.ap.knowledge_base_service.delete_kb_by_id(knowledge_base_uuid)
return self.success({}) return self.success({})
@self.route( @self.route(
@@ -57,7 +61,7 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
endpoint='get_knowledge_base_files', endpoint='get_knowledge_base_files',
) )
async def get_knowledge_base_files(knowledge_base_uuid: str) -> str: async def get_knowledge_base_files(knowledge_base_uuid: str) -> str:
files = await self.ap.knowledge_base_service.get_files_by_knowledge_base(int(knowledge_base_uuid)) files = await self.ap.knowledge_base_service.get_files_by_knowledge_base(knowledge_base_uuid)
return self.success( return self.success(
data={ data={
'files': [ 'files': [
@@ -77,7 +81,7 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
endpoint='delete_specific_file_in_kb', endpoint='delete_specific_file_in_kb',
) )
async def delete_specific_file_in_kb(file_id: str) -> str: async def delete_specific_file_in_kb(file_id: str) -> str:
await self.ap.knowledge_base_service.delete_data_by_file_id(int(file_id)) await self.ap.knowledge_base_service.delete_data_by_file_id(file_id)
return self.success({}) return self.success({})
@self.route( @self.route(
@@ -95,5 +99,5 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
return self.http_status(400, -1, 'File ID is required') return self.http_status(400, -1, 'File ID is required')
# 调用服务层方法将文件与知识库关联 # 调用服务层方法将文件与知识库关联
await self.ap.knowledge_base_service.relate_file_id_with_kb(int(knowledge_base_uuid), int(file_id)) await self.ap.knowledge_base_service.relate_file_id_with_kb(knowledge_base_uuid, file_id)
return self.success({}) return self.success({})