@dudousxd/nestjs-agent-core 0.9.0 → 0.11.0

package/dist/index.cjs

@@ -45,8 +45,11 @@ __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,
+  THREAD_DETAIL_CONTENT_CHARS: () => THREAD_DETAIL_CONTENT_CHARS,
   ToolForbiddenError: () => ToolForbiddenError,
   ToolInputInvalidError: () => ToolInputInvalidError,
   ToolNotFoundError: () => ToolNotFoundError,
@@ -61,6 +64,8 @@ __export(index_exports, {
   estimateCost: () => estimateCost,
   filterToolsByAllowList: () => filterToolsByAllowList,
   filterToolsByRole: () => filterToolsByRole,
+  invokeWithTransientRetry: () => invokeWithTransientRetry,
+  isTransientToolError: () => isTransientToolError,
   publishAgentDelegated: () => publishAgentDelegated,
   publishAgentMessage: () => publishAgentMessage,
   publishAgentQuotaExceeded: () => publishAgentQuotaExceeded,
@@ -69,10 +74,14 @@ __export(index_exports, {
   publishAgentRunFinished: () => publishAgentRunFinished,
   publishAgentRunStarted: () => publishAgentRunStarted,
   publishAgentToolCall: () => publishAgentToolCall,
+  publishAgentToolRetry: () => publishAgentToolRetry,
+  resolveToolTransientRetryNumbers: () => resolveToolTransientRetryNumbers,
+  rollupThreadUsage: () => rollupThreadUsage,
   runAgentLoop: () => runAgentLoop,
   seedModelPrices: () => seedModelPrices,
   traceLlmTurn: () => traceLlmTurn,
   traceToolExecution: () => traceToolExecution,
+  truncateDetailContent: () => truncateDetailContent,
   withToolTimeout: () => withToolTimeout
 });
 module.exports = __toCommonJS(index_exports);
@@ -128,6 +137,22 @@ async function seedModelPrices(store, prices) {
 }
 __name(seedModelPrices, "seedModelPrices");
 
+// src/spi/governance-queries.ts
+var THREAD_DETAIL_CONTENT_CHARS = 2e3;
+function truncateDetailContent(content) {
+  if (content.length <= THREAD_DETAIL_CONTENT_CHARS) {
+    return {
+      content,
+      truncated: false
+    };
+  }
+  return {
+    content: content.slice(0, THREAD_DETAIL_CONTENT_CHARS),
+    truncated: true
+  };
+}
+__name(truncateDetailContent, "truncateDetailContent");
+
 // src/governance/compute.ts
 function estimateCost(usage, price) {
   if (price === void 0) {
@@ -233,6 +258,24 @@ function bucketByThread(rows, prices, threads, options) {
   return result.slice(0, options.limit);
 }
 __name(bucketByThread, "bucketByThread");
+function rollupThreadUsage(rows, prices) {
+  const rollup = {
+    requests: 0,
+    inputTokens: 0,
+    outputTokens: 0,
+    totalTokens: 0,
+    costUsd: 0
+  };
+  for (const row of rows) {
+    rollup.requests += 1;
+    rollup.inputTokens += row.inputTokens;
+    rollup.outputTokens += row.outputTokens;
+    rollup.totalTokens += row.inputTokens + row.outputTokens;
+    rollup.costUsd += rowCost(row, prices);
+  }
+  return rollup;
+}
+__name(rollupThreadUsage, "rollupThreadUsage");
 function bucketUsageTrend(rows, prices) {
   const byDay = /* @__PURE__ */ new Map();
   for (const row of rows) {
@@ -439,6 +482,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 +500,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) {
@@ -1018,7 +1148,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 {
@@ -1026,7 +1159,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;
@@ -1149,8 +1294,11 @@ __name(runAgentLoop, "runAgentLoop");
   AGENT_TOOL_REGISTRY,
   AgentRegistry,
   AgentStreamError,
+  DEFAULT_TOOL_TRANSIENT_RETRY_ATTEMPTS,
+  DEFAULT_TOOL_TRANSIENT_RETRY_BACKOFF_MS,
   DefaultRolesPolicy,
   QuotaExceededError,
+  THREAD_DETAIL_CONTENT_CHARS,
   ToolForbiddenError,
   ToolInputInvalidError,
   ToolNotFoundError,
@@ -1165,6 +1313,8 @@ __name(runAgentLoop, "runAgentLoop");
   estimateCost,
   filterToolsByAllowList,
   filterToolsByRole,
+  invokeWithTransientRetry,
+  isTransientToolError,
   publishAgentDelegated,
   publishAgentMessage,
   publishAgentQuotaExceeded,
@@ -1173,10 +1323,14 @@ __name(runAgentLoop, "runAgentLoop");
   publishAgentRunFinished,
   publishAgentRunStarted,
   publishAgentToolCall,
+  publishAgentToolRetry,
+  resolveToolTransientRetryNumbers,
+  rollupThreadUsage,
   runAgentLoop,
   seedModelPrices,
   traceLlmTurn,
   traceToolExecution,
+  truncateDetailContent,
   withToolTimeout
 });
 //# sourceMappingURL=index.cjs.map