@tanstack/ai 0.38.0 → 0.39.1

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

@@ -4,12 +4,14 @@ import { stripToSpecMiddleware } from "../../strip-to-spec-middleware.js";
 import { streamToText } from "../../stream-to-response.js";
 import { resolveDebugOption } from "../../logger/resolve.js";
 import { normalizeToolResult } from "../../utilities/tool-result.js";
+import { isProviderExecutedToolCall } from "../../utilities/provider-executed.js";
 import { LazyToolManager } from "./tools/lazy-tool-manager.js";
 import { ToolCallManager, MiddlewareAbortError, executeToolCalls } from "./tools/tool-calls.js";
 import { convertSchemaForStructuredOutput, isStandardSchema, parseWithStandardSchema, convertSchemaToJsonSchema } from "./tools/schema-converter.js";
 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 +72,7 @@ class TextEngine {
   // Middleware support
   middlewareRunner;
   middlewareCtx;
+  sandboxFileQueue = [];
   deferredPromises = [];
   abortReason;
   middlewareAbortController;
@@ -185,6 +188,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 +412,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 +461,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) {
@@ -996,6 +1015,9 @@ class TextEngine {
     for (const message of this.messages) {
       if (message.role === "assistant" && message.toolCalls) {
         for (const toolCall of message.toolCalls) {
+          if (isProviderExecutedToolCall(toolCall)) {
+            continue;
+          }
           if (!completedToolIds.has(toolCall.id)) {
             pending.push(toolCall);
           }
@@ -1412,6 +1434,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.