mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 12:05:54 +00:00
100 lines
4.1 KiB
Python
100 lines
4.1 KiB
Python
import quart
|
|
from ... import group
|
|
|
|
|
|
@group.group_class('knowledge_base', '/api/v1/knowledge/bases')
|
|
class KnowledgeBaseRouterGroup(group.RouterGroup):
|
|
async def initialize(self) -> None:
|
|
@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 = [
|
|
{
|
|
'uuid': kb.id,
|
|
'name': kb.name,
|
|
'description': kb.description,
|
|
'embedding_model_uuid': kb.embedding_model_uuid,
|
|
'top_k': kb.top_k,
|
|
}
|
|
for kb in knowledge_bases
|
|
]
|
|
return self.success(data={'bases': bases_list})
|
|
|
|
elif quart.request.method == 'POST':
|
|
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'), json_data.get('embedding_model_uuid')
|
|
)
|
|
return self.success(data={'uuid': knowledge_base_uuid})
|
|
|
|
@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(int(knowledge_base_uuid))
|
|
|
|
if knowledge_base is None:
|
|
return self.http_status(404, -1, 'knowledge base not found')
|
|
|
|
return self.success(
|
|
data={
|
|
'name': knowledge_base.name,
|
|
'description': knowledge_base.description,
|
|
'uuid': knowledge_base.id,
|
|
},
|
|
)
|
|
elif quart.request.method == 'DELETE':
|
|
await self.ap.knowledge_base_service.delete_kb_by_id(int(knowledge_base_uuid))
|
|
return self.success({})
|
|
|
|
@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(
|
|
data={
|
|
'files': [
|
|
{
|
|
'id': file.id,
|
|
'file_name': file.file_name,
|
|
'status': file.status,
|
|
}
|
|
for file in files
|
|
],
|
|
}
|
|
)
|
|
|
|
@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({})
|
|
|
|
@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({})
|