huoyan-enterprise/backend/api/user_setting.py

55 lines
2.2 KiB
Python

"""
用户设置 API 路由模块
定义用户设置相关的 API 路由,包括联网搜索设置、深度思考设置等。
"""
from fastapi import APIRouter, Depends
from core.dependencies import get_current_user
from models.user import User
from models.chat import (
SearchSettingResponse,
UpdateSearchSettingRequest,
ReasonerSettingResponse,
UpdateReasonerSettingRequest,
)
from services.user_setting_service import UserSettingService
# 创建路由实例
user_setting_router = APIRouter(prefix="/api/user", tags=["用户设置"])
@user_setting_router.get("/search-setting", summary="获取用户联网搜索设置", response_model=SearchSettingResponse)
async def get_search_setting(current_user: User = Depends(get_current_user)):
"""获取当前用户的联网搜索设置"""
is_search = await UserSettingService.get_search_setting(current_user.id)
return SearchSettingResponse(is_search=is_search)
@user_setting_router.put("/search-setting", summary="更新用户联网搜索设置", response_model=SearchSettingResponse)
async def update_search_setting(
request: UpdateSearchSettingRequest,
current_user: User = Depends(get_current_user)
):
"""更新当前用户的联网搜索设置"""
is_search = await UserSettingService.update_search_setting(current_user.id, request.is_search)
return SearchSettingResponse(is_search=is_search)
@user_setting_router.get("/reasoner-setting", summary="获取用户深度思考设置", response_model=ReasonerSettingResponse)
async def get_reasoner_setting(current_user: User = Depends(get_current_user)):
"""获取当前用户的深度思考设置"""
is_reasoner = await UserSettingService.get_reasoner_setting(current_user.id)
return ReasonerSettingResponse(is_reasoner=is_reasoner)
@user_setting_router.put("/reasoner-setting", summary="更新用户深度思考设置", response_model=ReasonerSettingResponse)
async def update_reasoner_setting(
request: UpdateReasonerSettingRequest,
current_user: User = Depends(get_current_user)
):
"""更新当前用户的深度思考设置"""
is_reasoner = await UserSettingService.update_reasoner_setting(current_user.id, request.is_reasoner)
return ReasonerSettingResponse(is_reasoner=is_reasoner)