@desplega.ai/agent-swarm 1.52.1 → 1.53.0

package/src/types.ts

@@ -143,6 +143,12 @@ export const AgentTaskSchema = z.object({
 
   // Structured output schema (optional — JSON Schema that task output must conform to)
   outputSchema: z.record(z.string(), z.unknown()).optional(),
+
+  // Context usage aggregates
+  compactionCount: z.number().int().min(0).optional(),
+  peakContextPercent: z.number().min(0).max(100).optional(),
+  totalContextTokensUsed: z.number().int().min(0).optional(),
+  contextWindowSize: z.number().int().min(0).optional(),
 });
 
 export const AgentStatusSchema = z.enum(["idle", "busy", "offline"]);
@@ -363,6 +369,76 @@ export const SessionCostSchema = z.object({
 
 export type SessionCost = z.infer<typeof SessionCostSchema>;
 
+// ============================================================================
+// Events
+// ============================================================================
+
+export const EventCategorySchema = z.enum([
+  "tool",
+  "skill",
+  "session",
+  "api",
+  "task",
+  "workflow",
+  "system",
+]);
+
+export const EventStatusSchema = z.enum(["ok", "error", "timeout", "skipped"]);
+
+export const EventSourceSchema = z.enum(["worker", "api", "hook", "scheduler", "cli"]);
+
+export const EventNameSchema = z.enum([
+  // Tool events
+  "tool.start",
+  "tool.end",
+  // Skill events
+  "skill.invoke",
+  "skill.complete",
+  // Session events
+  "session.start",
+  "session.end",
+  "session.resume",
+  "session.cost",
+  // API events
+  "api.request",
+  "api.error",
+  // Task events
+  "task.poll",
+  "task.assign",
+  "task.timeout",
+  // Workflow events
+  "workflow.step.start",
+  "workflow.step.end",
+  "workflow.run.start",
+  "workflow.run.end",
+  // System events
+  "system.boot",
+  "system.migration",
+  "system.error",
+]);
+
+export const SwarmEventSchema = z.object({
+  id: z.uuid(),
+  category: EventCategorySchema,
+  event: EventNameSchema,
+  status: EventStatusSchema,
+  source: EventSourceSchema,
+  agentId: z.string().optional(),
+  taskId: z.string().optional(),
+  sessionId: z.string().optional(),
+  parentEventId: z.string().optional(),
+  numericValue: z.number().optional(),
+  durationMs: z.number().int().optional(),
+  data: z.record(z.string(), z.unknown()).optional(),
+  createdAt: z.iso.datetime(),
+});
+
+export type EventCategory = z.infer<typeof EventCategorySchema>;
+export type EventStatus = z.infer<typeof EventStatusSchema>;
+export type EventSource = z.infer<typeof EventSourceSchema>;
+export type EventName = z.infer<typeof EventNameSchema>;
+export type SwarmEvent = z.infer<typeof SwarmEventSchema>;
+
 // ============================================================================
 // Scheduled Task Types
 // ============================================================================
@@ -923,3 +999,80 @@ export const SkillWithInstallInfoSchema = SkillSchema.extend({
   installedAt: z.string(),
 });
 export type SkillWithInstallInfo = z.infer<typeof SkillWithInstallInfoSchema>;
+
+// ── MCP Servers ──────────────────────────────────────────────────────────
+
+export const McpServerTransportSchema = z.enum(["stdio", "http", "sse"]);
+export type McpServerTransport = z.infer<typeof McpServerTransportSchema>;
+
+export const McpServerScopeSchema = z.enum(["global", "swarm", "agent"]);
+export type McpServerScope = z.infer<typeof McpServerScopeSchema>;
+
+export const McpServerSchema = z.object({
+  id: z.string(),
+  name: z.string(),
+  description: z.string().nullable(),
+  scope: McpServerScopeSchema,
+  ownerAgentId: z.string().nullable(),
+  transport: McpServerTransportSchema,
+  command: z.string().nullable(),
+  args: z.string().nullable(),
+  url: z.string().nullable(),
+  headers: z.string().nullable(),
+  envConfigKeys: z.string().nullable(),
+  headerConfigKeys: z.string().nullable(),
+  isEnabled: z.boolean(),
+  version: z.number(),
+  createdAt: z.string(),
+  lastUpdatedAt: z.string(),
+});
+export type McpServer = z.infer<typeof McpServerSchema>;
+
+export const AgentMcpServerSchema = z.object({
+  id: z.string(),
+  agentId: z.string(),
+  mcpServerId: z.string(),
+  isActive: z.boolean(),
+  installedAt: z.string(),
+});
+export type AgentMcpServer = z.infer<typeof AgentMcpServerSchema>;
+
+export const McpServerWithInstallInfoSchema = McpServerSchema.extend({
+  isActive: z.boolean(),
+  installedAt: z.string(),
+});
+export type McpServerWithInstallInfo = z.infer<typeof McpServerWithInstallInfoSchema>;
+
+// ============================================================================
+// Context Usage Tracking Types
+// ============================================================================
+
+export const ContextSnapshotEventTypeSchema = z.enum(["progress", "compaction", "completion"]);
+export type ContextSnapshotEventType = z.infer<typeof ContextSnapshotEventTypeSchema>;
+
+export const ContextSnapshotSchema = z.object({
+  id: z.uuid(),
+  taskId: z.uuid(),
+  agentId: z.uuid(),
+  sessionId: z.string(),
+
+  // Context window state
+  contextUsedTokens: z.number().int().min(0).optional(),
+  contextTotalTokens: z.number().int().min(0).optional(),
+  contextPercent: z.number().min(0).max(100).optional(),
+
+  // Event metadata
+  eventType: ContextSnapshotEventTypeSchema,
+
+  // Compaction-specific (null for non-compaction)
+  compactTrigger: z.enum(["auto", "manual"]).optional(),
+  preCompactTokens: z.number().int().min(0).optional(),
+
+  // Cumulative counters at this point
+  cumulativeInputTokens: z.number().int().min(0).default(0),
+  cumulativeOutputTokens: z.number().int().min(0).default(0),
+
+  createdAt: z.iso.datetime(),
+});
+
+export type ContextSnapshot = z.infer<typeof ContextSnapshotSchema>;

package/src/utils/context-window.ts

@@ -0,0 +1,41 @@
+/**
+ * Context window size lookup and usage computation utilities.
+ *
+ * This module is safe for both API and worker code — it has NO database imports.
+ */
+
+const CONTEXT_WINDOW_DEFAULTS: Record<string, number> = {
+  "claude-opus-4-6": 1_000_000,
+  "claude-sonnet-4-6": 1_000_000,
+  "claude-haiku-4-5": 200_000,
+  opus: 1_000_000,
+  sonnet: 1_000_000,
+  haiku: 200_000,
+  default: 200_000,
+};
+
+/**
+ * Look up the context window size (in tokens) for a given model identifier.
+ * Falls back to the "default" entry when the model is not explicitly mapped.
+ */
+const DEFAULT_CONTEXT_WINDOW = 200_000;
+
+export function getContextWindowSize(model: string): number {
+  return CONTEXT_WINDOW_DEFAULTS[model] ?? DEFAULT_CONTEXT_WINDOW;
+}
+
+/**
+ * Compute the total context tokens used from a Claude API usage object.
+ * Sums input_tokens + cache_creation_input_tokens + cache_read_input_tokens.
+ */
+export function computeContextUsed(usage: {
+  input_tokens?: number | null;
+  cache_creation_input_tokens?: number | null;
+  cache_read_input_tokens?: number | null;
+}): number {
+  return (
+    (usage.input_tokens ?? 0) +
+    (usage.cache_creation_input_tokens ?? 0) +
+    (usage.cache_read_input_tokens ?? 0)
+  );
+}

package/src/workflows/executors/base.ts

@@ -61,7 +61,15 @@ export abstract class BaseExecutor<
       };
     }
 
-    const result = await this.execute(configResult.data, input.context, input.meta);
+    let result: ExecutorResult<z.infer<TOutput>>;
+    try {
+      result = await this.execute(configResult.data, input.context, input.meta);
+    } catch (err) {
+      return {
+        status: "failed",
+        error: `Executor threw: ${err instanceof Error ? err.message : String(err)}`,
+      };
+    }
 
     // Validate output for successful results
     if (result.status === "success" && result.output !== undefined) {