@dudousxd/nestjs-agent 1.12.1 → 1.14.0

package/dist/index.cjs

@@ -483,8 +483,22 @@ var AgentService = class {
     }
     return entry.id;
   }
-  /** The model a turn runs on: the send's own, else the thread's pinned one, else none. */
+  /**
+  * The model a turn runs on: the send's own (that turn only — it is never stored on the thread),
+  * else the thread's pinned one, else none. An agent the catalog locks to one model runs on it
+  * whatever was sent or pinned, and a send naming another model is refused.
+  */
   async resolveModel(actor, agent, requested, threadId) {
+    const locked = this.models !== void 0 ? (await this.models.list({
+      actor,
+      agent
+    })).locked : void 0;
+    if (locked !== void 0) {
+      if (requested !== void 0 && requested !== locked.model) {
+        throw new import_common2.BadRequestException(`model "${requested}" cannot be selected: ${locked.reason ?? `this agent always uses "${locked.model}"`}`);
+      }
+      return locked.model;
+    }
     const model = requested ?? (threadId !== void 0 ? await this.threadModel(threadId) : null);
     if (model === null || model === void 0) {
       return void 0;
@@ -633,12 +647,15 @@ var AgentService = class {
   * "the user picked exactly the defaults" persist as the same answers, and no client has to
   * re-send what it was shown.
   */
-  async answer(actor, toolCallId2, answers2 = {}) {
+  async answer(actor, toolCallId2, answers2 = {}, opts = {}) {
     await this.assertOwnsToolCall(actor, toolCallId2);
     await this.assertAnswersFit(toolCallId2, answers2);
     const reply = {
       answers: answers2,
-      answeredByRef: actor.id
+      answeredByRef: actor.id,
+      ...opts.via !== void 0 ? {
+        answeredVia: opts.via
+      } : {}
     };
     return this.signalToolCall(toolCallId2, reply);
   }
@@ -647,12 +664,15 @@ var AgentService = class {
   * them: the run records this as a rejection, so a reader auditing what the agent was told can
   * tell a choice the user made from one they refused to make.
   */
-  async skip(actor, toolCallId2) {
+  async skip(actor, toolCallId2, opts = {}) {
     await this.assertOwnsToolCall(actor, toolCallId2);
     const reply = {
       answers: {},
       skipped: true,
-      answeredByRef: actor.id
+      answeredByRef: actor.id,
+      ...opts.via !== void 0 ? {
+        answeredVia: opts.via
+      } : {}
     };
     return this.signalToolCall(toolCallId2, reply);
   }
@@ -2514,7 +2534,9 @@ var ToolCallController = class {
   */
   async answer(req, body) {
     const actor = await this.actorResolver.resolve(req);
-    await this.agent.answer(actor, toolCallId(body.toolCallId), answers(body.answers));
+    await this.agent.answer(actor, toolCallId(body.toolCallId), answers(body.answers), {
+      via: via(body.via)
+    });
     return {
       ok: true
     };
@@ -2522,7 +2544,9 @@ var ToolCallController = class {
   /** Decline to answer and let the agent proceed on the answers it pre-picked. */
   async skip(req, body) {
     const actor = await this.actorResolver.resolve(req);
-    await this.agent.skip(actor, toolCallId(body.toolCallId));
+    await this.agent.skip(actor, toolCallId(body.toolCallId), {
+      via: via(body.via)
+    });
     return {
       ok: true
     };
@@ -2624,19 +2648,34 @@ var ToolsController = class {
     this.agents = agents;
   }
   async list(req, agent) {
-    if (agent !== void 0 && this.agents.get(agent) === void 0) {
+    if (agent !== void 0 && agent !== import_nestjs_agent_core15.ALL_AGENTS && this.agents.get(agent) === void 0) {
       throw new import_common15.NotFoundException(`No agent named "${agent}"`);
     }
     const actor = await this.actorResolver.resolve(req);
-    const deps = this.depsFactory.forAgent(agent);
-    const specs = await deps.registry.visibleSpecs(actor, deps.rolesPolicy, deps.toolAllowList);
-    return specs.map((spec) => ({
-      name: spec.name,
-      kind: spec.kind,
-      ...spec.presentation !== void 0 ? {
-        presentation: spec.presentation
-      } : {}
-    }));
+    const names = agent === import_nestjs_agent_core15.ALL_AGENTS ? [
+      void 0,
+      ...this.agents.list().map((definition) => definition.name)
+    ] : [
+      agent
+    ];
+    const entries = /* @__PURE__ */ new Map();
+    for (const name of names) {
+      const deps = this.depsFactory.forAgent(name);
+      const specs = await deps.registry.visibleSpecs(actor, deps.rolesPolicy, deps.toolAllowList);
+      for (const spec of specs) {
+        if (entries.has(spec.name)) continue;
+        entries.set(spec.name, {
+          name: spec.name,
+          kind: spec.kind,
+          ...spec.presentation !== void 0 ? {
+            presentation: spec.presentation
+          } : {}
+        });
+      }
+    }
+    return [
+      ...entries.values()
+    ];
   }
 };
 _ts_decorate15([
@@ -3301,9 +3340,11 @@ var LedgerQuotaProvider = class {
   }
   store;
   limits;
-  constructor(store, limits = {}) {
+  options;
+  constructor(store, limits = {}, options = {}) {
     this.store = store;
     this.limits = limits;
+    this.options = options;
   }
   async report(query) {
     const now = query.now ?? /* @__PURE__ */ new Date();
@@ -3317,10 +3358,14 @@ var LedgerQuotaProvider = class {
       windows.push(this.window("month", used.usedTokens, used.costUsd, range.resetsAt));
     }
     const blocked = (0, import_nestjs_agent_core21.exhaustedWindow)(windows);
+    const warning = (0, import_nestjs_agent_core21.quotaWarning)(windows);
     return {
       windows,
       ...blocked !== void 0 ? {
         blocked
+      } : {},
+      ...warning !== void 0 ? {
+        warning
       } : {}
     };
   }
@@ -3341,7 +3386,10 @@ var LedgerQuotaProvider = class {
       ...limits.usd !== void 0 ? {
         limitUsd: limits.usd
       } : {},
-      resetsAt
+      resetsAt,
+      ...this.options.warnAt !== void 0 && (limits.tokens !== void 0 || limits.usd !== void 0) ? {
+        warnAt: this.options.warnAt
+      } : {}
     };
   }
 };
@@ -3946,7 +3994,9 @@ function sharedProviders(durable) {
       // `quota` is either a provider of the host's own or ceilings for the built-in ledger one;
       // omitted, the ledger provider still reports usage, with no ceiling to block on.
       provide: import_nestjs_agent_core24.AGENT_QUOTA_PROVIDER,
-      useFactory: /* @__PURE__ */ __name((o, store) => isQuotaProvider(o.quota) ? o.quota : new LedgerQuotaProvider(store, o.quota?.limits), "useFactory"),
+      useFactory: /* @__PURE__ */ __name((o, store) => isQuotaProvider(o.quota) ? o.quota : new LedgerQuotaProvider(store, o.quota?.limits, o.quota?.warnAt !== void 0 ? {
+        warnAt: o.quota.warnAt
+      } : {}), "useFactory"),
       inject: [
         import_nestjs_agent_core24.AGENT_OPTIONS,
         import_nestjs_agent_core24.AGENT_STORE