deepline 0.1.285 → 0.1.286

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

@@ -155,7 +155,7 @@ export const SDK_RELEASE = {
   // 0.1.253 makes play-page browser opening opt-in and retires --no-open.
   // 0.1.254 removes the internal operations tree from the published SDK CLI.
   // Operators use the checkout-local deepline-admin binary instead.
-  version: '0.1.285',
+  version: '0.1.286',
   contracts: {
     api: {
       name: 'sdk-http-api',

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

@@ -1737,7 +1737,10 @@ async function resetRuntimePostgresPool(postgresUrl: string): Promise<void> {
 
 async function withRuntimePostgres<T>(
   session: RuntimePostgresSession,
-  fn: (client: RuntimePoolClient) => Promise<T>,
+  fn: (
+    client: RuntimePoolClient,
+    transaction: { active: boolean },
+  ) => Promise<T>,
   options: {
     cachePool?: boolean;
     maxConnectAttempts?: number;
@@ -1746,6 +1749,7 @@ async function withRuntimePostgres<T>(
 ): Promise<T> {
   let client: RuntimePoolClient | null = null;
   let requestLocalPool: RuntimePool | null = null;
+  let roleTransactionStarted = false;
   const cachePool =
     (options.cachePool ?? true) && canReuseRuntimePostgresPoolsAcrossRequests();
   const maxConnectAttempts =
@@ -1768,13 +1772,19 @@ async function withRuntimePostgres<T>(
         options.signal,
       );
       if (session.executionRole) {
+        await client.query('BEGIN');
+        roleTransactionStarted = true;
         await client.query(
-          `SET ROLE ${quoteIdentifier(session.executionRole)}`,
+          `SET LOCAL ROLE ${quoteIdentifier(session.executionRole)}`,
         );
       }
       break;
     } catch (error) {
       if (client) {
+        if (roleTransactionStarted) {
+          await client.query('ROLLBACK').catch(() => {});
+          roleTransactionStarted = false;
+        }
         client.release();
         client = null;
       }
@@ -1813,15 +1823,20 @@ async function withRuntimePostgres<T>(
     throw new Error('Runtime Postgres connection was not acquired.');
   }
   try {
-    return await fn(client);
-  } finally {
-    try {
-      if (session.executionRole) {
-        await client.query('RESET ROLE');
-      }
-    } finally {
-      client.release();
+    const result = await fn(client, { active: roleTransactionStarted });
+    if (roleTransactionStarted) {
+      await client.query('COMMIT');
+      roleTransactionStarted = false;
     }
+    return result;
+  } catch (error) {
+    if (roleTransactionStarted) {
+      await client.query('ROLLBACK').catch(() => {});
+      roleTransactionStarted = false;
+    }
+    throw error;
+  } finally {
+    client.release();
     if (requestLocalPool) {
       await Promise.resolve(requestLocalPool.end()).catch(() => {});
     }
@@ -2010,14 +2025,15 @@ async function withRuntimeSheetQueryClient<T>(
 
       return await withRuntimePostgres(
         session,
-        async (client) => {
-          if (input.transactional) await client.query('BEGIN');
+        async (client, transaction) => {
+          const managesTransaction = input.transactional && !transaction.active;
+          if (managesTransaction) await client.query('BEGIN');
           try {
             const result = await operation(client);
-            if (input.transactional) await client.query('COMMIT');
+            if (managesTransaction) await client.query('COMMIT');
             return result;
           } catch (error) {
-            if (input.transactional) {
+            if (managesTransaction) {
               await client.query('ROLLBACK').catch(() => {});
             }
             throw error;
@@ -2378,19 +2394,37 @@ async function withRuntimeWorkReceiptClient<T>(
   // can let provider work continue after durable receipt persistence is gone.
   // Only the schema-migration branch below may repair ownership before its
   // retry. A genuinely missing schema or second failure also rethrows loudly.
-  const runWithSelfHeal = async (client: RuntimeQueryClient): Promise<T> => {
+  const withClient = async <R>(
+    run: (client: RuntimeQueryClient) => Promise<R>,
+  ): Promise<R> =>
+    isRuntimeOneShotQueryFactoryRegistered()
+      ? await withRuntimeOneShotPostgres(session, run)
+      : await withRuntimePostgres(session, run, {
+          cachePool: !context.disablePostgresPoolCache,
+          signal: context.abortSignal,
+        });
+  let selfHealAttempted = false;
+
+  for (
+    let attempt = 1;
+    attempt <= RUNTIME_WORK_RECEIPT_QUERY_MAX_ATTEMPTS;
+    attempt += 1
+  ) {
     try {
-      return await operation(client);
+      return await withClient(operation);
     } catch (error) {
-      if (
+      const missingStorage =
         isMissingRelationError(error) ||
-        isMissingRuntimeWorkReceiptSelfHealColumnError(error)
-      ) {
+        isMissingRuntimeWorkReceiptSelfHealColumnError(error);
+      if (missingStorage && !selfHealAttempted) {
+        selfHealAttempted = true;
         runtimeWorkReceiptEnsureCache.delete(
           runtimeWorkReceiptEnsureCacheKey(session),
         );
         try {
-          await ensureRuntimeWorkReceiptTable(session, client);
+          await withClient((client) =>
+            ensureRuntimeWorkReceiptTable(session, client),
+          );
         } catch (ensureError) {
           if (!isPostgresPermissionDeniedError(ensureError)) {
             throw ensureError;
@@ -2401,31 +2435,12 @@ async function withRuntimeWorkReceiptClient<T>(
           await repairRuntimeStorageGrants(context, {
             playName: context.playName?.trim() || session.playName,
           });
+          await withClient((client) =>
+            ensureRuntimeWorkReceiptTable(session, client),
+          );
         }
-      } else {
-        throw error;
+        continue;
       }
-      return await operation(client);
-    }
-  };
-
-  for (
-    let attempt = 1;
-    attempt <= RUNTIME_WORK_RECEIPT_QUERY_MAX_ATTEMPTS;
-    attempt += 1
-  ) {
-    try {
-      return isRuntimeOneShotQueryFactoryRegistered()
-        ? await withRuntimeOneShotPostgres(session, runWithSelfHeal)
-        : await withRuntimePostgres(
-            session,
-            (client) => runWithSelfHeal(client),
-            {
-              cachePool: !context.disablePostgresPoolCache,
-              signal: context.abortSignal,
-            },
-          );
-    } catch (error) {
       if (
         attempt >= RUNTIME_WORK_RECEIPT_QUERY_MAX_ATTEMPTS ||
         !isTransientRuntimePostgresOperationError(error)
@@ -2928,8 +2943,9 @@ async function writeRuntimeRows(
   const chunks = chunkValues(rowEntries, DIRECT_POSTGRES_BATCH_SIZE);
   const needsTransaction = input.mode === 'replace' || chunks.length > 1;
 
-  return await withRuntimePostgres(session, async (client) => {
-    if (needsTransaction) await client.query('BEGIN');
+  return await withRuntimePostgres(session, async (client, transaction) => {
+    const managesTransaction = needsTransaction && !transaction.active;
+    if (managesTransaction) await client.query('BEGIN');
     try {
       if (input.mode === 'replace') {
         // Collapse 4 cleanup statements into one CTE-shaped query. Postgres
@@ -3048,10 +3064,10 @@ async function writeRuntimeRows(
         writtenRows += Number(rows[0]?.inserted_count ?? 0);
       }
 
-      if (needsTransaction) await client.query('COMMIT');
+      if (managesTransaction) await client.query('COMMIT');
       return { disposition: 'completed', writtenRows };
     } catch (error) {
-      if (needsTransaction) {
+      if (managesTransaction) {
         await client.query('ROLLBACK').catch(() => {});
       }
       throw error;

package/dist/cli/index.js

@@ -718,7 +718,7 @@ var SDK_RELEASE = {
   // 0.1.253 makes play-page browser opening opt-in and retires --no-open.
   // 0.1.254 removes the internal operations tree from the published SDK CLI.
   // Operators use the checkout-local deepline-admin binary instead.
-  version: "0.1.285",
+  version: "0.1.286",
   contracts: {
     api: {
       name: "sdk-http-api",

package/dist/cli/index.mjs

@@ -703,7 +703,7 @@ var SDK_RELEASE = {
   // 0.1.253 makes play-page browser opening opt-in and retires --no-open.
   // 0.1.254 removes the internal operations tree from the published SDK CLI.
   // Operators use the checkout-local deepline-admin binary instead.
-  version: "0.1.285",
+  version: "0.1.286",
   contracts: {
     api: {
       name: "sdk-http-api",

package/dist/index.js

@@ -438,7 +438,7 @@ var SDK_RELEASE = {
   // 0.1.253 makes play-page browser opening opt-in and retires --no-open.
   // 0.1.254 removes the internal operations tree from the published SDK CLI.
   // Operators use the checkout-local deepline-admin binary instead.
-  version: "0.1.285",
+  version: "0.1.286",
   contracts: {
     api: {
       name: "sdk-http-api",

package/dist/index.mjs

@@ -367,7 +367,7 @@ var SDK_RELEASE = {
   // 0.1.253 makes play-page browser opening opt-in and retires --no-open.
   // 0.1.254 removes the internal operations tree from the published SDK CLI.
   // Operators use the checkout-local deepline-admin binary instead.
-  version: "0.1.285",
+  version: "0.1.286",
   contracts: {
     api: {
       name: "sdk-http-api",

package/package.json

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