117 lines
3.5 KiB
Python
117 lines
3.5 KiB
Python
"""知识库路由（A4）
|
||
|
||
POST /kb/documents/upload 上传文档
|
||
GET /kb/documents 文档列表
|
||
GET /kb/documents/{id}/status 文档处理状态
|
||
POST /kb/documents/{id}/retry 重试失败文档
|
||
PATCH /kb/documents/{id} 更新元数据
|
||
DELETE /kb/documents/{id} 删除文档
|
||
GET /kb/datasources 数据源列表
|
||
POST /kb/datasources 新建数据源
|
||
POST /kb/datasources/{id}/sync 手动同步
|
||
GET /kb/datasources/{id}/logs 同步日志
|
||
"""
|
||
from flask import Blueprint, request, jsonify
|
||
from insurance.kb.service import KBService
|
||
from insurance.middleware.auth_middleware import jwt_required, admin_required
|
||
|
||
kb_bp = Blueprint("kb", __name__)
|
||
kb_service = KBService()
|
||
|
||
|
||
@kb_bp.route("/documents/upload", methods=["POST"])
|
||
@admin_required
|
||
def upload_documents():
|
||
"""多文件上传，触发异步处理流水线。"""
|
||
files = request.files.getlist("files")
|
||
insurance_type = request.form.get("险种", "")
|
||
company = request.form.get("保司", "")
|
||
|
||
if not files:
|
||
return jsonify({"code": 1001, "message": "请选择文件", "data": None}), 400
|
||
|
||
result = kb_service.upload_documents(files, insurance_type, company)
|
||
return jsonify(result)
|
||
|
||
|
||
@kb_bp.route("/documents", methods=["GET"])
|
||
@jwt_required
|
||
def list_documents():
|
||
"""文档列表，支持多维筛选。"""
|
||
params = {
|
||
"page": request.args.get("page", 1, type=int),
|
||
"page_size": request.args.get("page_size", 20, type=int),
|
||
"险种": request.args.get("险种", ""),
|
||
"保司": request.args.get("保司", ""),
|
||
"status": request.args.get("status", ""),
|
||
"keyword": request.args.get("keyword", ""),
|
||
}
|
||
result = kb_service.list_documents(params)
|
||
return jsonify(result)
|
||
|
||
|
||
@kb_bp.route("/documents/<doc_id>/status", methods=["GET"])
|
||
@jwt_required
|
||
def get_document_status(doc_id):
|
||
"""查询文档处理流水线状态。"""
|
||
result = kb_service.get_document_status(doc_id)
|
||
return jsonify(result)
|
||
|
||
|
||
@kb_bp.route("/documents/<doc_id>/retry", methods=["POST"])
|
||
@admin_required
|
||
def retry_document(doc_id):
|
||
"""重试处理失败的文档。"""
|
||
result = kb_service.retry_document(doc_id)
|
||
return jsonify(result)
|
||
|
||
|
||
@kb_bp.route("/documents/<doc_id>", methods=["PATCH"])
|
||
@admin_required
|
||
def update_document(doc_id):
|
||
"""更新文档元数据。"""
|
||
data = request.get_json()
|
||
result = kb_service.update_document(doc_id, data)
|
||
return jsonify(result)
|
||
|
||
|
||
@kb_bp.route("/documents/<doc_id>", methods=["DELETE"])
|
||
@admin_required
|
||
def delete_document(doc_id):
|
||
"""归档/软删除文档。"""
|
||
result = kb_service.delete_document(doc_id)
|
||
return jsonify(result)
|
||
|
||
|
||
@kb_bp.route("/datasources", methods=["GET"])
|
||
@admin_required
|
||
def list_datasources():
|
||
"""获取数据源列表。"""
|
||
result = kb_service.list_datasources()
|
||
return jsonify(result)
|
||
|
||
|
||
@kb_bp.route("/datasources", methods=["POST"])
|
||
@admin_required
|
||
def create_datasource():
|
||
"""新建数据源配置。"""
|
||
data = request.get_json()
|
||
result = kb_service.create_datasource(data)
|
||
return jsonify(result)
|
||
|
||
|
||
@kb_bp.route("/datasources/<ds_id>/sync", methods=["POST"])
|
||
@admin_required
|
||
def sync_datasource(ds_id):
|
||
"""手动触发数据源同步。"""
|
||
result = kb_service.sync_datasource(ds_id)
|
||
return jsonify(result)
|
||
|
||
|
||
@kb_bp.route("/datasources/<ds_id>/logs", methods=["GET"])
|
||
@admin_required
|
||
def get_sync_logs(ds_id):
|
||
"""获取同步历史日志。"""
|
||
result = kb_service.get_sync_logs(ds_id)
|
||
return jsonify(result)
|