@skein-js/agent-protocol 0.9.1 → 0.10.0

package/dist/index.d.ts

@@ -441,6 +441,15 @@ declare function createProtocolHandlers(service: ProtocolService): ProtocolHandl
  * `--n-jobs-per-worker` default, so a project moving off `langgraph dev` keeps its throughput.
  */
 declare const DEFAULT_RUN_CONCURRENCY = 10;
+/**
+ * How long `stop()` lets in-flight runs finish before aborting them. Chosen to leave room inside the
+ * tightest common stop window — a container platform typically sends SIGTERM and SIGKILLs ~10s later
+ * (Cloud Run's default; `docker stop`'s too) — for this drain *plus* the abort-and-settle step that
+ * follows it. Raise it via `SKEIN_SHUTDOWN_GRACE_MS` where the platform allows a longer window
+ * (Railway, Kubernetes' `terminationGracePeriodSeconds`, ECS `stopTimeout`); the host must also wait
+ * longer than this before forcing exit, or the abort path never runs. See docs/deploy.md.
+ */
+declare const DEFAULT_SHUTDOWN_GRACE_MS = 5000;
 interface RunWorkerOptions {
     /**
      * Max queued runs this worker executes at once. Default {@link DEFAULT_RUN_CONCURRENCY}.
@@ -450,7 +459,10 @@ interface RunWorkerOptions {
      * the worker. See docs/runs-and-redis.md.
      */
     maxConcurrency?: number;
-    /** How long `stop()` waits for in-flight runs before aborting them (ms). Default 5000. */
+    /**
+     * How long `stop()` waits for in-flight runs before aborting them (ms). Default
+     * {@link DEFAULT_SHUTDOWN_GRACE_MS}.
+     */
     shutdownGraceMs?: number;
 }
 interface RunWorker {
@@ -613,4 +625,4 @@ declare function toSseEvents(frames: AsyncIterable<RunFrame>, finalStatus: () =>
  */
 declare function parseAfterSeq(lastEventId: string | undefined): number;
 
-export { type AssistantService, type Clock, type CommandInput, type CompiledGraphFactory, type CreateAssistantInput, type CreateRunInput, type CreateThreadInput, DEFAULT_INVOKE_PREFIX, DEFAULT_RUN_CONCURRENCY, type DeleteAssistantOptions, type DrawGraphOptions, type GraphInvokeHandlerName, type GraphInvokeOptions, type GraphResolver, type GraphSchemas, type HistoryOptions, type HttpMethod, type Logger, type PatchThreadInput, type ProtocolContext, type ProtocolDeps, type ProtocolHandler, type ProtocolHandlers, type ProtocolRequest, type ProtocolResponse, type ProtocolRuntime, type ProtocolRuntimeOptions, type ProtocolService, type ResolvedGraph, type RouteBinding, type RouteMatch, type RouteMatcher, type RunService, type RunWorker, type RunWorkerOptions, SSE_HEADERS, SkeinBaseStore, type StartedStream, type StoreService, type SubgraphsOptions, type ThreadService, type ThreadStreamInput, type ThreadStreamService, buildProtocolService, copyThreadIdIntoBody, createContext, createGraphInvokeHandler, createProtocolHandlers, createProtocolRuntime, createProtocolService, createProtocolServiceFromContext, createRouteMatcher, createRunWorker, encodeFrame, encodeTerminal, foldThreadId, graphInvokeRoutes, matchSkeinRoute, parseAfterSeq, skeinRoutes, toSseEvents };
+export { type AssistantService, type Clock, type CommandInput, type CompiledGraphFactory, type CreateAssistantInput, type CreateRunInput, type CreateThreadInput, DEFAULT_INVOKE_PREFIX, DEFAULT_RUN_CONCURRENCY, DEFAULT_SHUTDOWN_GRACE_MS, type DeleteAssistantOptions, type DrawGraphOptions, type GraphInvokeHandlerName, type GraphInvokeOptions, type GraphResolver, type GraphSchemas, type HistoryOptions, type HttpMethod, type Logger, type PatchThreadInput, type ProtocolContext, type ProtocolDeps, type ProtocolHandler, type ProtocolHandlers, type ProtocolRequest, type ProtocolResponse, type ProtocolRuntime, type ProtocolRuntimeOptions, type ProtocolService, type ResolvedGraph, type RouteBinding, type RouteMatch, type RouteMatcher, type RunService, type RunWorker, type RunWorkerOptions, SSE_HEADERS, SkeinBaseStore, type StartedStream, type StoreService, type SubgraphsOptions, type ThreadService, type ThreadStreamInput, type ThreadStreamService, buildProtocolService, copyThreadIdIntoBody, createContext, createGraphInvokeHandler, createProtocolHandlers, createProtocolRuntime, createProtocolService, createProtocolServiceFromContext, createRouteMatcher, createRunWorker, encodeFrame, encodeTerminal, foldThreadId, graphInvokeRoutes, matchSkeinRoute, parseAfterSeq, skeinRoutes, toSseEvents };

package/dist/index.js

@@ -2015,7 +2015,14 @@ import {
   isTerminalRunStatus as isTerminalRunStatus6
 } from "@skein-js/core";
 var DEFAULT_RUN_CONCURRENCY = 10;
-var sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
+var DEFAULT_SHUTDOWN_GRACE_MS = 5e3;
+function cancellableDelay(ms) {
+  let timer;
+  const elapsed = new Promise((resolve) => {
+    timer = setTimeout(resolve, ms);
+  });
+  return { elapsed, cancel: () => clearTimeout(timer) };
+}
 function logRunLifecycle(logger, run, status, startedAt, endedAt) {
   logger.info(status === "success" ? "Background run succeeded" : `Background run ${status}`, {
     run_id: run.run_id,
@@ -2030,9 +2037,15 @@ function logRunLifecycle(logger, run, status, startedAt, endedAt) {
 function createRunWorker(ctx, options = {}) {
   const { deps, control } = ctx;
   const maxConcurrency = options.maxConcurrency ?? DEFAULT_RUN_CONCURRENCY;
-  const shutdownGraceMs = options.shutdownGraceMs ?? 5e3;
+  const shutdownGraceMs = options.shutdownGraceMs ?? DEFAULT_SHUTDOWN_GRACE_MS;
   const inFlight = /* @__PURE__ */ new Set();
-  const process = async (queued) => {
+  const running = /* @__PURE__ */ new Set();
+  const executeQueuedRun = (queued) => {
+    const task = executeOne(queued);
+    running.add(task);
+    return task.finally(() => running.delete(task));
+  };
+  const executeOne = async (queued) => {
     const run = await deps.store.runs.get(queued.run_id);
     if (!run || isTerminalRunStatus6(run.status)) {
       ctx.rollbackPlans.delete(queued.run_id);
@@ -2053,20 +2066,37 @@ function createRunWorker(ctx, options = {}) {
   return {
     start() {
       if (consumer) return;
-      consumer = deps.queue.consume(process, { concurrency: maxConcurrency });
+      consumer = deps.queue.consume(executeQueuedRun, { concurrency: maxConcurrency });
     },
     async stop() {
       if (!consumer) return;
       const active = consumer;
       consumer = void 0;
       let drained = false;
-      const graceful = active.close().then(() => {
-        drained = true;
-      });
-      await Promise.race([graceful, sleep(shutdownGraceMs)]);
+      let drainFailure;
+      const graceful = active.close().then(
+        () => {
+          drained = true;
+        },
+        (error) => {
+          drainFailure = error;
+        }
+      );
+      const deadline = cancellableDelay(shutdownGraceMs);
+      try {
+        await Promise.race([graceful, deadline.elapsed]);
+      } finally {
+        deadline.cancel();
+      }
       if (!drained) {
         for (const runId of inFlight) control.abort(runId, "cancel");
         await graceful;
+        await Promise.allSettled([...running]);
+      }
+      if (drainFailure !== void 0) {
+        deps.logger.warn("Run queue did not close cleanly during shutdown", {
+          error: drainFailure instanceof Error ? drainFailure.message : String(drainFailure)
+        });
       }
     }
   };
@@ -2317,6 +2347,7 @@ function createGraphInvokeHandler(deps, options = {}) {
 export {
   DEFAULT_INVOKE_PREFIX,
   DEFAULT_RUN_CONCURRENCY,
+  DEFAULT_SHUTDOWN_GRACE_MS,
   SSE_HEADERS,
   SkeinBaseStore,
   buildProtocolService,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skein-js/agent-protocol",
-  "version": "0.9.1",
+  "version": "0.10.0",
   "description": "Framework-agnostic Agent Protocol engine for LangGraph.js — run engine, handlers, and SSE, driven entirely by injected dependencies.",
   "license": "Apache-2.0",
   "author": "Maina Wycliffe <wmmaina7@gmail.com>",
@@ -30,7 +30,8 @@
   "exports": {
     ".": {
       "types": "./dist/index.d.ts",
-      "import": "./dist/index.js"
+      "import": "./dist/index.js",
+      "default": "./dist/index.js"
     }
   },
   "files": [
@@ -44,14 +45,14 @@
   },
   "dependencies": {
     "zod": "^3.25.76",
-    "@skein-js/core": "0.9.1"
+    "@skein-js/core": "0.10.0"
   },
   "peerDependencies": {
     "@langchain/langgraph": "^1.4.0",
     "@langchain/langgraph-sdk": "^1.9.0"
   },
   "devDependencies": {
-    "@skein-js/storage-memory": "0.9.1"
+    "@skein-js/storage-memory": "0.10.0"
   },
   "publishConfig": {
     "access": "public"