xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-08-12 13:50:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: api bug

Browse Source
This commit is contained in:
Junyan Qin
2025-07-11 21:30:47 +08:00
parent 2ed3b68790
commit a79a22a74d
1 changed files with 25 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/api/http/controller/groups/knowledge/base.py
+25 -32
View File
@@ -59,23 +59,34 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
@self.route( @self.route(
'/<knowledge_base_uuid>/files', '/<knowledge_base_uuid>/files',
methods=['GET'], methods=['GET', 'POST'],
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(knowledge_base_uuid) if quart.request.method == 'GET':
return self.success( files = await self.ap.knowledge_base_service.get_files_by_knowledge_base(knowledge_base_uuid)
data={ return self.success(
'files': [ data={
{ 'files': [
'id': file.id, {
'file_name': file.file_name, 'id': file.id,
'status': file.status, 'file_name': file.file_name,
} 'status': file.status,
for file in files }
], for file in files
} ],
) }
)
elif quart.request.method == 'POST':
json_data = await quart.request.json
file_id = json_data.get('file_id')
if not file_id:
return self.http_status(400, -1, 'File ID is required')
# 调用服务层方法将文件与知识库关联
await self.ap.knowledge_base_service.relate_file_id_with_kb(knowledge_base_uuid, file_id)
return self.success({})
@self.route( @self.route(
'/<knowledge_base_uuid>/files/<file_id>', '/<knowledge_base_uuid>/files/<file_id>',
@@ -85,21 +96,3 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
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(file_id) await self.ap.knowledge_base_service.delete_data_by_file_id(file_id)
return self.success({}) return self.success({})
@self.route(
'/<knowledge_base_uuid>/files',
methods=['POST'],
endpoint='relate_file_with_kb',
)
async def relate_file_id_with_kb(knowledge_base_uuid: str, file_id: str) -> str:
if 'file' not in quart.request.files:
return self.http_status(400, -1, 'No file part in the request')
json_data = await quart.request.json
file_id = json_data.get('file_id')
if not file_id:
return self.http_status(400, -1, 'File ID is required')
# 调用服务层方法将文件与知识库关联
await self.ap.knowledge_base_service.relate_file_id_with_kb(knowledge_base_uuid, file_id)
return self.success({})