@axiom-lattice/core 2.1.67 → 2.1.69

package/dist/index.mjs

@@ -2393,6 +2393,10 @@ var InMemoryWorkflowTrackingStore = class {
     this.runs.set(runId, updated);
     return updated;
   }
+  async deleteWorkflowRun(runId) {
+    this.runs.delete(runId);
+    this.steps.delete(runId);
+  }
   async getWorkflowRunsByThreadId(tenantId, threadId) {
     const results = [];
     for (const run of this.runs.values()) {
@@ -5613,6 +5617,9 @@ var SandboxLatticeManager = class _SandboxLatticeManager extends BaseLatticeMana
       projectId: config.projectId,
       assistant_id: config.assistant_id
     });
+    console.log(
+      `[SandboxLattice] getSandboxFromConfig name=${name} isolation=${policy.vmIsolation} tenantId=${config.tenantId} workspaceId=${config.workspaceId} projectId=${config.projectId} assistant_id=${config.assistant_id} thread_id=${config.thread_id}`
+    );
     return this.createSandbox(name, config);
   }
   async getVolumeBackend(config) {
@@ -5723,7 +5730,11 @@ var createShellExecTool = ({
         const runConfig = exe_config?.configurable?.runConfig ?? {};
         const sandboxManager = getSandBoxManager();
         const sandbox = await sandboxManager.getSandboxFromConfig({
-          ...runConfig,
+          assistant_id: runConfig.assistant_id || "",
+          thread_id: runConfig.thread_id || "",
+          tenantId: runConfig.tenantId,
+          workspaceId: runConfig.workspaceId,
+          projectId: runConfig.projectId,
           vmIsolation
         });
         const executeResult = await sandbox.shell.execCommand({
@@ -5782,7 +5793,11 @@ registerToolLattice(
       const runConfig = exe_config.configurable?.runConfig || {};
       const sandboxManager = getSandBoxManager();
       const sandbox = await sandboxManager.getSandboxFromConfig({
-        ...runConfig,
+        assistant_id: runConfig.assistant_id || "",
+        thread_id: runConfig.thread_id || "",
+        tenantId: runConfig.tenantId,
+        workspaceId: runConfig.workspaceId,
+        projectId: runConfig.projectId,
         vmIsolation: "global"
       });
       let inputPath = input.uri;
@@ -17239,6 +17254,9 @@ function createTopologyMiddleware(options) {
       stepIdMap: z54.record(z54.string()).default({})
     }),
     beforeAgent: async (state, runtime) => {
+      if (state.runId) {
+        return {};
+      }
       const runConfig = runtime.context?.runConfig ?? {};
       const startingNode = edges[0]?.from ?? runConfig.assistant_id ?? "";
       let runId = "";
@@ -20938,6 +20956,9 @@ var MicrosandboxServiceClient = class {
     this.apiKey = config.apiKey;
   }
   async ensureSandbox(name, input) {
+    console.log(
+      `[MicrosandboxClient] ensureSandbox name=${name} image=${input.image ?? "not set"} envKeys=${Object.keys(input.env ?? {}).join(",") || "none"}`
+    );
     return this.request(`/api/sandboxes/${encodeURIComponent(name)}`, {
       method: "PUT",
       body: input
@@ -21185,8 +21206,12 @@ var MicrosandboxRemoteProvider = class {
   }
   buildEnsureInput(config) {
     const defaultMicrosandboxRemoteConfig = getDefaultMicrosandboxRemoteConfig();
+    const image = this.config.image ?? defaultMicrosandboxRemoteConfig.image;
+    console.log(
+      `[MicrosandboxRemote] buildEnsureInput image=${image} from=${this.config.image ? "providerConfig" : "defaultConfig"}`
+    );
     return {
-      image: this.config.image ?? defaultMicrosandboxRemoteConfig.image,
+      image,
       cpus: this.config.cpus ?? defaultMicrosandboxRemoteConfig.cpus,
       memoryMib: this.config.memoryMib ?? defaultMicrosandboxRemoteConfig.memoryMib,
       env: {
@@ -21351,7 +21376,7 @@ var RemoteSandboxProvider = class {
       environment: ""
     });
   }
-  async createSandbox(name) {
+  async createSandbox(name, _config) {
     const existing = this.instances.get(name);
     if (existing) {
       return existing;
@@ -21488,7 +21513,7 @@ var E2BProvider = class {
     this.instances = /* @__PURE__ */ new Map();
     this.creating = /* @__PURE__ */ new Map();
   }
-  async createSandbox(name) {
+  async createSandbox(name, _config) {
     const existing = this.instances.get(name);
     if (existing) {
       const running = await existing.getStatus();