@agent-assistant/core 0.2.18 → 0.2.20

package/dist/core.js

@@ -1,6 +1,6 @@
-const DEFAULT_HANDLER_TIMEOUT_MS = 30_000;
+const DEFAULT_HANDLER_TIMEOUT_MS = 5 * 60_000;
 const DEFAULT_MAX_CONCURRENT_HANDLERS = 10;
-const STOP_DRAIN_TIMEOUT_MS = 30_000;
+const DRAIN_MARGIN_MS = 5_000;
 export class AssistantDefinitionError extends Error {
     constructor(message) {
         super(message);
@@ -206,14 +206,23 @@ export function createAssistant(definition, adapters) {
             return;
         }
         drainWaiter ??= createDeferred();
-        await Promise.race([
-            drainWaiter.promise,
-            new Promise((_, reject) => {
-                setTimeout(() => {
-                    reject(new Error(`Timed out waiting ${STOP_DRAIN_TIMEOUT_MS}ms for in-flight handlers to drain`));
-                }, STOP_DRAIN_TIMEOUT_MS);
-            }),
-        ]);
+        const drainTimeoutMs = constraints.handlerTimeoutMs + DRAIN_MARGIN_MS;
+        let timeoutHandle;
+        try {
+            await Promise.race([
+                drainWaiter.promise,
+                new Promise((_, reject) => {
+                    timeoutHandle = setTimeout(() => {
+                        reject(new Error(`Timed out waiting ${drainTimeoutMs}ms for in-flight handlers to drain`));
+                    }, drainTimeoutMs);
+                }),
+            ]);
+        }
+        finally {
+            if (timeoutHandle) {
+                clearTimeout(timeoutHandle);
+            }
+        }
     }
     function runNext() {
         while (lifecycleState === 'started' &&

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agent-assistant/core",
-  "version": "0.2.18",
+  "version": "0.2.20",
   "description": "Assistant definition, lifecycle, and runtime composition for Agent Assistant SDK",
   "type": "module",
   "main": "dist/index.js",