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

feat: modify the rag.py

Browse Source
This commit is contained in:
WangCham
2025-07-09 22:09:46 +08:00
parent cd25340826
commit ac03a2dceb
4 changed files with 338 additions and 229 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
pkg/api/http/controller/groups/knowledge_base.py
View File

@@ -1,6 +1,6 @@
import quart
from .. import group
import os # 导入 os 用于文件操作
@group.group_class('knowledge_base', '/api/v1/knowledge/bases')
class KnowledgeBaseRouterGroup(group.RouterGroup):
@@ -9,8 +9,8 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
return quart.jsonify({'code': code, 'data': data or {}, 'msg': msg})
async def initialize(self) -> None:
@self.route('', methods=['POST', 'GET'])
async def _() -> str:
@self.route('', methods=['POST', 'GET'], endpoint='handle_knowledge_bases')
async def handle_knowledge_bases() -> str:
if quart.request.method == 'GET':
knowledge_bases = await self.ap.knowledge_base_service.get_all_knowledge_bases()
bases_list = [
@@ -23,17 +23,17 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
]
return self.success(code=0, data={'bases': bases_list}, msg='ok')
# POST: create a new knowledge base
json_data = await quart.request.json
knowledge_base_uuid = await self.ap.knowledge_base_service.create_knowledge_base(
json_data.get('name'), json_data.get('description')
)
_ = knowledge_base_uuid
return self.success(code=0, data={}, msg='ok')
return self.success(code=0, data={'uuid': knowledge_base_uuid}, msg='ok')
@self.route('/<knowledge_base_uuid>', methods=['GET', 'DELETE'])
async def _(knowledge_base_uuid: str) -> str:
@self.route('/<knowledge_base_uuid>', methods=['GET', 'DELETE'], endpoint='handle_specific_knowledge_base')
async def handle_specific_knowledge_base(knowledge_base_uuid: str) -> str:
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(int(knowledge_base_uuid))
if knowledge_base is None:
return self.http_status(404, -1, 'knowledge base not found')
@@ -48,28 +48,42 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
msg='ok',
)
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(int(knowledge_base_uuid))
return self.success(code=0, msg='ok')
@self.route('/<knowledge_base_uuid>/files', methods=['GET'])
async def _(knowledge_base_uuid: str) -> str:
if quart.request.method == 'GET':
files = await self.ap.knowledge_base_service.get_files_by_knowledge_base(knowledge_base_uuid)
return self.success(
code=0,
data=[
{
'id': file.id,
'file_name': file.file_name,
'status': file.status,
}
for file in files
],
msg='ok',
)
# delete specific file in knowledge base
@self.route('/<knowledge_base_uuid>/files/<file_id>', methods=['DELETE'])
async def _(knowledge_base_uuid: str, file_id: str) -> str:
await self.ap.knowledge_base_service.delete_data_by_file_id(file_id)
@self.route('/<knowledge_base_uuid>/files', methods=['GET'], endpoint='get_knowledge_base_files')
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))
return self.success(
code=0,
data=[
{
'id': file.id,
'file_name': file.file_name,
'status': file.status,
}
for file in files
],
msg='ok',
)
@self.route('/<knowledge_base_uuid>/files/<file_id>', methods=['DELETE'], endpoint='delete_specific_file_in_kb')
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))
return self.success(code=0, msg='ok')
@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(int(knowledge_base_uuid), int(file_id))
return self.success(code=0, data={}, msg='ok')