deepline 0.2.11 → 0.2.13

package/dist/bundling-sources/sdk/src/release.ts

@@ -160,7 +160,7 @@ export const SDK_RELEASE = {
   // 0.2.0 makes Dataset Handles uniformly async-only after 0.1.320 briefly
   // exposed storage-dependent synchronous access. This deliberate minor
   // release keeps lazy paging semantics independent of row residency.
-  version: '0.2.11',
+  version: '0.2.13',
   contracts: {
     api: {
       name: 'sdk-http-api',

package/dist/bundling-sources/shared_libs/play-runtime/context.ts

@@ -2276,23 +2276,16 @@ export class PlayContextImpl {
     if (!this.#options.heartbeatRuntimeStepReceipts) {
       return await this.assertRuntimeToolReceiptOwnership(requests);
     }
-    const byLeaseId = new Map<string, string[]>();
-    for (const target of targets) {
-      const keys = byLeaseId.get(target.leaseId) ?? [];
-      keys.push(target.receiptKey);
-      byLeaseId.set(target.leaseId, keys);
-    }
-    await Promise.all(
-      [...byLeaseId].map(async ([leaseId, keys]) => {
-        const receipts = await this.#options.heartbeatRuntimeStepReceipts!({
-          runId: this.currentReceiptOwnerRunId,
-          runAttempt: this.currentRunAttempt,
-          leaseId,
-          keys,
-        });
-        this.assertRuntimeToolReceiptHeartbeatResult(keys, leaseId, receipts);
-      }),
-    );
+    const uniqueTargets = [
+      ...new Map(targets.map((target) => [target.receiptKey, target])).values(),
+    ];
+    const receipts = await this.#options.heartbeatRuntimeStepReceipts({
+      runId: this.currentReceiptOwnerRunId,
+      runAttempt: this.currentRunAttempt,
+      keys: uniqueTargets.map((target) => target.receiptKey),
+      leaseIds: uniqueTargets.map((target) => target.leaseId),
+    });
+    this.assertRuntimeToolReceiptHeartbeatResults(uniqueTargets, receipts);
   }
 
   private async flushRuntimeToolOwnershipAssertionBatch(): Promise<void> {
@@ -2303,26 +2296,24 @@ export class PlayContextImpl {
 
     if (this.#options.heartbeatRuntimeStepReceipts) {
       try {
-        const byLeaseId = new Map<string, string[]>();
-        for (const target of targets) {
-          const keys = byLeaseId.get(target.leaseId) ?? [];
-          keys.push(target.receiptKey);
-          byLeaseId.set(target.leaseId, keys);
-        }
+        const uniqueTargets = [
+          ...new Map(
+            targets.map((target) => [target.receiptKey, target]),
+          ).values(),
+        ];
+        const receipts = await this.#options.heartbeatRuntimeStepReceipts({
+          runId: this.currentReceiptOwnerRunId,
+          runAttempt: this.currentRunAttempt,
+          keys: uniqueTargets.map((target) => target.receiptKey),
+          leaseIds: uniqueTargets.map((target) => target.leaseId),
+        });
         const renewed = new Map<string, RuntimeStepReceipt | null>();
-        await Promise.all(
-          [...byLeaseId].map(async ([leaseId, keys]) => {
-            const receipts = await this.#options.heartbeatRuntimeStepReceipts!({
-              runId: this.currentReceiptOwnerRunId,
-              runAttempt: this.currentRunAttempt,
-              leaseId,
-              keys,
-            });
-            for (let index = 0; index < keys.length; index += 1) {
-              renewed.set(keys[index]!, receipts[index] ?? null);
-            }
-          }),
-        );
+        for (let index = 0; index < uniqueTargets.length; index += 1) {
+          renewed.set(
+            uniqueTargets[index]!.receiptKey,
+            receipts[index] ?? null,
+          );
+        }
         for (const assertion of assertions) {
           const lost = assertion.targets.find((target) => {
             const receipt = renewed.get(target.receiptKey);
@@ -2385,21 +2376,21 @@ export class PlayContextImpl {
     for (const assertion of assertions) assertion.reject(error);
   }
 
-  private assertRuntimeToolReceiptHeartbeatResult(
-    keys: string[],
-    leaseId: string,
+  private assertRuntimeToolReceiptHeartbeatResults(
+    targets: Array<{ receiptKey: string; leaseId: string }>,
     receipts: Array<RuntimeStepReceipt | null>,
   ): void {
-    for (let index = 0; index < keys.length; index += 1) {
+    for (let index = 0; index < targets.length; index += 1) {
+      const target = targets[index]!;
       const receipt = this.normalizeRuntimeStepReceipt(
-        keys[index] ?? '',
+        target.receiptKey,
         receipts[index],
       );
-      if (!this.runtimeToolReceiptStillOwned(receipt, leaseId)) {
+      if (!this.runtimeToolReceiptStillOwned(receipt, target.leaseId)) {
         throw new RuntimeReceiptLeaseLostError({
-          receiptKey: keys[index] ?? 'unknown',
+          receiptKey: target.receiptKey,
           runId: this.currentReceiptOwnerRunId,
-          leaseId,
+          leaseId: target.leaseId,
         });
       }
     }
@@ -8040,58 +8031,50 @@ export class PlayContextImpl {
           this.toolCallResolvers.has(request.callId),
         );
         if (unsettled.length === 0) return 'terminal';
-        const byLeaseId = new Map<string, Map<string, ToolCallRequest[]>>();
+        const byReceiptKey = new Map<string, ToolCallRequest[]>();
         for (const request of unsettled) {
-          const leaseId = request.receiptLeaseId!;
-          const byReceiptKey = byLeaseId.get(leaseId) ?? new Map();
           const receiptKey = request.receiptKey!;
           const group = byReceiptKey.get(receiptKey) ?? [];
           group.push(request);
           byReceiptKey.set(receiptKey, group);
-          byLeaseId.set(leaseId, byReceiptKey);
         }
-        await Promise.all(
-          [...byLeaseId].map(async ([leaseId, byReceiptKey]) => {
-            // One content-addressed receipt can have many same-run followers.
-            // The store renews unique receipt rows, not request positions, so
-            // duplicate keys in one heartbeat would yield null duplicate
-            // positions and falsely stop the supervisor as if ownership were
-            // lost.
-            const entries = [...byReceiptKey];
-            const keys = entries.map(([receiptKey]) => receiptKey);
-            const receipts = await heartbeatReceipts({
+        // One content-addressed receipt can have many same-run followers. The
+        // store renews unique receipt rows, not request positions, so dedupe by
+        // key while preserving the matching per-row lease id.
+        const entries = [...byReceiptKey];
+        const keys = entries.map(([receiptKey]) => receiptKey);
+        const leaseIds = entries.map(
+          ([, requests]) => requests[0]!.receiptLeaseId!,
+        );
+        const receipts = await heartbeatReceipts({
+          runId: this.currentReceiptOwnerRunId,
+          runAttempt: this.currentRunAttempt,
+          keys,
+          leaseIds,
+        });
+        for (let index = 0; index < entries.length; index += 1) {
+          const [receiptKey, requests] = entries[index]!;
+          const leaseId = leaseIds[index]!;
+          // Completion can race this bulk heartbeat response. A settled
+          // receipt group no longer needs ownership and must not turn that
+          // race into a false lease-loss failure.
+          if (
+            !requests.some((request) =>
+              this.toolCallResolvers.has(request.callId),
+            )
+          ) {
+            continue;
+          }
+          if (
+            !this.runtimeToolReceiptStillOwned(receipts[index] ?? null, leaseId)
+          ) {
+            throw new RuntimeReceiptLeaseLostError({
+              receiptKey,
               runId: this.currentReceiptOwnerRunId,
-              runAttempt: this.currentRunAttempt,
               leaseId,
-              keys,
             });
-            for (let index = 0; index < entries.length; index += 1) {
-              const [receiptKey, requests] = entries[index]!;
-              // Completion can race this bulk heartbeat response. A settled
-              // receipt group no longer needs ownership and must not turn that
-              // race into a false lease-loss failure.
-              if (
-                !requests.some((request) =>
-                  this.toolCallResolvers.has(request.callId),
-                )
-              ) {
-                continue;
-              }
-              if (
-                !this.runtimeToolReceiptStillOwned(
-                  receipts[index] ?? null,
-                  leaseId,
-                )
-              ) {
-                throw new RuntimeReceiptLeaseLostError({
-                  receiptKey,
-                  runId: this.currentReceiptOwnerRunId,
-                  leaseId,
-                });
-              }
-            }
-          }),
-        );
+          }
+        }
         return 'active';
       },
       isLeaseLost: (error) => error instanceof RuntimeReceiptLeaseLostError,

package/dist/bundling-sources/shared_libs/play-runtime/ctx-types.ts

@@ -204,7 +204,8 @@ export interface HeartbeatRuntimeStepReceiptsInput {
   keys: string[];
   runId: string;
   runAttempt?: number | null;
-  leaseId: string;
+  leaseId?: string;
+  leaseIds?: string[];
 }
 
 export interface CsvOptions {

package/dist/bundling-sources/shared_libs/play-runtime/runtime-actions.ts

@@ -215,7 +215,8 @@ export type RuntimeReceiptAction =
       playName: string;
       runId: string;
       runAttempt?: number | null;
-      leaseId: string;
+      leaseId?: string;
+      leaseIds?: string[];
       keys: string[];
     };
 

package/dist/bundling-sources/shared_libs/play-runtime/runtime-api.ts

@@ -5799,15 +5799,30 @@ export async function heartbeatRuntimeWorkReceipts(
     playName: string;
     runId: string;
     runAttempt?: number | null;
-    leaseId: string;
+    leaseId?: string;
+    leaseIds?: string[];
     keys: string[];
     leaseTtlMs?: number | null;
   },
 ): Promise<Array<WorkReceipt | null>> {
-  const keys = input.keys.map((key) => key.trim()).filter(Boolean);
+  if (
+    input.leaseIds !== undefined &&
+    input.leaseIds.length !== input.keys.length
+  ) {
+    throw new Error(
+      `Runtime receipt bulk heartbeat requires one lease ID per key. Received ${input.leaseIds.length} lease IDs for ${input.keys.length} keys.`,
+    );
+  }
+  const entries = input.keys
+    .map((key, index) => ({
+      key: key.trim(),
+      leaseId: input.leaseIds?.[index]?.trim() || input.leaseId?.trim() || '',
+    }))
+    .filter((entry) => Boolean(entry.key));
+  const keys = entries.map((entry) => entry.key);
   if (keys.length === 0) return [];
-  const leaseId = input.leaseId.trim();
-  if (!leaseId) return keys.map(() => null);
+  const leaseIds = entries.map((entry) => entry.leaseId);
+  if (leaseIds.some((leaseId) => !leaseId)) return keys.map(() => null);
   const runAttempt = normalizeRuntimeRunAttempt(input.runAttempt);
   const leaseTtlMs = normalizeRuntimeLeaseTtlMs(
     input.leaseTtlMs,
@@ -5824,8 +5839,10 @@ export async function heartbeatRuntimeWorkReceipts(
       const { rows } = await client.query<Record<string, unknown>>(
         `
       WITH input_keys AS (
-        SELECT key_values.key_hex, key_values.ord
+        SELECT key_values.key_hex, lease_values.lease_id, key_values.ord
         FROM unnest($1::text[]) WITH ORDINALITY AS key_values(key_hex, ord)
+        JOIN unnest($3::text[]) WITH ORDINALITY AS lease_values(lease_id, ord)
+          ON lease_values.ord = key_values.ord
       ),
       renewed AS (
         UPDATE ${workReceiptTable(session)} AS target
@@ -5837,7 +5854,7 @@ export async function heartbeatRuntimeWorkReceipts(
              receiptTable: 'target',
              ownerRunIdSql: '$2::text',
              ownerRunAttemptSql: '$5',
-             leaseIdSql: '$3::text',
+             leaseIdSql: 'input_keys.lease_id',
            })}
         RETURNING target.k,
                   target.status,
@@ -5888,7 +5905,7 @@ export async function heartbeatRuntimeWorkReceipts(
         [
           keys.map(workReceiptKeyHex),
           input.runId,
-          leaseId,
+          leaseIds,
           leaseTtlMs,
           runAttempt,
         ],

package/dist/bundling-sources/shared_libs/play-runtime/runtime-contract.ts

@@ -1,4 +1,4 @@
-export const PLAY_RUNTIME_CONTRACT = 3;
+export const PLAY_RUNTIME_CONTRACT = 4;
 export const PLAY_RUNTIME_CONTRACT_MIN_SUPPORTED = 1;
 export const PLAY_RUNTIME_CONTRACT_HEADER = 'x-deepline-runtime-contract';
 /**
@@ -25,6 +25,13 @@ export function resolvePlayRuntimeTransportAttemptId(
 }
 
 export const RUNTIME_CONTRACT_CHANGELOG = [
+  {
+    contract: 4,
+    changed:
+      'Runtime receipt heartbeat requests can carry one positional lease ID per receipt key.',
+    compatOwner: 'runtime',
+    compatRemoval: 'contract 3 runtime lane drain',
+  },
   {
     contract: 3,
     changed:

package/dist/bundling-sources/shared_libs/play-runtime/runtime-receipt-store-adapter.ts

@@ -236,13 +236,13 @@ export class RuntimeReceiptStoreAdapter {
         ),
       heartbeatRuntimeStepReceipts: (input) =>
         writeMany(
-          input.keys.map((key) => ({
+          input.keys.map((key, index) => ({
             kind: 'heartbeat' as const,
             input: {
               key,
               runId: input.runId,
               runAttempt: input.runAttempt,
-              leaseId: input.leaseId,
+              leaseId: input.leaseIds?.[index] ?? input.leaseId ?? '',
             },
           })),
         ),
@@ -387,7 +387,6 @@ function receiptCommandBatchKey(command: ReceiptCommand): string {
         command.kind,
         command.input.runId,
         command.input.runAttempt ?? null,
-        command.input.leaseId,
       ]);
     case 'acquire_execution_lock':
     case 'release_execution_lock':
@@ -491,11 +490,15 @@ async function sendReceiptCommands(input: {
         ReceiptCommand,
         { kind: 'heartbeat' }
       >[];
+      const leaseIds = commands.map((command) => command.input.leaseId);
+      const sharedLeaseId = leaseIds.every((leaseId) => leaseId === leaseIds[0])
+        ? leaseIds[0]
+        : undefined;
       return await heartbeatRuntimeStepReceiptsViaAppRuntime(input.context, {
         playName: input.playName,
         runId: first.input.runId,
         runAttempt: first.input.runAttempt,
-        leaseId: first.input.leaseId,
+        ...(sharedLeaseId ? { leaseId: sharedLeaseId } : { leaseIds }),
         keys: commands.map((command) => command.input.key),
       });
     }

package/dist/cli/index.js

@@ -1040,7 +1040,7 @@ var SDK_RELEASE = {
   // 0.2.0 makes Dataset Handles uniformly async-only after 0.1.320 briefly
   // exposed storage-dependent synchronous access. This deliberate minor
   // release keeps lazy paging semantics independent of row residency.
-  version: "0.2.11",
+  version: "0.2.13",
   contracts: {
     api: {
       name: "sdk-http-api",

package/dist/cli/index.mjs

@@ -1025,7 +1025,7 @@ var SDK_RELEASE = {
   // 0.2.0 makes Dataset Handles uniformly async-only after 0.1.320 briefly
   // exposed storage-dependent synchronous access. This deliberate minor
   // release keeps lazy paging semantics independent of row residency.
-  version: "0.2.11",
+  version: "0.2.13",
   contracts: {
     api: {
       name: "sdk-http-api",

package/dist/index.js

@@ -763,7 +763,7 @@ var SDK_RELEASE = {
   // 0.2.0 makes Dataset Handles uniformly async-only after 0.1.320 briefly
   // exposed storage-dependent synchronous access. This deliberate minor
   // release keeps lazy paging semantics independent of row residency.
-  version: "0.2.11",
+  version: "0.2.13",
   contracts: {
     api: {
       name: "sdk-http-api",

package/dist/index.mjs

@@ -689,7 +689,7 @@ var SDK_RELEASE = {
   // 0.2.0 makes Dataset Handles uniformly async-only after 0.1.320 briefly
   // exposed storage-dependent synchronous access. This deliberate minor
   // release keeps lazy paging semantics independent of row residency.
-  version: "0.2.11",
+  version: "0.2.13",
   contracts: {
     api: {
       name: "sdk-http-api",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "deepline",
-  "version": "0.2.11",
+  "version": "0.2.13",
   "description": "Deepline SDK + CLI — B2B data enrichment powered by durable cloud execution",
   "license": "MIT",
   "repository": {