@byok-sdk/server 0.13.0 → 0.15.0

package/dist/index.js

@@ -280,6 +280,7 @@ var TaskEventRelay = class {
     switch (envelope.type) {
       case "task.claim":
         if (envelope.payload.runtime !== void 0) state.claimedRuntime = envelope.payload.runtime;
+        if (envelope.payload.harnessId !== void 0) state.claimedHarnessId = envelope.payload.harnessId;
         this.#transition(state, taskId, "Claimed", at);
         return;
       case "task.started":
@@ -367,6 +368,7 @@ var TaskEventRelay = class {
       taskId,
       state: next,
       at,
+      ...state.claimedHarnessId === void 0 ? {} : { claimedHarnessId: state.claimedHarnessId },
       ...state.claimedRuntime === void 0 ? {} : { claimedRuntime: state.claimedRuntime }
     });
     if (next !== "Complete" && next !== "Failed" && next !== "Cancelled") return;
@@ -462,6 +464,7 @@ function toTaskSnapshot(attempt, pending, terminal, dispatched) {
     updatedAt: attempt.updatedAt,
     ...result === void 0 ? {} : { result },
     ...pending?.approvalId === void 0 ? {} : { pendingApprovalId: pending.approvalId },
+    ...attempt.claimedHarnessId === void 0 ? {} : { claimedHarnessId: attempt.claimedHarnessId },
     ...attempt.claimedRuntime === void 0 ? {} : { claimedRuntime: attempt.claimedRuntime },
     ...attempt.claimedRuntimeCapabilities === void 0 ? {} : { claimedRuntimeCapabilities: attempt.claimedRuntimeCapabilities }
   };
@@ -473,6 +476,7 @@ function toMachineInfo(device, connection) {
     connected: connection?.connected ?? false,
     lastSeen: connection?.lastSeen,
     ...connection?.clientVersion === void 0 ? {} : { clientVersion: connection.clientVersion },
+    harnesses: device.harnesses,
     runtimes: connection?.runtimes === void 0 ? void 0 : [...connection.runtimes],
     configuredToolsets: connection?.configuredToolsets === void 0 ? void 0 : [...connection.configuredToolsets]
   };
@@ -990,7 +994,7 @@ var SqliteTaskAttemptStore = class {
       db.prepare(
         `UPDATE task_attempt SET owner_device_id = ?, claimed_runtime = ?, claimed_runtime_capabilities_json = ?,
           status = 'claimed', updated_at = ?
-         WHERE tenant_id = ? AND task_id = ? AND owner_device_id IS NULL
+         WHERE tenant_id = ? AND task_id = ? AND device_id = ? AND owner_device_id IS NULL
            AND cancellation_requested_at IS NULL AND status = 'offered'`
       ).run(
         input.deviceId,
@@ -998,7 +1002,8 @@ var SqliteTaskAttemptStore = class {
         input.capabilities === void 0 ? null : JSON.stringify(input.capabilities),
         this.#now(),
         tenant,
-        input.taskId
+        input.taskId,
+        input.deviceId
       );
       return readTask(db, tenant, input.taskId);
     });
@@ -1891,6 +1896,7 @@ function createByokServer(opts) {
     const policy = input.policy ?? { mode: "confirm" };
     const runtime = dispatchSelection?.runtimeId ?? input.runtime;
     const common = {
+      ...input.harnessId === void 0 ? {} : { harnessId: input.harnessId },
       instruction: input.instruction,
       policy,
       ...runtime === void 0 ? {} : { runtime },