@dudousxd/nestjs-agent 1.37.0 → 1.39.0

package/dist/durable/index.js

@@ -69,7 +69,7 @@ var init_kit = __esm({
 });
 
 // src/durable/agent-run.workflow.ts
-import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY2, AGENT_STORE, RUN_ENDED_BEFORE_TOOL_CALL, RunCancelledError, agentFailureCode, agentRunContext, encodeStreamEvent as encodeStreamEvent2, isReplayIntegrityError as isReplayIntegrityError2, publishAgentRunFailed, releaseThreadRun, runAgentLoop, runDurationMs, settleAll, settleDeadRun, settleUnsettledDelegation, streamFailure } from "@dudousxd/nestjs-agent-core";
+import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY2, AGENT_STORE, RUN_ENDED_BEFORE_TOOL_CALL, RunCancelledError, agentFailureCode, agentRunContext, encodeStreamEvent as encodeStreamEvent2, isReplayIntegrityError as isReplayIntegrityError2, publishAgentRunFailed, releaseThreadRun, runAgentLoop, runDurationMs, runEndTimings, settleAll, settleDeadRun, settleUnsettledDelegation, streamFailure } from "@dudousxd/nestjs-agent-core";
 import { RUN_GATEWAY, Workflow, WorkflowService } from "@dudousxd/nestjs-durable";
 import { isWorkflowControlFlowSignal } from "@dudousxd/nestjs-durable-core";
 import { Inject as Inject2, Injectable as Injectable2, Logger, Optional } from "@nestjs/common";
