@tanstack/ai 0.37.0 → 0.39.0

package/dist/esm/activities/chat/index.js

@@ -10,6 +10,7 @@ import { convertSchemaForStructuredOutput, isStandardSchema, parseWithStandardSc
 import { maxIterations } from "./agent-loop-strategies.js";
 import { convertMessagesToModelMessages, generateMessageId } from "./messages.js";
 import { MiddlewareRunner } from "./middleware/compose.js";
+import { provideSandboxRuntime } from "./middleware/sandbox-runtime.js";
 import { CapabilityRegistry } from "./middleware/capabilities.js";
 import { validateCapabilities } from "./middleware/validate.js";
 import { MCPManager } from "./mcp/manager.js";
@@ -70,6 +71,7 @@ class TextEngine {
   // Middleware support
   middlewareRunner;
   middlewareCtx;
+  sandboxFileQueue = [];
   deferredPromises = [];
   abortReason;
   middlewareAbortController;
@@ -185,6 +187,16 @@ class TextEngine {
       getOptional: (capability) => capability[0](this.middlewareCtx, { optional: true }),
       provide: (capability, value) => capability[1](this.middlewareCtx, value)
     };
+    provideSandboxRuntime(this.middlewareCtx, {
+      logger: this.logger,
+      emit: (event) => {
+        this.logger.sandbox(`file ${event.type} ${event.path}`, { event });
+        void this.middlewareRunner.runSandboxFile(this.middlewareCtx, event);
+        this.sandboxFileQueue.push(
+          this.createCustomEventChunk("sandbox.file", { ...event })
+        );
+      }
+    });
   }
   /** Get the accumulated content after the chat loop completes */
   getAccumulatedContent() {
@@ -399,6 +411,10 @@ class TextEngine {
       threadId: this.threadId,
       runId: this.runIdOverride,
       parentRunId: this.parentRunIdOverride,
+      // Expose provided capabilities (e.g. sandbox) to harness adapters.
+      capabilities: this.middlewareCtx,
+      // Client approval decisions, for harness interactive-approval resolution.
+      approvals: this.initialApprovals,
       ...combinedSchema ? { outputSchema: combinedSchema } : {}
     })) {
       if (this.isCancelled()) {
@@ -444,10 +460,12 @@ class TextEngine {
       if (chunk.type === "RUN_FINISHED" && chunk.usage) {
         await this.middlewareRunner.runOnUsage(this.middlewareCtx, chunk.usage);
       }
+      yield* this.drainSandboxFileQueue();
       if (this.earlyTermination) {
         break;
       }
     }
+    yield* this.drainSandboxFileQueue();
   }
   handleStreamChunk(chunk) {
     switch (chunk.type) {
@@ -1412,6 +1430,16 @@ class TextEngine {
       this.middlewareCtx.chunkIndex++;
     }
   }
+  /**
+   * Drain queued `sandbox.file` chunks (emitted via the SandboxRuntime sink)
+   * through the middleware pipeline and into the public stream.
+   */
+  async *drainSandboxFileQueue() {
+    while (this.sandboxFileQueue.length > 0) {
+      const chunk = this.sandboxFileQueue.shift();
+      if (chunk) yield* this.pipeThroughMiddleware(chunk);
+    }
+  }
   /**
    * Drain an executeToolCalls async generator, yielding any CustomEvent chunks
    * through the middleware pipeline and returning the final ExecuteToolCallsResult.