@dudousxd/nestjs-agent 0.3.2 → 0.3.3

package/dist/durable/index.js

@@ -124,7 +124,9 @@ AgentRunWorkflow = _ts_decorate([
 ], AgentRunWorkflow);
 
 // src/durable/durable-agent-runner.ts
+import { randomUUID } from "node:crypto";
 import { WorkflowService } from "@dudousxd/nestjs-durable";
+import { WorkflowSuspended as WorkflowSuspended2 } from "@dudousxd/nestjs-durable-core";
 import { Injectable as Injectable2 } from "@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;
@@ -150,9 +152,16 @@ var DurableAgentRunner = class {
       ...input,
       day: input.day ?? utcDay()
     };
-    const result = await this.workflows.start(AgentRunWorkflow, stamped);
+    const runId = randomUUID();
+    try {
+      await this.workflows.start(AgentRunWorkflow, stamped, runId);
+    } catch (error) {
+      if (!(error instanceof WorkflowSuspended2)) {
+        throw error;
+      }
+    }
     return {
-      runId: result.runId
+      runId
     };
   }
   async signal(runId, toolCallId, decision) {
@@ -427,6 +436,9 @@ var AgentService = class {
       ...params.regenerate === true ? {
         regenerate: true
       } : {},
+      ...params.attachments !== void 0 ? {
+        attachments: params.attachments
+      } : {},
       ...params.pageContext !== void 0 ? {
         pageContext: params.pageContext
       } : {}
@@ -681,6 +693,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
       } : {},
@@ -719,11 +734,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}
 
 `);
       }
@@ -1782,6 +1812,7 @@ var AgentModule = class _AgentModule {
   }
   static forRootAsync(options) {
     const path = options.path ?? DEFAULT_PATH;
+    const includeStore = options.externalStore !== true;
     return {
       module: _AgentModule,
       global: true,
@@ -1797,11 +1828,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)
     };
   }
 };