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

perf: format

Browse Source
This commit is contained in:
Junyan Qin
2025-07-05 21:56:17 +08:00
parent 0e5c9e19e1
commit 39c062f73e
2 changed files with 30 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/api/http/controller/groups/knowledge_base.py
+29 -36
View File
@@ -1,49 +1,36 @@
import quart import quart
from .. import group from .. import group
@group.group_class('knowledge_base', '/api/v1/knowledge/bases') @group.group_class('knowledge_base', '/api/v1/knowledge/bases')
class KnowledgeBaseRouterGroup(group.RouterGroup): class KnowledgeBaseRouterGroup(group.RouterGroup):
# 定义成功方法 # 定义成功方法
def success(self, code=0, data=None, msg: str = 'ok') -> quart.Response: def success(self, code=0, data=None, msg: str = 'ok') -> quart.Response:
return quart.jsonify({ return quart.jsonify({'code': code, 'data': data or {}, 'msg': msg})
"code": code,
"data": data or {},
"msg": msg
})
async def initialize(self) -> None: async def initialize(self) -> None:
@self.route('', methods=['POST', 'GET']) @self.route('', methods=['POST', 'GET'])
async def _() -> str: async def _() -> str:
if quart.request.method == 'GET': if quart.request.method == 'GET':
knowledge_bases = await self.ap.knowledge_base_service.get_all_knowledge_bases() knowledge_bases = await self.ap.knowledge_base_service.get_all_knowledge_bases()
bases_list = [ bases_list = [
{ {
"uuid": kb.id, 'uuid': kb.id,
"name": kb.name, 'name': kb.name,
"description": kb.description, 'description': kb.description,
} for kb in knowledge_bases }
for kb in knowledge_bases
] ]
return self.success(code=0, return self.success(code=0, data={'bases': bases_list}, msg='ok')
data={'bases': bases_list},
msg='ok')
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('name'), json_data.get('description')
json_data.get('description')
) )
return self.success(code=0, _ = knowledge_base_uuid
data={}, return self.success(code=0, data={}, msg='ok')
msg='ok')
@self.route('/<knowledge_base_uuid>', methods=['GET', 'DELETE'])
@self.route('/<knowledge_base_uuid>', methods=['GET','DELETE'])
async def _(knowledge_base_uuid: str) -> str: async def _(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(knowledge_base_uuid) knowledge_base = await self.ap.knowledge_base_service.get_knowledge_base_by_id(knowledge_base_uuid)
@@ -54,11 +41,11 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
return self.success( return self.success(
code=0, code=0,
data={ data={
"name": knowledge_base.name, 'name': knowledge_base.name,
"description": knowledge_base.description, 'description': knowledge_base.description,
"uuid": knowledge_base.id 'uuid': knowledge_base.id,
}, },
msg='ok' msg='ok',
) )
elif quart.request.method == 'DELETE': elif quart.request.method == 'DELETE':
await self.ap.knowledge_base_service.delete_kb_by_id(knowledge_base_uuid) await self.ap.knowledge_base_service.delete_kb_by_id(knowledge_base_uuid)
@@ -68,15 +55,21 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
async def _(knowledge_base_uuid: str) -> str: async def _(knowledge_base_uuid: str) -> str:
if quart.request.method == 'GET': if quart.request.method == 'GET':
files = await self.ap.knowledge_base_service.get_files_by_knowledge_base(knowledge_base_uuid) files = await self.ap.knowledge_base_service.get_files_by_knowledge_base(knowledge_base_uuid)
return self.success(code=0,data=[{ return self.success(
"id": file.id, code=0,
"file_name": file.file_name, data=[
"status": file.status {
} for file in files],msg='ok') 'id': file.id,
'file_name': file.file_name,
'status': file.status,
}
for file in files
],
msg='ok',
)
# delete specific file in knowledge base # delete specific file in knowledge base
@self.route('/<knowledge_base_uuid>/files/<file_id>', methods=['DELETE']) @self.route('/<knowledge_base_uuid>/files/<file_id>', methods=['DELETE'])
async def _(knowledge_base_uuid: str, file_id: str) -> str: async def _(knowledge_base_uuid: str, file_id: str) -> str:
await self.ap.knowledge_base_service.delete_data_by_file_id(file_id) await self.ap.knowledge_base_service.delete_data_by_file_id(file_id)
return self.success(code=0, msg='ok') return self.success(code=0, msg='ok')
pkg/api/http/controller/groups/pipelines/pipelines.py
+1 -1
View File
@@ -2,7 +2,7 @@ from __future__ import annotations
import quart import quart
from .. import group from ... import group
@group.group_class('pipelines', '/api/v1/pipelines') @group.group_class('pipelines', '/api/v1/pipelines')