@dudousxd/nestjs-agent 1.22.0 → 1.23.1

package/dist/index.cjs

@@ -1175,6 +1175,9 @@ var ChatQueueService = class _ChatQueueService {
       } : {},
       ...input.uiCapabilities !== void 0 ? {
         uiCapabilities: input.uiCapabilities
+      } : {},
+      ...input.hostContext !== void 0 ? {
+        hostContext: input.hostContext
       } : {}
     });
     await store.setQueuePause(threadId, {
@@ -1260,6 +1263,9 @@ var ChatQueueService = class _ChatQueueService {
       ...message.uiCapabilities !== void 0 ? {
         uiCapabilities: message.uiCapabilities
       } : {},
+      ...message.hostContext !== void 0 ? {
+        hostContext: message.hostContext
+      } : {},
       ...message.model !== void 0 ? {
         model: message.model
       } : {}
@@ -1630,6 +1636,9 @@ var AgentService = class {
       }
       const created = await this.store.createThread({
         actor: params.actor,
+        ...agentName !== void 0 ? {
+          agentName
+        } : {},
         ...params.transient === true ? {
           transient: true
         } : {},
@@ -1642,7 +1651,7 @@ var AgentService = class {
       });
       threadId = created.id;
     } else {
-      await this.assertOwnsThread(params.actor, threadId);
+      if (params.authorized !== true) await this.assertOwnsThread(params.actor, threadId);
       if (params.personaId !== void 0) {
         await this.pinThreadPersona(threadId, params.personaId);
       }
@@ -1668,6 +1677,9 @@ var AgentService = class {
       ...params.uiCapabilities !== void 0 ? {
         uiCapabilities: params.uiCapabilities
       } : {},
+      ...params.hostContext !== void 0 ? {
+        hostContext: params.hostContext
+      } : {},
       ...model !== void 0 ? {
         model
       } : {}
@@ -1690,7 +1702,7 @@ var AgentService = class {
       });
     }
     if (live === null && mode !== "queue") {
-      const runId = crypto.randomUUID();
+      const runId = this.runner.runIdFor?.(input) ?? crypto.randomUUID();
       if (await queue.queueStore().claimActiveStream(threadId, runId, stale !== null ? {
         replacing: stale
       } : {})) {
@@ -1758,6 +1770,9 @@ var AgentService = class {
       ...input.uiCapabilities !== void 0 ? {
         uiCapabilities: input.uiCapabilities
       } : {},
+      ...input.hostContext !== void 0 ? {
+        hostContext: input.hostContext
+      } : {},
       ...interrupting !== void 0 ? {
         interrupt: true,
         at: "head"
@@ -5969,11 +5984,11 @@ function isQuotaProvider(quota) {
 __name(isQuotaProvider, "isQuotaProvider");
 function resolveCatalog(options) {
   if (options.models !== void 0) return options.models;
-  const carried = options.model.catalog;
+  const carried = options.model?.catalog;
   return carried !== void 0 && typeof carried.list === "function" ? carried : void 0;
 }
 __name(resolveCatalog, "resolveCatalog");
-function sharedProviders(durable) {
+function sharedProviders(durable, engine) {
   const providers = [
     {
       provide: import_nestjs_agent_core33.AGENT_TOOL_REGISTRY,
@@ -6104,7 +6119,12 @@ function sharedProviders(durable) {
       import_core4.ModulesContainer
     ]
   });
-  if (durable) {
+  if (engine !== void 0) {
+    providers.push(...engine.providers, {
+      provide: import_nestjs_agent_core33.AGENT_RUNNER,
+      useExisting: engine.runner
+    });
+  } else if (durable) {
     providers.push({
       provide: import_nestjs_agent_core33.AGENT_RUNNER,
       useFactory: /* @__PURE__ */ __name((durableRunner) => {
@@ -6129,8 +6149,9 @@ function sharedProviders(durable) {
   return providers;
 }
 __name(sharedProviders, "sharedProviders");
-function exportsFor() {
+function exportsFor(engine) {
   return [
+    ...engine?.exports ?? [],
     import_nestjs_agent_core33.AGENT_OPTIONS,
     import_nestjs_agent_core33.AGENT_TOOL_REGISTRY,
     import_nestjs_agent_core33.AGENT_REGISTRY,
@@ -6180,10 +6201,11 @@ function adapterProviders(adapters) {
   return (adapters ?? []).flatMap((adapter) => adapter.providers ?? []);
 }
 __name(adapterProviders, "adapterProviders");
-function controllersForSurface(surface, adapters) {
+function controllersForSurface(surface, adapters, engine) {
   return surface === "engine" ? [] : [
     ...BASE_CONTROLLERS,
-    ...adapterControllers(adapters)
+    ...adapterControllers(adapters),
+    ...engine?.controllers ?? []
   ];
 }
 __name(controllersForSurface, "controllersForSurface");
@@ -6213,12 +6235,22 @@ function routerFor(path) {
   ]);
 }
 __name(routerFor, "routerFor");
+function assertRunnable(options) {
+  if (options.engine !== void 0 && options.durable === true) {
+    throw new Error(`AgentModule: \`durable: true\` runs turns on the library's loop, but the \`${options.engine.name}\` engine runs them instead. Drop \`durable\` (the engine decides how its turns survive a restart).`);
+  }
+  if (options.engine === void 0 && options.model === void 0) {
+    throw new Error("AgentModule: set `model` (the loop runs the turns) or `engine` (something else does).");
+  }
+}
+__name(assertRunnable, "assertRunnable");
 var AgentModule = class _AgentModule {
   static {
     __name(this, "AgentModule");
   }
   static forRoot(options) {
     const path = options.path ?? DEFAULT_PATH;
+    assertRunnable(options);
     applyGuards(options.guards, options.adapters);
     return {
       module: _AgentModule,
@@ -6227,21 +6259,27 @@ var AgentModule = class _AgentModule {
         import_core4.DiscoveryModule,
         routerFor(path)
       ],
-      controllers: controllersForSurface(options.surface, options.adapters),
+      controllers: controllersForSurface(options.surface, options.adapters, options.engine),
       providers: [
         {
           provide: import_nestjs_agent_core33.AGENT_OPTIONS,
           useValue: options
         },
-        ...sharedProviders(options.durable ?? false),
+        ...sharedProviders(options.durable ?? false, options.engine),
         ...guardProviders(options.guards),
         ...adapterProviders(options.adapters)
       ],
-      exports: exportsFor()
+      exports: exportsFor(options.engine)
     };
   }
   static forRootAsync(options) {
     const path = options.path ?? DEFAULT_PATH;
+    if (options.engine !== void 0) {
+      assertRunnable({
+        engine: options.engine,
+        durable: options.durable ?? false
+      });
+    }
     applyGuards(options.guards, options.adapters);
     return {
       module: _AgentModule,
@@ -6251,7 +6289,7 @@ var AgentModule = class _AgentModule {
         routerFor(path),
         ...options.imports ?? []
       ],
-      controllers: controllersForSurface(options.surface, options.adapters),
+      controllers: controllersForSurface(options.surface, options.adapters, options.engine),
       providers: [
         {
           provide: import_nestjs_agent_core33.AGENT_OPTIONS,
@@ -6261,18 +6299,25 @@ var AgentModule = class _AgentModule {
           // anything today (AgentDurableModule takes its own mirror option — see its doc for why),
           // but it's stamped here too so the EFFECTIVE value is always observable off
           // AGENT_OPTIONS, sync or async alike.
-          useFactory: /* @__PURE__ */ __name(async (...args) => ({
-            ...await options.useFactory(...args),
-            durable: options.durable ?? false,
-            surface: options.surface ?? "both"
-          }), "useFactory"),
+          useFactory: /* @__PURE__ */ __name(async (...args) => {
+            const resolved = {
+              ...await options.useFactory(...args),
+              durable: options.durable ?? false,
+              surface: options.surface ?? "both",
+              ...options.engine !== void 0 ? {
+                engine: options.engine
+              } : {}
+            };
+            assertRunnable(resolved);
+            return resolved;
+          }, "useFactory"),
           inject: options.inject ?? []
         },
-        ...sharedProviders(options.durable ?? false),
+        ...sharedProviders(options.durable ?? false, options.engine),
         ...guardProviders(options.guards),
         ...adapterProviders(options.adapters)
       ],
-      exports: exportsFor()
+      exports: exportsFor(options.engine)
     };
   }
 };
@@ -6407,7 +6452,10 @@ var AgUiRunHandler = class {
     if (owner !== null && owner !== actor.id) {
       throw new import_common28.ForbiddenException("thread belongs to another actor");
     }
-    if (owner !== null && turn.media.length === 0) {
+    if (forwarded.regenerate === true && owner === null) {
+      throw invalid("invalid_input", "regenerate needs a thread that already has an exchange");
+    }
+    if (owner !== null && forwarded.regenerate !== true && turn.media.length === 0 && turn.staged.length === 0) {
       const decision = await this.agent.handleTextDecision(input.threadId, actor, turn.text);
       if ("proposalDecision" in decision) {
         this.writeEvents(res, [
@@ -6422,7 +6470,7 @@ var AgUiRunHandler = class {
         return;
       }
     }
-    const refs = await this.stage(actor, turn.media, warnings);
+    const refs = await this.stage(actor, turn.media, warnings, turn.staged);
     if (turn.text.trim().length === 0 && refs.length === 0) {
       throw invalid("no_user_message", "the user message to answer is empty");
     }
@@ -6458,6 +6506,9 @@ var AgUiRunHandler = class {
       } : {},
       ...refs.length > 0 ? {
         attachments: refs
+      } : {},
+      ...forwarded.regenerate === true ? {
+        regenerate: true
       } : {}
     });
     await this.pipe(res, started.runId, {
@@ -6469,10 +6520,25 @@ var AgUiRunHandler = class {
       preamble: warningEvents(warnings)
     });
   }
-  /** Stage the turn's inline media through the bound store, within its limits; say what was not used. */
-  async stage(actor, media, warnings) {
+  /**
+  * The turn's attachments: the ones already staged (by mediaId), then its inline media staged
+  * through the bound store — all within the store's limits; say what was not used.
+  */
+  async stage(actor, media, warnings, staged = []) {
     const refs = [];
     const limits = attachmentLimits(this.options, this.staging);
+    for (const ref of staged) {
+      const label = `The attachment ${ref.mediaId} was not used`;
+      if (this.staging === void 0) {
+        warnings.push(`${label}: attachments are not enabled on this agent.`);
+      } else if (refs.length >= limits.maxPerMessage) {
+        warnings.push(`${label}: a message carries at most ${limits.maxPerMessage} attachments.`);
+      } else {
+        refs.push({
+          mediaId: ref.mediaId
+        });
+      }
+    }
     for (const item of media) {
       const label = `The ${item.kind} "${item.filename}" was not used`;
       if (this.staging === void 0) {
@@ -6485,7 +6551,7 @@ var AgUiRunHandler = class {
         warnings.push(`${label}: it is larger than ${limits.maxBytes} bytes.`);
       } else {
         try {
-          const staged = await this.staging.stage({
+          const staged2 = await this.staging.stage({
             data: item.data,
             filename: item.filename,
             contentType: item.contentType,
@@ -6493,7 +6559,7 @@ var AgUiRunHandler = class {
             actor
           });
           refs.push({
-            mediaId: staged.mediaId
+            mediaId: staged2.mediaId
           });
         } catch {
           warnings.push(`${label}: it could not be stored.`);
@@ -6611,11 +6677,17 @@ var AgUiRunHandler = class {
     res.setHeader("X-Agent-Run-Id", runId);
     res.flushHeaders();
     let terminal = false;
+    const cursor = {
+      seq: 0
+    };
     try {
       const frames = (0, import_ag_ui.agUiFramesFromNdjson)(this.agent.subscribe(runId));
-      for await (const event of (0, import_ag_ui.agUiEvents)(frames, options)) {
+      for await (const event of (0, import_ag_ui.agUiEvents)(frames, {
+        ...options,
+        cursor
+      })) {
         if (event.type === "RUN_FINISHED" || event.type === "RUN_ERROR") terminal = true;
-        res.write((0, import_ag_ui.agUiSse)(event));
+        res.write((0, import_ag_ui.agUiSse)(event, cursor.seq));
       }
     } catch {
       if (!terminal) {