25 lines
939 B
Python
25 lines
939 B
Python
"""大屏 Agent 启动与澄清恢复接口"""
|
|
|
|
from fastapi import APIRouter, Request
|
|
|
|
from app.common.sse import AgentEventResponse
|
|
from app.schemas.agent import ResumeRequest
|
|
from app.schemas.request import DesignRequest
|
|
|
|
router = APIRouter(prefix="/agent", tags=["Agent"])
|
|
|
|
|
|
@router.post("/run", response_class=AgentEventResponse, status_code=200)
|
|
async def stream_agent(body: DesignRequest, request: Request):
|
|
service = request.app.state.agent_service
|
|
# 在发送流响应头之前完成会话校验并占用任务
|
|
execution = await service.prepare_run(body)
|
|
return AgentEventResponse(service, execution, request)
|
|
|
|
|
|
@router.post("/resume", response_class=AgentEventResponse, status_code=200)
|
|
async def stream_resume_agent(body: ResumeRequest, request: Request):
|
|
service = request.app.state.agent_service
|
|
execution = await service.prepare_resume(body)
|
|
return AgentEventResponse(service, execution, request)
|