@dudousxd/nestjs-agent 0.3.2 → 0.3.3

package/dist/{agent-deps.factory-DpVkZYCy.d.cts → agent-deps.factory-D4KyREOv.d.cts}

@@ -140,6 +140,17 @@ interface AgentModuleAsyncOptions extends Pick<ModuleMetadata, 'imports'> {
      * the runner. Requires importing `AgentDurableModule` and a configured `DurableModule`.
      */
     durable?: boolean;
+    /**
+     * Set when `AGENT_STORE` is bound by a globally-imported store module (e.g.
+     * `MikroOrmAgentStoreModule.forFeature()`) instead of returned as `store` from `useFactory`.
+     *
+     * `forRoot` decides this synchronously (`store` present or not), but the async factory resolves
+     * too late to inspect — so by default `forRootAsync` binds `AGENT_STORE` from the factory result,
+     * which would shadow a global store binding with `undefined` when the factory omits `store`. Set
+     * this to `true` to skip the local binding and defer to the global one. Leave it unset (and return
+     * `store` from the factory) when the store is constructed inside the factory.
+     */
+    externalStore?: boolean;
 }
 
 /** The synthesized `agent`-kind tool name an orchestrator uses to hand off to `target`. */

package/dist/{agent-deps.factory-DpVkZYCy.d.ts → agent-deps.factory-D4KyREOv.d.ts}

@@ -140,6 +140,17 @@ interface AgentModuleAsyncOptions extends Pick<ModuleMetadata, 'imports'> {
      * the runner. Requires importing `AgentDurableModule` and a configured `DurableModule`.
      */
     durable?: boolean;
+    /**
+     * Set when `AGENT_STORE` is bound by a globally-imported store module (e.g.
+     * `MikroOrmAgentStoreModule.forFeature()`) instead of returned as `store` from `useFactory`.
+     *
+     * `forRoot` decides this synchronously (`store` present or not), but the async factory resolves
+     * too late to inspect — so by default `forRootAsync` binds `AGENT_STORE` from the factory result,
+     * which would shadow a global store binding with `undefined` when the factory omits `store`. Set
+     * this to `true` to skip the local binding and defer to the global one. Leave it unset (and return
+     * `store` from the factory) when the store is constructed inside the factory.
+     */
+    externalStore?: boolean;
 }
 
 /** The synthesized `agent`-kind tool name an orchestrator uses to hand off to `target`. */

package/dist/durable/index.cjs

@@ -151,7 +151,9 @@ AgentRunWorkflow = _ts_decorate([
 ], AgentRunWorkflow);
 
 // src/durable/durable-agent-runner.ts
+var import_node_crypto = require("crypto");
 var import_nestjs_durable2 = require("@dudousxd/nestjs-durable");
+var import_nestjs_durable_core2 = require("@dudousxd/nestjs-durable-core");
 var import_common2 = require("@nestjs/common");
 function _ts_decorate2(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
@@ -177,9 +179,16 @@ var DurableAgentRunner = class {
       ...input,
       day: input.day ?? utcDay()
     };
-    const result = await this.workflows.start(AgentRunWorkflow, stamped);
+    const runId = (0, import_node_crypto.randomUUID)();
+    try {
+      await this.workflows.start(AgentRunWorkflow, stamped, runId);
+    } catch (error) {
+      if (!(error instanceof import_nestjs_durable_core2.WorkflowSuspended)) {
+        throw error;
+      }
+    }
     return {
-      runId: result.runId
+      runId
     };
   }
   async signal(runId, toolCallId, decision) {
@@ -454,6 +463,9 @@ var AgentService = class {
       ...params.regenerate === true ? {
         regenerate: true
       } : {},
+      ...params.attachments !== void 0 ? {
+        attachments: params.attachments
+      } : {},
       ...params.pageContext !== void 0 ? {
         pageContext: params.pageContext
       } : {}
@@ -708,6 +720,9 @@ var ChatController = class {
       ...body.agent !== void 0 ? {
         agentName: body.agent
       } : {},
+      ...body.attachments !== void 0 ? {
+        attachments: body.attachments
+      } : {},
       ...body.pageContext !== void 0 ? {
         pageContext: body.pageContext
       } : {},
@@ -746,11 +761,26 @@ data: ${JSON.stringify({
 
 `);
     const decoder = new TextDecoder();
+    let buffer = "";
     try {
       for await (const chunk of this.agent.subscribe(runId)) {
-        res.write(`data: ${JSON.stringify({
-          delta: decoder.decode(chunk)
-        })}
+        buffer += decoder.decode(chunk, {
+          stream: true
+        });
+        let newline = buffer.indexOf("\n");
+        while (newline !== -1) {
+          const line = buffer.slice(0, newline);
+          buffer = buffer.slice(newline + 1);
+          if (line.length > 0) {
+            res.write(`data: ${line}
+
+`);
+          }
+          newline = buffer.indexOf("\n");
+        }
+      }
+      if (buffer.length > 0) {
+        res.write(`data: ${buffer}
 
 `);
       }
@@ -1809,6 +1839,7 @@ var AgentModule = class _AgentModule {
   }
   static forRootAsync(options) {
     const path = options.path ?? DEFAULT_PATH;
+    const includeStore = options.externalStore !== true;
     return {
       module: _AgentModule,
       global: true,
@@ -1824,11 +1855,9 @@ var AgentModule = class _AgentModule {
           useFactory: options.useFactory,
           inject: options.inject ?? []
         },
-        // The async factory resolves too late to inspect `store`, so we always bind AGENT_STORE here
-        // (its factory reads the resolved options). Omit-and-import-a-store-module is a `forRoot` path.
-        ...sharedProviders(options.durable ?? false, true)
+        ...sharedProviders(options.durable ?? false, includeStore)
       ],
-      exports: exportsFor(true)
+      exports: exportsFor(includeStore)
     };
   }
 };