@probelabs/probe 0.6.0-rc274 → 0.6.0-rc275

package/build/agent/index.js

@@ -31292,13 +31292,44 @@ var init_TaskManager = __esm({
         this.tasks = newTasks;
       }
       /**
-       * Create multiple tasks in batch
+       * Create multiple tasks in batch, resolving user-provided IDs to auto-generated IDs.
+       * If any task fails validation, no tasks are created (atomic operation).
        * @param {Object[]} tasksData - Array of task data objects
        * @returns {Task[]} Created tasks
        */
       createTasks(tasksData) {
-        const createdTasks = [];
+        const idMap = /* @__PURE__ */ new Map();
+        const batchAutoIds = /* @__PURE__ */ new Set();
+        let nextCounter = this.taskCounter;
         for (const taskData of tasksData) {
+          nextCounter++;
+          const autoId = `task-${nextCounter}`;
+          batchAutoIds.add(autoId);
+          if (taskData.id) {
+            idMap.set(taskData.id, autoId);
+          }
+        }
+        const resolvedTasksData = tasksData.map((taskData) => {
+          const resolved = { ...taskData };
+          delete resolved.id;
+          if (resolved.dependencies) {
+            resolved.dependencies = resolved.dependencies.map((depId) => {
+              if (idMap.has(depId)) return idMap.get(depId);
+              if (this.tasks.has(depId) || batchAutoIds.has(depId)) return depId;
+              throw new Error(`Dependency "${depId}" does not exist. Available tasks: ${this._getAvailableTaskIds()}${idMap.size > 0 ? `, batch IDs: ${Array.from(idMap.keys()).join(", ")}` : ""}`);
+            });
+          }
+          if (resolved.after) {
+            if (idMap.has(resolved.after)) {
+              resolved.after = idMap.get(resolved.after);
+            } else if (!this.tasks.has(resolved.after) && !batchAutoIds.has(resolved.after)) {
+              throw new Error(`Task "${resolved.after}" does not exist. Cannot insert after non-existent task. Available tasks: ${this._getAvailableTaskIds()}${idMap.size > 0 ? `, batch IDs: ${Array.from(idMap.keys()).join(", ")}` : ""}`);
+            }
+          }
+          return resolved;
+        });
+        const createdTasks = [];
+        for (const taskData of resolvedTasksData) {
           const task = this.createTask(taskData);
           createdTasks.push(task);
         }

package/build/agent/tasks/TaskManager.js

@@ -132,14 +132,53 @@ export class TaskManager {
   }
 
   /**
-   * Create multiple tasks in batch
+   * Create multiple tasks in batch, resolving user-provided IDs to auto-generated IDs.
+   * If any task fails validation, no tasks are created (atomic operation).
    * @param {Object[]} tasksData - Array of task data objects
    * @returns {Task[]} Created tasks
    */
   createTasks(tasksData) {
-    const createdTasks = [];
-
+    // Build a mapping of user-provided IDs to future auto-generated IDs
+    const idMap = new Map();
+    const batchAutoIds = new Set();
+    let nextCounter = this.taskCounter;
     for (const taskData of tasksData) {
+      nextCounter++;
+      const autoId = `task-${nextCounter}`;
+      batchAutoIds.add(autoId);
+      if (taskData.id) {
+        idMap.set(taskData.id, autoId);
+      }
+    }
+
+    // Resolve dependencies and "after" references, then validate before creating anything
+    const resolvedTasksData = tasksData.map(taskData => {
+      const resolved = { ...taskData };
+      delete resolved.id; // Don't pass user ID to createTask
+
+      if (resolved.dependencies) {
+        resolved.dependencies = resolved.dependencies.map(depId => {
+          if (idMap.has(depId)) return idMap.get(depId);
+          // Check if it's already an existing task ID or a batch auto-generated ID
+          if (this.tasks.has(depId) || batchAutoIds.has(depId)) return depId;
+          throw new Error(`Dependency "${depId}" does not exist. Available tasks: ${this._getAvailableTaskIds()}${idMap.size > 0 ? `, batch IDs: ${Array.from(idMap.keys()).join(', ')}` : ''}`);
+        });
+      }
+
+      if (resolved.after) {
+        if (idMap.has(resolved.after)) {
+          resolved.after = idMap.get(resolved.after);
+        } else if (!this.tasks.has(resolved.after) && !batchAutoIds.has(resolved.after)) {
+          throw new Error(`Task "${resolved.after}" does not exist. Cannot insert after non-existent task. Available tasks: ${this._getAvailableTaskIds()}${idMap.size > 0 ? `, batch IDs: ${Array.from(idMap.keys()).join(', ')}` : ''}`);
+        }
+      }
+
+      return resolved;
+    });
+
+    // All validation passed — create tasks
+    const createdTasks = [];
+    for (const taskData of resolvedTasksData) {
       const task = this.createTask(taskData);
       createdTasks.push(task);
     }

package/cjs/agent/ProbeAgent.cjs

@@ -58232,13 +58232,44 @@ var init_TaskManager = __esm({
         this.tasks = newTasks;
       }
       /**
-       * Create multiple tasks in batch
+       * Create multiple tasks in batch, resolving user-provided IDs to auto-generated IDs.
+       * If any task fails validation, no tasks are created (atomic operation).
        * @param {Object[]} tasksData - Array of task data objects
        * @returns {Task[]} Created tasks
        */
       createTasks(tasksData) {
-        const createdTasks = [];
+        const idMap = /* @__PURE__ */ new Map();
+        const batchAutoIds = /* @__PURE__ */ new Set();
+        let nextCounter = this.taskCounter;
         for (const taskData of tasksData) {
+          nextCounter++;
+          const autoId = `task-${nextCounter}`;
+          batchAutoIds.add(autoId);
+          if (taskData.id) {
+            idMap.set(taskData.id, autoId);
+          }
+        }
+        const resolvedTasksData = tasksData.map((taskData) => {
+          const resolved = { ...taskData };
+          delete resolved.id;
+          if (resolved.dependencies) {
+            resolved.dependencies = resolved.dependencies.map((depId) => {
+              if (idMap.has(depId)) return idMap.get(depId);
+              if (this.tasks.has(depId) || batchAutoIds.has(depId)) return depId;
+              throw new Error(`Dependency "${depId}" does not exist. Available tasks: ${this._getAvailableTaskIds()}${idMap.size > 0 ? `, batch IDs: ${Array.from(idMap.keys()).join(", ")}` : ""}`);
+            });
+          }
+          if (resolved.after) {
+            if (idMap.has(resolved.after)) {
+              resolved.after = idMap.get(resolved.after);
+            } else if (!this.tasks.has(resolved.after) && !batchAutoIds.has(resolved.after)) {
+              throw new Error(`Task "${resolved.after}" does not exist. Cannot insert after non-existent task. Available tasks: ${this._getAvailableTaskIds()}${idMap.size > 0 ? `, batch IDs: ${Array.from(idMap.keys()).join(", ")}` : ""}`);
+            }
+          }
+          return resolved;
+        });
+        const createdTasks = [];
+        for (const taskData of resolvedTasksData) {
           const task = this.createTask(taskData);
           createdTasks.push(task);
         }

package/cjs/index.cjs

@@ -31368,13 +31368,44 @@ var init_TaskManager = __esm({
         this.tasks = newTasks;
       }
       /**
-       * Create multiple tasks in batch
+       * Create multiple tasks in batch, resolving user-provided IDs to auto-generated IDs.
+       * If any task fails validation, no tasks are created (atomic operation).
        * @param {Object[]} tasksData - Array of task data objects
        * @returns {Task[]} Created tasks
        */
       createTasks(tasksData) {
-        const createdTasks = [];
+        const idMap = /* @__PURE__ */ new Map();
+        const batchAutoIds = /* @__PURE__ */ new Set();
+        let nextCounter = this.taskCounter;
         for (const taskData of tasksData) {
+          nextCounter++;
+          const autoId = `task-${nextCounter}`;
+          batchAutoIds.add(autoId);
+          if (taskData.id) {
+            idMap.set(taskData.id, autoId);
+          }
+        }
+        const resolvedTasksData = tasksData.map((taskData) => {
+          const resolved = { ...taskData };
+          delete resolved.id;
+          if (resolved.dependencies) {
+            resolved.dependencies = resolved.dependencies.map((depId) => {
+              if (idMap.has(depId)) return idMap.get(depId);
+              if (this.tasks.has(depId) || batchAutoIds.has(depId)) return depId;
+              throw new Error(`Dependency "${depId}" does not exist. Available tasks: ${this._getAvailableTaskIds()}${idMap.size > 0 ? `, batch IDs: ${Array.from(idMap.keys()).join(", ")}` : ""}`);
+            });
+          }
+          if (resolved.after) {
+            if (idMap.has(resolved.after)) {
+              resolved.after = idMap.get(resolved.after);
+            } else if (!this.tasks.has(resolved.after) && !batchAutoIds.has(resolved.after)) {
+              throw new Error(`Task "${resolved.after}" does not exist. Cannot insert after non-existent task. Available tasks: ${this._getAvailableTaskIds()}${idMap.size > 0 ? `, batch IDs: ${Array.from(idMap.keys()).join(", ")}` : ""}`);
+            }
+          }
+          return resolved;
+        });
+        const createdTasks = [];
+        for (const taskData of resolvedTasksData) {
           const task = this.createTask(taskData);
           createdTasks.push(task);
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe",
-  "version": "0.6.0-rc274",
+  "version": "0.6.0-rc275",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",

package/src/agent/tasks/TaskManager.js

@@ -132,14 +132,53 @@ export class TaskManager {
   }
 
   /**
-   * Create multiple tasks in batch
+   * Create multiple tasks in batch, resolving user-provided IDs to auto-generated IDs.
+   * If any task fails validation, no tasks are created (atomic operation).
    * @param {Object[]} tasksData - Array of task data objects
    * @returns {Task[]} Created tasks
    */
   createTasks(tasksData) {
-    const createdTasks = [];
-
+    // Build a mapping of user-provided IDs to future auto-generated IDs
+    const idMap = new Map();
+    const batchAutoIds = new Set();
+    let nextCounter = this.taskCounter;
     for (const taskData of tasksData) {
+      nextCounter++;
+      const autoId = `task-${nextCounter}`;
+      batchAutoIds.add(autoId);
+      if (taskData.id) {
+        idMap.set(taskData.id, autoId);
+      }
+    }
+
+    // Resolve dependencies and "after" references, then validate before creating anything
+    const resolvedTasksData = tasksData.map(taskData => {
+      const resolved = { ...taskData };
+      delete resolved.id; // Don't pass user ID to createTask
+
+      if (resolved.dependencies) {
+        resolved.dependencies = resolved.dependencies.map(depId => {
+          if (idMap.has(depId)) return idMap.get(depId);
+          // Check if it's already an existing task ID or a batch auto-generated ID
+          if (this.tasks.has(depId) || batchAutoIds.has(depId)) return depId;
+          throw new Error(`Dependency "${depId}" does not exist. Available tasks: ${this._getAvailableTaskIds()}${idMap.size > 0 ? `, batch IDs: ${Array.from(idMap.keys()).join(', ')}` : ''}`);
+        });
+      }
+
+      if (resolved.after) {
+        if (idMap.has(resolved.after)) {
+          resolved.after = idMap.get(resolved.after);
+        } else if (!this.tasks.has(resolved.after) && !batchAutoIds.has(resolved.after)) {
+          throw new Error(`Task "${resolved.after}" does not exist. Cannot insert after non-existent task. Available tasks: ${this._getAvailableTaskIds()}${idMap.size > 0 ? `, batch IDs: ${Array.from(idMap.keys()).join(', ')}` : ''}`);
+        }
+      }
+
+      return resolved;
+    });
+
+    // All validation passed — create tasks
+    const createdTasks = [];
+    for (const taskData of resolvedTasksData) {
       const task = this.createTask(taskData);
       createdTasks.push(task);
     }