@dudousxd/nestjs-agent 1.29.0 → 1.31.0

package/dist/index.cjs

@@ -1,9 +1,14 @@
 "use strict";
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __esm = (fn, res) => function __init() {
+  return fn && (res = (0, fn[__getOwnPropNames(fn)[0]])(fn = 0)), res;
+};
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -17,8 +22,66 @@ var __copyProps = (to, from, except, desc) => {
   return to;
 };
 var __reExport = (target, mod, secondTarget) => (__copyProps(target, mod, "default"), secondTarget && __copyProps(secondTarget, mod, "default"));
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
+// src/vite/index.ts
+var import_kit, NEST_SANDBOX_KIT_DESCRIPTOR;
+var init_vite = __esm({
+  "src/vite/index.ts"() {
+    "use strict";
+    import_kit = require("@dudousxd/nestjs-agent-core/genui/kit");
+    NEST_SANDBOX_KIT_DESCRIPTOR = ".genui/sandbox-kit.json";
+  }
+});
+
+// src/genui/kit.ts
+var kit_exports = {};
+__export(kit_exports, {
+  NEST_SANDBOX_KIT_MANIFEST: () => NEST_SANDBOX_KIT_MANIFEST,
+  nestSandboxKitDiscovery: () => nestSandboxKitDiscovery
+});
+function nestSandboxKitDiscovery(options = {}) {
+  const manifest = options.manifest ?? NEST_SANDBOX_KIT_MANIFEST;
+  return (0, import_kit2.sandboxKitDiscovery)({
+    root: (0, import_node_path.resolve)(options.root ?? process.cwd()),
+    descriptor: options.descriptor ?? NEST_SANDBOX_KIT_DESCRIPTOR,
+    ...manifest !== false ? {
+      manifest: {
+        file: manifest.file,
+        outDir: manifest.outDir
+      }
+    } : {},
+    ...options.kitUrl !== void 0 ? {
+      kitUrl: options.kitUrl
+    } : {},
+    ...options.production !== void 0 ? {
+      production: options.production
+    } : {}
+  });
+}
+var import_node_path, import_kit2, NEST_SANDBOX_KIT_MANIFEST;
+var init_kit = __esm({
+  "src/genui/kit.ts"() {
+    "use strict";
+    import_node_path = require("path");
+    import_kit2 = require("@dudousxd/nestjs-agent-core/genui/kit");
+    init_vite();
+    NEST_SANDBOX_KIT_MANIFEST = Object.freeze({
+      file: "public/.vite/manifest.json",
+      outDir: "public"
+    });
+    __name(nestSandboxKitDiscovery, "nestSandboxKitDiscovery");
+  }
+});
+
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
@@ -143,6 +206,26 @@ function provideAgentTools(factory, inject) {
 __name(provideAgentTools, "provideAgentTools");
 var defineTool = import_nestjs_agent_core.createFunctionalTool;
 
+// src/genui/agent-genui.token.ts
+var AGENT_GENUI = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:genui");
+var AgentGenui = class {
+  static {
+    __name(this, "AgentGenui");
+  }
+  catalog;
+  resolveCatalog;
+  channels;
+  base;
+  sandboxClient;
+  constructor(catalog, resolveCatalog2, channels, base = {}, sandboxClient) {
+    this.catalog = catalog;
+    this.resolveCatalog = resolveCatalog2;
+    this.channels = channels;
+    this.base = base;
+    this.sandboxClient = sandboxClient;
+  }
+};
+
 // src/genui/agent-genui.module.ts
 function _ts_decorate(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
@@ -160,6 +243,7 @@ function _ts_decorate(decorators, target, key, desc) {
 __name(_ts_decorate, "_ts_decorate");
 var GENUI_CATALOG = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:genui-catalog");
 var GENUI_OPTIONS = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:genui-options");
+var GENUI_RESOLVED = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:genui-resolved");
 var GenuiCatalogResolver = class {
   static {
     __name(this, "GenuiCatalogResolver");
@@ -178,20 +262,99 @@ function resolverProviders(resolver) {
   ];
 }
 __name(resolverProviders, "resolverProviders");
-function toolOptions(options, resolver) {
-  const { catalog: _catalog, sandbox: _sandbox, ...rest } = options;
+function mergeSandboxClients(clients) {
+  const merged = {
+    theme: false
+  };
+  for (const client of clients) {
+    const one = client();
+    merged.theme ||= one.theme;
+    if (one.tailwind !== void 0) merged.tailwind ??= one.tailwind;
+    if (one.kit !== void 0) merged.kit ??= one.kit;
+  }
+  return merged;
+}
+__name(mergeSandboxClients, "mergeSandboxClients");
+async function resolveGenui(options) {
+  const { sandbox: asked, channels: askedChannels } = options;
+  (0, import_genui.assertGenuiChannels)(askedChannels);
+  const clients = [];
+  let discovery;
+  const resolveSandbox = /* @__PURE__ */ __name(async (value) => {
+    if (value === void 0 || value === false) return value;
+    const [{ nestSandboxKitDiscovery: nestSandboxKitDiscovery2 }, { resolveSandboxServer }] = await Promise.all([
+      Promise.resolve().then(() => (init_kit(), kit_exports)),
+      import("@dudousxd/nestjs-agent-core/genui/kit")
+    ]);
+    discovery ??= nestSandboxKitDiscovery2(options.sandboxKit);
+    const resolved = resolveSandboxServer(value, discovery);
+    clients.push(resolved.client);
+    return resolved.define;
+  }, "resolveSandbox");
+  const sandbox = await resolveSandbox(asked);
+  const own = options.catalog?.components.find((each) => (0, import_genui.sandboxPolicyOf)(each) !== void 0);
+  const ownOptions = asked === void 0 && own?.sandboxOptions !== void 0 && (own.sandboxView?.kit === true || own.sandboxView?.tailwind === true) ? await resolveSandbox({
+    ...own.sandboxOptions,
+    name: own.name
+  }) : void 0;
+  let channels;
+  if (askedChannels !== void 0) {
+    channels = {};
+    for (const [name, entry] of Object.entries(askedChannels)) {
+      if (entry === void 0) continue;
+      const own2 = await resolveSandbox(entry.sandbox);
+      channels[name] = {
+        ...entry,
+        ...own2 !== void 0 ? {
+          sandbox: own2
+        } : {}
+      };
+    }
+  }
+  return {
+    ...sandbox !== void 0 ? {
+      sandbox
+    } : {},
+    ...typeof ownOptions === "object" ? {
+      catalogSandbox: ownOptions
+    } : {},
+    ...channels !== void 0 ? {
+      channels
+    } : {},
+    ...clients.length > 0 ? {
+      sandboxClient: /* @__PURE__ */ __name(() => mergeSandboxClients(clients), "sandboxClient")
+    } : {}
+  };
+}
+__name(resolveGenui, "resolveGenui");
+function perRequest(resolver) {
+  return resolver === void 0 ? void 0 : (scope) => resolver.resolve(scope);
+}
+__name(perRequest, "perRequest");
+function toolOptions(options, resolved, resolver) {
+  const { catalog: _catalog, sandbox: _sandbox, channels: _channels, sandboxKit: _sandboxKit, ...rest } = options;
+  const resolveCatalog2 = perRequest(resolver);
   return {
     ...rest,
-    ...resolver !== void 0 ? {
-      resolveCatalog: /* @__PURE__ */ __name((scope) => resolver.resolve(scope), "resolveCatalog")
+    ...resolved.channels !== void 0 ? {
+      channels: resolved.channels
+    } : {},
+    ...resolved.sandbox !== void 0 ? {
+      sandbox: resolved.sandbox
+    } : {},
+    ...resolveCatalog2 !== void 0 ? {
+      resolveCatalog: resolveCatalog2
     } : {}
   };
 }
 __name(toolOptions, "toolOptions");
-function withSandbox(options) {
+function withSandbox(options, resolved) {
   const base = options.catalog ?? (0, import_genui.defineCatalog)([]);
-  const { sandbox } = options;
-  return sandbox === void 0 || sandbox === false ? base : base.extend([
+  const { sandbox, catalogSandbox } = resolved;
+  if (sandbox === void 0 || sandbox === false) return catalogSandbox !== void 0 ? base.extend([
+    (0, import_genui.defineSandbox)(catalogSandbox)
+  ]) : base;
+  return base.extend([
     (0, import_genui.defineSandbox)(sandbox === true ? {} : sandbox)
   ]);
 }
@@ -199,18 +362,50 @@ __name(withSandbox, "withSandbox");
 function coreProviders(resolver) {
   return [
     {
-      provide: GENUI_CATALOG,
-      useFactory: /* @__PURE__ */ __name((options) => withSandbox(options), "useFactory"),
+      provide: GENUI_RESOLVED,
+      useFactory: /* @__PURE__ */ __name((options) => resolveGenui(options), "useFactory"),
       inject: [
         GENUI_OPTIONS
       ]
     },
+    {
+      provide: GENUI_CATALOG,
+      useFactory: /* @__PURE__ */ __name((options, resolved) => withSandbox(options, resolved), "useFactory"),
+      inject: [
+        GENUI_OPTIONS,
+        GENUI_RESOLVED
+      ]
+    },
+    {
+      provide: AGENT_GENUI,
+      useFactory: /* @__PURE__ */ __name((options, catalog, resolved, resolverInstance) => new AgentGenui(catalog, perRequest(resolverInstance), resolved.channels, {
+        ...options.mode !== void 0 ? {
+          mode: options.mode
+        } : {},
+        ...options.streaming !== void 0 ? {
+          streaming: options.streaming
+        } : {},
+        ...resolved.sandbox !== void 0 ? {
+          sandbox: resolved.sandbox
+        } : {}
+      }, resolved.sandboxClient), "useFactory"),
+      inject: [
+        GENUI_OPTIONS,
+        GENUI_CATALOG,
+        GENUI_RESOLVED,
+        {
+          token: GenuiCatalogResolver,
+          optional: true
+        }
+      ]
+    },
     ...resolverProviders(resolver),
     // The generated tools, registered by the agent module's discovery like any `provideAgentTool`.
     // Built from the injected catalog, so `overrideProvider(GENUI_CATALOG)` changes them too.
-    provideAgentTools((options, catalog, perRequest) => (0, import_genui.genuiTools)(catalog, toolOptions(options, perRequest)), [
+    provideAgentTools((options, catalog, resolved, perRequestResolver) => (0, import_genui.genuiTools)(catalog, toolOptions(options, resolved, perRequestResolver)), [
       GENUI_OPTIONS,
       GENUI_CATALOG,
+      GENUI_RESOLVED,
       {
         token: GenuiCatalogResolver,
         optional: true
@@ -221,7 +416,8 @@ function coreProviders(resolver) {
 __name(coreProviders, "coreProviders");
 var EXPORTS = [
   GENUI_CATALOG,
-  GENUI_OPTIONS
+  GENUI_OPTIONS,
+  AGENT_GENUI
 ];
 var AgentGenuiModule = class _AgentGenuiModule {
   static {
@@ -553,7 +749,7 @@ var AgentDepsFactory2 = class {
       id: persona.id,
       label: persona.label
     };
-    const resolve = /* @__PURE__ */ __name(async (prompt, ctx) => typeof prompt === "function" ? prompt(ctx) : prompt, "resolve");
+    const resolve2 = /* @__PURE__ */ __name(async (prompt, ctx) => typeof prompt === "function" ? prompt(ctx) : prompt, "resolve");
     const own = persona.systemPrompt;
     const systemPrompt = /* @__PURE__ */ __name(async (ctx) => {
       const scoped = {
@@ -561,10 +757,10 @@ var AgentDepsFactory2 = class {
         persona: ref
       };
       if (own === void 0) {
-        return resolve(base, scoped);
+        return resolve2(base, scoped);
       }
-      const basePrompt = typeof own === "function" ? await resolve(base, scoped) : void 0;
-      return resolve(own, {
+      const basePrompt = typeof own === "function" ? await resolve2(base, scoped) : void 0;
+      return resolve2(own, {
         ...scoped,
         ...basePrompt !== void 0 ? {
           basePrompt
@@ -1485,8 +1681,16 @@ function sendableParams(params) {
       message: "nothing to answer: send a non-empty message, an attachment, or regenerate"
     });
   }
-  return text === message ? params : {
-    ...params,
+  const { channel, ...rest } = params;
+  const stamped = channel === void 0 ? params : {
+    ...rest,
+    pageContext: {
+      ...params.pageContext,
+      channel
+    }
+  };
+  return text === message ? stamped : {
+    ...stamped,
     message: text
   };
 }
@@ -3408,11 +3612,13 @@ var ConfigController = class {
   actorResolver;
   staging;
   models;
-  constructor(options, actorResolver, staging, models) {
+  genui;
+  constructor(options, actorResolver, staging, models, genui) {
     this.options = options;
     this.actorResolver = actorResolver;
     this.staging = staging;
     this.models = models;
+    this.genui = genui;
   }
   async config(req) {
     await this.actorResolver.resolve(req);
@@ -3426,7 +3632,12 @@ var ConfigController = class {
       },
       identity: {
         anonymous: this.actorResolver instanceof AnonymousActorResolver
-      }
+      },
+      ...this.genui?.sandboxClient !== void 0 ? {
+        genui: {
+          sandbox: this.genui.sandboxClient()
+        }
+      } : {}
     };
   }
 };
@@ -3447,11 +3658,14 @@ ConfigController = _ts_decorate10([
   _ts_param9(2, (0, import_common11.Inject)(import_nestjs_agent_core13.AGENT_ATTACHMENT_STAGING)),
   _ts_param9(3, (0, import_common11.Optional)()),
   _ts_param9(3, (0, import_common11.Inject)(import_nestjs_agent_core13.AGENT_MODEL_CATALOG)),
+  _ts_param9(4, (0, import_common11.Optional)()),
+  _ts_param9(4, (0, import_common11.Inject)(AGENT_GENUI)),
   _ts_metadata9("design:type", Function),
   _ts_metadata9("design:paramtypes", [
     typeof AgentModuleOptions === "undefined" ? Object : AgentModuleOptions,
     typeof ActorResolver === "undefined" ? Object : ActorResolver,
     Object,
+    Object,
     Object
   ])
 ], ConfigController);
@@ -5133,8 +5347,8 @@ var InProcessTokenStreamSink = class {
     return buf;
   }
   wake(buf) {
-    for (const resolve of buf.notify) {
-      resolve();
+    for (const resolve2 of buf.notify) {
+      resolve2();
     }
     buf.notify.clear();
   }
@@ -5173,7 +5387,7 @@ var InProcessTokenStreamSink = class {
         }
         return;
       }
-      await new Promise((resolve) => buf.notify.add(resolve));
+      await new Promise((resolve2) => buf.notify.add(resolve2));
     }
   }
   close(runId) {
@@ -5850,12 +6064,12 @@ var InlineAgentRunner = class _InlineAgentRunner {
     }
     const key = `${runId}:${toolCallId2}`;
     let timer;
-    const lapse = new Promise((resolve) => {
+    const lapse = new Promise((resolve2) => {
       timer = setTimeout(() => {
         const entry = this.pending.get(key);
         if (entry !== void 0) {
           this.pending.delete(key);
-          resolve({
+          resolve2({
             approved: false,
             expired: true
           });
@@ -5869,9 +6083,9 @@ var InlineAgentRunner = class _InlineAgentRunner {
     ]).finally(() => clearTimeout(timer));
   }
   park(runId, toolCallId2) {
-    return new Promise((resolve, reject) => {
+    return new Promise((resolve2, reject) => {
       this.pending.set(`${runId}:${toolCallId2}`, {
-        resolve,
+        resolve: resolve2,
         reject
       });
     });