@sema-agent/server 1.250.0 → 1.252.0

package/dist/http/server.js

@@ -21,6 +21,7 @@ import { looksLikeJwt } from "../auth-bridge.js";
 import { redactSteerIn } from "../orchestration/workflow-agent-steer.js";
 import { emitPendingWorkflowCompletions, taskNotificationInboxEntry, taskNotificationFoldKey, taskNotificationStreamKey, NotifiedKeys } from "../orchestration/workflow-completion-inbox.js";
 import { fleetRunPublisher, fleetRunLabels } from "../fleet/fleet-bus.js";
+import { defaultSubagentTailBus, projectTailFrame } from "../fleet/subagent-tail-bus.js";
 import { RateLimiter } from "../observability/rate-limit.js";
 import { withPrincipal } from "../observability/principal-context.js";
 import { projectEvents, runSummary, mapTraceEvent, toolStartEventData, toolEndEventData, taskProgressEventData, taskNotificationEventData, compactedEventData, diagnosticsEventData, brainStatusEventData, appendModelUsageDelta, appendPromptManifest, attachModelUsage } from "../trace/project.js";
@@ -168,6 +169,7 @@ const WORKFLOW_AGENT_STEER_RE = /^\/v1\/workflows\/([^/]+)\/agents\/([^/]+)\/ste
 const RUN_SUBAGENT_STEER_RE = /^\/v1\/runs\/([^/]+)\/subagents\/([^/]+)\/steer$/;
 const RUN_SUBAGENT_RESUME_RE = /^\/v1\/runs\/([^/]+)\/subagents\/([^/]+)\/resume$/;
 const RUN_SUBAGENT_OUTPUT_RE = /^\/v1\/runs\/([^/]+)\/subagents\/([^/]+)\/output$/;
+const RUN_SUBAGENT_STREAM_RE = /^\/v1\/runs\/([^/]+)\/subagents\/([^/]+)\/stream$/;
 const RUN_TASK_OUTPUT_RE = /^\/v1\/runs\/([^/]+)\/tasks\/([^/]+)\/output$/;
 const RUN_TASK_STOP_RE = /^\/v1\/runs\/([^/]+)\/tasks\/([^/]+)\/stop$/;
 const HAND_TOOL_NAMES = Object.keys(HAND_TOOL_EFFECTS);
