6f64eecb43
- fix(api): heartbeat endpoint no longer requires request body, fix spa_fallback 404 handler causing 500 on API routes, add db.refresh() in agent update/heartbeat - feat(time): unify all frontend time display to Beijing time (Asia/Shanghai) via new beijing() utility; backend alert messages also use BJT - feat(ui): collapse AI agent instruction panel on Tasks page by default - chore: add backend/app/utils.py and frontend/src/utils/time.js
88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
"""API router — Agent management."""
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.database import get_db
|
|
from app.schemas import AgentBatchRegister, AgentBatchRegisterResult, AgentCreate, AgentHeartbeat, AgentOut, AgentUpdate
|
|
from app.services import AgentService, TaskService
|
|
|
|
router = APIRouter(prefix="/api/agents", tags=["agents"])
|
|
|
|
|
|
@router.post("", response_model=AgentOut, status_code=201)
|
|
async def register_agent(body: AgentCreate, db: AsyncSession = Depends(get_db)):
|
|
svc = AgentService(db)
|
|
agent = await svc.create(name=body.name, description=body.description)
|
|
task_count = await svc.get_task_count(agent.id)
|
|
return AgentOut(**{**agent.__dict__, "task_count": task_count})
|
|
|
|
|
|
@router.get("", response_model=list[AgentOut])
|
|
async def list_agents(db: AsyncSession = Depends(get_db)):
|
|
svc = AgentService(db)
|
|
agents = await svc.list_all()
|
|
result = []
|
|
for a in agents:
|
|
cnt = await svc.get_task_count(a.id)
|
|
result.append(AgentOut(**{**a.__dict__, "task_count": cnt}))
|
|
return result
|
|
|
|
|
|
@router.get("/{agent_id}", response_model=AgentOut)
|
|
async def get_agent(agent_id: int, db: AsyncSession = Depends(get_db)):
|
|
svc = AgentService(db)
|
|
agent = await svc.get(agent_id)
|
|
if not agent:
|
|
raise HTTPException(404, "Agent not found")
|
|
cnt = await svc.get_task_count(agent.id)
|
|
return AgentOut(**{**agent.__dict__, "task_count": cnt})
|
|
|
|
|
|
@router.put("/{agent_id}", response_model=AgentOut)
|
|
async def update_agent(agent_id: int, body: AgentUpdate, db: AsyncSession = Depends(get_db)):
|
|
svc = AgentService(db)
|
|
agent = await svc.update(agent_id, **body.model_dump(exclude_none=True))
|
|
if not agent:
|
|
raise HTTPException(404, "Agent not found")
|
|
cnt = await svc.get_task_count(agent.id)
|
|
return AgentOut(**{**agent.__dict__, "task_count": cnt})
|
|
|
|
|
|
@router.post("/{agent_id}/heartbeat", response_model=AgentOut)
|
|
async def heartbeat(agent_id: int, db: AsyncSession = Depends(get_db)):
|
|
svc = AgentService(db)
|
|
agent = await svc.heartbeat(agent_id)
|
|
if not agent:
|
|
raise HTTPException(404, "Agent not found")
|
|
cnt = await svc.get_task_count(agent.id)
|
|
return AgentOut(**{**agent.__dict__, "task_count": cnt})
|
|
|
|
|
|
# ── Batch registration: AI agent self-registers with all its tasks ─────
|
|
|
|
@router.post("/register-with-tasks", response_model=AgentBatchRegisterResult, status_code=201)
|
|
async def register_agent_with_tasks(body: AgentBatchRegister, db: AsyncSession = Depends(get_db)):
|
|
"""AI Agent 一次性注册自己 + 所有定时任务。"""
|
|
agent_svc = AgentService(db)
|
|
task_svc = TaskService(db)
|
|
|
|
agent = await agent_svc.create(name=body.name, description=body.description)
|
|
tasks_created = 0
|
|
for t in body.tasks:
|
|
await task_svc.create(
|
|
agent_id=agent.id,
|
|
name=t.name,
|
|
cron_expression=t.cron_expression,
|
|
description=t.description,
|
|
grace_period=t.grace_period,
|
|
)
|
|
tasks_created += 1
|
|
|
|
await db.commit()
|
|
task_count = await agent_svc.get_task_count(agent.id)
|
|
return AgentBatchRegisterResult(
|
|
agent=AgentOut(**{**agent.__dict__, "task_count": task_count}),
|
|
tasks_created=tasks_created,
|
|
)
|