@bastani/atomic 0.9.23-alpha.1 → 0.9.24-alpha.1

package/dist/builtin/workflows/src/extension/index.bundle.mjs

@@ -11864,7 +11864,7 @@ var require_query = __commonJS(function(exports, module) {
       if (typeof this.text !== "string" && typeof this.name !== "string") {
         return new Error("A query must have either text or a name. Supplying neither is unsupported.");
       }
-      const previous = connection.parsedStatements[this.name];
+      const previous = connection.parsedStatements[this.name] || connection.submittedNamedStatements[this.name];
       if (this.text && previous && this.text !== previous) {
         return new Error(`Prepared statements must be unique - '${this.name}' was used for a different statement`);
       }
@@ -11884,7 +11884,7 @@ var require_query = __commonJS(function(exports, module) {
       return null;
     }
     hasBeenParsed(connection) {
-      return this.name && connection.parsedStatements[this.name];
+      return this.name && (connection.parsedStatements[this.name] || connection.submittedNamedStatements[this.name]);
     }
     handlePortalSuspended(connection) {
       this._getRows(connection, this.rows);
@@ -11907,6 +11907,9 @@ var require_query = __commonJS(function(exports, module) {
           name: this.name,
           types: this.types
         });
+        if (this.name) {
+          connection.submittedNamedStatements[this.name] = this.text;
+        }
       }
       try {
         connection.bind({
@@ -12638,7 +12641,7 @@ var require_parser2 = __commonJS(function(exports) {
     const parameterCount = reader.int16();
     const message = new messages_1.ParameterDescriptionMessage(LATEINIT_LENGTH, parameterCount);
     for (let i = 0;i < parameterCount; i++) {
-      message.dataTypeIDs[i] = reader.int32();
+      message.dataTypeIDs[i] = reader.uint32();
     }
     return message;
   };
@@ -12840,6 +12843,7 @@ var require_connection = __commonJS(function(exports, module) {
       this._keepAlive = config.keepAlive;
       this._keepAliveInitialDelayMillis = config.keepAliveInitialDelayMillis;
       this.parsedStatements = {};
+      this.submittedNamedStatements = {};
       this.ssl = config.ssl || false;
       this.sslNegotiation = config.sslNegotiation || "postgres";
       this._ending = false;
@@ -13360,6 +13364,8 @@ var require_client = __commonJS(function(exports, module) {
         encoding: this.connectionParameters.client_encoding || "utf8"
       });
       this._queryQueue = [];
+      this._sentQueryQueue = [];
+      this.pipeline = Boolean(c.pipeline);
       this.binary = c.binary || defaults.binary;
       this.processID = null;
       this.secretKey = null;
@@ -13394,6 +13400,8 @@ var require_client = __commonJS(function(exports, module) {
         enqueueError(activeQuery);
         this._activeQuery = null;
       }
+      this._sentQueryQueue.forEach(enqueueError);
+      this._sentQueryQueue.length = 0;
       this._queryQueue.forEach(enqueueError);
       this._queryQueue.length = 0;
     }
@@ -13621,6 +13629,9 @@ var require_client = __commonJS(function(exports, module) {
         return;
       }
       this._activeQuery = null;
+      if (activeQuery.name) {
+        delete this.connection.submittedNamedStatements[activeQuery.name];
+      }
       activeQuery.handleError(msg, this.connection);
     }
     _handleRowDescription(msg) {
@@ -13677,6 +13688,7 @@ var require_client = __commonJS(function(exports, module) {
       }
       if (activeQuery.name) {
         this.connection.parsedStatements[activeQuery.name] = activeQuery.text;
+        delete this.connection.submittedNamedStatements[activeQuery.name];
       }
     }
     _handleCopyInResponse(msg) {
@@ -13743,6 +13755,8 @@ var require_client = __commonJS(function(exports, module) {
         });
       } else if (client._queryQueue.indexOf(query) !== -1) {
         client._queryQueue.splice(client._queryQueue.indexOf(query), 1);
+      } else if (client._sentQueryQueue.indexOf(query) !== -1) {
+        query.callback = () => {};
       }
     }
     setTypeParser(oid, format, parseFn) {
@@ -13758,6 +13772,10 @@ var require_client = __commonJS(function(exports, module) {
       return utils.escapeLiteral(str);
     }
     _pulseQueryQueue() {
+      if (this.pipeline) {
+        this._pulsePipelinedQueryQueue();
+        return;
+      }
       if (this.readyForQuery === true) {
         this._activeQuery = this._queryQueue.shift();
         const activeQuery = this._getActiveQuery();
@@ -13778,6 +13796,30 @@ var require_client = __commonJS(function(exports, module) {
         }
       }
     }
+    _pulsePipelinedQueryQueue() {
+      if (!this._connected || !this._queryable) {
+        return;
+      }
+      while (this._queryQueue.length > 0) {
+        const query = this._queryQueue.shift();
+        this.hasExecuted = true;
+        const queryError = query.submit(this.connection);
+        if (queryError) {
+          process.nextTick(() => {
+            query.handleError(queryError, this.connection);
+          });
+          continue;
+        }
+        this._sentQueryQueue.push(query);
+      }
+      if (this.readyForQuery && !this._activeQuery && this._sentQueryQueue.length > 0) {
+        this._activeQuery = this._sentQueryQueue.shift();
+        this.readyForQuery = false;
+      }
+      if (!this._activeQuery && this._sentQueryQueue.length === 0 && this._queryQueue.length === 0 && this.hasExecuted) {
+        this.emit("drain");
+      }
+    }
     query(config, values, callback) {
       let query;
       let result;
@@ -13819,6 +13861,9 @@ var require_client = __commonJS(function(exports, module) {
           const index = this._queryQueue.indexOf(query);
           if (index > -1) {
             this._queryQueue.splice(index, 1);
+          } else if (this.pipeline) {
+            this.connection.stream.destroy();
+            return;
           }
           this._pulseQueryQueue();
         }, readTimeout);
@@ -13845,7 +13890,7 @@ var require_client = __commonJS(function(exports, module) {
         });
         return result;
       }
-      if (this._queryQueue.length > 0) {
+      if (this._queryQueue.length > 0 && !this.pipeline) {
         queryQueueLengthDeprecationNotice();
       }
       this._queryQueue.push(query);
@@ -13871,7 +13916,11 @@ var require_client = __commonJS(function(exports, module) {
           return this._Promise.resolve();
         }
       }
-      if (this._getActiveQuery() || !this._queryable) {
+      if (!this._queryable) {
+        this.connection.stream.destroy();
+      } else if (this.pipeline && (this._getActiveQuery() || this._sentQueryQueue.length > 0 || this._queryQueue.length > 0)) {
+        this.once("drain", () => this.connection.end());
+      } else if (this._getActiveQuery()) {
         this.connection.stream.destroy();
       } else {
         this.connection.end();
@@ -14342,7 +14391,7 @@ var require_query2 = __commonJS(function(exports, module) {
     sourceFunction: "routine"
   };
   NativeQuery.prototype.handleError = function(err) {
-    const fields = this.native.pq.resultErrorFields();
+    const fields = this.native && this.native.pq.resultErrorFields();
     if (fields) {
       for (const key in fields) {
         const normalizedFieldName = errorFieldMap[key] || key;
@@ -14469,6 +14518,8 @@ var require_client2 = __commonJS(function(exports, module) {
     this._connecting = false;
     this._connected = false;
     this._queryable = true;
+    this.pipeline = Boolean(config.pipeline);
+    this._pipelineInFlight = false;
     const cp = this.connectionParameters = new ConnectionParameters(config);
     if (config.nativeConnectionString)
       cp.nativeConnectionString = config.nativeConnectionString;
@@ -14613,7 +14664,7 @@ var require_client2 = __commonJS(function(exports, module) {
       });
       return result;
     }
-    if (this._queryQueue.length > 0) {
+    if (this._queryQueue.length > 0 && !this.pipeline) {
       queryQueueLengthDeprecationNotice();
     }
     this._queryQueue.push(query);
@@ -14634,15 +14685,22 @@ var require_client2 = __commonJS(function(exports, module) {
         cb = (err) => err ? reject(err) : resolve();
       });
     }
-    this.native.end(function() {
-      self._connected = false;
-      self._errorAllQueries(new Error("Connection terminated"));
-      process.nextTick(() => {
-        self.emit("end");
-        if (cb)
-          cb();
+    const doEnd = function() {
+      self.native.end(function() {
+        self._connected = false;
+        self._errorAllQueries(new Error("Connection terminated"));
+        process.nextTick(() => {
+          self.emit("end");
+          if (cb)
+            cb();
+        });
       });
-    });
+    };
+    if (this.pipeline && (this._pipelineInFlight || this._queryQueue.length > 0)) {
+      this.once("drain", doEnd);
+    } else {
+      doEnd();
+    }
     return result;
   };
   Client.prototype._hasActiveQuery = function() {
@@ -14652,6 +14710,9 @@ var require_client2 = __commonJS(function(exports, module) {
     if (!this._connected) {
       return;
     }
+    if (this.pipeline && !initialConnection) {
+      return this._pulsePipelinedQueryQueue();
+    }
     if (this._hasActiveQuery()) {
       return;
     }
@@ -14669,6 +14730,69 @@ var require_client2 = __commonJS(function(exports, module) {
       self._pulseQueryQueue();
     });
   };
+  Client.prototype._pulsePipelinedQueryQueue = function() {
+    if (!this._connected || this._pipelineInFlight) {
+      return;
+    }
+    if (this._queryQueue.length === 0) {
+      if (this.hasExecuted) {
+        this.emit("drain");
+      }
+      return;
+    }
+    this._pipelineInFlight = true;
+    const self = this;
+    const queries = [];
+    const nativeQueries = [];
+    const utils = require_utils2();
+    while (this._queryQueue.length > 0) {
+      const query = this._queryQueue.shift();
+      this.hasExecuted = true;
+      nativeQueries.push(query);
+      const values = query.values ? query.values.map(utils.prepareValue) : null;
+      const pipelineEntry = { text: query.text, name: query.name };
+      if (values) {
+        pipelineEntry.values = values;
+      }
+      if (query.name && this.namedQueries[query.name]) {
+        pipelineEntry._alreadyPrepared = true;
+      }
+      queries.push(pipelineEntry);
+    }
+    this.native.pipeline(queries, function(err, results) {
+      self._pipelineInFlight = false;
+      if (err) {
+        for (let i = 0;i < nativeQueries.length; i++) {
+          const q = nativeQueries[i];
+          q.native = self.native;
+          q.handleError(err);
+        }
+        self._pulsePipelinedQueryQueue();
+        return;
+      }
+      for (let i = 0;i < nativeQueries.length; i++) {
+        const q = nativeQueries[i];
+        const r = results[i];
+        q.native = self.native;
+        if (r.err) {
+          q.handleError(r.err);
+        } else {
+          if (q.name) {
+            self.namedQueries[q.name] = q.text;
+          }
+          q.state = "end";
+          q.emit("end", r.result);
+          if (q.callback) {
+            q.callback(null, r.result);
+          }
+        }
+        setImmediate(function() {
+          q.emit("_done");
+        });
+      }
+      self._pulsePipelinedQueryQueue();
+    });
+  };
   Client.prototype.cancel = function(query) {
     if (this._activeQuery === query) {
       this.native.cancel(function() {});
@@ -93808,9 +93932,21 @@ function registerWorkflowLifecycleHandlers(pi, deps) {
     return confirmSessionSwitch(reason, ctx);
   });
   pi.on("session_before_fork", async (_event, ctx) => confirmSessionSwitch("fork", ctx));
+  const notifyDiscoveryDiagnostics = (ctx) => {
+    if (!ctx?.ui)
+      return;
+    const diagnostics = formatStartupDiagnostics(null, runtimeState.discoveryRef.current);
+    if (diagnostics !== null)
+      ctx.ui.notify?.(diagnostics, "warning");
+  };
+  const warmWorkflowDiscovery = (ctx) => {
+    runtimeState.startWorkflowDiscoveryWarmup(() => notifyDiscoveryDiagnostics(ctx));
+  };
+  let startupResourcesDiscovered = false;
   pi.on("session_start", async (event, ctx) => {
     lifetime.release ??= getDurableBackendProcessOwner().injectedBackend === undefined ? acquireDbosLease() : async () => {};
     const reason = typeof event === "object" && event !== null && "reason" in event ? event.reason : undefined;
+    startupResourcesDiscovered = false;
     runtimeState.resetWorkflowDiscoveryForSession();
     await runtimeState.ensureWorkflowConfigLoaded();
     if (replacementStopsWorkflows(reason)) {
@@ -93825,13 +93961,7 @@ function registerWorkflowLifecycleHandlers(pi, deps) {
     else
       await stageControlRegistry.clearDetached();
     runtimeState.setNotificationsActive(true);
-    runtimeState.startWorkflowDiscoveryWarmup(() => {
-      if (!ctx?.ui)
-        return;
-      const diagnostics = formatStartupDiagnostics(null, runtimeState.discoveryRef.current);
-      if (diagnostics !== null)
-        ctx.ui.notify?.(diagnostics, "warning");
-    });
+    warmWorkflowDiscovery(ctx);
     if (ctx?.ui) {
       const diagnostics = formatStartupDiagnostics(runtimeState.configLoadRef.current, null);
       if (diagnostics !== null)
@@ -93841,6 +93971,16 @@ function registerWorkflowLifecycleHandlers(pi, deps) {
     }
     runtimeState.updateHostStageSessionDir(ctx?.sessionManager ?? pi.sessionManager);
   });
+  pi.on("resources_discover", (event, ctx) => {
+    if (eventReason(event) !== "startup")
+      return;
+    if (!startupResourcesDiscovered) {
+      startupResourcesDiscovered = true;
+      return;
+    }
+    runtimeState.refreshWorkflowDiscovery(() => notifyDiscoveryDiagnostics(ctx));
+    return;
+  });
   installCompactionHook(pi, store);
   pi.on("session_shutdown", async (event) => {
     const reason = eventReason(event);
@@ -111170,7 +111310,7 @@ class StageSessionController {
     await session?.closeWorkflowStageGeneration?.();
     this.artifactCapture.close();
   }
-  async promptWithFallback(text, sdkOptions, consumer = "prompt") {
+  async promptWithFallback(text, sdkOptions, consumer = "prompt", newSessionText) {
     if (this.effectiveStageOptions?.model === "auto")
       await this.preparePrompt(text);
     if (this.session !== undefined && this.activeCreation === undefined && this.ownedCreationPromise === undefined && this.startupWait.signal.aborted) {
@@ -111229,6 +111369,8 @@ class StageSessionController {
     let promptText = resumedText ?? text;
     if (await this.tryResumeCurrentSession(promptText, sdkOptions, candidates))
       return;
+    if (resumedText === undefined && newSessionText !== undefined)
+      promptText = newSessionText;
     let index = this.activeCandidateIndex ?? 0;
     while (index < candidates.length) {
       if (this.ownedCreationPromise !== undefined || this.activeCreation !== undefined) {
@@ -112529,11 +112671,12 @@ function createStageContext2(opts) {
         let structuredOutputError = STRUCTURED_OUTPUT_MISSING_ERROR;
         while (!structuredOutputCapture.called) {
           let nextPrompt = promptText;
+          let newSessionPrompt;
           let correctiveAttempts = 0;
           controller.beginStructuredOutputCandidateCycle();
           while (!structuredOutputCapture.called) {
             controller.resetStructuredOutputToolError();
-            await controller.promptWithFallback(nextPrompt, sdkOptions);
+            await controller.promptWithFallback(nextPrompt, sdkOptions, "prompt", newSessionPrompt);
             if (structuredOutputCapture.called)
               break;
             structuredOutputError = controller.structuredOutputFailureReason();
@@ -112541,6 +112684,9 @@ function createStageContext2(opts) {
               break;
             correctiveAttempts += 1;
             nextPrompt = formatStructuredOutputCorrectionPrompt(structuredOutputError, correctiveAttempts, hasOutputArtifact);
+            newSessionPrompt = `${promptText}
+
+${nextPrompt}`;
           }
           if (structuredOutputCapture.called)
             break;
@@ -117033,6 +117179,20 @@ function createWorkflowExtensionRuntimeState(pi, adapters, options = {}) {
       }
     }).catch(() => {});
   }
+  function refreshWorkflowDiscovery(onApplied) {
+    workflowDiscoveryGeneration += 1;
+    lazyDiscoveryPromise = null;
+    if (discoveryRef.current === null) {
+      startWorkflowDiscoveryWarmup(onApplied);
+      return;
+    }
+    const discoveryStart = workflowDiscoveryGeneration;
+    reloadWorkflowResources().then((report) => {
+      if (report.outcome === "applied" && isWorkflowDiscoveryCurrent(discoveryStart) && notificationsActive) {
+        onApplied?.();
+      }
+    }).catch(() => {});
+  }
   return {
     persistenceRef,
     mcpPort,
@@ -117049,6 +117209,7 @@ function createWorkflowExtensionRuntimeState(pi, adapters, options = {}) {
     ensureWorkflowResourcesLoaded,
     reloadWorkflowResources,
     startWorkflowDiscoveryWarmup,
+    refreshWorkflowDiscovery,
     runWithLifecycleSuppressedForPolicy(policy, fn) {
       return policy.mode !== "non_interactive" || policy.awaitTerminalRun !== true ? fn() : withWorkflowLifecycleNotificationsSuppressedAsync(lifecycleNotificationState, fn);
     },
@@ -117844,7 +118005,8 @@ function clonePackageSource(source) {
     ...source.skills === undefined ? {} : { skills: [...source.skills] },
     ...source.prompts === undefined ? {} : { prompts: [...source.prompts] },
     ...source.themes === undefined ? {} : { themes: [...source.themes] },
-    ...source.workflows === undefined ? {} : { workflows: [...source.workflows] }
+    ...source.workflows === undefined ? {} : { workflows: [...source.workflows] },
+    ...source.mcpServers === undefined ? {} : { mcpServers: [...source.mcpServers] }
   };
 }
 function packageSourcePath(source) {