127 lines
4.2 KiB
Python
127 lines
4.2 KiB
Python
from fastapi import APIRouter, HTTPException, Query, Depends
|
||
from typing import Dict, List, Optional
|
||
from pydantic import BaseModel
|
||
from services.role_service import role_service
|
||
from services.device_service import device_service
|
||
from services.history_service import history_service
|
||
from api.auth import api_auth
|
||
|
||
router = APIRouter(prefix="/api/roles")
|
||
|
||
class RoleResponse(BaseModel):
|
||
role_key: str
|
||
name: str
|
||
description: Optional[str] = None
|
||
languages: Optional[List[str]] = None
|
||
|
||
class RoleSummaryResponse(BaseModel):
|
||
role_key: str
|
||
name: str
|
||
|
||
class DeviceRoleUpdate(BaseModel):
|
||
role_key: str
|
||
language: Optional[str] = None
|
||
|
||
class ConversationMessage(BaseModel):
|
||
user: str
|
||
assistant: str
|
||
timestamp: Optional[float] = None
|
||
|
||
class PaginatedRoleResponse(BaseModel):
|
||
total: int
|
||
page: int
|
||
page_size: int
|
||
data: List[RoleResponse]
|
||
|
||
class PaginatedHistoryResponse(BaseModel):
|
||
total: int
|
||
page: int
|
||
page_size: int
|
||
data: List[ConversationMessage]
|
||
role_key: str
|
||
role_name: str
|
||
|
||
@router.get("/list", response_model=PaginatedRoleResponse)
|
||
async def list_roles(
|
||
authenticated_device_id: str = Depends(api_auth),
|
||
page: int = Query(1, ge=1),
|
||
page_size: int = Query(20, ge=1, le=100),
|
||
search: Optional[str] = None
|
||
):
|
||
"""获取角色分页列表，支持搜索"""
|
||
return await role_service.get_roles_paginated(page, page_size, search)
|
||
|
||
@router.get("/summaries", response_model=List[RoleSummaryResponse])
|
||
async def get_role_summaries(
|
||
authenticated_device_id: str = Depends(api_auth)
|
||
):
|
||
"""获取角色简要信息列表（仅包含key和name）"""
|
||
return await role_service.get_role_summaries()
|
||
|
||
@router.get("/device/{device_id}")
|
||
async def get_device_role(
|
||
device_id: str,
|
||
authenticated_device_id: str = Depends(api_auth)
|
||
):
|
||
"""获取设备当前使用的角色"""
|
||
if device_id != authenticated_device_id:
|
||
raise HTTPException(status_code=403, detail="无权访问其他设备的配置")
|
||
|
||
return await device_service.get_device_role(device_id)
|
||
|
||
@router.put("/device/{device_id}")
|
||
async def update_device_role(
|
||
device_id: str,
|
||
role_update: DeviceRoleUpdate,
|
||
authenticated_device_id: str = Depends(api_auth)
|
||
):
|
||
"""更新设备的角色配置"""
|
||
if device_id != authenticated_device_id:
|
||
raise HTTPException(status_code=403, detail="无权修改其他设备的配置")
|
||
|
||
try:
|
||
return await device_service.update_device_role(
|
||
device_id,
|
||
role_update.role_key,
|
||
role_update.language
|
||
)
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=404, detail=str(e))
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=f"更新角色时出错: {str(e)}")
|
||
|
||
@router.get("/history/{device_id}", response_model=PaginatedHistoryResponse)
|
||
async def get_conversation_history(
|
||
device_id: str,
|
||
authenticated_device_id: str = Depends(api_auth),
|
||
role_key: Optional[str] = None,
|
||
page: int = Query(1, ge=1),
|
||
page_size: int = Query(20, ge=1, le=100),
|
||
since: Optional[float] = None,
|
||
until: Optional[float] = None
|
||
):
|
||
"""获取设备的对话历史记录，支持分页和时间范围筛选"""
|
||
if device_id != authenticated_device_id:
|
||
raise HTTPException(status_code=403, detail="无权访问其他设备的对话历史")
|
||
|
||
try:
|
||
return await history_service.get_history_paginated(
|
||
device_id, role_key, page, page_size, since, until
|
||
)
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=f"获取对话历史失败: {str(e)}")
|
||
|
||
@router.get("/history-summary/{device_id}")
|
||
async def get_conversation_history_summary(
|
||
device_id: str,
|
||
authenticated_device_id: str = Depends(api_auth),
|
||
days: int = Query(7, ge=1, le=30)
|
||
):
|
||
"""获取设备的对话历史摘要信息，包括每个角色的最后交互时间和消息数量"""
|
||
if device_id != authenticated_device_id:
|
||
raise HTTPException(status_code=403, detail="无权访问其他设备的对话历史摘要")
|
||
|
||
try:
|
||
return await history_service.get_history_summary(device_id, days)
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=f"获取对话历史摘要失败: {str(e)}") |