@agent-assistant/core 0.2.19 → 0.2.21

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' &&
@@ -227,6 +236,14 @@ export function createAssistant(definition, adapters) {
             void executeDispatch(nextDispatch);
         }
     }
+    function reportError(error, message) {
+        const userHandler = frozenDefinition.hooks?.onError;
+        if (typeof userHandler === 'function') {
+            userHandler(error, message);
+            return;
+        }
+        console.error(`[${frozenDefinition.id ?? 'agent'}] unhandled capability handler error (capability=${message.capability}, id=${message.id}):`, error);
+    }
     async function executeDispatch(dispatchJob) {
         const { message, resolve, reject } = dispatchJob;
         try {
@@ -237,7 +254,7 @@ export function createAssistant(definition, adapters) {
             }
             const handler = capabilityMap.get(message.capability);
             if (!handler) {
-                frozenDefinition.hooks?.onError?.(new Error(`No capability registered for '${message.capability}'`), message);
+                reportError(new Error(`No capability registered for '${message.capability}'`), message);
                 resolve();
                 return;
             }
@@ -253,13 +270,13 @@ export function createAssistant(definition, adapters) {
                 await withTimeout(handler, message, context, constraints.handlerTimeoutMs);
             }
             catch (error) {
-                frozenDefinition.hooks?.onError?.(error instanceof Error ? error : new Error(String(error)), message);
+                reportError(error instanceof Error ? error : new Error(String(error)), message);
             }
             resolve();
         }
         catch (error) {
             const wrappedError = error instanceof Error ? error : new Error(String(error));
-            frozenDefinition.hooks?.onError?.(wrappedError, message);
+            reportError(wrappedError, message);
             reject(wrappedError);
         }
         finally {

package/package.json

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