@opengeni/db 4.6.1 → 4.6.2-canary.39853438886001

package/dist/canonical-human-identities.js

@@ -11,10 +11,10 @@ import {
   getCanonicalHumanIdentityProjection,
   synchronizeCanonicalHumanLoginBindings,
   validateCanonicalHumanSession
-} from "./chunk-XX5M3TJJ.js";
-import "./chunk-K2YN562I.js";
+} from "./chunk-LLMVN6ZL.js";
+import "./chunk-RXKWURPN.js";
 import "./chunk-SMCA7PNJ.js";
-import "./chunk-KMBU4KDY.js";
+import "./chunk-PKY6KWEP.js";
 import "./chunk-PZ5AY32C.js";
 export {
   CanonicalHumanIdentityAuthorityError,

package/dist/capability-reconciliation.js

@@ -1,8 +1,8 @@
 import {
   withWorkspaceRls
-} from "./chunk-K2YN562I.js";
+} from "./chunk-RXKWURPN.js";
 import "./chunk-SMCA7PNJ.js";
-import "./chunk-KMBU4KDY.js";
+import "./chunk-PKY6KWEP.js";
 import "./chunk-PZ5AY32C.js";
 
 // src/capability-reconciliation.ts

package/dist/{chunk-575FUJE2.js → chunk-2AYATAKX.js}

@@ -1,14 +1,14 @@
 import {
   rawRows,
   withWorkspaceRls
-} from "./chunk-K2YN562I.js";
+} from "./chunk-RXKWURPN.js";
 import {
   LOSSLESS_JSON_STRING_PREFIX,
   fromPostgresLosslessJson,
   sessionEvents,
   sessionTurns,
   toPostgresLosslessJson
-} from "./chunk-KMBU4KDY.js";
+} from "./chunk-PKY6KWEP.js";
 
 // src/session-event-slices.ts
 import { and, asc, desc, eq, gt, inArray, lt, notInArray, sql as sql2 } from "drizzle-orm";
@@ -275,4 +275,4 @@ export {
   readSessionMessageSliceInScope,
   listSessionEventSlices
 };
-//# sourceMappingURL=chunk-575FUJE2.js.map
+//# sourceMappingURL=chunk-2AYATAKX.js.map

package/dist/{chunk-ZWXHVIPE.js → chunk-2ND6WB3I.js}

@@ -2,14 +2,14 @@ import {
   buildConnectionTokenResolver,
   connectionScopeKey,
   refreshOAuthConnectionCredential
-} from "./chunk-C3FI3DJ5.js";
+} from "./chunk-HT7WH7RH.js";
 import {
   retainedProviderCommandPersistence
-} from "./chunk-TLCTS7CZ.js";
+} from "./chunk-LJA7IRDK.js";
 import {
   excludeUnclaimedHumanPromptEventFilter,
   readSessionMessageSliceInScope
-} from "./chunk-575FUJE2.js";
+} from "./chunk-2AYATAKX.js";
 import {
   PROVIDER_RECOVERY_SUSPENDED_AT,
   recoverProviderClockAfterDependencyWait
@@ -40,7 +40,7 @@ import {
   throwIfSubscriptionSettingLocked,
   updateSubscriptionCoreConnectionAccess,
   withSubscriptionCoreAccessScope
-} from "./chunk-MSAWMGWJ.js";
+} from "./chunk-3SPJSY7W.js";
 import {
   buildSubscriptionCoreCredentialResolver,
   decryptEnvironmentValue,
@@ -53,15 +53,15 @@ import {
   withSubscriptionCoreProviderPlacementWorld,
   withSubscriptionCoreRefreshLock,
   withSubscriptionUsageReading
-} from "./chunk-BH2HBICR.js";
+} from "./chunk-XNTQSEZT.js";
 import {
   memoByProvider,
   subscriptionCoreConnectionKind,
   subscriptionCoreProviderId
-} from "./chunk-GYAQYECK.js";
+} from "./chunk-EWOI5BFP.js";
 import {
   transactionallyAuthorizeEditableArtifactActor
-} from "./chunk-GSZZDYPV.js";
+} from "./chunk-HFJRCJWS.js";
 import {
   InteractionResourceConflictError,
   InteractionResourceNotFoundError,
@@ -108,11 +108,11 @@ import {
   withEffectiveSessionPolicy,
   withSessionListModelPolicy,
   withoutLapsedWorkspaceAccountBindings
-} from "./chunk-OFQDECMZ.js";
+} from "./chunk-QAG55O3L.js";
 import {
   withOrganizationIntegrationAcquisition,
   withOrganizationIntegrationPolicyFence
-} from "./chunk-4LVV4R4F.js";
+} from "./chunk-CUFE2VRQ.js";
 import {
   SESSION_DISCOVERY_CONTROL_TITLE_MAX_CHARS,
   SessionCommandIdempotencyError,
@@ -209,7 +209,7 @@ import {
   withPoolWakeServiceScopeInTransaction,
   workspaceControlLockBudget,
   workspaceControlRequestLockTimeoutMs
-} from "./chunk-7P67EF5K.js";
+} from "./chunk-SLTEPCXD.js";
 import {
   ClaudeSubscriptionReconnectRequired,
   SUBSCRIPTION_CORE_CLAUDE,
@@ -253,7 +253,7 @@ import {
   withCodexTokenDeadline,
   withoutRetiredProviderKeys,
   writeSubscriptionSessionBinding
-} from "./chunk-VXCIFQX7.js";
+} from "./chunk-PB7JKMHJ.js";
 import {
   CreditDebitAttribution,
   beginPlanCreditSettlement,
@@ -272,7 +272,7 @@ import {
   planPostUseCreditAllocations,
   readSubscriptionPlanCatalog,
   upsertOrganizationPlan
-} from "./chunk-TIF52FMJ.js";
+} from "./chunk-JSXB6FWX.js";
 import {
   external_exports
 } from "./chunk-4UJCXBR2.js";
