27 lines
817 B
Python
27 lines
817 B
Python
"""知识库检索路由（独立 Blueprint）
|
||
|
||
POST /retrieval/search 向量检索
|
||
"""
|
||
from flask import Blueprint, request, jsonify
|
||
from insurance.kb.service import KBService
|
||
from insurance.middleware.auth_middleware import jwt_required
|
||
|
||
retrieval_bp = Blueprint("retrieval", __name__)
|
||
kb_service = KBService()
|
||
|
||
|
||
@retrieval_bp.route("/search", methods=["POST"])
|
||
@jwt_required
|
||
def retrieval_search():
|
||
"""向量检索接口：独立于对话，直接检索知识库。"""
|
||
data = request.get_json()
|
||
query = data.get("query", "")
|
||
filters = data.get("filters", {})
|
||
top_k = data.get("top_k", 5)
|
||
|
||
if not query:
|
||
return jsonify({"code": 1001, "message": "请输入搜索内容", "data": None}), 400
|
||
|
||
result = kb_service.retrieval_search(query, filters, top_k)
|
||
return jsonify(result)
|