stitchkit 0.69.0 → 0.70.1

package/dist/index-bd17ytae.js

@@ -0,0 +1,192 @@
+import {
+  AppError
+} from "./index-0w9abg87.js";
+import {
+  isRecord
+} from "./index-qyrqwr4c.js";
+
+// src/internal/errors.ts
+import { z } from "zod";
+function issuePath(path) {
+  return path.length > 0 ? path.map(String).join(".") : "(root)";
+}
+function formatZodError(error) {
+  const issues = error.issues.slice(0, 5);
+  const lines = issues.map((issue) => `${issuePath(issue.path)}: ${issue.message}`);
+  const suffix = error.issues.length > 5 ? `
+...and ${error.issues.length - 5} more issues` : "";
+  return lines.join(`
+`) + suffix;
+}
+function zodIssues(error) {
+  return error.issues.map((issue) => ({
+    path: issuePath(issue.path),
+    code: issue.code,
+    message: issue.message
+  }));
+}
+var MAX_DETAIL_ISSUES = 20;
+function errorCode(err) {
+  if (AppError.is(err))
+    return err.code;
+  if (err instanceof z.ZodError)
+    return "VALIDATION_ERROR";
+  return;
+}
+function recordedErrorMessage(code, envelopeMessage, thrown) {
+  if ((code === "INTERNAL_SERVER_ERROR" || code === "REALTIME_CONTRACT_VIOLATION") && thrown !== undefined) {
+    if (thrown instanceof Error)
+      return thrown.message;
+    if (typeof thrown === "string")
+      return thrown;
+  }
+  return envelopeMessage;
+}
+function normalizeError(err) {
+  if (AppError.is(err)) {
+    if (err.code === "REALTIME_CONTRACT_VIOLATION") {
+      return new AppError("REALTIME_CONTRACT_VIOLATION", "Realtime contract violation", 500);
+    }
+    return err;
+  }
+  if (err instanceof z.ZodError) {
+    return new AppError("VALIDATION_ERROR", formatZodError(err), 400, {
+      issues: zodIssues(err).slice(0, MAX_DETAIL_ISSUES)
+    });
+  }
+  console.error("[stitchkit] unhandled error:", err);
+  return new AppError("INTERNAL_SERVER_ERROR", "Internal server error", 500);
+}
+function strippedPaths(before, after, prefix) {
+  if (Array.isArray(before)) {
+    if (!Array.isArray(after))
+      return [];
+    return before.flatMap((item, i) => strippedPaths(item, after[i], `${prefix}[${i}]`));
+  }
+  if (!isRecord(before) || !isRecord(after))
+    return [];
+  const paths = [];
+  for (const [key, value] of Object.entries(before)) {
+    const path = prefix ? `${prefix}.${key}` : key;
+    if (!(key in after)) {
+      paths.push(path);
+      continue;
+    }
+    paths.push(...strippedPaths(value, after[key], path));
+  }
+  return paths;
+}
+function validateHandlerOutput(schema, data, onStripped) {
+  const parsed = schema.safeParse(data);
+  if (parsed.success) {
+    if (onStripped) {
+      const paths = strippedPaths(data, parsed.data, "");
+      if (paths.length > 0)
+        onStripped(paths);
+    }
+    return { ok: true, data: parsed.data };
+  }
+  return {
+    ok: false,
+    message: `Handler output does not match the contract: ${formatZodError(parsed.error)}`
+  };
+}
+function validateDeclaredOutput(schema, data, onStripped) {
+  if (!schema) {
+    if (data === undefined || data === null)
+      return { ok: true, data };
+    return {
+      ok: false,
+      message: "Handler returned data but the contract declares no output"
+    };
+  }
+  if (data === undefined) {
+    return {
+      ok: false,
+      message: "Handler returned undefined but the contract declares an output"
+    };
+  }
+  return validateHandlerOutput(schema, data, onStripped);
+}
+
+// src/observability/trace.ts
+var TRACEPARENT_RE = /^([0-9a-f]{2})-([0-9a-f]{32})-([0-9a-f]{16})-([0-9a-f]{2})(.*)$/i;
+function randomHex(bytes) {
+  const arr = new Uint8Array(bytes);
+  crypto.getRandomValues(arr);
+  let hex = "";
+  for (const byte of arr)
+    hex += byte.toString(16).padStart(2, "0");
+  return hex;
+}
+function createTraceContext() {
+  return { traceId: randomHex(16), spanId: randomHex(8) };
+}
+function parseTraceparent(header) {
+  if (!header)
+    return null;
+  const match = TRACEPARENT_RE.exec(header.trim());
+  if (!match?.[1] || !match[2] || !match[3] || !match[4] || match[5] === undefined)
+    return null;
+  const version = match[1].toLowerCase();
+  if (version === "ff")
+    return null;
+  const suffix = match[5];
+  if (version === "00" ? suffix !== "" : suffix !== "" && !/^(?:-[0-9a-f]{2,})+$/i.test(suffix)) {
+    return null;
+  }
+  const traceId = match[2].toLowerCase();
+  const parentSpanId = match[3].toLowerCase();
+  const traceFlags = match[4].toLowerCase();
+  if (/^0+$/.test(traceId) || /^0+$/.test(parentSpanId))
+    return null;
+  return { traceId, spanId: randomHex(8), parentSpanId, traceFlags };
+}
+function formatTraceparent(ctx) {
+  return `00-${ctx.traceId}-${ctx.spanId}-${ctx.traceFlags ?? "01"}`;
+}
+function resolveTraceContext(req) {
+  return parseTraceparent(req.headers.get("traceparent")) ?? createTraceContext();
+}
+function childSpan(parent) {
+  return {
+    traceId: parent.traceId,
+    spanId: randomHex(8),
+    parentSpanId: parent.spanId,
+    ...parent.tracestate !== undefined && { tracestate: parent.tracestate },
+    ...parent.baggage !== undefined && { baggage: parent.baggage },
+    ...parent.traceFlags !== undefined && { traceFlags: parent.traceFlags }
+  };
+}
+var encoder = new TextEncoder;
+function boundedPropagationValue(value, maxBytes, maxMembers) {
+  if (typeof value !== "string")
+    return;
+  const trimmed = value.trim();
+  if (!trimmed || encoder.encode(trimmed).byteLength > maxBytes)
+    return;
+  for (const character of trimmed) {
+    const code = character.charCodeAt(0);
+    if (code <= 31 || code === 127)
+      return;
+  }
+  const members = trimmed.split(",");
+  if (members.length > maxMembers || members.some((member) => !member.trim())) {
+    return;
+  }
+  return trimmed;
+}
+function resolvePropagationContext(metadata, fallback) {
+  const traceparent = metadata?.traceparent;
+  const parsed = typeof traceparent === "string" ? parseTraceparent(traceparent) : null;
+  const trace = traceparent === undefined ? fallback ?? createTraceContext() : parsed ?? createTraceContext();
+  const tracestate = parsed ? boundedPropagationValue(metadata?.tracestate, 512, 32) : undefined;
+  const baggage = boundedPropagationValue(metadata?.baggage, 8192, 180);
+  return {
+    ...trace,
+    ...tracestate !== undefined && { tracestate },
+    ...baggage !== undefined && { baggage }
+  };
+}
+
+export { formatZodError, zodIssues, errorCode, recordedErrorMessage, normalizeError, validateDeclaredOutput, createTraceContext, parseTraceparent, formatTraceparent, resolveTraceContext, childSpan, resolvePropagationContext };

