修复 画布,mcp服务,搜索,文档的接口

This commit is contained in:
2025-11-07 09:34:35 +08:00
parent c5f8fe06e7
commit 54532747d2
21 changed files with 1023 additions and 272 deletions

View File

@@ -14,8 +14,18 @@
# limitations under the License.
#
from flask import request
from flask_login import current_user, login_required
from typing import Optional
from fastapi import APIRouter, Depends, Query
from api.apps.models.auth_dependencies import get_current_user
from api.apps.models.search_models import (
CreateSearchRequest,
UpdateSearchRequest,
DeleteSearchRequest,
ListSearchAppsQuery,
ListSearchAppsBody,
)
from api import settings
from api.constants import DATASET_NAME_LIMIT
@@ -25,14 +35,23 @@ from api.db.services import duplicate_name
from api.db.services.search_service import SearchService
from api.db.services.user_service import TenantService, UserTenantService
from api.utils import get_uuid
from api.utils.api_utils import get_data_error_result, get_json_result, not_allowed_parameters, server_error_response, validate_request
from api.utils.api_utils import (
get_data_error_result,
get_json_result,
server_error_response,
)
# 创建路由器
router = APIRouter()
@manager.route("/create", methods=["post"]) # noqa: F821
@login_required
@validate_request("name")
def create():
req = request.get_json()
@router.post('/create')
async def create(
request: CreateSearchRequest,
current_user = Depends(get_current_user)
):
"""创建搜索应用"""
req = request.model_dump(exclude_unset=True)
search_name = req["name"]
description = req.get("description", "")
if not isinstance(search_name, str):
@@ -62,12 +81,13 @@ def create():
return server_error_response(e)
@manager.route("/update", methods=["post"]) # noqa: F821
@login_required
@validate_request("search_id", "name", "search_config", "tenant_id")
@not_allowed_parameters("id", "created_by", "create_time", "update_time", "create_date", "update_date", "created_by")
def update():
req = request.get_json()
@router.post('/update')
async def update(
request: UpdateSearchRequest,
current_user = Depends(get_current_user)
):
"""更新搜索应用"""
req = request.model_dump(exclude_unset=True)
if not isinstance(req["name"], str):
return get_data_error_result(message="Search name must be string.")
if req["name"].strip() == "":
@@ -84,6 +104,12 @@ def update():
if not SearchService.accessible4deletion(search_id, current_user.id):
return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
# 验证不允许的参数
not_allowed = ["id", "created_by", "create_time", "update_time", "create_date", "update_date"]
for key in not_allowed:
if key in req:
del req[key]
try:
search_app = SearchService.query(tenant_id=tenant_id, id=search_id)[0]
if not search_app:
@@ -119,10 +145,12 @@ def update():
return server_error_response(e)
@manager.route("/detail", methods=["GET"]) # noqa: F821
@login_required
def detail():
search_id = request.args["search_id"]
@router.get('/detail')
async def detail(
search_id: str = Query(..., description="搜索应用ID"),
current_user = Depends(get_current_user)
):
"""获取搜索应用详情"""
try:
tenants = UserTenantService.query(user_id=current_user.id)
for tenant in tenants:
@@ -139,20 +167,23 @@ def detail():
return server_error_response(e)
@manager.route("/list", methods=["POST"]) # noqa: F821
@login_required
def list_search_app():
keywords = request.args.get("keywords", "")
page_number = int(request.args.get("page", 0))
items_per_page = int(request.args.get("page_size", 0))
orderby = request.args.get("orderby", "create_time")
if request.args.get("desc", "true").lower() == "false":
desc = False
else:
desc = True
@router.post('/list')
async def list_search_app(
query: ListSearchAppsQuery = Depends(),
body: Optional[ListSearchAppsBody] = None,
current_user = Depends(get_current_user)
):
"""列出搜索应用"""
if body is None:
body = ListSearchAppsBody()
keywords = query.keywords or ""
page_number = int(query.page or 0)
items_per_page = int(query.page_size or 0)
orderby = query.orderby or "create_time"
desc = query.desc.lower() == "true" if query.desc else True
req = request.get_json()
owner_ids = req.get("owner_ids", [])
owner_ids = body.owner_ids or [] if body else []
try:
if not owner_ids:
# tenants = TenantService.get_joined_tenants_by_user_id(current_user.id)
@@ -171,12 +202,13 @@ def list_search_app():
return server_error_response(e)
@manager.route("/rm", methods=["post"]) # noqa: F821
@login_required
@validate_request("search_id")
def rm():
req = request.get_json()
search_id = req["search_id"]
@router.post('/rm')
async def rm(
request: DeleteSearchRequest,
current_user = Depends(get_current_user)
):
"""删除搜索应用"""
search_id = request.search_id
if not SearchService.accessible4deletion(search_id, current_user.id):
return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)