@dudousxd/nestjs-agent 1.22.0 → 1.23.0

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)
     };
   }
 };