@dudousxd/nestjs-agent 1.27.1 → 1.29.0

package/dist/index.js

@@ -13,7 +13,7 @@ import { AGENT_OPTIONS } from "@dudousxd/nestjs-agent-core";
 import { BadRequestException, Inject as Inject2, Injectable, Optional } from "@nestjs/common";
 
 // src/genui/agent-genui.module.ts
-import { defineCatalog, genuiTools } from "@dudousxd/nestjs-agent-core/genui";
+import { defineCatalog, defineSandbox, genuiTools } from "@dudousxd/nestjs-agent-core/genui";
 import { Inject, Module } from "@nestjs/common";
 
 // src/functional-tool.ts
@@ -92,7 +92,7 @@ function resolverProviders(resolver) {
 }
 __name(resolverProviders, "resolverProviders");
 function toolOptions(options, resolver) {
-  const { catalog: _catalog, ...rest } = options;
+  const { catalog: _catalog, sandbox: _sandbox, ...rest } = options;
   return {
     ...rest,
     ...resolver !== void 0 ? {
@@ -101,11 +101,19 @@ function toolOptions(options, resolver) {
   };
 }
 __name(toolOptions, "toolOptions");
+function withSandbox(options) {
+  const base = options.catalog ?? defineCatalog([]);
+  const { sandbox } = options;
+  return sandbox === void 0 || sandbox === false ? base : base.extend([
+    defineSandbox(sandbox === true ? {} : sandbox)
+  ]);
+}
+__name(withSandbox, "withSandbox");
 function coreProviders(resolver) {
   return [
     {
       provide: GENUI_CATALOG,
-      useFactory: /* @__PURE__ */ __name((options) => options.catalog ?? defineCatalog([]), "useFactory"),
+      useFactory: /* @__PURE__ */ __name((options) => withSandbox(options), "useFactory"),
       inject: [
         GENUI_OPTIONS
       ]
@@ -564,6 +572,9 @@ AgentDepsFactory2 = _ts_decorate2([
   ])
 ], AgentDepsFactory2);
 
+// src/agent-service.token.ts
+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_FACTORY2, AGENT_MODEL_CATALOG, AGENT_OPTIONS as AGENT_OPTIONS4, AGENT_QUOTA_PROVIDER as AGENT_QUOTA_PROVIDER2, AGENT_RUNNER, AGENT_STORE as AGENT_STORE4, DEFAULT_TEXT_ACTION_PROPOSAL_VOCABULARY as DEFAULT_TEXT_ACTION_PROPOSAL_VOCABULARY2, REQUESTER_APPROVER, findCatalogModel, mayDecideApproval as mayDecideApproval2, readElicitationQuestions, settleDeadRun, textActionProposalReply as textActionProposalReply2, validateElicitationAnswer } from "@dudousxd/nestjs-agent-core";
 import { BadRequestException as BadRequestException2, ConflictException as ConflictException2, ForbiddenException as ForbiddenException2, GoneException, HttpException, HttpStatus, Inject as Inject5, Injectable as Injectable4, NotFoundException as NotFoundException2, NotImplementedException as NotImplementedException3, Optional as Optional3, PayloadTooLargeException, UnsupportedMediaTypeException } from "@nestjs/common";
@@ -1393,7 +1404,7 @@ function sendableParams(params) {
   };
 }
 __name(sendableParams, "sendableParams");
-var AgentService = class {
+var AgentService2 = class {
   static {
     __name(this, "AgentService");
   }
@@ -2576,7 +2587,7 @@ var AgentService = class {
     }
   }
 };
-AgentService = _ts_decorate5([
+AgentService2 = _ts_decorate5([
   Injectable4(),
   _ts_param4(0, Inject5(AGENT_RUNNER)),
   _ts_param4(1, Inject5(AGENT_STORE4)),
@@ -2603,7 +2614,7 @@ AgentService = _ts_decorate5([
     typeof ChatQueueService === "undefined" ? Object : ChatQueueService,
     typeof ActionProposalService === "undefined" ? Object : ActionProposalService
   ])
-], AgentService);
+], AgentService2);
 
 // src/approval-port.adapter.ts
 import { Injectable as Injectable5, NotImplementedException as NotImplementedException4, Optional as Optional4 } from "@nestjs/common";
@@ -2697,7 +2708,7 @@ AgentApprovalPortAdapter = _ts_decorate6([
   _ts_param5(1, Optional4()),
   _ts_metadata5("design:type", Function),
   _ts_metadata5("design:paramtypes", [
-    typeof AgentService === "undefined" ? Object : AgentService,
+    typeof AgentService2 === "undefined" ? Object : AgentService2,
     typeof ActionProposalService === "undefined" ? Object : ActionProposalService
   ])
 ], AgentApprovalPortAdapter);
@@ -2898,7 +2909,7 @@ AttachmentsController = _ts_decorate8([
   _ts_metadata7("design:paramtypes", [
     typeof AgentModuleOptions === "undefined" ? Object : AgentModuleOptions,
     typeof ActorResolver === "undefined" ? Object : ActorResolver,
-    typeof AgentService === "undefined" ? Object : AgentService,
+    typeof AgentService2 === "undefined" ? Object : AgentService2,
     Object
   ])
 ], AttachmentsController);
@@ -3170,7 +3181,7 @@ ChatController = _ts_decorate9([
   _ts_param8(1, Inject8(AGENT_ACTOR_RESOLVER2)),
   _ts_metadata8("design:type", Function),
   _ts_metadata8("design:paramtypes", [
-    typeof AgentService === "undefined" ? Object : AgentService,
+    typeof AgentService2 === "undefined" ? Object : AgentService2,
     typeof ActorResolver === "undefined" ? Object : ActorResolver
   ])
 ], ChatController);
@@ -3568,7 +3579,7 @@ MessagesController = _ts_decorate12([
   _ts_param11(1, Inject11(AGENT_ACTOR_RESOLVER5)),
   _ts_metadata11("design:type", Function),
   _ts_metadata11("design:paramtypes", [
-    typeof AgentService === "undefined" ? Object : AgentService,
+    typeof AgentService2 === "undefined" ? Object : AgentService2,
     typeof ActorResolver === "undefined" ? Object : ActorResolver
   ])
 ], MessagesController);
@@ -3637,7 +3648,7 @@ ModelsController = _ts_decorate13([
   _ts_param12(1, Inject12(AGENT_ACTOR_RESOLVER6)),
   _ts_metadata12("design:type", Function),
   _ts_metadata12("design:paramtypes", [
-    typeof AgentService === "undefined" ? Object : AgentService,
+    typeof AgentService2 === "undefined" ? Object : AgentService2,
     typeof ActorResolver === "undefined" ? Object : ActorResolver
   ])
 ], ModelsController);
@@ -3797,7 +3808,7 @@ QueueController = _ts_decorate14([
   _ts_param13(1, Inject13(AGENT_ACTOR_RESOLVER7)),
   _ts_metadata13("design:type", Function),
   _ts_metadata13("design:paramtypes", [
-    typeof AgentService === "undefined" ? Object : AgentService,
+    typeof AgentService2 === "undefined" ? Object : AgentService2,
     typeof ActorResolver === "undefined" ? Object : ActorResolver
   ])
 ], QueueController);
@@ -3864,7 +3875,7 @@ QuotaController = _ts_decorate15([
   _ts_param14(1, Inject14(AGENT_ACTOR_RESOLVER8)),
   _ts_metadata14("design:type", Function),
   _ts_metadata14("design:paramtypes", [
-    typeof AgentService === "undefined" ? Object : AgentService,
+    typeof AgentService2 === "undefined" ? Object : AgentService2,
     typeof ActorResolver === "undefined" ? Object : ActorResolver
   ])
 ], QuotaController);
@@ -4127,7 +4138,7 @@ ThreadsController = _ts_decorate17([
   _ts_param16(1, Inject16(AGENT_ACTOR_RESOLVER10)),
   _ts_metadata16("design:type", Function),
   _ts_metadata16("design:paramtypes", [
-    typeof AgentService === "undefined" ? Object : AgentService,
+    typeof AgentService2 === "undefined" ? Object : AgentService2,
     typeof ActorResolver === "undefined" ? Object : ActorResolver
   ])
 ], ThreadsController);
@@ -4342,7 +4353,7 @@ ToolCallController = _ts_decorate18([
   _ts_param17(1, Inject17(AGENT_ACTOR_RESOLVER11)),
   _ts_metadata17("design:type", Function),
   _ts_metadata17("design:paramtypes", [
-    typeof AgentService === "undefined" ? Object : AgentService,
+    typeof AgentService2 === "undefined" ? Object : AgentService2,
     typeof ActorResolver === "undefined" ? Object : ActorResolver
   ])
 ], ToolCallController);
@@ -6276,7 +6287,11 @@ function sharedProviders(durable, engine) {
       useExisting: ChatQueueService
     },
     InlineAgentRunner,
-    AgentService,
+    AgentService2,
+    {
+      provide: AGENT_SERVICE,
+      useExisting: AgentService2
+    },
     ActionProposalService,
     ActionProposalWorkerService,
     // Bound ALWAYS (durable or inline) — the console's cross-thread approvals inbox routes decisions
@@ -6343,7 +6358,8 @@ function exportsFor(engine) {
     AGENT_MEMORY3,
     AGENT_DEPS_FACTORY7,
     AgentDepsFactory2,
-    AgentService,
+    AgentService2,
+    AGENT_SERVICE,
     ActionProposalService,
     ActionProposalWorkerService,
     ChatQueueService,
@@ -6504,7 +6520,9 @@ AgentModule = _ts_decorate27([
 
 // src/ag-ui/ag-ui.adapter.ts
 import { AGENT_ACTOR_RESOLVER as AGENT_ACTOR_RESOLVER15, AGENT_ATTACHMENT_STAGING as AGENT_ATTACHMENT_STAGING5, AGENT_OPTIONS as AGENT_OPTIONS11 } from "@dudousxd/nestjs-agent-core";
+import { A2UI_LEGACY_BASIC_CATALOG_ID, negotiateA2uiCatalog, readAgUiA2uiCatalogIds } from "@dudousxd/nestjs-agent-core/a2ui";
 import { AG_UI_CUSTOM, actionProposalDecisionEvents, agUiEvents, agUiFramesFromNdjson, agUiSse, parseRunInput, planResume, readAnswersPayload, readApprovalPayload, readContext, readForwardedProps, readUserTurn } from "@dudousxd/nestjs-agent-core/ag-ui";
+import { uiActionText } from "@dudousxd/nestjs-agent-core/genui";
 import { BadRequestException as BadRequestException8, Body as Body7, Controller as Controller15, ForbiddenException as ForbiddenException4, HttpCode as HttpCode2, Inject as Inject26, Optional as Optional11, Post as Post8, Req as Req14, Res as Res3 } 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;
@@ -6560,12 +6578,14 @@ var AgUiRunHandler = class {
   options;
   staging;
   quietMs;
-  constructor(agent, actorResolver, options, staging, quietMs) {
+  a2ui;
+  constructor(agent, actorResolver, options, staging, quietMs, a2ui = void 0) {
     this.agent = agent;
     this.actorResolver = actorResolver;
     this.options = options;
     this.staging = staging;
     this.quietMs = quietMs;
+    this.a2ui = a2ui;
   }
   async handle(req, res, body) {
     const input = parseRunInput(body);
@@ -6614,12 +6634,22 @@ var AgUiRunHandler = class {
           skip: first.address.position,
           answered: plan.decisions.map((decision) => decision.address.toolCallId),
           ...quiet,
+          ...this.a2uiFor(input),
           preamble: warningEvents(warnings)
         });
         return;
       }
     }
-    const turn = readUserTurn(input.messages);
+    const action = forwarded.uiAction;
+    if (typeof action === "string") {
+      throw invalid("invalid_ui_action", action);
+    }
+    const turn = action !== void 0 ? {
+      text: uiActionText(action),
+      media: [],
+      staged: [],
+      dropped: []
+    } : readUserTurn(input.messages);
     if (turn === null) {
       throw invalid(NO_USER_MESSAGE_CODE, "messages carries no user message to answer");
     }
@@ -6631,7 +6661,7 @@ var AgUiRunHandler = class {
     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) {
+    if (owner !== null && forwarded.regenerate !== true && action === void 0 && 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, [
@@ -6693,6 +6723,7 @@ var AgUiRunHandler = class {
       streamRunId: started.runId,
       streamThreadId: started.threadId,
       ...quiet,
+      ...this.a2uiFor(input),
       preamble: warningEvents(warnings)
     });
   }
@@ -6839,6 +6870,13 @@ var AgUiRunHandler = class {
     for (const event of events) res.write(agUiSse(event));
     res.end();
   }
+  /** A2UI for this request's client: the basic catalog under the id it advertises. Off when off. */
+  a2uiFor(input) {
+    if (this.a2ui === void 0) return {};
+    return {
+      a2ui: negotiateA2uiCatalog(this.a2ui, readAgUiA2uiCatalogIds(input), A2UI_LEGACY_BASIC_CATALOG_ID)
+    };
+  }
   /**
   * Pipe a run to the client as AG-UI 1.0 over SSE: one event per `data:` line, from `RUN_STARTED`
   * to the run's terminal event. It ends when the run stops to ask — the library run stays parked,
@@ -6880,12 +6918,13 @@ var AgUiRunHandler = class {
 function agUiAdapter(options = {}) {
   const path = (options.path ?? "ag-ui").replace(/^\/+|\/+$/g, "");
   const quietMs = options.quietMs;
+  const a2ui = options.a2ui === void 0 || options.a2ui === false ? void 0 : options.a2ui === true ? {} : options.a2ui;
   let AgUiController = class AgUiController extends AgUiRunHandler {
     static {
       __name(this, "AgUiController");
     }
-    constructor(agent, actorResolver, moduleOptions, staging) {
-      super(agent, actorResolver, moduleOptions, staging, quietMs);
+    constructor(agent, actorResolver, moduleOptions, staging, catalog) {
+      super(agent, actorResolver, moduleOptions, staging, quietMs, a2ui !== void 0 ? withCatalog(a2ui, catalog) : void 0);
     }
     run(req, res, body) {
       return this.handle(req, res, body);
@@ -6907,17 +6946,20 @@ function agUiAdapter(options = {}) {
   ], AgUiController.prototype, "run", null);
   AgUiController = _ts_decorate28([
     Controller15(),
-    _ts_param26(0, Inject26(AgentService)),
+    _ts_param26(0, Inject26(AgentService2)),
     _ts_param26(1, Inject26(AGENT_ACTOR_RESOLVER15)),
     _ts_param26(2, Inject26(AGENT_OPTIONS11)),
     _ts_param26(3, Optional11()),
     _ts_param26(3, Inject26(AGENT_ATTACHMENT_STAGING5)),
+    _ts_param26(4, Optional11()),
+    _ts_param26(4, Inject26(GENUI_CATALOG)),
     _ts_metadata26("design:type", Function),
     _ts_metadata26("design:paramtypes", [
-      typeof AgentService === "undefined" ? Object : AgentService,
+      typeof AgentService2 === "undefined" ? Object : AgentService2,
       typeof ActorResolver === "undefined" ? Object : ActorResolver,
       typeof AgentModuleOptions === "undefined" ? Object : AgentModuleOptions,
-      typeof AttachmentStagingStore === "undefined" ? Object : AttachmentStagingStore
+      typeof AttachmentStagingStore === "undefined" ? Object : AttachmentStagingStore,
+      typeof Catalog === "undefined" ? Object : Catalog
     ])
   ], AgUiController);
   return {
@@ -6928,6 +6970,359 @@ function agUiAdapter(options = {}) {
   };
 }
 __name(agUiAdapter, "agUiAdapter");
+function withCatalog(options, catalog) {
+  if (options.catalog !== void 0 || catalog === void 0) return options;
+  return {
+    ...options,
+    catalog
+  };
+}
+__name(withCatalog, "withCatalog");
+
+// src/a2ui/a2ui.adapter.ts
+import { AGENT_ACTOR_RESOLVER as AGENT_ACTOR_RESOLVER16 } from "@dudousxd/nestjs-agent-core";
+import { A2UI_APPROVE_ACTION, A2UI_REJECT_ACTION, A2uiProjector, a2uiThreadReplay, negotiateA2uiCatalog as negotiateA2uiCatalog2, readA2uiAction, readA2uiClientCapabilities } from "@dudousxd/nestjs-agent-core/a2ui";
+import { agUiEvents as agUiEvents2, agUiFramesFromNdjson as agUiFramesFromNdjson2, decodeInterruptId } from "@dudousxd/nestjs-agent-core/ag-ui";
+import { uiActionText as uiActionText2, validateUiCapabilities as validateUiCapabilities2 } from "@dudousxd/nestjs-agent-core/genui";
+import { BadRequestException as BadRequestException9, Body as Body8, Controller as Controller16, ForbiddenException as ForbiddenException5, Get as Get13, HttpCode as HttpCode3, Inject as Inject27, Optional as Optional12, Param as Param7, Post as Post9, Query as Query7, Req as Req15, Res as Res4 } from "@nestjs/common";
+function _ts_decorate29(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") {
+    r = Reflect.decorate(decorators, target, key, desc);
+  } else {
+    for (var i = decorators.length - 1; i >= 0; i--) {
+      if (d = decorators[i]) {
+        r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+      }
+    }
+  }
+  return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+__name(_ts_decorate29, "_ts_decorate");
+function _ts_metadata27(metadataKey, metadataValue) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function") {
+    return Reflect.metadata(metadataKey, metadataValue);
+  }
+}
+__name(_ts_metadata27, "_ts_metadata");
+function _ts_param27(paramIndex, decorator) {
+  return function(target, key) {
+    decorator(target, key, paramIndex);
+  };
+}
+__name(_ts_param27, "_ts_param");
+var A2UI_VIA = "a2ui";
+function isRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+__name(isRecord, "isRecord");
+function invalid2(code, message) {
+  return new BadRequestException9({
+    statusCode: 400,
+    code,
+    message
+  });
+}
+__name(invalid2, "invalid");
+function bodyField(body, key) {
+  if (body[key] !== void 0) return body[key];
+  return isRecord(body.metadata) ? body.metadata[key] : void 0;
+}
+__name(bodyField, "bodyField");
+function readDataModel(raw, maxBytes) {
+  if (raw === void 0) return void 0;
+  if (!isRecord(raw) || !isRecord(raw.surfaces)) {
+    return "a2uiClientDataModel carries { version, surfaces }";
+  }
+  let size;
+  try {
+    size = new TextEncoder().encode(JSON.stringify(raw.surfaces)).length;
+  } catch {
+    return "a2uiClientDataModel must be JSON";
+  }
+  if (size > maxBytes) {
+    return `a2uiClientDataModel may be at most ${maxBytes} bytes (this one is ${size})`;
+  }
+  return Object.keys(raw.surfaces).length > 0 ? raw.surfaces : void 0;
+}
+__name(readDataModel, "readDataModel");
+var A2uiRunHandler = class {
+  static {
+    __name(this, "A2uiRunHandler");
+  }
+  agent;
+  actorResolver;
+  projection;
+  quietMs;
+  maxActionBytes;
+  maxDataModelBytes;
+  constructor(agent, actorResolver, projection, quietMs, maxActionBytes, maxDataModelBytes = void 0) {
+    this.agent = agent;
+    this.actorResolver = actorResolver;
+    this.projection = projection;
+    this.quietMs = quietMs;
+    this.maxActionBytes = maxActionBytes;
+    this.maxDataModelBytes = maxDataModelBytes;
+  }
+  /**
+  * `GET <path>/a2ui/threads/:threadId`: a stored thread as `{ threadId, entries }`
+  * (`a2uiThreadReplay`) — what a client draws again after a reload. Owner-scoped; a thread nobody
+  * has answers `{ entries: [] }`. `?catalog=<id>` names the client's basic catalog id.
+  */
+  async replay(req, threadId, catalog) {
+    const actor = await this.actorResolver.resolve(req);
+    const owner = await this.agent.threadOwner(threadId);
+    if (owner === null) return {
+      threadId,
+      entries: []
+    };
+    const thread = await this.agent.getThread(actor, threadId);
+    const supported = typeof catalog === "string" ? [
+      catalog
+    ] : Array.isArray(catalog) ? catalog.filter((id) => typeof id === "string") : void 0;
+    return {
+      threadId,
+      entries: a2uiThreadReplay(thread?.messages ?? [], negotiateA2uiCatalog2(this.projection, supported))
+    };
+  }
+  async handle(req, res, raw) {
+    const body = isRecord(raw) ? raw : {};
+    const agentName = typeof body.agent === "string" && body.agent.length > 0 ? body.agent : void 0;
+    const actor = await this.actorResolver.resolve(req);
+    const projection = negotiateA2uiCatalog2(this.projection, readA2uiClientCapabilities(bodyField(body, "a2uiClientCapabilities")));
+    const dataModel = readDataModel(bodyField(body, "a2uiClientDataModel"), this.maxDataModelBytes ?? 32768);
+    if (typeof dataModel === "string") throw invalid2("invalid_data_model", dataModel);
+    let action;
+    if (body.action !== void 0) {
+      const read = isRecord(body.action) ? readA2uiAction(body.action, {
+        ...this.maxActionBytes !== void 0 ? {
+          maxBytes: this.maxActionBytes
+        } : {}
+      }) : "action must be an A2UI action message";
+      if (typeof read === "string") throw invalid2("invalid_action", read);
+      action = read;
+    }
+    if (action?.name === A2UI_APPROVE_ACTION || action?.name === A2UI_REJECT_ACTION) {
+      const interruptId = action.context.interruptId;
+      const address = typeof interruptId === "string" ? decodeInterruptId(interruptId) : void 0;
+      if (address === void 0 || address === null || address.kind === "elicitation") {
+        throw invalid2("invalid_action", "an approval action carries the interruptId it decides");
+      }
+      if (address.kind === "proposal") {
+        if (address.proposalId === void 0 || address.threadId === void 0) {
+          throw invalid2("invalid_action", "the interruptId names no proposal");
+        }
+        const decided = await this.agent.decideActionProposal(actor, address.threadId, address.proposalId, {
+          decision: action.name === A2UI_APPROVE_ACTION ? "approved" : "rejected",
+          via: A2UI_VIA
+        });
+        this.writeHead(res, {
+          "X-Agent-Thread-Id": address.threadId
+        });
+        const projector = new A2uiProjector(projection);
+        const messageId = `proposal-${address.proposalId}`;
+        for (const event of [
+          {
+            type: "TEXT_MESSAGE_START",
+            messageId,
+            role: "assistant"
+          },
+          {
+            type: "TEXT_MESSAGE_CONTENT",
+            messageId,
+            delta: decided.text
+          }
+        ]) {
+          for (const message2 of projector.project(event)) res.write(`${JSON.stringify(message2)}
+`);
+        }
+        res.end();
+        return;
+      }
+      await this.agent.assertResumable(actor, address.stream);
+      await this.agent.checkDecision(actor, address.toolCallId);
+      if (action.name === A2UI_APPROVE_ACTION) {
+        await this.agent.approve(actor, address.toolCallId, {
+          via: A2UI_VIA
+        });
+      } else {
+        await this.agent.reject(actor, address.toolCallId, void 0, {
+          via: A2UI_VIA
+        });
+      }
+      await this.stream(res, address.stream, {
+        skip: address.position,
+        answered: [
+          address.toolCallId
+        ],
+        projection
+      });
+      return;
+    }
+    const message = typeof body.message === "string" ? body.message : "";
+    const text = action !== void 0 ? uiActionText2(action) : message;
+    if (text.trim().length === 0) {
+      throw invalid2("no_user_message", "send a message or an action");
+    }
+    const threadId = typeof body.threadId === "string" && body.threadId.length > 0 ? body.threadId : void 0;
+    const owner = threadId !== void 0 ? await this.agent.threadOwner(threadId) : null;
+    if (owner !== null && owner !== actor.id) {
+      throw new ForbiddenException5("thread belongs to another actor");
+    }
+    let uiCapabilities;
+    if (Object.hasOwn(body, "uiCapabilities")) {
+      try {
+        uiCapabilities = validateUiCapabilities2(body.uiCapabilities);
+      } catch (error) {
+        throw invalid2("invalid_input", error.message);
+      }
+    }
+    const started = await this.agent.chat({
+      actor,
+      message: text,
+      ...threadId === void 0 ? {} : owner !== null ? {
+        threadId
+      } : {
+        newThreadId: threadId
+      },
+      ...agentName !== void 0 ? {
+        agentName
+      } : {},
+      ...uiCapabilities !== void 0 ? {
+        uiCapabilities
+      } : {},
+      ...dataModel !== void 0 ? {
+        pageContext: {
+          a2uiDataModel: dataModel
+        }
+      } : {}
+    });
+    await this.stream(res, started.runId, {
+      threadId: started.threadId,
+      projection
+    });
+  }
+  /** Open the JSON Lines response. */
+  writeHead(res, extra) {
+    res.status(200);
+    res.setHeader("Content-Type", "application/jsonl; charset=utf-8");
+    res.setHeader("Cache-Control", "no-cache, no-transform");
+    res.setHeader("Connection", "keep-alive");
+    res.setHeader("X-Accel-Buffering", "no");
+    for (const [name, value] of Object.entries(extra)) res.setHeader(name, value);
+    res.flushHeaders();
+  }
+  /** Write one run as A2UI JSON Lines, until it ends or stops to ask. */
+  async stream(res, runId, options) {
+    this.writeHead(res, {
+      "X-Agent-Run-Id": runId,
+      ...options.threadId !== void 0 ? {
+        "X-Agent-Thread-Id": options.threadId
+      } : {}
+    });
+    const projector = new A2uiProjector(options.projection ?? this.projection);
+    const write = /* @__PURE__ */ __name((messages) => {
+      for (const message of messages) res.write(`${JSON.stringify(message)}
+`);
+    }, "write");
+    try {
+      const frames = agUiFramesFromNdjson2(this.agent.subscribe(runId));
+      for await (const event of agUiEvents2(frames, {
+        threadId: options.threadId ?? runId,
+        runId,
+        streamRunId: runId,
+        ...options.threadId !== void 0 ? {
+          streamThreadId: options.threadId
+        } : {},
+        ...options.skip !== void 0 ? {
+          skip: options.skip
+        } : {},
+        ...options.answered !== void 0 ? {
+          answered: options.answered
+        } : {},
+        ...this.quietMs !== void 0 ? {
+          quietMs: this.quietMs
+        } : {}
+      })) {
+        write(projector.project(event));
+      }
+    } catch {
+      write(projector.project({
+        type: "RUN_ERROR",
+        message: "The run could not be followed."
+      }));
+    }
+    res.end();
+  }
+};
+function a2uiAdapter(options = {}) {
+  const path = (options.path ?? "a2ui").replace(/^\/+|\/+$/g, "");
+  const { path: _path, quietMs, maxActionBytes, maxDataModelBytes, ...projection } = options;
+  let A2uiController = class A2uiController extends A2uiRunHandler {
+    static {
+      __name(this, "A2uiController");
+    }
+    constructor(agent, actorResolver, catalog) {
+      super(agent, actorResolver, projection.catalog === void 0 && catalog !== void 0 ? {
+        ...projection,
+        catalog
+      } : projection, quietMs, maxActionBytes, maxDataModelBytes);
+    }
+    thread(req, threadId, catalog) {
+      return this.replay(req, threadId, catalog);
+    }
+    run(req, res, body) {
+      return this.handle(req, res, body);
+    }
+  };
+  _ts_decorate29([
+    Get13(`${path}/threads/:threadId`),
+    _ts_param27(0, Req15()),
+    _ts_param27(1, Param7("threadId")),
+    _ts_param27(2, Query7("catalog")),
+    _ts_metadata27("design:type", Function),
+    _ts_metadata27("design:paramtypes", [
+      typeof Request === "undefined" ? Object : Request,
+      String,
+      Object
+    ]),
+    _ts_metadata27("design:returntype", typeof ReturnType === "undefined" ? Object : ReturnType)
+  ], A2uiController.prototype, "thread", null);
+  _ts_decorate29([
+    Post9(path),
+    HttpCode3(200),
+    _ts_param27(0, Req15()),
+    _ts_param27(1, Res4()),
+    _ts_param27(2, Body8()),
+    _ts_metadata27("design:type", Function),
+    _ts_metadata27("design:paramtypes", [
+      typeof Request === "undefined" ? Object : Request,
+      typeof Response === "undefined" ? Object : Response,
+      Object
+    ]),
+    _ts_metadata27("design:returntype", typeof Promise === "undefined" ? Object : Promise)
+  ], A2uiController.prototype, "run", null);
+  A2uiController = _ts_decorate29([
+    Controller16(),
+    _ts_param27(0, Inject27(AGENT_SERVICE)),
+    _ts_param27(1, Inject27(AGENT_ACTOR_RESOLVER16)),
+    _ts_param27(2, Optional12()),
+    _ts_param27(2, Inject27(GENUI_CATALOG)),
+    _ts_metadata27("design:type", Function),
+    _ts_metadata27("design:paramtypes", [
+      typeof AgentService === "undefined" ? Object : AgentService,
+      typeof ActorResolver === "undefined" ? Object : ActorResolver,
+      typeof Catalog === "undefined" ? Object : Catalog
+    ])
+  ], A2uiController);
+  return {
+    name: "a2ui",
+    controllers: [
+      A2uiController
+    ]
+  };
+}
+__name(a2uiAdapter, "a2uiAdapter");
 
 // src/resolver/header-actor-resolver.ts
 import { UnauthorizedException } from "@nestjs/common";
@@ -7006,8 +7401,10 @@ __name(requestUserActorResolver, "requestUserActorResolver");
 // src/index.ts
 export * from "@dudousxd/nestjs-agent-core";
 export {
+  A2uiRunHandler,
   AGENT_CHAT_QUEUE,
   AGENT_METADATA,
+  AGENT_SERVICE,
   AGENT_TOOL_BRAND,
   AI_TOOL_METADATA,
   ATTACHMENT_PAGE_SIZE,
@@ -7020,7 +7417,7 @@ export {
   AgentDepsFactory2 as AgentDepsFactory,
   AgentDiscoveryService,
   AgentModule,
-  AgentService,
+  AgentService2 as AgentService,
   AiTool,
   AiToolDiscoveryService,
   AnonymousActorResolver,
@@ -7049,6 +7446,7 @@ export {
   SystemPrompt,
   SystemPromptContributor,
   ToolsController,
+  a2uiAdapter,
   agUiAdapter,
   attachmentLimits,
   declaredSkillProvider,