@axiom-lattice/core 2.1.17 → 2.1.18

package/dist/index.mjs

@@ -2074,6 +2074,9 @@ var DefaultSandboxManager = class {
     this.creatingSandboxes = /* @__PURE__ */ new Map();
     this.baseURL = baseURL;
   }
+  getBaseURL() {
+    return this.baseURL;
+  }
   async createSandbox(sandboxName) {
     if (sandboxName === "global") {
       const client = new SandboxClient({ baseUrl: `${this.baseURL}/sandbox/global`, environment: "" });
@@ -2959,7 +2962,33 @@ registerToolLattice(
       if (!result.ok) {
         return `Error taking screenshot: ${JSON.stringify(result.error.content)}`;
       }
-      return result.body?.data?.content?.map((item) => item.text).join("\n") ?? "Screenshot taken successfully";
+      const lines = [];
+      for (const item of result.body?.data?.content ?? []) {
+        if (item.type === "image") {
+          const base64Data = item.data;
+          const buffer = Buffer.from(base64Data, "base64");
+          const screenshotPath = `/home/gem/screenshots/screenshot_${input.name}.png`;
+          const uploadResult = await sandbox.file.uploadFile({
+            file: buffer,
+            path: screenshotPath
+          });
+          if (uploadResult.ok) {
+            const screenshot_md = [
+              "",
+              "Screenshot saved!",
+              "```attachments",
+              JSON.stringify([{ id: screenshotPath, name: `screenshot_${input.name}.png` }]),
+              "```"
+            ].join("\n");
+            lines.push(screenshot_md);
+          } else {
+            lines.push(`Failed to save screenshot: ${JSON.stringify(uploadResult.error)}`);
+          }
+        } else {
+          lines.push(`${item.text}`);
+        }
+      }
+      return lines.length > 0 ? lines.join("\n") : "No Content";
     } catch (e) {
       return `Error taking screenshot: ${e instanceof Error ? e.message : String(e)}`;
     }
@@ -5651,18 +5680,17 @@ var DeepAgentGraphBuilder = class {
         if (!sandboxManager) {
           throw new Error("Sandbox manager not found");
         }
-        let sandboxName = "global";
-        if (params.connectedSandbox.isolatedLevel === "agent") {
-          sandboxName = agentLattice.config.key;
-          filesystemBackend = async (agentAndState) => new SandboxFilesystem({
-            sandboxInstance: await sandboxManager.createSandbox(sandboxName)
-          });
-        } else if (params.connectedSandbox.isolatedLevel === "thread") {
-          filesystemBackend = async (agentAndState) => new SandboxFilesystem({
+        filesystemBackend = async (agentAndState) => {
+          let sandboxName = "global";
+          if (params.connectedSandbox?.isolatedLevel === "agent") {
+            sandboxName = agentLattice.config.key;
+          } else if (params.connectedSandbox?.isolatedLevel === "thread") {
+            sandboxName = agentAndState.threadId ?? "global";
+          }
+          return new SandboxFilesystem({
             sandboxInstance: await sandboxManager.createSandbox(sandboxName)
-            //TODO: add threadId to sandbox
           });
-        }
+        };
       }
       if (availabledModules.includes("code_eval")) {
         middlewares.push(createCodeEvalMiddleware());