diff --git a/apps/api/src/orchestrator/orchestrator.controller.ts b/apps/api/src/orchestrator/orchestrator.controller.ts index e17d403..3cd25ac 100644 --- a/apps/api/src/orchestrator/orchestrator.controller.ts +++ b/apps/api/src/orchestrator/orchestrator.controller.ts @@ -1,4 +1,4 @@ -import { Controller, Get, Res, UseGuards } from "@nestjs/common"; +import { Controller, Get, Query, Res, UseGuards } from "@nestjs/common"; import { AgentStatus } from "@prisma/client"; import type { Response } from "express"; import { AuthGuard } from "../auth/guards/auth.guard"; @@ -6,6 +6,7 @@ import { PrismaService } from "../prisma/prisma.service"; const AGENT_POLL_INTERVAL_MS = 5_000; const SSE_HEARTBEAT_MS = 15_000; +const DEFAULT_EVENTS_LIMIT = 25; interface OrchestratorAgentDto { id: string; @@ -15,6 +16,26 @@ interface OrchestratorAgentDto { createdAt: Date; } +interface OrchestratorEventDto { + type: string; + timestamp: string; + agentId?: string; + taskId?: string; + data?: Record; +} + +interface OrchestratorHealthDto { + status: "healthy" | "degraded" | "unhealthy"; + database: "connected" | "disconnected"; + agents: { + total: number; + working: number; + idle: number; + errored: number; + }; + timestamp: string; +} + @Controller("orchestrator") @UseGuards(AuthGuard) export class OrchestratorController { @@ -25,6 +46,81 @@ export class OrchestratorController { return this.fetchActiveAgents(); } + @Get("events/recent") + async getRecentEvents( + @Query("limit") limit?: string + ): Promise<{ events: OrchestratorEventDto[] }> { + const eventsLimit = limit ? parseInt(limit, 10) : DEFAULT_EVENTS_LIMIT; + const safeLimit = Math.min(Math.max(eventsLimit, 1), 100); + + // Fetch recent agent activity to derive events + const agents = await this.prisma.agent.findMany({ + where: { + status: { + not: AgentStatus.TERMINATED, + }, + }, + orderBy: { + createdAt: "desc", + }, + take: safeLimit, + }); + + // Derive events from agent status changes + const events: OrchestratorEventDto[] = agents.map((agent) => ({ + type: `agent:${agent.status.toLowerCase()}`, + timestamp: agent.createdAt.toISOString(), + agentId: agent.id, + data: { + name: agent.name, + role: agent.role, + model: agent.model, + }, + })); + + return { events }; + } + + @Get("health") + async getHealth(): Promise { + let databaseConnected = false; + let agents: OrchestratorAgentDto[] = []; + + try { + // Check database connectivity + await this.prisma.$queryRaw`SELECT 1`; + databaseConnected = true; + + // Get agent counts + agents = await this.fetchActiveAgents(); + } catch { + databaseConnected = false; + } + + const working = agents.filter((a) => a.status === AgentStatus.WORKING).length; + const idle = agents.filter((a) => a.status === AgentStatus.IDLE).length; + const errored = agents.filter((a) => a.status === AgentStatus.ERROR).length; + + let status: OrchestratorHealthDto["status"] = "healthy"; + if (!databaseConnected) { + status = "unhealthy"; + } else if (errored > 0) { + status = "degraded"; + } + + return { + status, + database: databaseConnected ? "connected" : "disconnected", + agents: { + total: agents.length, + working, + idle, + errored, + }, + timestamp: new Date().toISOString(), + }; + } + @Get("events") async streamEvents(@Res() res: Response): Promise { res.setHeader("Content-Type", "text/event-stream");