package/dist/{index-q7gdep2c.js → index-da1aqnhb.js}

@@ -2,13 +2,13 @@ import {
   contextContributionDelta,
   mergeContextContribution,
   prepareContextContribution
-} from "./index-d1jsvkyk.js";
+} from "./index-vbf2p6me.js";
 import {
   base64UrlToBytes,
   bytesToBase64Url,
   isUnsafeKey,
   safeJsonParse
-} from "./index-x18ndp4p.js";
+} from "./index-vj3vvpaa.js";
 import {
   forbidden,
   unauthorized

package/dist/{index-p1be103g.js → index-ktsps64f.js}

@@ -1,7 +1,9 @@
 import {
-  isUnsafeKey,
   resolveTraceContext
-} from "./index-x18ndp4p.js";
+} from "./index-bd17ytae.js";
+import {
+  isUnsafeKey
+} from "./index-vj3vvpaa.js";
 import {
   isRecord
 } from "./index-qyrqwr4c.js";

package/dist/{index-mbxds404.js → index-mjx0wt4c.js}

@@ -1,14 +1,14 @@
 import {
   collectToolSurface
-} from "./index-xxsq3ca1.js";
+} from "./index-vc498pst.js";
 import {
   createToolRunner,
   formatToolError
-} from "./index-kg9xx84n.js";
+} from "./index-7rey2b8s.js";
 import {
   coerceJsonArgs,
   toolResultFromError
-} from "./index-h8vn1jcr.js";
+} from "./index-qzbg46b6.js";
 import {
   isWithinDir
 } from "./index-sbdmyz75.js";
