1015 lines
34 KiB
Python
1015 lines
34 KiB
Python
|
# =============================================================================
|
|||
|
|
# 企微IT智能服务台 — 管理后台 API 路由
|
|||
|
|
# =============================================================================
|
|||
|
|
# 说明：管理后台的全部路由端点，统一 /admin/ 前缀
|
|||
|
|
# 包含 7 组 API：
|
|||
|
|
# 1. 运营总览仪表盘
|
|||
|
|
# 2. 功能开关/参数管理
|
|||
|
|
# 3. 坐席人员管理
|
|||
|
|
# 4. 外部系统集成配置
|
|||
|
|
# 5. 快速回复管理（审核）
|
|||
|
|
# 6. 消息分配模式
|
|||
|
|
# 7. 会话监控
|
|||
|
|
# 8. 全局搜索
|
|||
|
|
# 所有接口需要 admin 角色权限
|
|||
|
|
# =============================================================================
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
from typing import Optional
|
|||
|
|
from uuid import UUID
|
|||
|
|
|
|||
|
|
from fastapi import APIRouter, Depends, Query
|
|||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|||
|
|
|
|||
|
|
from app.api.agents import get_current_agent
|
|||
|
|
from app.database import get_db
|
|||
|
|
from app.models.agent import Agent
|
|||
|
|
from app.schemas.admin import (
|
|||
|
|
AgentCreateRequest,
|
|||
|
|
AgentUpdateRequest,
|
|||
|
|
AssignmentModeUpdateRequest,
|
|||
|
|
ConfigUpdateRequest,
|
|||
|
|
IntegrationUpdateRequest,
|
|||
|
|
QuickReplyReviewRequest,
|
|||
|
|
)
|
|||
|
|
from app.services import admin_service
|
|||
|
|
from app.utils.response import AppException, success_response
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
# 创建路由器
|
|||
|
|
router = APIRouter(prefix="/admin")
|
|||
|
|
|
|||
|
|
|
|||
|
|
# --------------------------------------------------------------------------
|
|||
|
|
# 管理后台权限校验依赖
|
|||
|
|
# --------------------------------------------------------------------------
|
|||
|
|
async def require_admin(
|
|||
|
|
agent: Agent = Depends(get_current_agent),
|
|||
|
|
) -> Agent:
|
|||
|
|
"""管理后台权限校验：仅 role='admin' 可访问。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
agent: 当前坐席（通过认证依赖注入）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Agent: 具有管理权限的坐席对象
|
|||
|
|
|
|||
|
|
Raises:
|
|||
|
|
AppException: 非管理员角色（错误码 1004）
|
|||
|
|
"""
|
|||
|
|
if agent.role != "admin":
|
|||
|
|
raise AppException(1004, "无管理权限")
|
|||
|
|
return agent
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 1. 运营总览仪表盘
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/dashboard/overview ----------
|
|||
|
|
@router.get("/dashboard/overview")
|
|||
|
|
async def get_dashboard_overview(
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取仪表盘统计数据。
|
|||
|
|
|
|||
|
|
返回在线坐席数、今日会话数、待审核数、集成健康状态等。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
admin: 管理员（权限校验）
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含仪表盘数据
|
|||
|
|
"""
|
|||
|
|
overview = await admin_service.get_dashboard_overview(db)
|
|||
|
|
return success_response(data=overview.model_dump())
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 2. 功能开关/参数管理
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/configs ----------
|
|||
|
|
@router.get("/configs")
|
|||
|
|
async def get_configs(
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取全部配置项（按功能分组）。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含配置分组列表
|
|||
|
|
"""
|
|||
|
|
groups = await admin_service.get_config_groups(db)
|
|||
|
|
return success_response(data={
|
|||
|
|
"groups": [g.model_dump() for g in groups]
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- PUT /api/admin/configs/{key} ----------
|
|||
|
|
@router.put("/configs/{key}")
|
|||
|
|
async def update_config(
|
|||
|
|
key: str,
|
|||
|
|
body: ConfigUpdateRequest,
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""更新单个配置项（同时记录变更日志）。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
key: 配置键
|
|||
|
|
body: 更新请求体
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含变更信息
|
|||
|
|
"""
|
|||
|
|
result = await admin_service.update_config(db, key, body.value, admin.id)
|
|||
|
|
return success_response(data=result)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/configs/{key}/history ----------
|
|||
|
|
@router.get("/configs/{key}/history")
|
|||
|
|
async def get_config_history(
|
|||
|
|
key: str,
|
|||
|
|
limit: int = Query(20, ge=1, le=100, description="返回条数上限"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取指定配置项的变更历史。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
key: 配置键
|
|||
|
|
limit: 返回条数上限
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含变更历史列表
|
|||
|
|
"""
|
|||
|
|
items = await admin_service.get_config_history(db, key, limit)
|
|||
|
|
return success_response(data={
|
|||
|
|
"items": [item.model_dump() for item in items]
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 3. 坐席人员管理
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/agents ----------
|
|||
|
|
@router.get("/agents")
|
|||
|
|
async def list_admin_agents(
|
|||
|
|
status: Optional[str] = Query(None, description="按状态筛选: online/offline/busy"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取坐席列表（管理视图，含角色/技能标签）。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
status: 按状态筛选（可选）
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含坐席列表
|
|||
|
|
"""
|
|||
|
|
items = await admin_service.list_admin_agents(db, status)
|
|||
|
|
return success_response(data={
|
|||
|
|
"items": [item.model_dump() for item in items]
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- POST /api/admin/agents ----------
|
|||
|
|
@router.post("/agents")
|
|||
|
|
async def create_agent(
|
|||
|
|
body: AgentCreateRequest,
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""添加坐席。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
body: 创建请求体
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含创建的坐席信息
|
|||
|
|
"""
|
|||
|
|
agent = await admin_service.create_agent(
|
|||
|
|
db,
|
|||
|
|
user_id=body.user_id,
|
|||
|
|
name=body.name,
|
|||
|
|
role=body.role,
|
|||
|
|
skill_tags=body.skill_tags,
|
|||
|
|
max_load=body.max_load,
|
|||
|
|
)
|
|||
|
|
return success_response(data=agent.model_dump())
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- PUT /api/admin/agents/{id} ----------
|
|||
|
|
@router.put("/agents/{agent_id}")
|
|||
|
|
async def update_agent(
|
|||
|
|
agent_id: str,
|
|||
|
|
body: AgentUpdateRequest,
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""编辑坐席（角色/技能标签/负载上限）。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
agent_id: 坐席ID
|
|||
|
|
body: 更新请求体
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含更新后的坐席信息
|
|||
|
|
"""
|
|||
|
|
result = await admin_service.update_agent(
|
|||
|
|
db,
|
|||
|
|
agent_id=agent_id,
|
|||
|
|
role=body.role,
|
|||
|
|
skill_tags=body.skill_tags,
|
|||
|
|
max_load=body.max_load,
|
|||
|
|
)
|
|||
|
|
return success_response(data=result)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- DELETE /api/admin/agents/{id} ----------
|
|||
|
|
@router.delete("/agents/{agent_id}")
|
|||
|
|
async def delete_agent(
|
|||
|
|
agent_id: str,
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""移除坐席。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
agent_id: 坐席ID
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式
|
|||
|
|
"""
|
|||
|
|
await admin_service.delete_agent(db, agent_id)
|
|||
|
|
return success_response(data=None, message="坐席已移除")
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- POST /api/admin/agents/{id}/otp-unbind ----------
|
|||
|
|
@router.post("/agents/{agent_id}/otp-unbind")
|
|||
|
|
async def admin_unbind_agent_otp(
|
|||
|
|
agent_id: str,
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""强制解绑坐席的OTP。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
agent_id: 坐席ID
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式
|
|||
|
|
"""
|
|||
|
|
from app.models.agent import Agent as AgentModel
|
|||
|
|
from sqlalchemy import select
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
stmt = select(AgentModel).where(AgentModel.id == agent_id)
|
|||
|
|
result = await db.execute(stmt)
|
|||
|
|
agent = result.scalars().first()
|
|||
|
|
|
|||
|
|
if not agent:
|
|||
|
|
raise AppException(1001, "坐席不存在")
|
|||
|
|
|
|||
|
|
agent.otp_secret = None
|
|||
|
|
agent.otp_enabled = 0
|
|||
|
|
agent.updated_at = datetime.now()
|
|||
|
|
db.add(agent)
|
|||
|
|
await db.flush()
|
|||
|
|
|
|||
|
|
logger.info(f"管理员强制解绑OTP: agent={agent.user_id}")
|
|||
|
|
return success_response(data={"message": "OTP已解绑"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 4. 外部系统集成配置
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/integrations ----------
|
|||
|
|
@router.get("/integrations")
|
|||
|
|
async def get_integrations(
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取集成系统列表及配置状态。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含集成系统列表
|
|||
|
|
"""
|
|||
|
|
items = await admin_service.get_integrations(db)
|
|||
|
|
return success_response(data={
|
|||
|
|
"items": [item.model_dump() for item in items]
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- PUT /api/admin/integrations/{id} ----------
|
|||
|
|
@router.put("/integrations/{integration_id}")
|
|||
|
|
async def update_integration(
|
|||
|
|
integration_id: str,
|
|||
|
|
body: IntegrationUpdateRequest,
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""更新集成配置（支持 url_key 和 access_key 两种模式）。
|
|||
|
|
|
|||
|
|
- url_key 模式（Dify / RAGFlow）：传入 api_url + api_key
|
|||
|
|
- access_key 模式（火绒安全）：传入 access_key_id + access_key_secret + base_url
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
integration_id: 集成系统ID（如 dify/ragflow/huorong）
|
|||
|
|
body: 更新请求体
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含更新后的集成信息
|
|||
|
|
"""
|
|||
|
|
result = await admin_service.update_integration(
|
|||
|
|
db,
|
|||
|
|
integration_id=integration_id,
|
|||
|
|
# url_key 模式字段
|
|||
|
|
api_url=body.api_url or "",
|
|||
|
|
api_key=body.api_key or "",
|
|||
|
|
# access_key 模式字段（火绒）
|
|||
|
|
access_key_id=body.access_key_id or "",
|
|||
|
|
access_key_secret=body.access_key_secret or "",
|
|||
|
|
# account_password 模式字段（联软）
|
|||
|
|
api_account=body.api_account or "",
|
|||
|
|
api_password=body.api_password or "",
|
|||
|
|
validate_key=body.validate_key or "",
|
|||
|
|
base_url=body.base_url or "",
|
|||
|
|
agent_id=admin.id,
|
|||
|
|
)
|
|||
|
|
return success_response(data=result.model_dump())
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 5. 快速回复管理
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/quick-replies/pending ----------
|
|||
|
|
@router.get("/quick-replies/pending")
|
|||
|
|
async def list_pending_quick_replies(
|
|||
|
|
category: Optional[str] = Query(None, description="按分类筛选"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取待审核快速回复模板列表。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
category: 按分类筛选（可选）
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含待审核模板列表
|
|||
|
|
"""
|
|||
|
|
items = await admin_service.list_pending_quick_replies(db, category)
|
|||
|
|
return success_response(data={
|
|||
|
|
"items": [item.model_dump() for item in items]
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- PUT /api/admin/quick-replies/{id}/review ----------
|
|||
|
|
@router.put("/quick-replies/{template_id}/review")
|
|||
|
|
async def review_quick_reply(
|
|||
|
|
template_id: str,
|
|||
|
|
body: QuickReplyReviewRequest,
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""审核快速回复模板（通过/驳回）。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
template_id: 模板ID
|
|||
|
|
body: 审核请求体
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含审核结果
|
|||
|
|
"""
|
|||
|
|
result = await admin_service.review_quick_reply(
|
|||
|
|
db,
|
|||
|
|
template_id=template_id,
|
|||
|
|
action=body.action,
|
|||
|
|
reason=body.reason,
|
|||
|
|
agent_id=admin.id,
|
|||
|
|
)
|
|||
|
|
return success_response(data=result)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 6. 消息分配模式
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/assignment-mode ----------
|
|||
|
|
@router.get("/assignment-mode")
|
|||
|
|
async def get_assignment_mode(
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取当前分配模式。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含分配模式信息
|
|||
|
|
"""
|
|||
|
|
result = await admin_service.get_assignment_mode(db)
|
|||
|
|
return success_response(data=result.model_dump())
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- PUT /api/admin/assignment-mode ----------
|
|||
|
|
@router.put("/assignment-mode")
|
|||
|
|
async def update_assignment_mode(
|
|||
|
|
body: AssignmentModeUpdateRequest,
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""切换分配模式（阶段一仅允许手动接单）。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
body: 更新请求体
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含更新后的分配模式
|
|||
|
|
"""
|
|||
|
|
result = await admin_service.update_assignment_mode(db, body.mode, admin.id)
|
|||
|
|
return success_response(data=result.model_dump())
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 7. 会话监控
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/monitor/sessions ----------
|
|||
|
|
@router.get("/monitor/sessions")
|
|||
|
|
async def get_monitor_sessions(
|
|||
|
|
status: Optional[str] = Query(None, description="按会话状态筛选"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取实时会话列表（Demo预览）。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
status: 按会话状态筛选（可选）
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含会话监控数据
|
|||
|
|
"""
|
|||
|
|
result = await admin_service.get_monitor_sessions(db, status)
|
|||
|
|
return success_response(data=result.model_dump())
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 8. 全局搜索
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/search ----------
|
|||
|
|
@router.get("/search")
|
|||
|
|
async def global_search(
|
|||
|
|
q: str = Query(..., min_length=1, description="搜索关键词"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""搜索配置项、坐席、快速回复。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
q: 搜索关键词
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 统一响应格式，包含搜索结果
|
|||
|
|
"""
|
|||
|
|
items = await admin_service.global_search(db, q)
|
|||
|
|
return success_response(data={
|
|||
|
|
"items": [item.model_dump() for item in items]
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 9. 火绒安全集成 API
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 说明：火绒终端安全系统的数据代理端点
|
|||
|
|
# - 测试连接：验证 AccessKey 签名是否正确
|
|||
|
|
# - 终端列表：代理查询火绒终端
|
|||
|
|
# - 终端详情：代理查询终端详细信息
|
|||
|
|
# - 漏洞信息：代理查询高危漏洞
|
|||
|
|
# - 病毒事件：代理查询病毒事件统计
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- POST /api/admin/integrations/huorong/test ----------
|
|||
|
|
@router.post("/integrations/huorong/test")
|
|||
|
|
async def test_huorong_connection(
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""测试火绒API连接。
|
|||
|
|
|
|||
|
|
使用当前保存的 AccessKey 配置调用火绒 _list 接口验证连接。
|
|||
|
|
仅请求1条数据，最小化对火绒系统的影响。
|
|||
|
|
失败时返回调试信息帮助排查凭据问题。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 包含 success(bool) 和 message(str)，失败时含 debug 信息
|
|||
|
|
"""
|
|||
|
|
from app.integrations.huorong.config import get_huorong_client
|
|||
|
|
from app.integrations.huorong.exceptions import HuorongConfigError, HuorongError
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
client = await get_huorong_client(db)
|
|||
|
|
result = await client.test_connection()
|
|||
|
|
# 附带凭据摘要（脱敏），方便排查
|
|||
|
|
result["debug"] = {
|
|||
|
|
"base_url": client.base_url,
|
|||
|
|
"access_key_id": client.access_key_id,
|
|||
|
|
"key_length": len(client.access_key_secret),
|
|||
|
|
}
|
|||
|
|
# 失败时附加排查提示
|
|||
|
|
if not result.get("success"):
|
|||
|
|
result["debug_hint"] = (
|
|||
|
|
"请检查: 1) 登录火绒控制中心 → 中心设置 → 通用设置 → API接口 → 管理密钥，"
|
|||
|
|
"确认 AccessKey ID 和 Secret 正确; 2) 确认 AccessKey 未过期或被撤销; "
|
|||
|
|
"3) 确认当前网络可访问火绒控制中心内网地址"
|
|||
|
|
)
|
|||
|
|
return success_response(data=result)
|
|||
|
|
except HuorongConfigError as e:
|
|||
|
|
return success_response(data={
|
|||
|
|
"success": False,
|
|||
|
|
"message": e.message,
|
|||
|
|
})
|
|||
|
|
except HuorongError as e:
|
|||
|
|
# 认证失败时返回调试信息帮助用户排查
|
|||
|
|
return success_response(data={
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"测试失败: {e.message}",
|
|||
|
|
"debug_hint": "请检查: 1) 火绒控制中心 > 中心设置 > API接口 > 管理密钥 确认凭据正确; 2) AccessKey 是否已过期或被撤销",
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/integrations/huorong/terminals ----------
|
|||
|
|
@router.get("/integrations/huorong/terminals")
|
|||
|
|
async def list_huorong_terminals(
|
|||
|
|
group_id: Optional[str] = Query(None, description="分组ID"),
|
|||
|
|
page: int = Query(1, ge=1, description="页码"),
|
|||
|
|
per_page: int = Query(20, ge=1, le=100, description="每页条数"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""查询火绒终端列表。
|
|||
|
|
|
|||
|
|
代理火绒 /api/clnts/_list 接口，返回终端基本信息。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
group_id: 分组ID（可选）
|
|||
|
|
page: 页码
|
|||
|
|
per_page: 每页条数
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 终端列表数据
|
|||
|
|
"""
|
|||
|
|
from app.integrations.huorong.config import get_huorong_client
|
|||
|
|
from app.integrations.huorong.exceptions import HuorongConfigError, HuorongError
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
client = await get_huorong_client(db)
|
|||
|
|
result = await client.list_terminals(group_id=group_id, page=page, per_page=per_page)
|
|||
|
|
# 序列化 Pydantic 模型
|
|||
|
|
result["items"] = [item.model_dump() for item in result["items"]]
|
|||
|
|
return success_response(data=result)
|
|||
|
|
except HuorongConfigError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "config_missing"})
|
|||
|
|
except HuorongError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "api_error"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/integrations/huorong/terminals/{client_id} ----------
|
|||
|
|
@router.get("/integrations/huorong/terminals/{client_id}")
|
|||
|
|
async def get_huorong_terminal_detail(
|
|||
|
|
client_id: str,
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""查询火绒终端详细信息。
|
|||
|
|
|
|||
|
|
代理火绒 /api/clnts/_info2 接口，返回终端硬件/软件/资产/网络配置。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
client_id: 终端唯一ID
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 终端详细信息
|
|||
|
|
"""
|
|||
|
|
from app.integrations.huorong.config import get_huorong_client
|
|||
|
|
from app.integrations.huorong.exceptions import HuorongConfigError, HuorongError
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
client = await get_huorong_client(db)
|
|||
|
|
detail = await client.get_terminal_detail(client_id)
|
|||
|
|
return success_response(data=detail.model_dump())
|
|||
|
|
except HuorongConfigError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "config_missing"})
|
|||
|
|
except HuorongError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "api_error"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/integrations/huorong/leaks ----------
|
|||
|
|
@router.get("/integrations/huorong/leaks")
|
|||
|
|
async def list_huorong_leaks(
|
|||
|
|
group_id: Optional[str] = Query(None, description="分组ID"),
|
|||
|
|
page: int = Query(1, ge=1, description="页码"),
|
|||
|
|
per_page: int = Query(20, ge=1, le=100, description="每页条数"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""查询火绒高危漏洞终端。
|
|||
|
|
|
|||
|
|
代理火绒 /api/clnts/_leak 接口。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
group_id: 分组ID（可选）
|
|||
|
|
page: 页码
|
|||
|
|
per_page: 每页条数
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 漏洞终端列表
|
|||
|
|
"""
|
|||
|
|
from app.integrations.huorong.config import get_huorong_client
|
|||
|
|
from app.integrations.huorong.exceptions import HuorongConfigError, HuorongError
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
client = await get_huorong_client(db)
|
|||
|
|
result = await client.list_terminal_leaks(group_id=group_id, page=page, per_page=per_page)
|
|||
|
|
result["items"] = [item.model_dump() for item in result["items"]]
|
|||
|
|
return success_response(data=result)
|
|||
|
|
except HuorongConfigError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "config_missing"})
|
|||
|
|
except HuorongError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "api_error"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/integrations/huorong/virus-events ----------
|
|||
|
|
@router.get("/integrations/huorong/virus-events")
|
|||
|
|
async def list_huorong_virus_events(
|
|||
|
|
client_id: Optional[str] = Query(None, description="终端ID（type=0时需提供）"),
|
|||
|
|
group_id: Optional[str] = Query(None, description="分组ID（type=1时需提供）"),
|
|||
|
|
query_type: int = Query(2, ge=0, le=2, description="查询类型: 0=按终端ID, 1=按分组ID, 2=全部"),
|
|||
|
|
page: int = Query(1, ge=1, description="页码"),
|
|||
|
|
per_page: int = Query(20, ge=1, le=100, description="每页条数"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""查询火绒病毒事件统计。
|
|||
|
|
|
|||
|
|
代理火绒 /api/clnts/_virus_events 接口。
|
|||
|
|
火绒API要求 type 参数: 0=按client_id查, 1=按group_id查, 2=查全部。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
client_id: 终端ID（type=0时需提供）
|
|||
|
|
group_id: 分组ID（type=1时需提供）
|
|||
|
|
query_type: 查询类型，默认2(全部)
|
|||
|
|
page: 页码
|
|||
|
|
per_page: 每页条数
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 病毒事件统计数据
|
|||
|
|
"""
|
|||
|
|
from app.integrations.huorong.config import get_huorong_client
|
|||
|
|
from app.integrations.huorong.exceptions import HuorongConfigError, HuorongError
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
client = await get_huorong_client(db)
|
|||
|
|
result = await client.get_virus_events(
|
|||
|
|
client_id=client_id,
|
|||
|
|
group_id=group_id,
|
|||
|
|
query_type=query_type,
|
|||
|
|
page=page,
|
|||
|
|
per_page=per_page,
|
|||
|
|
)
|
|||
|
|
result["items"] = [item.model_dump() for item in result["items"]]
|
|||
|
|
return success_response(data=result)
|
|||
|
|
except HuorongConfigError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "config_missing"})
|
|||
|
|
except HuorongError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "api_error"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 10. 联软LV7000 安全集成 API
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 说明：联软终端管理系统的数据代理端点
|
|||
|
|
# - 测试连接：验证账号密码+Token获取
|
|||
|
|
# - 终端查询：按员工账号/IP/MAC查终端（核心映射接口）
|
|||
|
|
# - 终端详情：极详细硬件+软件+资产+网络信息
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- POST /api/admin/integrations/lianruan/test ----------
|
|||
|
|
@router.post("/integrations/lianruan/test")
|
|||
|
|
async def test_lianruan_connection(
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""测试联软API连接。
|
|||
|
|
|
|||
|
|
通过获取Token验证：IP白名单 + 账号密码 + Token。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 包含 success(bool) 和 message(str)
|
|||
|
|
"""
|
|||
|
|
from app.integrations.lianruan.config import get_lianruan_client
|
|||
|
|
from app.integrations.lianruan.exceptions import LianruanConfigError
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
client = await get_lianruan_client(db)
|
|||
|
|
result = await client.test_connection()
|
|||
|
|
return success_response(data=result)
|
|||
|
|
except LianruanConfigError as e:
|
|||
|
|
return success_response(data={
|
|||
|
|
"success": False,
|
|||
|
|
"message": e.message,
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/integrations/lianruan/terminals ----------
|
|||
|
|
@router.get("/integrations/lianruan/terminals")
|
|||
|
|
async def query_lianruan_terminals(
|
|||
|
|
strusername: Optional[str] = Query(None, description="员工账号（核心映射字段）"),
|
|||
|
|
strdevname: Optional[str] = Query(None, description="计算机名"),
|
|||
|
|
strdevip: Optional[str] = Query(None, description="IP地址"),
|
|||
|
|
page: int = Query(1, ge=1, description="页码"),
|
|||
|
|
per_page: int = Query(20, ge=1, le=100, description="每页条数"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""查询联软终端设备（核心映射接口）。
|
|||
|
|
|
|||
|
|
⭐ strusername 参数可直接按员工账号查终端！这是联软最大的优势。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
strusername: 员工账号（映射金钥匙）
|
|||
|
|
strdevname: 计算机名
|
|||
|
|
strdevip: IP地址
|
|||
|
|
page: 页码
|
|||
|
|
per_page: 每页条数
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 终端列表数据
|
|||
|
|
"""
|
|||
|
|
from app.integrations.lianruan.config import get_lianruan_client
|
|||
|
|
from app.integrations.lianruan.exceptions import LianruanConfigError, LianruanError
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
client = await get_lianruan_client(db)
|
|||
|
|
result = await client.query_dev_by_params(
|
|||
|
|
strusername=strusername or "",
|
|||
|
|
strdevname=strdevname or "",
|
|||
|
|
strdevip=strdevip or "",
|
|||
|
|
page=page,
|
|||
|
|
per_page=per_page,
|
|||
|
|
)
|
|||
|
|
result["items"] = [item.model_dump() for item in result["items"]]
|
|||
|
|
return success_response(data=result)
|
|||
|
|
except LianruanConfigError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "config_missing"})
|
|||
|
|
except LianruanError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "api_error"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 11. P2: 会话审计
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/audit/conversations ----------
|
|||
|
|
@router.get("/audit/conversations")
|
|||
|
|
async def list_audit_conversations(
|
|||
|
|
status: Optional[str] = Query(None, description="按状态筛选"),
|
|||
|
|
agent_id: Optional[str] = Query(None, description="按坐席筛选"),
|
|||
|
|
keyword: Optional[str] = Query(None, description="按员工姓名/消息摘要搜索"),
|
|||
|
|
date_from: Optional[str] = Query(None, description="开始日期 YYYY-MM-DD"),
|
|||
|
|
date_to: Optional[str] = Query(None, description="结束日期 YYYY-MM-DD"),
|
|||
|
|
page: int = Query(1, ge=1, description="页码"),
|
|||
|
|
page_size: int = Query(20, ge=1, le=100, description="每页条数"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取会话审计列表（支持分页+多条件筛选）。"""
|
|||
|
|
result = await admin_service.list_audit_conversations(
|
|||
|
|
db, status=status, agent_id=agent_id, keyword=keyword,
|
|||
|
|
date_from=date_from, date_to=date_to, page=page, page_size=page_size,
|
|||
|
|
)
|
|||
|
|
return success_response(data=result)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/audit/conversations/{id} ----------
|
|||
|
|
@router.get("/audit/conversations/{conversation_id}")
|
|||
|
|
async def get_audit_conversation_detail(
|
|||
|
|
conversation_id: str,
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取会话审计详情（含消息列表）。"""
|
|||
|
|
result = await admin_service.get_audit_conversation_detail(db, conversation_id)
|
|||
|
|
if not result:
|
|||
|
|
raise AppException(3004, "会话不存在")
|
|||
|
|
return success_response(data=result)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 12. P2: 坐席绩效统计
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/agent-performance ----------
|
|||
|
|
@router.get("/agent-performance")
|
|||
|
|
async def get_agent_performance(
|
|||
|
|
date_from: Optional[str] = Query(None, description="开始日期 YYYY-MM-DD"),
|
|||
|
|
date_to: Optional[str] = Query(None, description="结束日期 YYYY-MM-DD"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取坐席绩效统计。"""
|
|||
|
|
items = await admin_service.get_agent_performance(db, date_from=date_from, date_to=date_to)
|
|||
|
|
return success_response(data={"items": items})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 13. P2: 系统日志
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/system-logs ----------
|
|||
|
|
@router.get("/system-logs")
|
|||
|
|
async def get_system_logs(
|
|||
|
|
page: int = Query(1, ge=1, description="页码"),
|
|||
|
|
page_size: int = Query(50, ge=1, le=200, description="每页条数"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""获取系统日志（配置变更日志）。"""
|
|||
|
|
result = await admin_service.get_system_logs(db, page=page, page_size=page_size)
|
|||
|
|
return success_response(data=result)# ---------- GET /api/admin/integrations/lianruan/terminals/{devname}/detail ----------
|
|||
|
|
@router.get("/integrations/lianruan/terminals/{devname}/detail")
|
|||
|
|
async def get_lianruan_terminal_detail(
|
|||
|
|
devname: str,
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""查询联软终端详细信息（极详细）。
|
|||
|
|
|
|||
|
|
比火绒_info2更丰富，含逻辑磁盘使用率、显示器、内存条详情。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
devname: 计算机名
|
|||
|
|
admin: 管理员
|
|||
|
|
db: 数据库会话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict: 终端详细信息
|
|||
|
|
"""
|
|||
|
|
from app.integrations.lianruan.config import get_lianruan_client
|
|||
|
|
from app.integrations.lianruan.exceptions import LianruanConfigError, LianruanError
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
client = await get_lianruan_client(db)
|
|||
|
|
detail = await client.get_dev_all_info(strdevname=devname)
|
|||
|
|
return success_response(data=detail.model_dump())
|
|||
|
|
except LianruanConfigError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "config_missing"})
|
|||
|
|
except LianruanError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "api_error"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==========================================================================
|
|||
|
|
# 14. RAGFlow 知识检索集成 API
|
|||
|
|
# ==========================================================================
|
|||
|
|
|
|||
|
|
# ---------- POST /api/admin/integrations/ragflow/test ----------
|
|||
|
|
@router.post("/integrations/ragflow/test")
|
|||
|
|
async def test_ragflow_connection(
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""测试 RAGFlow API 连接。"""
|
|||
|
|
from app.integrations.ragflow.config import get_ragflow_client
|
|||
|
|
from app.integrations.ragflow.exceptions import RagflowConfigError
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
client = await get_ragflow_client(db)
|
|||
|
|
result = await client.test_connection()
|
|||
|
|
return success_response(data=result)
|
|||
|
|
except RagflowConfigError as e:
|
|||
|
|
return success_response(data={
|
|||
|
|
"success": False,
|
|||
|
|
"message": e.message,
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- GET /api/admin/integrations/ragflow/datasets ----------
|
|||
|
|
@router.get("/integrations/ragflow/datasets")
|
|||
|
|
async def list_ragflow_datasets(
|
|||
|
|
page: int = Query(1, ge=1, description="页码"),
|
|||
|
|
page_size: int = Query(20, ge=1, le=100, description="每页条数"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""列出 RAGFlow 知识库（数据集）。"""
|
|||
|
|
from app.integrations.ragflow.config import get_ragflow_client
|
|||
|
|
from app.integrations.ragflow.exceptions import RagflowConfigError, RagflowError
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
client = await get_ragflow_client(db)
|
|||
|
|
result = await client.list_datasets(page=page, page_size=page_size)
|
|||
|
|
result["items"] = [item.model_dump() for item in result["items"]]
|
|||
|
|
return success_response(data=result)
|
|||
|
|
except RagflowConfigError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "config_missing"})
|
|||
|
|
except RagflowError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "api_error"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------- POST /api/admin/integrations/ragflow/retrieval ----------
|
|||
|
|
@router.post("/integrations/ragflow/retrieval")
|
|||
|
|
async def ragflow_retrieval(
|
|||
|
|
question: str = Query(..., min_length=1, description="查询问题"),
|
|||
|
|
dataset_ids: Optional[str] = Query(None, description="数据集ID列表（逗号分隔）"),
|
|||
|
|
top_k: int = Query(5, ge=1, le=20, description="返回结果数量"),
|
|||
|
|
admin: Agent = Depends(require_admin),
|
|||
|
|
db: AsyncSession = Depends(get_db),
|
|||
|
|
):
|
|||
|
|
"""RAGFlow 知识检索测试。"""
|
|||
|
|
from app.integrations.ragflow.config import get_ragflow_client
|
|||
|
|
from app.integrations.ragflow.exceptions import RagflowConfigError, RagflowError
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
client = await get_ragflow_client(db)
|
|||
|
|
ds_ids = None
|
|||
|
|
if dataset_ids:
|
|||
|
|
ds_ids = [id.strip() for id in dataset_ids.split(",") if id.strip()]
|
|||
|
|
|
|||
|
|
result = await client.retrieval(
|
|||
|
|
question=question,
|
|||
|
|
dataset_ids=ds_ids,
|
|||
|
|
top_k=top_k,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
return success_response(data={
|
|||
|
|
"chunks": [chunk.model_dump() for chunk in result.chunks],
|
|||
|
|
"doc_aggs": [agg.model_dump() for agg in result.doc_aggs],
|
|||
|
|
"total": result.total,
|
|||
|
|
})
|
|||
|
|
except RagflowConfigError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "config_missing"})
|
|||
|
|
except RagflowError as e:
|
|||
|
|
return success_response(data={"error": e.message, "error_code": "api_error"})
|