@@ -677,6 +679,7 @@ export function createHttpServer(deps) {
                 appendSystemPrompt: true,
                 compactionModel: true,
                 subagentOutput: Boolean(deps.subagentTaskOutput) && Boolean(deps.runStore),
+                subagentStream: Boolean(deps.subagentTaskOutput) && Boolean(deps.runStore),
                 taskHandles: Boolean(deps.taskHandleOutput) && Boolean(deps.taskHandleStop) && Boolean(deps.runStore),
                 taskAgents: deps.config.requirePrincipal !== true,
                 retainBackgroundProcesses: deps.config.requirePrincipal !== true,
@@ -982,6 +985,14 @@ export function createHttpServer(deps) {
                                         onForwardEvent: (e) => {
                                             fleetPub?.onForwardEvent(e);
                                             ledgerSink?.onForwardEvent(e);
+                                            {
+                                                const bg = e.bgAgentId;
+                                                if (bg !== undefined && defaultSubagentTailBus.hasSubscribers(bg)) {
+                                                    const f = projectTailFrame(e);
+                                                    if (f)
+                                                        defaultSubagentTailBus.publish(bg, f);
+                                                }
+                                            }
                                             if (res.writableEnded)
                                                 return;
                                             const t = e.type;
@@ -1008,6 +1019,9 @@ export function createHttpServer(deps) {
                                         onTaskNotification: (n) => {
                                             if (n.task_type === "workflow")
                                                 return;
+                                            if (defaultSubagentTailBus.hasSubscribers(n.task_id)) {
+                                                defaultSubagentTailBus.publish(n.task_id, { type: "task_settled", taskId: n.task_id, status: n.status, ...(n.summary ? { summary: redactSecrets(n.summary) } : {}) });
+                                            }
                                             const hadRow = fleetPub?.onChildTerminal(n.sessionId ?? n.task_id, n.status, n.task_id, n.toolUseId) ?? false;
                                             const deliverable = syncLegLive && !res.writableEnded && !res.destroyed;
                                             const parked = !deliverable && Boolean(deps.workflowCompletionInbox && prepared.spec.sessionId);
@@ -1975,6 +1989,76 @@ export function createHttpServer(deps) {
             sendJson(res, 200, { taskId: runId, target, content: out.content, output: out.details });
             return;
         }
+        const subStreamMatch = req.method === "GET" ? RUN_SUBAGENT_STREAM_RE.exec(url) : null;
+        if (subStreamMatch) {
+            if (!deps.runStore || !deps.subagentTaskOutput) {
+                sendJson(res, 501, { error: "subagent streams require the run store" });
+                return;
+            }
+            const principal = gatedPrincipal(req, deps.config);
+            if (deps.config.requirePrincipal && principal === undefined) {
+                sendJson(res, 401, { error: `missing principal header '${deps.config.principalHeader}'` });
+                return;
+            }
+            const runId = safeDecode(subStreamMatch[1]);
+            const target = safeDecode(subStreamMatch[2]);
+            if (runId === null || target === null) {
+                sendJson(res, 400, { error: "malformed subagent path (invalid percent-encoding)" });
+                return;
+            }
+            const trusted = explicitOperatorOk(principal, deps.config.operatorPrincipals);
+            const run = await deps.runStore.getRun(runId);
+            if (!run || (!trusted && run.owner !== null && run.owner !== principal)) {
+                sendJson(res, 404, { error: "run not found" });
+                return;
+            }
+            const callerSession = new URL(req.url ?? "", "http://x").searchParams.get("session");
+            if (!trusted && run.sessionId && callerSession !== run.sessionId) {
+                sendJson(res, 404, { error: "run not found" });
+                return;
+            }
+            const probe = await deps.subagentTaskOutput(target, { owner: runId, scope: run.owner ?? "default", ...(run.sessionId ? { sessionId: run.sessionId } : {}) });
+            const probeDetails = probe.details;
+            if (probeDetails?.error === "not_found" || probeDetails?.type !== "background_agent") {
+                sendJson(res, 404, { error: `no background agent "${target}" under this run (unknown handle, not this run's child, or already reaped — bg children live in the replica-local registry for the parent's lifetime; wa… workflow-agent rows are read via the workflow journal, not this verb)` });
+                return;
+            }
+            sseHeaders(res);
+            res.write(`event: meta\ndata: ${JSON.stringify({ version: 1, runId, target, status: probeDetails.status ?? "running", live: "replica-local", replayFace: "GET /v1/runs/:id/subagents/:handle/output" })}\n\n`);
+            if (probeDetails.status !== "running" && probeDetails.status !== "pending") {
+                res.end();
+                return;
+            }
+            const it = defaultSubagentTailBus.subscribe(target);
+            let closed = false;
+            const hb = setInterval(() => {
+                if (!res.writableEnded)
+                    res.write(`event: heartbeat\ndata: {}\n\n`);
+            }, 15_000);
+            if (typeof hb.unref === "function")
+                hb.unref();
+            req.on("close", () => {
+                closed = true;
+                void it.return?.();
+            });
+            try {
+                for (;;) {
+                    const n = await it.next();
+                    if (n.done || closed || res.writableEnded)
+                        break;
+                    res.write(`event: forward\ndata: ${JSON.stringify(n.value)}\n\n`);
+                    if (n.value.type === "task_settled")
+                        break;
+                }
+            }
+            finally {
+                clearInterval(hb);
+                void it.return?.();
+                if (!res.writableEnded)
+                    res.end();
+            }
+            return;
+        }
         const taskVerbMatch = req.method === "GET" ? RUN_TASK_OUTPUT_RE.exec(url) : req.method === "POST" ? RUN_TASK_STOP_RE.exec(url) : null;
         if (taskVerbMatch) {
             const stopVerb = req.method === "POST";
@@ -4875,6 +4959,14 @@ export function createHttpServer(deps) {
                     const stream = await deps.runner.resumeStream(token, outcome, { ...taskConfig, signal: cancelCtrl.signal, preemptSignal: preemptCtrl.signal }, {
                         onForwardEvent: (e) => {
                             fleetPub?.onForwardEvent(e);
+                            {
+                                const bg = e.bgAgentId;
+                                if (bg !== undefined && defaultSubagentTailBus.hasSubscribers(bg)) {
+                                    const f = projectTailFrame(e);
+                                    if (f)
+                                        defaultSubagentTailBus.publish(bg, f);
+                                }
+                            }
                             if (!rs || !taskId)
                                 return;
                             const t = e.type;
@@ -4891,6 +4983,9 @@ export function createHttpServer(deps) {
                         onTaskNotification: (n) => {
                             if (n.task_type === "workflow")
                                 return;
+                            if (defaultSubagentTailBus.hasSubscribers(n.task_id)) {
+                                defaultSubagentTailBus.publish(n.task_id, { type: "task_settled", taskId: n.task_id, status: n.status, ...(n.summary ? { summary: redactSecrets(n.summary) } : {}) });
+                            }
                             const hadRow = fleetPub?.onChildTerminal(n.sessionId ?? n.task_id, n.status, n.task_id, n.toolUseId) ?? false;
                             const parked = !resumeLegLive && Boolean(deps.workflowCompletionInbox && sessionId);
                             deps.logger?.info?.("task_notification_observed", { route: "resume", taskId: n.task_id, taskType: n.task_type, status: n.status, hadFleetRow: hadRow, legLive: resumeLegLive, parkedDurable: parked });