@@ -22,10 +22,12 @@ import {
 } from "./index-cby4ar3v.js";
 import {
   formatZodError,
-  isUnsafeKey,
-  safeJsonParse,
   validateDeclaredOutput
-} from "./index-x18ndp4p.js";
+} from "./index-bd17ytae.js";
+import {
+  isUnsafeKey,
+  safeJsonParse
+} from "./index-vj3vvpaa.js";
 import {
   isRecord
 } from "./index-qyrqwr4c.js";

package/dist/{index-h8vn1jcr.js → index-qzbg46b6.js}

@@ -1,17 +1,19 @@
 import {
   getRequestContext,
   runWithRequestContext
-} from "./index-p1be103g.js";
+} from "./index-ktsps64f.js";
 import {
   objectShapeKeys
 } from "./index-22by16v6.js";
 import {
   formatZodError,
-  isUnsafeKey,
   normalizeError,
-  safeJsonParse,
   validateDeclaredOutput
-} from "./index-x18ndp4p.js";
+} from "./index-bd17ytae.js";
+import {
+  isUnsafeKey,
+  safeJsonParse
+} from "./index-vj3vvpaa.js";
 import {
   AppError,
   STITCH_ERROR_STATUS,

package/dist/{index-h7ctj1kp.js → index-s8nt8c22.js}

@@ -1,14 +1,17 @@
+import {
+  AgentToolError
+} from "./index-3xnq72rz.js";
 import {
   collectToolSurface
-} from "./index-xxsq3ca1.js";
+} from "./index-vc498pst.js";
 import {
   createToolRunner,
   formatToolError
-} from "./index-kg9xx84n.js";
+} from "./index-7rey2b8s.js";
 import {
   isToolExecutionControlError,
   toolResultFromError
-} from "./index-h8vn1jcr.js";
+} from "./index-qzbg46b6.js";
 import {
   isRecord
 } from "./index-qyrqwr4c.js";
@@ -40,16 +43,16 @@ function mountAgent(services, config = {}) {
     });
     const execute = async (rawArgs, options) => {
       const args = isRecord(rawArgs) ? rawArgs : {};
-      try {
-        const result = await runTool(mountable, args, { signal: options.abortSignal });
-        if (result.ok)
-          return result.data;
-        return formatToolError(result, mountable.name, config.errorHint);
-      } catch (err) {
+      const result = await runTool(mountable, args, {
+        signal: options.abortSignal
+      }).catch((err) => {
         if (isToolExecutionControlError(err))
           throw err;
-        return formatToolError(toolResultFromError(err), mountable.name, config.errorHint);
-      }
+        throw new AgentToolError(formatToolError(toolResultFromError(err), mountable.name, config.errorHint), err);
+      });
+      if (result.ok)
+        return result.data;
+      throw new AgentToolError(formatToolError(result, mountable.name, config.errorHint));
     };
     const presenter = entry.kind === "runtime" ? entry.definition.present?.agent : undefined;
     if (entry.kind === "runtime" && presenter) {

package/dist/{index-d1jsvkyk.js → index-vbf2p6me.js}

@@ -1,7 +1,7 @@
 import {
   isUnsafeKey,
   mergeMeta
-} from "./index-x18ndp4p.js";
+} from "./index-vj3vvpaa.js";
 import {
   callRuntimeHandler,
   isRecord,

package/dist/{index-xxsq3ca1.js → index-vc498pst.js}

@@ -3,7 +3,7 @@ import {
 } from "./index-wwst0td5.js";
 import {
   contractToolMountable
-} from "./index-kg9xx84n.js";
+} from "./index-7rey2b8s.js";
 import {
   projectToolSurface
 } from "./index-22by16v6.js";

package/dist/{index-x18ndp4p.js → index-vj3vvpaa.js}

@@ -1,10 +1,3 @@
-import {
-  AppError
-} from "./index-0w9abg87.js";
-import {
-  isRecord
-} from "./index-qyrqwr4c.js";
-
 // src/contract/define.ts
 import { z } from "zod";
 
