@dudousxd/nestjs-agent-core 0.8.0 → 0.10.0

package/dist/index.cjs

@@ -45,6 +45,8 @@ __export(index_exports, {
   AGENT_TOOL_REGISTRY: () => AGENT_TOOL_REGISTRY,
   AgentRegistry: () => AgentRegistry,
   AgentStreamError: () => AgentStreamError,
+  DEFAULT_TOOL_TRANSIENT_RETRY_ATTEMPTS: () => DEFAULT_TOOL_TRANSIENT_RETRY_ATTEMPTS,
+  DEFAULT_TOOL_TRANSIENT_RETRY_BACKOFF_MS: () => DEFAULT_TOOL_TRANSIENT_RETRY_BACKOFF_MS,
   DefaultRolesPolicy: () => DefaultRolesPolicy,
   QuotaExceededError: () => QuotaExceededError,
   ToolForbiddenError: () => ToolForbiddenError,
@@ -61,6 +63,8 @@ __export(index_exports, {
   estimateCost: () => estimateCost,
   filterToolsByAllowList: () => filterToolsByAllowList,
   filterToolsByRole: () => filterToolsByRole,
+  invokeWithTransientRetry: () => invokeWithTransientRetry,
+  isTransientToolError: () => isTransientToolError,
   publishAgentDelegated: () => publishAgentDelegated,
   publishAgentMessage: () => publishAgentMessage,
   publishAgentQuotaExceeded: () => publishAgentQuotaExceeded,
@@ -69,6 +73,8 @@ __export(index_exports, {
   publishAgentRunFinished: () => publishAgentRunFinished,
   publishAgentRunStarted: () => publishAgentRunStarted,
   publishAgentToolCall: () => publishAgentToolCall,
+  publishAgentToolRetry: () => publishAgentToolRetry,
+  resolveToolTransientRetryNumbers: () => resolveToolTransientRetryNumbers,
   runAgentLoop: () => runAgentLoop,
   seedModelPrices: () => seedModelPrices,
   traceLlmTurn: () => traceLlmTurn,
@@ -439,6 +445,10 @@ function publishAgentRetrieved(payload) {
   (0, import_nestjs_diagnostics.emit)("agent", "retrieved", payload);
 }
 __name(publishAgentRetrieved, "publishAgentRetrieved");
+function publishAgentToolRetry(payload) {
+  (0, import_nestjs_diagnostics.emit)("agent", "tool.retry", payload);
+}
+__name(publishAgentToolRetry, "publishAgentToolRetry");
 var AGENT_SPAN_EVENTS = [
   "llm.turn",
   "tool.execution",
@@ -453,13 +463,96 @@ var AGENT_DIAGNOSTIC_EVENTS = [
   "run.finished",
   "run.failed",
   "delegated",
-  "retrieved"
+  "retrieved",
+  "tool.retry"
 ];
 function agentDiagnosticKey(event) {
   return `agent:${event}`;
 }
 __name(agentDiagnosticKey, "agentDiagnosticKey");
 
+// src/tool-retry.ts
+function hasTransientShape(error) {
+  if (typeof error !== "object" || error === null) {
+    return false;
+  }
+  const code = "code" in error ? error.code : void 0;
+  const errno = "errno" in error ? error.errno : void 0;
+  const sqlState = "sqlState" in error ? error.sqlState : void 0;
+  if (code === 1213 || code === 1205 || errno === 1213 || errno === 1205) {
+    return true;
+  }
+  if (code === "ER_LOCK_DEADLOCK" || code === "ER_LOCK_WAIT_TIMEOUT") {
+    return true;
+  }
+  if (code === "40001" || code === "40P01" || sqlState === "40001" || sqlState === "40P01") {
+    return true;
+  }
+  if (code === "SQLITE_BUSY") {
+    return true;
+  }
+  const message = "message" in error ? error.message : void 0;
+  return typeof message === "string" && /deadlock|lock wait timeout|serialization failure/i.test(message);
+}
+__name(hasTransientShape, "hasTransientShape");
+function isTransientToolError(error) {
+  if (hasTransientShape(error)) {
+    return true;
+  }
+  if (typeof error === "object" && error !== null && "cause" in error) {
+    const cause = error.cause;
+    if (cause !== void 0 && cause !== error && hasTransientShape(cause)) {
+      return true;
+    }
+  }
+  return false;
+}
+__name(isTransientToolError, "isTransientToolError");
+var DEFAULT_TOOL_TRANSIENT_RETRY_ATTEMPTS = 2;
+var DEFAULT_TOOL_TRANSIENT_RETRY_BACKOFF_MS = 150;
+function resolveToolTransientRetryNumbers(setting) {
+  if (setting === false) {
+    return false;
+  }
+  return {
+    attempts: setting?.attempts ?? DEFAULT_TOOL_TRANSIENT_RETRY_ATTEMPTS,
+    backoffMs: setting?.backoffMs ?? DEFAULT_TOOL_TRANSIENT_RETRY_BACKOFF_MS
+  };
+}
+__name(resolveToolTransientRetryNumbers, "resolveToolTransientRetryNumbers");
+function delay(ms) {
+  return new Promise((resolve) => {
+    setTimeout(resolve, ms);
+  });
+}
+__name(delay, "delay");
+async function invokeWithTransientRetry(fn, setting, options) {
+  if (setting === false) {
+    return fn();
+  }
+  const attempts = setting.attempts ?? DEFAULT_TOOL_TRANSIENT_RETRY_ATTEMPTS;
+  const backoffMs = setting.backoffMs ?? DEFAULT_TOOL_TRANSIENT_RETRY_BACKOFF_MS;
+  const classify = setting.classify ?? isTransientToolError;
+  let attempt = 1;
+  for (; ; ) {
+    try {
+      return await fn();
+    } catch (error) {
+      if (options?.isControlFlowError?.(error) === true) {
+        throw error;
+      }
+      const attemptsRemain = attempt < attempts;
+      if (!attemptsRemain || !classify(error)) {
+        throw error;
+      }
+      options?.onRetry?.(attempt, error);
+      await delay(backoffMs * attempt);
+      attempt += 1;
+    }
+  }
+}
+__name(invokeWithTransientRetry, "invokeWithTransientRetry");
+
 // src/agent-loop.ts
 function resolveCostUsd(usage, reportedCostUsd, price) {
   if (reportedCostUsd !== void 0) {
@@ -861,7 +954,8 @@ ${buildContextBlock(passages)}`;
           input: {
             query: input.userText
           },
-          status: "auto_executed"
+          status: "auto_executed",
+          runId: hooks.runId
         });
         await deps.store.updateToolCall({
           toolCallId,
@@ -910,7 +1004,8 @@ ${buildContextBlock(passages)}`;
           toolName: call.name,
           toolType: "read",
           input: call.input,
-          status: "auto_executed"
+          status: "auto_executed",
+          runId: hooks.runId
         }));
         const overDepth = (input.delegationDepth ?? 0) >= MAX_DELEGATION_DEPTH;
         publishAgentDelegated({
@@ -952,7 +1047,8 @@ ${buildContextBlock(passages)}`;
           toolName: call.name,
           toolType: "action",
           input: call.input,
-          status: "pending_approval"
+          status: "pending_approval",
+          runId: hooks.runId
         }));
         const decision = await hooks.awaitApproval(call, ctx);
         deciderRef = decision.executedByRef ?? input.actor.id;
@@ -989,7 +1085,8 @@ ${buildContextBlock(passages)}`;
           toolName: call.name,
           toolType: "read",
           input: call.input,
-          status: "auto_executed"
+          status: "auto_executed",
+          runId: hooks.runId
         }));
       }
       const startedAt2 = Date.now();
@@ -1014,7 +1111,10 @@ ${buildContextBlock(passages)}`;
             ctx: stepCtx,
             ...deps.toolTimeoutMs !== void 0 ? {
               timeoutMs: deps.toolTimeoutMs
-            } : {}
+            } : {},
+            // Numeric-only: the handler applies withToolTimeout AND its own local `classify` — see
+            // ToolStepEnvelope.transientRetry.
+            transientRetry: resolveToolTransientRetryNumbers(deps.toolTransientRetry)
           };
           output = await hooks.dispatchTool(call, envelope);
         } else {
@@ -1022,7 +1122,19 @@ ${buildContextBlock(passages)}`;
             toolCallId: call.id,
             toolName: call.name,
             toolType
-          }, () => deps.registry.invoke(call.name, call.input, ctx, deps.rolesPolicy)));
+          }, () => invokeWithTransientRetry(() => deps.registry.invoke(call.name, call.input, ctx, deps.rolesPolicy), deps.toolTransientRetry ?? {}, {
+            ...hooks.isControlFlowError !== void 0 ? {
+              isControlFlowError: hooks.isControlFlowError
+            } : {},
+            onRetry: /* @__PURE__ */ __name((attempt, retryError) => {
+              publishAgentToolRetry({
+                toolName: call.name,
+                toolCallId: call.id,
+                attempt,
+                message: retryError instanceof Error ? retryError.message : String(retryError)
+              });
+            }, "onRetry")
+          })));
           output = deps.toolTimeoutMs !== void 0 ? await withToolTimeout(invocation, deps.toolTimeoutMs, call.name) : await invocation;
         }
         const executionMs = Date.now() - startedAt2;
@@ -1145,6 +1257,8 @@ __name(runAgentLoop, "runAgentLoop");
   AGENT_TOOL_REGISTRY,
   AgentRegistry,
   AgentStreamError,
+  DEFAULT_TOOL_TRANSIENT_RETRY_ATTEMPTS,
+  DEFAULT_TOOL_TRANSIENT_RETRY_BACKOFF_MS,
   DefaultRolesPolicy,
   QuotaExceededError,
   ToolForbiddenError,
@@ -1161,6 +1275,8 @@ __name(runAgentLoop, "runAgentLoop");
   estimateCost,
   filterToolsByAllowList,
   filterToolsByRole,
+  invokeWithTransientRetry,
+  isTransientToolError,
   publishAgentDelegated,
   publishAgentMessage,
   publishAgentQuotaExceeded,
@@ -1169,6 +1285,8 @@ __name(runAgentLoop, "runAgentLoop");
   publishAgentRunFinished,
   publishAgentRunStarted,
   publishAgentToolCall,
+  publishAgentToolRetry,
+  resolveToolTransientRetryNumbers,
   runAgentLoop,
   seedModelPrices,
   traceLlmTurn,