@annals/agent-mesh 0.17.2 → 0.17.3

package/dist/index.js

@@ -1251,6 +1251,32 @@ var BridgeManager = class {
     }, 5 * 6e4);
     timer.unref?.();
     this.pendingTransfers.set(offer.transfer_id, { sender, timer });
+    this.pushZipToWorker(offer.transfer_id, zipBuffer);
+  }
+  /** Push ZIP buffer to Worker DO via chunked transfer_upload messages */
+  pushZipToWorker(transferId, zipBuffer) {
+    const CHUNK_SIZE2 = 64 * 1024;
+    const totalChunks = Math.ceil(zipBuffer.length / CHUNK_SIZE2);
+    log.debug(`Pushing ZIP to Worker: transfer=${transferId.slice(0, 8)}... size=${zipBuffer.length} chunks=${totalChunks}`);
+    for (let i = 0; i < totalChunks; i++) {
+      const start = i * CHUNK_SIZE2;
+      const end = Math.min(start + CHUNK_SIZE2, zipBuffer.length);
+      const chunk = zipBuffer.subarray(start, end);
+      const msg = {
+        type: "transfer_upload",
+        transfer_id: transferId,
+        chunk_index: i,
+        total_chunks: totalChunks,
+        data: chunk.toString("base64")
+      };
+      this.wsClient.send(msg);
+    }
+    const complete = {
+      type: "transfer_upload_complete",
+      transfer_id: transferId
+    };
+    this.wsClient.send(complete);
+    log.info(`ZIP pushed to Worker: transfer=${transferId.slice(0, 8)}... (${totalChunks} chunks)`);
   }
   handleRtcSignalRelay(msg) {
     const entry = this.pendingTransfers.get(msg.transfer_id);
@@ -4502,7 +4528,10 @@ function registerDiscoverCommand(program2) {
 }
 
 // src/commands/call.ts
-import { readFileSync as readFileSync2, writeFileSync as writeFileSync3 } from "fs";
+import { readFileSync as readFileSync2, writeFileSync as writeFileSync3, mkdirSync as mkdirSync5 } from "fs";
+import { createHash as createHash2 } from "crypto";
+import { execSync as execSync4 } from "child_process";
+import { join as join11 } from "path";
 var DEFAULT_BASE_URL4 = "https://agents.hot";
 async function submitRating(baseUrl, token, agentId, callId, rating) {
   const res = await fetch(`${baseUrl}/api/agents/${agentId}/rate`, {
@@ -4534,6 +4563,75 @@ function handleError2(err) {
   }
   process.exit(1);
 }
+async function downloadTransferFiles(agentId, offer, token, outputDir, json) {
+  const url = `${DEFAULT_BASE_URL4}/api/agents/${agentId}/transfer/${offer.transfer_id}`;
+  if (!json) {
+    log.info(`Downloading ${offer.file_count} file(s) (${(offer.zip_size / 1024).toFixed(1)} KB)...`);
+  }
+  await sleep6(1e3);
+  let lastError = null;
+  for (let attempt = 0; attempt < 5; attempt++) {
+    try {
+      const res = await fetch(url, {
+        headers: { Authorization: `Bearer ${token}` }
+      });
+      if (res.status === 202) {
+        await sleep6(2e3);
+        continue;
+      }
+      if (res.status === 404) {
+        if (attempt < 4) {
+          await sleep6(2e3);
+          continue;
+        }
+        log.warn("Transfer expired or not found");
+        return;
+      }
+      if (!res.ok) {
+        log.warn(`Download failed: HTTP ${res.status}`);
+        return;
+      }
+      const arrayBuffer = await res.arrayBuffer();
+      const zipBuffer = Buffer.from(arrayBuffer);
+      const hash = createHash2("sha256").update(zipBuffer).digest("hex");
+      if (hash !== offer.zip_sha256) {
+        log.warn(`SHA-256 mismatch: expected ${offer.zip_sha256.slice(0, 12)}... got ${hash.slice(0, 12)}...`);
+        return;
+      }
+      mkdirSync5(outputDir, { recursive: true });
+      const zipPath = join11(outputDir, `.transfer-${offer.transfer_id.slice(0, 8)}.zip`);
+      writeFileSync3(zipPath, zipBuffer);
+      try {
+        execSync4(`unzip -o -q "${zipPath}" -d "${outputDir}"`);
+        try {
+          execSync4(`rm "${zipPath}"`);
+        } catch {
+        }
+      } catch {
+        log.warn(`Failed to extract ZIP. Saved to: ${zipPath}`);
+        return;
+      }
+      if (json) {
+        console.log(JSON.stringify({
+          type: "files_downloaded",
+          file_count: offer.file_count,
+          output_dir: outputDir,
+          zip_size: zipBuffer.length,
+          sha256_verified: true
+        }));
+      } else {
+        log.success(`${offer.file_count} file(s) extracted to ${outputDir}`);
+      }
+      return;
+    } catch (err) {
+      lastError = err;
+      if (attempt < 4) {
+        await sleep6(2e3);
+      }
+    }
+  }
+  log.warn(`Download failed after retries: ${lastError?.message || "unknown error"}`);
+}
 async function asyncCall(opts) {
   const selfAgentId = process.env.AGENT_BRIDGE_AGENT_ID;
   const res = await fetch(`${DEFAULT_BASE_URL4}/api/agents/${opts.id}/call`, {
@@ -4598,6 +4696,7 @@ async function asyncCall(opts) {
 `);
       }
       const result = task.result || "";
+      const offer = task.file_transfer_offer;
       if (opts.json) {
         console.log(JSON.stringify({
           call_id,
@@ -4606,7 +4705,7 @@ async function asyncCall(opts) {
           status: "completed",
           result,
           ...task.attachments?.length ? { attachments: task.attachments } : {},
-          ...task.file_transfer_offer ? { file_transfer_offer: task.file_transfer_offer } : {},
+          ...offer ? { file_transfer_offer: offer } : {},
           rate_hint: `POST /api/agents/${opts.id}/rate  body: { call_id: "${call_id}", rating: 1-5 }`
         }));
       } else {
@@ -4616,10 +4715,6 @@ async function asyncCall(opts) {
             log.info(`  ${GRAY}File:${RESET} ${att.name}  ${GRAY}${att.url}${RESET}`);
           }
         }
-        const offer = task.file_transfer_offer;
-        if (offer) {
-          log.info(`  ${GRAY}Files:${RESET} ${offer.file_count} file(s) available via WebRTC`);
-        }
         if (session_key) {
           log.info(`  ${GRAY}Session:${RESET} ${session_key}`);
         }
@@ -4628,6 +4723,10 @@ async function asyncCall(opts) {
         writeFileSync3(opts.outputFile, result);
         if (!opts.json) log.info(`Saved to ${opts.outputFile}`);
       }
+      if (offer && opts.withFiles) {
+        const outputDir = opts.outputFile ? join11(opts.outputFile, "..", "files") : join11(process.cwd(), "agent-output");
+        await downloadTransferFiles(opts.id, offer, opts.token, outputDir, opts.json);
+      }
       if (!opts.json) {
         log.info(`${GRAY}Rate this call: agent-mesh rate ${call_id} <1-5> --agent ${opts.id}${RESET}`);
       }
@@ -4715,6 +4814,7 @@ async function streamCall(opts) {
   let inThinkingBlock = false;
   let callId = res.headers.get("X-Call-Id") || "";
   let sessionKey = res.headers.get("X-Session-Key") || "";
+  let fileOffer = null;
   while (true) {
     const { done, value } = await reader.read();
     if (done) break;
@@ -4756,7 +4856,7 @@ async function streamCall(opts) {
               }
             }
             if (event.file_transfer_offer) {
-              log.info(`  ${GRAY}Files:${RESET} ${event.file_transfer_offer.file_count} file(s) available via WebRTC`);
+              fileOffer = event.file_transfer_offer;
             }
           } else if (event.type === "error") {
             process.stderr.write(`
@@ -4785,6 +4885,9 @@ Error: ${event.message}
             }
           }
         }
+        if (event.type === "done" && event.file_transfer_offer) {
+          fileOffer = event.file_transfer_offer;
+        }
       } catch {
       }
     }
@@ -4793,6 +4896,11 @@ Error: ${event.message}
     writeFileSync3(opts.outputFile, outputBuffer);
     if (!opts.json) log.info(`Saved to ${opts.outputFile}`);
   }
+  if (fileOffer && opts.withFiles) {
+    console.log("");
+    const outputDir = opts.outputFile ? join11(opts.outputFile, "..", "files") : join11(process.cwd(), "agent-output");
+    await downloadTransferFiles(opts.id, fileOffer, opts.token, outputDir, opts.json);
+  }
   if (!opts.json) {
     console.log("\n");
     log.success("Call completed");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@annals/agent-mesh",
-  "version": "0.17.2",
+  "version": "0.17.3",
   "description": "CLI bridge connecting local AI agents to the Agents.Hot platform",
   "type": "module",
   "bin": {