@@ -311,188 +304,4 @@ var ContractStreamFrameSchema = z4.discriminatedUnion("type", [
   }).strict(),
   z4.object({ type: z4.literal("end") }).strict()
 ]);
-// src/internal/errors.ts
-import { z as z5 } from "zod";
-function issuePath(path) {
-  return path.length > 0 ? path.map(String).join(".") : "(root)";
-}
-function formatZodError(error) {
-  const issues = error.issues.slice(0, 5);
-  const lines = issues.map((issue) => `${issuePath(issue.path)}: ${issue.message}`);
-  const suffix = error.issues.length > 5 ? `
-...and ${error.issues.length - 5} more issues` : "";
-  return lines.join(`
-`) + suffix;
-}
-function zodIssues(error) {
-  return error.issues.map((issue) => ({
-    path: issuePath(issue.path),
-    code: issue.code,
-    message: issue.message
-  }));
-}
-var MAX_DETAIL_ISSUES = 20;
-function errorCode(err) {
-  if (AppError.is(err))
-    return err.code;
-  if (err instanceof z5.ZodError)
-    return "VALIDATION_ERROR";
-  return;
-}
-function recordedErrorMessage(code, envelopeMessage, thrown) {
-  if ((code === "INTERNAL_SERVER_ERROR" || code === "REALTIME_CONTRACT_VIOLATION") && thrown !== undefined) {
-    if (thrown instanceof Error)
-      return thrown.message;
-    if (typeof thrown === "string")
-      return thrown;
-  }
-  return envelopeMessage;
-}
-function normalizeError(err) {
-  if (AppError.is(err)) {
-    if (err.code === "REALTIME_CONTRACT_VIOLATION") {
-      return new AppError("REALTIME_CONTRACT_VIOLATION", "Realtime contract violation", 500);
-    }
-    return err;
-  }
-  if (err instanceof z5.ZodError) {
-    return new AppError("VALIDATION_ERROR", formatZodError(err), 400, {
-      issues: zodIssues(err).slice(0, MAX_DETAIL_ISSUES)
-    });
-  }
-  console.error("[stitchkit] unhandled error:", err);
-  return new AppError("INTERNAL_SERVER_ERROR", "Internal server error", 500);
-}
-function strippedPaths(before, after, prefix) {
-  if (Array.isArray(before)) {
-    if (!Array.isArray(after))
-      return [];
-    return before.flatMap((item, i) => strippedPaths(item, after[i], `${prefix}[${i}]`));
-  }
-  if (!isRecord(before) || !isRecord(after))
-    return [];
-  const paths = [];
-  for (const [key, value] of Object.entries(before)) {
-    const path = prefix ? `${prefix}.${key}` : key;
-    if (!(key in after)) {
-      paths.push(path);
-      continue;
-    }
-    paths.push(...strippedPaths(value, after[key], path));
-  }
-  return paths;
-}
-function validateHandlerOutput(schema, data, onStripped) {
-  const parsed = schema.safeParse(data);
-  if (parsed.success) {
-    if (onStripped) {
-      const paths = strippedPaths(data, parsed.data, "");
-      if (paths.length > 0)
-        onStripped(paths);
-    }
-    return { ok: true, data: parsed.data };
-  }
-  return {
-    ok: false,
-    message: `Handler output does not match the contract: ${formatZodError(parsed.error)}`
-  };
-}
-function validateDeclaredOutput(schema, data, onStripped) {
-  if (!schema) {
-    if (data === undefined || data === null)
-      return { ok: true, data };
-    return {
-      ok: false,
-      message: "Handler returned data but the contract declares no output"
-    };
-  }
-  if (data === undefined) {
-    return {
-      ok: false,
-      message: "Handler returned undefined but the contract declares an output"
-    };
-  }
-  return validateHandlerOutput(schema, data, onStripped);
-}
-
-// src/observability/trace.ts
-var TRACEPARENT_RE = /^([0-9a-f]{2})-([0-9a-f]{32})-([0-9a-f]{16})-([0-9a-f]{2})(.*)$/i;
-function randomHex(bytes) {
-  const arr = new Uint8Array(bytes);
-  crypto.getRandomValues(arr);
-  let hex = "";
-  for (const byte of arr)
-    hex += byte.toString(16).padStart(2, "0");
-  return hex;
-}
-function createTraceContext() {
-  return { traceId: randomHex(16), spanId: randomHex(8) };
-}
-function parseTraceparent(header) {
-  if (!header)
-    return null;
-  const match = TRACEPARENT_RE.exec(header.trim());
-  if (!match?.[1] || !match[2] || !match[3] || !match[4] || match[5] === undefined)
-    return null;
-  const version = match[1].toLowerCase();
-  if (version === "ff")
-    return null;
-  const suffix = match[5];
-  if (version === "00" ? suffix !== "" : suffix !== "" && !/^(?:-[0-9a-f]{2,})+$/i.test(suffix)) {
-    return null;
-  }
-  const traceId = match[2].toLowerCase();
-  const parentSpanId = match[3].toLowerCase();
-  const traceFlags = match[4].toLowerCase();
-  if (/^0+$/.test(traceId) || /^0+$/.test(parentSpanId))
-    return null;
-  return { traceId, spanId: randomHex(8), parentSpanId, traceFlags };
-}
-function formatTraceparent(ctx) {
-  return `00-${ctx.traceId}-${ctx.spanId}-${ctx.traceFlags ?? "01"}`;
-}
-function resolveTraceContext(req) {
-  return parseTraceparent(req.headers.get("traceparent")) ?? createTraceContext();
-}
-function childSpan(parent) {
-  return {
-    traceId: parent.traceId,
-    spanId: randomHex(8),
-    parentSpanId: parent.spanId,
-    ...parent.tracestate !== undefined && { tracestate: parent.tracestate },
-    ...parent.baggage !== undefined && { baggage: parent.baggage },
-    ...parent.traceFlags !== undefined && { traceFlags: parent.traceFlags }
-  };
-}
-var encoder = new TextEncoder;
-function boundedPropagationValue(value, maxBytes, maxMembers) {
-  if (typeof value !== "string")
-    return;
-  const trimmed = value.trim();
-  if (!trimmed || encoder.encode(trimmed).byteLength > maxBytes)
-    return;
-  for (const character of trimmed) {
-    const code = character.charCodeAt(0);
-    if (code <= 31 || code === 127)
-      return;
-  }
-  const members = trimmed.split(",");
-  if (members.length > maxMembers || members.some((member) => !member.trim())) {
-    return;
-  }
-  return trimmed;
-}
-function resolvePropagationContext(metadata, fallback) {
-  const traceparent = metadata?.traceparent;
-  const parsed = typeof traceparent === "string" ? parseTraceparent(traceparent) : null;
-  const trace = traceparent === undefined ? fallback ?? createTraceContext() : parsed ?? createTraceContext();
-  const tracestate = parsed ? boundedPropagationValue(metadata?.tracestate, 512, 32) : undefined;
-  const baggage = boundedPropagationValue(metadata?.baggage, 8192, 180);
-  return {
-    ...trace,
-    ...tracestate !== undefined && { tracestate },
-    ...baggage !== undefined && { baggage }
-  };
-}
-
-export { joinRoutePath, parseTrailingWildcard, isUnsafeKey, safeJsonParse, defineContract, mergeMeta, bytesToBase64Url, base64UrlToBytes, DEFAULT_CONTRACT_STREAM_FRAME_BYTES, ContractStreamFrameSchema, formatZodError, zodIssues, errorCode, recordedErrorMessage, normalizeError, validateDeclaredOutput, createTraceContext, parseTraceparent, formatTraceparent, resolveTraceContext, childSpan, resolvePropagationContext };
+export { joinRoutePath, parseTrailingWildcard, isUnsafeKey, safeJsonParse, defineContract, mergeMeta, bytesToBase64Url, base64UrlToBytes, DEFAULT_CONTRACT_STREAM_FRAME_BYTES, ContractStreamFrameSchema };

