heyatlas 1.8.2 → 1.9.1

package/dist/cli.js

@@ -5559,6 +5559,22 @@ class AtlasTunnel {
       };
     });
   }
+  async waitForState(timeoutMs = 1e4) {
+    if (this.currentState)
+      return;
+    return new Promise((resolve, reject) => {
+      const timeout = setTimeout(() => reject(new Error("Timed out waiting for state")), timeoutMs);
+      const check = () => {
+        if (this.currentState) {
+          clearTimeout(timeout);
+          resolve();
+        } else {
+          setTimeout(check, 100);
+        }
+      };
+      check();
+    });
+  }
   handleStateUpdate(state, source) {
     if (source === "server") {
       this.currentState = state;
@@ -52193,19 +52209,14 @@ You coordinate the following specialized subagents — delegate to them via the
 
 - Always create a plan before executing
 - Update your plan as tasks progress
-- Use absolute paths for all file operations
+- Use absolute paths for all file operations — save outputs under /home/user/output/
 - If you encounter auth barriers (logins, CAPTCHAs), ask the user to complete the manual step
-- Provide a comprehensive summary when done
+- **MANDATORY**: After ALL file-producing tasks, you MUST upload outputs to R2 before summarizing. This is not optional.
 
-## Output Upload
+## Output Upload (REQUIRED)
 
-After completing a task that produces files, upload them to cloud storage using rclone:
+You MUST upload every output file to cloud storage. Tasks are not complete without this step.
 
-1. Read \\\`/home/user/agents/task-meta.json\\\` to get the bucket name, userId, and taskId
-2. Upload all output files using: \\\`rclone copy <file-or-dir> r2:<bucket>/<userId>/<taskId>/\\\`
-3. Include the uploaded file paths in your summary
-
-Example:
 \\\`\\\`\\\`bash
 # Read task metadata
 META=$(cat /home/user/agents/task-meta.json)
@@ -52214,12 +52225,14 @@ USER_ID=$(echo $META | jq -r '.userId')
 TASK_ID=$(echo $META | jq -r '.taskId')
 PUBLIC_URL=$(echo $META | jq -r '.publicUrl')
 
-# Upload output files
-rclone copy /home/user/output.docx r2:$BUCKET/$USER_ID/$TASK_ID/
+# Upload ALL output files
+rclone copy /home/user/output/ r2:$BUCKET/$USER_ID/$TASK_ID/
+
+# Report public URLs for each file
+echo "File available at: $PUBLIC_URL/$USER_ID/$TASK_ID/<filename>"
+\\\`\\\`\\\`
 
-# Share the public URL
-echo "File available at: $PUBLIC_URL/$USER_ID/$TASK_ID/output.docx"
-\\\`\\\`\\\``,
+Your final summary MUST include the public URL for every uploaded file.`,
   "smith-browser.md": `---
 description: Browser automation expert — navigates websites, fills forms, extracts data, and performs interactive web workflows
 mode: subagent
@@ -52339,7 +52352,7 @@ You are a Documentation Specialist responsible for creating, modifying, and mana
 
 - If no format is specified, create an HTML file
 - For data-heavy documents, generate charts using Python and embed them
-- Use absolute paths for all file operations
+- Save ALL output files under /home/user/output/ (create directory if needed)
 - When complete, provide the file path and a summary
 
 ## Rules
@@ -52347,6 +52360,7 @@ You are a Documentation Specialist responsible for creating, modifying, and mana
 - Primary output should be files, not just text in response
 - Use bash tools for data processing, visualization, and file operations
 - For charts: write a Python script using plotly/matplotlib, execute it, save output as image
+- **ALWAYS save files to /home/user/output/** — this is required for upload to work
 - Provide a clear summary of work done and paths to created files`
 };
 var SMITH_OPENCODE_JSONC = `{
@@ -52574,6 +52588,7 @@ async function connect(agentType, options = {}) {
     const taskData = JSON.parse(fs.readFileSync(options.taskFile, "utf-8"));
     console.log(`Task: ${taskData.description.slice(0, 80)}...`);
     await tunnel.connect(credentials.userId, agent.name);
+    await tunnel.waitForState();
     console.log("Tunnel established");
     try {
       await handleTask(taskData, agent, tunnel);
@@ -52645,7 +52660,7 @@ async function handleTask(task, agent, tunnel) {
         { type: "ui_message", timestamp: Date.now(), data: { id: crypto.randomUUID(), role: "assistant", parts } }
       ]);
     }
-    const outputs = extractOutputUrls(parts);
+    const outputs = await uploadTaskOutputs(task.id);
     await tunnel.updateTask(task.id, {
       state: "completed",
       result: "end_turn",
@@ -52777,40 +52792,77 @@ Please continue based on the above context.`;
   }
   return { prompt, latestUserMessage };
 }
-function extractOutputUrls(parts) {
-  const outputs = [];
-  const urlPattern = /https?:\/\/[^\s"'<>]+/g;
-  for (const part of parts) {
-    if (part.type === "text" && typeof part.text === "string") {
-      const matches = part.text.match(urlPattern);
-      if (!matches)
-        continue;
-      for (const url3 of matches) {
-        const filename = url3.split("/").pop() || "";
-        const ext = filename.includes(".") ? filename.split(".").pop()?.toLowerCase() : null;
-        if (!ext)
+async function uploadTaskOutputs(taskId) {
+  const fs = await import("fs");
+  const path = await import("path");
+  const { execSync } = await import("child_process");
+  const metaPath = "/home/user/agents/task-meta.json";
+  if (!fs.existsSync(metaPath))
+    return [];
+  let meta3;
+  try {
+    meta3 = JSON.parse(fs.readFileSync(metaPath, "utf-8"));
+  } catch {
+    return [];
+  }
+  if (!meta3.publicUrl || !meta3.bucket)
+    return [];
+  const outputFiles = [];
+  const scanDirs = ["/home/user/output", "/home/user"];
+  const skipDirs = new Set(["agents", ".opencode", ".heyatlas", ".config", ".npm", ".local", ".cache", "node_modules"]);
+  const outputExts = new Set(["docx", "xlsx", "pptx", "pdf", "html", "csv", "json", "png", "jpg", "jpeg", "gif", "svg", "zip", "tar", "md", "txt"]);
+  for (const dir of scanDirs) {
+    if (!fs.existsSync(dir))
+      continue;
+    const entries = fs.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      if (entry.isDirectory() && dir === "/home/user") {
+        if (skipDirs.has(entry.name) || entry.name.startsWith("."))
           continue;
-        const typeMap = {
-          pptx: "application/vnd.openxmlformats-officedocument.presentationml.presentation",
-          docx: "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
-          xlsx: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
-          pdf: "application/pdf",
-          html: "text/html",
-          png: "image/png",
-          jpg: "image/jpeg",
-          jpeg: "image/jpeg",
-          csv: "text/csv",
-          json: "application/json"
-        };
-        outputs.push({
-          url: url3,
-          filename: decodeURIComponent(filename),
-          type: typeMap[ext] || undefined
-        });
+        const subDir = path.join(dir, entry.name);
+        const subEntries = fs.readdirSync(subDir, { withFileTypes: true });
+        for (const sub of subEntries) {
+          if (sub.isFile()) {
+            const ext = sub.name.split(".").pop()?.toLowerCase();
+            if (ext && outputExts.has(ext))
+              outputFiles.push(path.join(subDir, sub.name));
+          }
+        }
+      } else if (entry.isFile()) {
+        const ext = entry.name.split(".").pop()?.toLowerCase();
+        if (ext && outputExts.has(ext))
+          outputFiles.push(path.join(dir, entry.name));
       }
     }
   }
-  return [...new Map(outputs.map((o) => [o.url, o])).values()];
+  if (outputFiles.length === 0)
+    return [];
+  const r2Dest = `r2:${meta3.bucket}/${meta3.userId}/${meta3.taskId}`;
+  const outputs = [];
+  const typeMap = {
+    pptx: "application/vnd.openxmlformats-officedocument.presentationml.presentation",
+    docx: "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+    xlsx: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+    pdf: "application/pdf",
+    html: "text/html",
+    png: "image/png",
+    jpg: "image/jpeg",
+    jpeg: "image/jpeg",
+    csv: "text/csv"
+  };
+  for (const filePath of outputFiles) {
+    const filename = path.basename(filePath);
+    try {
+      execSync(`rclone copyto "${filePath}" "${r2Dest}/${filename}"`, { stdio: "pipe", timeout: 30000 });
+      const ext = filename.split(".").pop()?.toLowerCase() || "";
+      const publicUrl = `${meta3.publicUrl.replace(/\/$/, "")}/${meta3.userId}/${meta3.taskId}/${filename}`;
+      outputs.push({ url: publicUrl, filename, type: typeMap[ext] });
+      console.log(`Uploaded: ${filename}`);
+    } catch (e) {
+      console.error(`Failed to upload ${filename}:`, e instanceof Error ? e.message : e);
+    }
+  }
+  return outputs;
 }
 function openBrowser(url3) {
   try {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "heyatlas",
-  "version": "1.8.2",
+  "version": "1.9.1",
   "description": "Tunnel local AI agents to the cloud for voice-powered interactions",
   "author": "Bishwendu Kundu <bishwenduk029@gmail.com>",
   "license": "MIT",