"""大屏 Agent 启动与澄清恢复接口"""

from fastapi import APIRouter, Request

from app.common.response import ApiResponse, success_response
from app.schemas.agent import AgentResult, ResumeRequest
from app.schemas.request import DesignRequest

router = APIRouter(prefix="/agent", tags=["Agent"])


@router.post("/run", response_model=ApiResponse[AgentResult])
async def run_agent(body: DesignRequest, request: Request):
    return success_response(await request.app.state.agent_service.run(body))


@router.post("/resume", response_model=ApiResponse[AgentResult])
async def resume_agent(body: ResumeRequest, request: Request):
    return success_response(await request.app.state.agent_service.resume(body))