package/dist/{index-3eqrkqhg.js → index-vy5bjy07.js}

@@ -6,10 +6,13 @@ import {
 import {
   createBoundedSinkManager
 } from "./index-a59da114.js";
+import {
+  isAgentToolError
+} from "./index-3xnq72rz.js";
 import {
   ToolExecutionControlError,
   isToolExecutionControlError
-} from "./index-h8vn1jcr.js";
+} from "./index-qzbg46b6.js";
 import {
   AgentAssistantPlaceholderSchema,
   AgentJsonObjectSchema,
@@ -1678,12 +1681,13 @@ function createRunExecutor(dependencies) {
             });
             throw part.error;
           }
+          const output = isAgentToolError(part.error) ? jsonValue(part.error.output) : { message: "Tool execution failed" };
           parts.push(AgentMessagePartSchema.parse({
             type: "tool-result",
             callId: part.toolCallId,
             toolName: part.toolName,
             outcome: "error",
-            output: { message: "Tool execution failed" }
+            output
           }));
           await publish({
             type: "tool-status",
@@ -1694,7 +1698,7 @@ function createRunExecutor(dependencies) {
             callId: part.toolCallId,
             toolName: part.toolName,
             status: "failed",
-            output: { message: "Tool execution failed" },
+            output,
             emittedAt: now().toISOString()
           });
         } else if (part.type === "tool-output-denied") {

package/dist/{index-6wzr93cg.js → index-w0445741.js}

@@ -8,7 +8,7 @@ import {
 } from "./index-8eywc9zv.js";
 import {
   ShutdownOptionsSchema
-} from "./index-dk6e56g0.js";
+} from "./index-7wpgrqa8.js";
 import {
   AppError
 } from "./index-0w9abg87.js";