@@ -308,7 +308,7 @@ import {
   withWorkspaceSubjectSessionActivityRls,
   withWorkspaceUsageLock,
   withoutSessionRlsActor
-} from "./chunk-K2YN562I.js";
+} from "./chunk-RXKWURPN.js";
 import {
   databaseBootConnectOptions
 } from "./chunk-SMCA7PNJ.js";
@@ -478,6 +478,7 @@ import {
   sessionHumanInputRequests,
   sessionListSnapshots,
   sessionMcpServers,
+  sessionMessageSearchTexts,
   sessionPendingToolCalls,
   sessionPins,
   sessionRealtimeConnections,
@@ -555,7 +556,7 @@ import {
   workspaceWebhookDeliveries,
   workspaceWebhooks,
   workspaces
-} from "./chunk-KMBU4KDY.js";
+} from "./chunk-PKY6KWEP.js";
 
 // src/migrate.ts
 import { readdir, readFile } from "fs/promises";
@@ -1299,6 +1300,9 @@ function batchedBackfillTransactionLocalSetting(file) {
       value: "1"
     };
   }
+  if (file === "0910_session_message_search_backfill.sql") {
+    return { guc: "opengeni.session_message_search_backfill_v1", value: "1" };
+  }
   return null;
 }
 
@@ -15970,6 +15974,7 @@ var Cursor = external_exports.object({
   matched: external_exports.number().int().nonnegative().safe(),
   occurrences: external_exports.number().int().nonnegative().safe()
 }).strict();