@@ -613,6 +613,18 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
     ];
     const hooks = {
       runId: ctx.runId,
+      // The runtime's own enqueue stamp: the run row's `createdAt` is written when the run is
+      // started (a send, a parent's `ctx.child`, the queue's `ctx.startChild`), so now − createdAt,
+      // read on the body's first execution (inside `persist:run:start`), is how long it waited for a
+      // worker. The clocks are the enqueuing process's and the worker's.
+      queueMs: /* @__PURE__ */ __name(async () => {
+        const runs = this.runs;
+        const run = typeof runs?.getRun === "function" ? await runs.getRun(ctx.runId) : (await runs?.getRunDetail(ctx.runId))?.run;
+        const createdAt = run?.createdAt;
+        if (createdAt === void 0) return void 0;
+        const at = new Date(createdAt).getTime();
+        return Number.isFinite(at) ? Date.now() - at : void 0;
+      }, "queueMs"),
       // A child forwards into the top-level sink but must not end/fail it (the top-level run owns it).
       // A thread's own turn hands the thread to its queue just before the loop ends the stream, so
       // the reader learns what runs next from the `queue` frame, before the end.
@@ -783,6 +795,7 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
       const timed = durationMs !== void 0 ? {
         durationMs
       } : {};
+      const measured = runEndTimings(error);
       if (isReplayIntegrityError2(error)) {
         const detail = error.message;
         const code2 = agentFailureCode(error);
@@ -793,6 +806,7 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
           code: code2,
           message: detail,
           ...timed,
+          ...measured,
           context: agentRunContext(input)
         });
         await settleDeadRun(this.store, {
@@ -816,6 +830,7 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
         code,
         message,
         ...timed,
+        ...measured,
         context: agentRunContext(input)
       });
       await stepOf(ctx)("persist:run:fail", async () => {
@@ -1458,7 +1473,7 @@ AgentDurableModule = _ts_decorate4([
 
 // src/agent.module.ts
 import { AGENT_ACTOR_RESOLVER as AGENT_ACTOR_RESOLVER14, AGENT_APPROVAL_PORT, AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY9, AGENT_DURABLE_RUNNER as AGENT_DURABLE_RUNNER2, AGENT_MEMORY as AGENT_MEMORY3, AGENT_MODEL as AGENT_MODEL3, AGENT_MODEL_CATALOG as AGENT_MODEL_CATALOG3, AGENT_OPTIONS as AGENT_OPTIONS10, AGENT_PROMPT_CONTRIBUTORS as AGENT_PROMPT_CONTRIBUTORS3, AGENT_QUOTA_PROVIDER as AGENT_QUOTA_PROVIDER3, AGENT_REGISTRY as AGENT_REGISTRY6, AGENT_ROLES_POLICY as AGENT_ROLES_POLICY2, AGENT_RUNNER as AGENT_RUNNER2, AGENT_SINK as AGENT_SINK4, AGENT_SKILLS as AGENT_SKILLS3, AGENT_SKILL_SOURCES as AGENT_SKILL_SOURCES2, AGENT_STORE as AGENT_STORE9, AGENT_TOOL_REGISTRY as AGENT_TOOL_REGISTRY3, AgentRegistry as AgentRegistry6, ClosedRolesPolicy, DefaultRolesPolicy, InMemoryAgentStore, ToolRegistry as ToolRegistry3, personalAgentGate, personalAgentScopes } from "@dudousxd/nestjs-agent-core";
-import { Global as Global2, Logger as Logger10, Module as Module3 } from "@nestjs/common";
+import { Global as Global2, Logger as Logger11, Module as Module3 } from "@nestjs/common";
 import { DiscoveryModule, ModulesContainer, RouterModule } from "@nestjs/core";
 
 // src/agent-deps.factory.ts
@@ -2179,8 +2194,8 @@ AgentDepsFactory2 = _ts_decorate6([
 var AGENT_SERVICE = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:agent-service");
 
 // src/agent.service.ts
-import { AGENT_ATTACHMENT_STAGING as AGENT_ATTACHMENT_STAGING2, AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY4, AGENT_MODEL_CATALOG, AGENT_OPTIONS as AGENT_OPTIONS4, AGENT_QUOTA_PROVIDER as AGENT_QUOTA_PROVIDER2, AGENT_RUNNER, AGENT_STORE as AGENT_STORE6, DEFAULT_TEXT_ACTION_PROPOSAL_VOCABULARY as DEFAULT_TEXT_ACTION_PROPOSAL_VOCABULARY2, REQUESTER_APPROVER, findCatalogModel, mayDecideApproval as mayDecideApproval2, readElicitationQuestions, settleDeadRun as settleDeadRun2, textActionProposalReply as textActionProposalReply2, validateElicitationAnswer } from "@dudousxd/nestjs-agent-core";
-import { BadRequestException as BadRequestException2, ConflictException as ConflictException2, ForbiddenException as ForbiddenException2, GoneException, HttpException, HttpStatus, Inject as Inject8, Injectable as Injectable7, NotFoundException as NotFoundException2, NotImplementedException as NotImplementedException3, Optional as Optional5, PayloadTooLargeException, UnsupportedMediaTypeException } from "@nestjs/common";
+import { AGENT_ATTACHMENT_STAGING as AGENT_ATTACHMENT_STAGING2, AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY4, AGENT_MODEL_CATALOG, AGENT_OPTIONS as AGENT_OPTIONS4, AGENT_QUOTA_PROVIDER as AGENT_QUOTA_PROVIDER2, AGENT_RUNNER, AGENT_STORE as AGENT_STORE6, DEFAULT_TEXT_ACTION_PROPOSAL_VOCABULARY as DEFAULT_TEXT_ACTION_PROPOSAL_VOCABULARY2, REQUESTER_APPROVER, findCatalogModel, mayDecideApproval as mayDecideApproval2, readElicitationQuestions, recordAiUsage, settleDeadRun as settleDeadRun2, textActionProposalReply as textActionProposalReply2, validateElicitationAnswer } from "@dudousxd/nestjs-agent-core";
+import { BadRequestException as BadRequestException2, ConflictException as ConflictException2, ForbiddenException as ForbiddenException2, GoneException, HttpException, HttpStatus, Inject as Inject8, Injectable as Injectable7, Logger as Logger4, NotFoundException as NotFoundException2, NotImplementedException as NotImplementedException3, Optional as Optional5, PayloadTooLargeException, UnsupportedMediaTypeException } from "@nestjs/common";
 
 // src/attachment-limits.ts
 var DEFAULT_MAX_ATTACHMENT_BYTES = 20 * 1024 * 1024;
@@ -2981,6 +2996,7 @@ function _ts_param7(paramIndex, decorator) {
   };
 }
 __name(_ts_param7, "_ts_param");
+var usageLogger = new Logger4("AgentService");
 var NO_USER_MESSAGE_CODE = "no_user_message";
 function sendableParams(params) {
   const message = params.message;
@@ -4224,6 +4240,31 @@ var AgentService = class {
       throw new ForbiddenException2("run belongs to another actor");
     }
   }
+  /**
+  * Record AI spend that did not come from an agent turn — a transcription, a TTS render, a batch of
+  * embeddings, a direct OpenAI/OpenRouter SDK call — on the same ledger the agent's turns write, so
+  * cost, quota and the dashboard see one figure.
+  *
+  * Never throws into the caller's flow by default: a failed write or an invalid input is logged and
+  * answered with `{ recorded: false, error }`. Pass `{ throwOnError: true }` where losing the row
+  * must fail the caller.
+  *
+  * Counted against `actorRef` like an agent turn: the ledger is the budget (`LedgerQuotaProvider`),
+  * so its tokens count in the token windows (a row with no tokens adds 0 there) and its `costUsd` in
+  * the USD windows. A `QuotaStore` counter is bumped only where loop deps carry one (`AgentModule`
+  * wires none). `actorRef: null` is platform cost, recorded under `PLATFORM_ACTOR_REF` and counted
+  * against nobody.
+  */
+  recordUsage(input, options = {}) {
+    return recordAiUsage({
+      store: this.store,
+      quota: this.deps.forAgent().quota,
+      day: utcDay(),
+      input,
+      options,
+      warn: /* @__PURE__ */ __name((message) => usageLogger.warn(message), "warn")
+    });
+  }
 };
 AgentService = _ts_decorate9([
   Injectable7(),
@@ -6130,7 +6171,7 @@ ToolsController = _ts_decorate23([
 
 // src/discovery/agent-discovery.service.ts
 import { AGENT_PROMPT_CONTRIBUTORS as AGENT_PROMPT_CONTRIBUTORS2, AGENT_REGISTRY as AGENT_REGISTRY4, AgentRegistry as AgentRegistry5 } from "@dudousxd/nestjs-agent-core";
-import { Inject as Inject22, Injectable as Injectable9, Logger as Logger4 } from "@nestjs/common";
+import { Inject as Inject22, Injectable as Injectable9, Logger as Logger5 } from "@nestjs/common";
 import { DiscoveryService, MetadataScanner } from "@nestjs/core";
 
 // src/decorator/agent.decorator.ts
@@ -6181,7 +6222,7 @@ var AgentDiscoveryService = class _AgentDiscoveryService {
   scanner;
   registry;
   contributors;
-  logger = new Logger4(_AgentDiscoveryService.name);
+  logger = new Logger5(_AgentDiscoveryService.name);
   constructor(discovery, scanner, registry, contributors) {
     this.discovery = discovery;
     this.scanner = scanner;
@@ -6363,7 +6404,7 @@ AgentDiscoveryService = _ts_decorate24([
 
 // src/discovery/ai-tool-discovery.service.ts
 import { AGENT_OPTIONS as AGENT_OPTIONS7, AGENT_REGISTRY as AGENT_REGISTRY5, AGENT_TOOL_REGISTRY as AGENT_TOOL_REGISTRY2, normalizeDelegation as normalizeDelegation2 } from "@dudousxd/nestjs-agent-core";
-import { Inject as Inject23, Injectable as Injectable10, Logger as Logger5 } from "@nestjs/common";
+import { Inject as Inject23, Injectable as Injectable10, Logger as Logger6 } from "@nestjs/common";
 import { DiscoveryService as DiscoveryService2 } from "@nestjs/core";
 import { z } from "zod";
 
@@ -6410,7 +6451,7 @@ var AiToolDiscoveryService = class _AiToolDiscoveryService {
   registry;
   agents;
   options;
-  logger = new Logger5(_AiToolDiscoveryService.name);
+  logger = new Logger6(_AiToolDiscoveryService.name);
   constructor(discovery, registry, agents, options) {
     this.discovery = discovery;
     this.registry = registry;
@@ -6576,7 +6617,7 @@ AiToolDiscoveryService = _ts_decorate25([
 
 // src/discovery/skill-discovery.service.ts
 import { AGENT_SKILL_SOURCES } from "@dudousxd/nestjs-agent-core";
-import { Inject as Inject24, Injectable as Injectable11, Logger as Logger6 } from "@nestjs/common";
+import { Inject as Inject24, Injectable as Injectable11, Logger as Logger7 } from "@nestjs/common";
 import { DiscoveryService as DiscoveryService3 } from "@nestjs/core";
 
 // src/decorator/skill.decorator.ts
@@ -6625,7 +6666,7 @@ var SkillDiscoveryService = class _SkillDiscoveryService {
   }
   discovery;
   declared;
-  logger = new Logger6(_SkillDiscoveryService.name);
+  logger = new Logger7(_SkillDiscoveryService.name);
   constructor(discovery, declared) {
     this.discovery = discovery;
     this.declared = declared;
@@ -6750,7 +6791,7 @@ var LedgerQuotaProvider = class {
 
 // src/pricing-boot.service.ts
 import { AGENT_MODEL as AGENT_MODEL2, AGENT_OPTIONS as AGENT_OPTIONS8, AGENT_PRICING_STORE as AGENT_PRICING_STORE2, ensureModelPricing } from "@dudousxd/nestjs-agent-core";
-import { Inject as Inject25, Injectable as Injectable12, Logger as Logger7, Optional as Optional11 } from "@nestjs/common";
+import { Inject as Inject25, Injectable as Injectable12, Logger as Logger8, Optional as Optional11 } from "@nestjs/common";
 function _ts_decorate27(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function") {
@@ -6777,7 +6818,7 @@ function _ts_param25(paramIndex, decorator) {
   };
 }
 __name(_ts_param25, "_ts_param");
-var logger = new Logger7("AgentPricing");
+var logger = new Logger8("AgentPricing");
 var PricingBootService = class {
   static {
     __name(this, "PricingBootService");
@@ -6840,7 +6881,7 @@ PricingBootService = _ts_decorate27([
 // src/proposals/action-proposal-worker.service.ts
 import { randomUUID as randomUUID2 } from "crypto";
 import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY7, AGENT_OPTIONS as AGENT_OPTIONS9, AGENT_STORE as AGENT_STORE7, ActionProposalExecutor, ActionProposalWorker, assertIndependentActionCapabilities as assertIndependentActionCapabilities2 } from "@dudousxd/nestjs-agent-core";
-import { Inject as Inject26, Injectable as Injectable13, Logger as Logger8 } from "@nestjs/common";
+import { Inject as Inject26, Injectable as Injectable13, Logger as Logger9 } from "@nestjs/common";
 function _ts_decorate28(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function") {
@@ -6875,7 +6916,7 @@ var ActionProposalWorkerService = class _ActionProposalWorkerService {
   options;
   deps;
   worker;
-  logger = new Logger8(_ActionProposalWorkerService.name);
+  logger = new Logger9(_ActionProposalWorkerService.name);
   settledListeners = /* @__PURE__ */ new Set();
   admittedListeners = /* @__PURE__ */ new Set();
   /**
@@ -7132,8 +7173,8 @@ ActionProposalController = _ts_decorate29([
 ], ActionProposalController);
 
 // src/runner/inline-agent-runner.ts
-import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY8, AGENT_STORE as AGENT_STORE8, RUN_ENDED_BEFORE_TOOL_CALL as RUN_ENDED_BEFORE_TOOL_CALL2, RunCancelledError as RunCancelledError2, agentFailureCode as agentFailureCode2, agentRunContext as agentRunContext2, encodeStreamEvent as encodeStreamEvent5, publishAgentRunCancelled as publishAgentRunCancelled2, publishAgentRunFailed as publishAgentRunFailed2, releaseThreadRun as releaseThreadRun3, runAgentLoop as runAgentLoop2, runDurationMs as runDurationMs2, settleAll as settleAll2, settleUnsettledDelegation as settleUnsettledDelegation3, streamFailure as streamFailure3 } from "@dudousxd/nestjs-agent-core";
-import { Inject as Inject28, Injectable as Injectable14, Logger as Logger9, Optional as Optional12 } from "@nestjs/common";
+import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY8, AGENT_STORE as AGENT_STORE8, RUN_ENDED_BEFORE_TOOL_CALL as RUN_ENDED_BEFORE_TOOL_CALL2, RunCancelledError as RunCancelledError2, agentFailureCode as agentFailureCode2, agentRunContext as agentRunContext2, encodeStreamEvent as encodeStreamEvent5, publishAgentRunCancelled as publishAgentRunCancelled2, publishAgentRunFailed as publishAgentRunFailed2, releaseThreadRun as releaseThreadRun3, runAgentLoop as runAgentLoop2, runDurationMs as runDurationMs2, runEndTimings as runEndTimings2, settleAll as settleAll2, settleUnsettledDelegation as settleUnsettledDelegation3, streamFailure as streamFailure3 } from "@dudousxd/nestjs-agent-core";
+import { Inject as Inject28, Injectable as Injectable14, Logger as Logger10, Optional as Optional12 } from "@nestjs/common";
 function _ts_decorate30(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function") {
@@ -7167,7 +7208,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
   factory;
   store;
   queue;
-  logger = new Logger9(_InlineAgentRunner.name);
+  logger = new Logger10(_InlineAgentRunner.name);
   pending = /* @__PURE__ */ new Map();
   /**
   * Runs someone has asked to stop. In-process, like `pending`, and for the same reason: this runner
@@ -7204,10 +7245,12 @@ var InlineAgentRunner = class _InlineAgentRunner {
   }
   async start(input, options = {}) {
     const runId = options.runId ?? crypto.randomUUID();
+    const acceptedAt = Date.now();
     const day = input.day ?? utcDay();
     const deps = this.factory.forAgent(input.agentName);
     this.live.add(runId);
     const hooks = this.topLevelHooks({
+      acceptedAt,
       runId,
       deps,
       actor: input.actor,
@@ -7229,8 +7272,9 @@ var InlineAgentRunner = class _InlineAgentRunner {
       const timed = durationMs !== void 0 ? {
         durationMs
       } : {};
+      const measured = runEndTimings2(error);
       if (error instanceof RunCancelledError2) {
-        await this.settleCancelled(runId, input, deps, durationMs);
+        await this.settleCancelled(runId, input, deps, durationMs, measured);
         return;
       }
       const message = error instanceof Error ? error.message : String(error);
@@ -7242,6 +7286,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
         code,
         message,
         ...timed,
+        ...measured,
         context: agentRunContext2(input)
       });
       await this.store.recordRunEnd?.({
@@ -7319,7 +7364,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
   * the stream ENDED after a `cancelled` frame rather than failed, so a client that retries failed
   * streams does not retry this one.
   */
-  async settleCancelled(runId, input, deps, durationMs) {
+  async settleCancelled(runId, input, deps, durationMs, measured = {}) {
     this.logger.log(`agent run ${runId} cancelled`);
     const timed = durationMs !== void 0 ? {
       durationMs
@@ -7328,6 +7373,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
       runId,
       threadId: input.threadId,
       ...timed,
+      ...measured,
       context: agentRunContext2(input)
     });
     const writer = await deps.sink.open(runId);
@@ -7420,9 +7466,10 @@ var InlineAgentRunner = class _InlineAgentRunner {
     });
   }
   topLevelHooks(args) {
-    const { runId, deps, actor, day, threadId, chainBelow, input } = args;
+    const { runId, deps, actor, day, threadId, chainBelow, input, acceptedAt } = args;
     return {
       runId,
+      queueMs: /* @__PURE__ */ __name(() => Date.now() - acceptedAt, "queueMs"),
       // The loop ends the stream when the turn completes. Just before, the thread passes to the next
       // queued message (or is released), and the reader is told which — so by the time a client
       // sees the run end, the thread is already free or already running what it queued.
@@ -7547,10 +7594,12 @@ var InlineAgentRunner = class _InlineAgentRunner {
       const timed = durationMs !== void 0 ? {
         durationMs
       } : {};
+      const measured = runEndTimings2(error);
       const settled = {
         runId,
         threadId: subThread.id,
         ...timed,
+        ...measured,
         context: agentRunContext2(childInput)
       };
       if (cancelled) {
@@ -7739,7 +7788,7 @@ function _ts_decorate31(decorators, target, key, desc) {
 }
 __name(_ts_decorate31, "_ts_decorate");
 var DEFAULT_PATH = "agent";
-var logger2 = new Logger10("AgentModule");
+var logger2 = new Logger11("AgentModule");
 function externalWrapper(modules, token) {
   for (const module of modules.values()) {
     if (module.metatype === AgentModule) continue;