package/dist/{index-kxxyvkka.js → index-wc80at9n.js}

@@ -3,12 +3,12 @@ import {
   inputIsQuery,
   parseSSE,
   readBoundedUtf8Lines
-} from "./index-t4dpby6e.js";
+} from "./index-0yphgata.js";
 import {
   ContractStreamFrameSchema,
   DEFAULT_CONTRACT_STREAM_FRAME_BYTES,
   parseTrailingWildcard
-} from "./index-x18ndp4p.js";
+} from "./index-vj3vvpaa.js";
 import {
   isRecord,
   mapObject,

package/dist/{index-gqdfpv4n.js → index-y3w12g8d.js}

@@ -1,10 +1,10 @@
 import {
   collectTools,
   createToolRunner
-} from "./index-kg9xx84n.js";
+} from "./index-7rey2b8s.js";
 import {
   toolErrorFromResult
-} from "./index-h8vn1jcr.js";
+} from "./index-qzbg46b6.js";
 import {
   assertUniqueToolName
 } from "./index-22by16v6.js";

package/dist/node.js

@@ -5,7 +5,7 @@ import {
   createHandler,
   createSocketIOServer,
   createUnixClientTransport
-} from "./index-444747ww.js";
+} from "./index-4g196jmf.js";
 import {
   createImplement,
   createImplementRegistry,
@@ -14,8 +14,8 @@ import {
   createScopedImplementRegistry,
   implement,
   implementRegistry
-} from "./index-d1jsvkyk.js";
-import"./index-p1be103g.js";
+} from "./index-vbf2p6me.js";
+import"./index-ktsps64f.js";
 import"./index-7b188kmz.js";
 import {
   ShutdownOptionsSchema,
@@ -23,8 +23,9 @@ import {
   ShutdownStateSchema,
   ShutdownStatusSchema,
   createServerLifecycle
-} from "./index-dk6e56g0.js";
-import"./index-x18ndp4p.js";
+} from "./index-7wpgrqa8.js";
+import"./index-bd17ytae.js";
+import"./index-vj3vvpaa.js";
 import {
   AppError,
   appError,
@@ -151,6 +152,18 @@ async function serveNode(config) {
       return Promise.resolve();
     return new Promise((resolve) => socketDrainWaiters.add(resolve));
   };
+  const waitForUpgradedSocketDrain = () => {
+    if (upgradedSockets.size === 0)
+      return Promise.resolve();
+    return Promise.all([...upgradedSockets].map((activeSocket) => new Promise((resolve) => {
+      if (activeSocket.closed)
+        resolve();
+      else
+        activeSocket.once("close", () => resolve());
+    }))).then(() => {
+      return;
+    });
+  };
   const requireRuntime = () => {
     if (!runtime)
       throw new Error("[stitchkit] Node server lifecycle started before srvx");
@@ -166,6 +179,14 @@ async function serveNode(config) {
     pendingRequests: () => responses.size,
     pendingWebSockets: () => upgradedSockets.size,
     closeRealtime: () => socket?.close() ?? Promise.resolve(),
+    async terminateRealtime() {
+      const count = upgradedSockets.size;
+      const physicalClose = waitForUpgradedSocketDrain();
+      for (const activeSocket of [...upgradedSockets])
+        activeSocket.destroy();
+      await physicalClose;
+      return count;
+    },
     async stopGracefully() {
       const runtimeClose = socket ? Promise.resolve() : requireRuntime().close(false);
       requireNodeServer().closeIdleConnections?.();