+var SESSION_MESSAGE_SEARCH_WINDOW = 2e3;
 var SessionMessageSearchCursorError = class extends Error {
   constructor() {
     super("Invalid session message search cursor or changed search scope");
@@ -16074,31 +16079,50 @@ async function scanSessionMessages(db, workspaceId, request, filters, authorityI
   const matches = [];
   let exhausted = false;
   const e = sessionEvents;
-  {
-    const boundary = position ? position.skipSession ? sql46`${e.sessionId} > ${position.sessionId}::uuid` : sql46`(${e.sessionId} > ${position.sessionId}::uuid or (${e.sessionId} = ${position.sessionId}::uuid and ${position.done ? sql46`${e.sequence} > ${position.sequence}` : sql46`${e.sequence} >= ${position.sequence}`}))` : sql46`true`;
+  const c = sessionMessageSearchTexts;
+  await db.execute(
+    sql46`select pg_catalog.set_config('opengeni.session_message_search_pattern', ${sessionMessagePostgresPattern(request.query) ?? ""}, true)`
+  );
+  const limit = request.limit ?? 20;
+  const budgetStarted = performance.now();
+  let slots = 0;
+  for (; ; ) {
+    signal?.throwIfAborted();
+    const boundary = position ? position.skipSession ? sql46`${c.sessionId} > ${position.sessionId}::uuid` : sql46`(${c.sessionId}, ${c.sequence}) ${position.done ? sql46`>` : sql46`>=`} (${position.sessionId}::uuid, ${position.sequence}::int)` : sql46`true`;
     const textMatch = sessionMessageTextMatchSql(request.query, sql46`${e.payload}->>'text'`);
     const rows2 = await db.execute(sql46`
       with eligible as materialized (
         select ${e.id} as id, ${e.sessionId} as session_id, ${e.sequence} as sequence,
           ${textMatch.matched} as matched, ${textMatch.encoded} as encoded
-        from ${e}
+        from ${c}
+        inner join ${e} on ${e.workspaceId} = ${c.workspaceId}
+          and ${e.sessionId} = ${c.sessionId} and ${e.sequence} = ${c.sequence}
         inner join ${sessions}
           on ${sessions.workspaceId} = ${e.workspaceId} and ${sessions.id} = ${e.sessionId}
         where ${and19(
+      eq20(c.workspaceId, workspaceId),
       eq20(e.workspaceId, workspaceId),
       ...filters,
       boundary,
-      request.sessionId ? eq20(e.sessionId, request.sessionId) : sql46`true`,
+      request.sessionId ? eq20(c.sessionId, request.sessionId) : sql46`true`,
       sql46`${e.type} in ('user.message', 'agent.message.completed')`,
       sql46`(${e.turnAssociation} is null or ${e.turnAssociation} = 'current')`,
       sql46`${e.duplicateOfEventId} is null`,
       sql46`jsonb_typeof(${e.payload}->'text') = 'string'`
     )}
+        order by ${c.sessionId}, ${c.sequence}
+        limit ${sql46.raw(String(SESSION_MESSAGE_SEARCH_WINDOW))}
       ), visited as (
         select count(*)::int as visited from eligible
+      ), last_visited as (
+        select eligible.session_id as "lastSessionId", eligible.sequence as "lastSequence"
+        from eligible
+        order by eligible.session_id desc, eligible.sequence desc
+        limit 1
       )
-      select visited.visited, candidate.*
+      select visited.visited, last_visited.*, candidate.*
       from visited
+      left join last_visited on true
       left join lateral (
         select * from (
           select valid.*${request.groupBy === "session" ? sql46`, (count(*) filter (where valid.definite) over (partition by valid."sessionId"
@@ -16153,6 +16177,8 @@ async function scanSessionMessages(db, workspaceId, request, filters, authorityI
       order by candidate.rank
     `);
     const visited = rows2[0]?.visited ?? 0;
+    const lastSessionId = rows2[0]?.lastSessionId ?? null;
+    const lastSequence = rows2[0]?.lastSequence ?? null;
     const identities = rows2.flatMap(
       (row) => row.rank === null || row.sessionId === null || row.eventId === null || row.sequence === null || row.type === null ? [] : [
         {
@@ -16166,16 +16192,16 @@ async function scanSessionMessages(db, workspaceId, request, filters, authorityI
       ]
     );
     const visitedBefore = scanned - (position && !position.done ? 1 : 0);
-    const budgetStarted = performance.now();
     let identityIndex = 0;
-    for (let window = 0; window < 32 && matches.length < (request.limit ?? 20); window++) {
+    let interrupted = false;
+    for (; matches.length < limit; slots++) {
       signal?.throwIfAborted();
-      if (window > 0 && performance.now() - budgetStarted >= 1500) break;
-      const identity = identities[identityIndex];
-      if (!identity) {
-        exhausted = identities.length < 33;
+      if (slots >= 32 || slots > 0 && performance.now() - budgetStarted >= 1500) {
+        interrupted = true;
         break;
       }
+      const identity = identities[identityIndex];
+      if (!identity) break;
       const continuing = position && !position.done && position.sessionId === identity.sessionId && position.sequence === identity.sequence;
       if (!continuing) {
         scanned = visitedBefore + identity.rank;
@@ -16192,7 +16218,7 @@ async function scanSessionMessages(db, workspaceId, request, filters, authorityI
       const offset = position.offset;
       const utf16Offset = position.utf16Offset;
       const smallText = fromPostgresLosslessJson(identity.smallText, identity.codec);
-      const scalarWindows = typeof smallText === "string" ? 1 : Math.min(4, 32 - window);
+      const scalarWindows = typeof smallText === "string" ? 1 : Math.min(4, 32 - slots);
       const slice = typeof smallText === "string" ? {
         offset: 0,
         total: smallText.length,
@@ -16214,7 +16240,7 @@ async function scanSessionMessages(db, workspaceId, request, filters, authorityI
         offset,
         scalarWindows
       );
-      window += scalarWindows - 1;
+      slots += scalarWindows - 1;
       if (!slice || slice.omitted) {
         position = { ...position, done: true };
         identityIndex++;
@@ -16223,7 +16249,7 @@ async function scanSessionMessages(db, workspaceId, request, filters, authorityI
       pattern.lastIndex = Math.max(0, position.minimumMatchOffset - utf16Offset);
       let found;
       const messageId = fromPostgresLosslessJson(identity.messageId, identity.codec);
-      while (matches.length < (request.limit ?? 20) && (found = pattern.exec(slice.text))) {
+      while (matches.length < limit && (found = pattern.exec(slice.text))) {
         matches.push({
           sessionId: identity.sessionId,
           sessionTitle: identity.sessionTitle,
@@ -16254,7 +16280,7 @@ async function scanSessionMessages(db, workspaceId, request, filters, authorityI
         continue;
       }
       const units = slice.unit === "utf16" ? slice.text.length : Array.from(slice.text).length;
-      if (matches.length >= (request.limit ?? 20)) {
+      if (matches.length >= limit) {
         if (offset + units >= slice.total && position.minimumMatchOffset >= utf16Offset + slice.text.length) {
           position = { ...position, done: true };
           identityIndex++;
@@ -16281,8 +16307,27 @@ async function scanSessionMessages(db, workspaceId, request, filters, authorityI
         done: false
       };
     }
-    if (identityIndex >= identities.length && identities.length < 33) exhausted = true;
-    if (exhausted) scanned = visitedBefore + visited;
+    const drained = identityIndex >= identities.length && identities.length < 33;
+    if (drained && (visited < SESSION_MESSAGE_SEARCH_WINDOW || lastSessionId === null)) {
+      scanned = visitedBefore + visited;
+      exhausted = true;
+      break;
+    }
+    if (matches.length >= limit || interrupted) break;
+    if (drained && lastSessionId !== null && lastSequence !== null) {
+      scanned = visitedBefore + visited;
+      if (!(position?.skipSession && position.sessionId === lastSessionId))
+        position = {
+          sessionId: lastSessionId,
+          sequence: lastSequence,
+          offset: 0,
+          utf16Offset: 0,
+          minimumMatchOffset: 0,
+          matched: false,
+          done: true
+        };
+    }
+    if (performance.now() - budgetStarted >= 1500) break;
   }
   signal?.throwIfAborted();
   return {
@@ -70132,10 +70177,10 @@ async function isCodexBilledTurn(input) {
   if (!input.settings.codexSubscriptionEnabled) return false;
   if (input.acceptedTurn) {
     const accountId = input.accountId ?? (await rlsContextForWorkspace(input.db, input.workspaceId)).accountId;
-    const { readCodexCutoverDisposition: readCodexCutoverDisposition2 } = await import("./subscription-core-codex-compat-AK2UMLE3.js");
+    const { readCodexCutoverDisposition: readCodexCutoverDisposition2 } = await import("./subscription-core-codex-compat-SLRJTJ4T.js");
     const disposition = await readCodexCutoverDisposition2(input.db, accountId, input.workspaceId);
     if (disposition !== "core") return false;
-    const { subscriptionCoreAcceptedCodexTurnIsFunded } = await import("./subscription-core-codex-L7NPFWAH.js");
+    const { subscriptionCoreAcceptedCodexTurnIsFunded } = await import("./subscription-core-codex-D3IQ4SYO.js");
     return subscriptionCoreAcceptedCodexTurnIsFunded(input.db, {
       accountId,
       workspaceId: input.workspaceId,
@@ -114727,4 +114772,4 @@ export {
   isCodexBilledModel2 as isCodexBilledModel,
   sql151 as sql
 };
-//# sourceMappingURL=chunk-ZWXHVIPE.js.map
+//# sourceMappingURL=chunk-2ND6WB3I.js.map