23 lines
839 B
Python
23 lines
839 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
|
|
return AgentEventResponse(service, service.stream_run(body), 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)
|