@opengeni/db 0.27.7 → 0.27.9

package/src/insights.ts

@@ -1,7 +1,7 @@
 import { and, desc, eq, gte, inArray, lt, sql } from "drizzle-orm";
 import { alias } from "drizzle-orm/pg-core";
-import type { Database } from "./index";
-import { rlsContextForWorkspace, withRlsContext } from "./index";
+import type { Database } from "./database";
+import { rlsContextForWorkspace, withRlsContext } from "./database";
 import * as schema from "./schema";
 
 const UUID_RE = /^[0-9a-f]{8}-[0-9a-f]{4}-[1-8][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i;

package/src/memory-governance.ts

@@ -1,6 +1,6 @@
 import { sql } from "drizzle-orm";
-import type { Database } from "./index";
-import { setSubjectRlsContext, withWorkspaceRls } from "./index";
+import type { Database } from "./database";
+import { setSubjectRlsContext, withWorkspaceRls } from "./database";
 import {
   hashMemoryOperationPlan,
   hashMemoryRevertPlan,

package/src/new-session-drafts.ts

@@ -7,7 +7,7 @@ import type {
   ToolRef,
 } from "@opengeni/contracts";
 import { and, eq } from "drizzle-orm";
-import type { Database } from "./index";
+import type { Database } from "./database";
 import * as schema from "./schema";
 
 export type NewSessionDraftRow = typeof schema.newSessionDrafts.$inferSelect;

package/src/preference-registry.ts

@@ -21,8 +21,8 @@ import {
   type PreferenceRegistryTrust,
 } from "@opengeni/contracts";
 import { and, asc, desc, eq, gt, inArray, isNull, lte, or, sql, type SQL } from "drizzle-orm";
-import type { Database } from "./index";
-import { setSubjectRlsContext, withWorkspaceRls, withWorkspaceSubjectRls } from "./index";
+import type { Database } from "./database";
+import { setSubjectRlsContext, withWorkspaceRls, withWorkspaceSubjectRls } from "./database";
 import { nestedPostgresSqlState, safeDatabaseErrorFacts } from "./persistence-errors";
 import * as schema from "./schema";
 

package/src/provision-roles.ts

@@ -1,5 +1,5 @@
 import postgres from "postgres";
-import type { RlsStrategy } from "./index";
+import type { RlsStrategy } from "./database";
 import {
   RUNTIME_FULL_DML_TABLES,
   RUNTIME_READ_INSERT_TABLES,

package/src/runtime-posture.ts

@@ -1,5 +1,5 @@
 import { sql } from "drizzle-orm";
-import type { Database, RlsStrategy } from "./index";
+import type { Database, RlsStrategy } from "./database";
 import {
   classifyRoleRelationships,
   roleRelationshipsCatalogQuery,

package/src/schema.ts

@@ -3246,6 +3246,7 @@ export const sessionWorkflowWakeOutbox = pgTable(
     temporalWorkflowId: text("temporal_workflow_id").notNull(),
     wakeRevision: bigint("wake_revision", { mode: "number" }).notNull().default(1),
     deliveredRevision: bigint("delivered_revision", { mode: "number" }).notNull().default(0),
+    controlRevision: bigint("control_revision", { mode: "number" }).notNull().default(0),
     reason: text("reason").notNull(),
     attempts: integer("attempts").notNull().default(0),
     nextAttemptAt: timestamp("next_attempt_at", { withTimezone: true }).notNull().defaultNow(),
@@ -3262,6 +3263,10 @@ export const sessionWorkflowWakeOutbox = pgTable(
       "session_workflow_wake_outbox_revision_safe_check",
       sql`${table.wakeRevision} <= 9007199254740991 and ${table.deliveredRevision} <= 9007199254740991`,
     ),
+    controlRevisionValid: check(
+      "session_workflow_wake_outbox_control_revision_check",
+      sql`${table.controlRevision} >= 0 and ${table.controlRevision} <= ${table.wakeRevision} and ${table.controlRevision} <= 9007199254740991`,
+    ),
     workspaceAccount: foreignKey({
       name: "session_workflow_wake_outbox_workspace_account_fk",
       columns: [table.workspaceId, table.accountId],
@@ -5823,6 +5828,7 @@ export const socialConnections = pgTable(
     workspaceId: uuid("workspace_id")
       .notNull()
       .references(() => workspaces.id, { onDelete: "cascade" }),
+    subjectId: text("subject_id"),
     provider: text("provider").notNull(),
     accountHandle: text("account_handle").notNull(),
     accountName: text("account_name"),
@@ -5840,13 +5846,20 @@ export const socialConnections = pgTable(
     updatedAt: timestamp("updated_at", { withTimezone: true }).notNull().defaultNow(),
   },
   (table) => ({
-    workspaceProviderHandle: uniqueIndex("social_connections_workspace_provider_handle_idx").on(
+    workspaceProviderHandle: uniqueIndex("social_connections_workspace_provider_handle_idx")
+      .on(table.workspaceId, table.provider, table.accountHandle)
+      .where(sql`${table.subjectId} is null`),
+    subjectProviderHandle: uniqueIndex("social_connections_subject_provider_handle_idx")
+      .on(table.workspaceId, table.subjectId, table.provider)
+      .where(sql`${table.subjectId} is not null`),
+    providerStatus: index("social_connections_workspace_provider_status_idx").on(
       table.workspaceId,
       table.provider,
-      table.accountHandle,
+      table.status,
     ),
-    providerStatus: index("social_connections_workspace_provider_status_idx").on(
+    subjectProviderStatus: index("social_connections_subject_provider_status_idx").on(
       table.workspaceId,
+      table.subjectId,
       table.provider,
       table.status,
     ),

package/src/scoped-knowledge.ts

@@ -21,8 +21,8 @@ import type {
   ScopedKnowledgeScope,
 } from "@opengeni/contracts";
 import { and, eq, inArray, sql } from "drizzle-orm";
-import type { Database } from "./index";
-import { setSubjectRlsContext, withRlsContext } from "./index";
+import type { Database } from "./database";
+import { setSubjectRlsContext, withRlsContext } from "./database";
 import { nestedPostgresSqlState, safeDatabaseErrorFacts } from "./persistence-errors";
 import * as schema from "./schema";
 

package/src/session-control.ts

@@ -7,7 +7,7 @@ import {
   type TurnInitiatorContext,
 } from "@opengeni/contracts";
 import { and, eq, inArray, sql } from "drizzle-orm";
-import type { Database } from "./index";
+import type { Database } from "./database";
 import * as schema from "./schema";
 import { closePendingSessionToolCallsInTransaction } from "./session-tool-call-settlement";
 import {
@@ -1494,12 +1494,13 @@ async function registerContinuableWakes(
       )
     ), upserted as (
       insert into ${schema.sessionWorkflowWakeOutbox} (
-        session_id, account_id, workspace_id, temporal_workflow_id, reason
+        session_id, account_id, workspace_id, temporal_workflow_id, reason, control_revision
       )
-      select session_id, account_id, workspace_id, temporal_workflow_id, ${input.reason}
+      select session_id, account_id, workspace_id, temporal_workflow_id, ${input.reason}, 1
       from eligible
       on conflict (session_id) do update set
         wake_revision = ${schema.sessionWorkflowWakeOutbox}.wake_revision + 1,
+        control_revision = ${schema.sessionWorkflowWakeOutbox}.wake_revision + 1,
         temporal_workflow_id = excluded.temporal_workflow_id,
         reason = excluded.reason,
         attempts = 0,
@@ -1611,12 +1612,13 @@ async function registerCancellationWakes(
         and session.id in (${sessionIds})
     ), upserted as (
       insert into ${schema.sessionWorkflowWakeOutbox} (
-        session_id, account_id, workspace_id, temporal_workflow_id, reason
+        session_id, account_id, workspace_id, temporal_workflow_id, reason, control_revision
       )
-      select session_id, account_id, workspace_id, temporal_workflow_id, 'session_cancelled'
+      select session_id, account_id, workspace_id, temporal_workflow_id, 'session_cancelled', 1
       from eligible
       on conflict (session_id) do update set
         wake_revision = ${schema.sessionWorkflowWakeOutbox}.wake_revision + 1,
+        control_revision = ${schema.sessionWorkflowWakeOutbox}.wake_revision + 1,
         temporal_workflow_id = excluded.temporal_workflow_id,
         reason = excluded.reason,
         attempts = 0,
@@ -1641,6 +1643,7 @@ export async function registerSessionWorkflowWakeInTransaction(
     sessionId: string;
     temporalWorkflowId: string;
     reason: string;
+    controlRequested?: boolean;
   },
 ): Promise<number> {
   const [row] = await db
@@ -1650,6 +1653,7 @@ export async function registerSessionWorkflowWakeInTransaction(
       workspaceId: input.workspaceId,
       sessionId: input.sessionId,
       temporalWorkflowId: input.temporalWorkflowId,
+      controlRevision: input.controlRequested ? 1 : 0,
       reason: input.reason,
     })
     .onConflictDoUpdate({
@@ -1657,6 +1661,9 @@ export async function registerSessionWorkflowWakeInTransaction(
       set: {
         temporalWorkflowId: input.temporalWorkflowId,
         wakeRevision: sql`${schema.sessionWorkflowWakeOutbox.wakeRevision} + 1`,
+        controlRevision: input.controlRequested
+          ? sql`${schema.sessionWorkflowWakeOutbox.wakeRevision} + 1`
+          : sql`${schema.sessionWorkflowWakeOutbox.controlRevision}`,
         reason: input.reason,
         attempts: 0,
         nextAttemptAt: new Date(),

package/src/session-queue-commands.ts

@@ -12,7 +12,7 @@ import {
   type TurnExecutionPolicyV1,
 } from "@opengeni/contracts";
 import { and, asc, eq, inArray, sql } from "drizzle-orm";
-import type { Database } from "./index";
+import type { Database } from "./database";
 import { sanitizeEventPayload } from "./event-payload-sanitizer";
 import { closePendingSessionToolCallsInTransaction } from "./session-tool-call-settlement";
 import {
@@ -1180,6 +1180,7 @@ export async function steerQueuedTurnInTransaction(
     sessionId: input.sessionId,
     temporalWorkflowId: session.temporalWorkflowId ?? `session-${input.sessionId}`,
     reason: "queue_steer",
+    controlRequested: true,
   });
   const receipt = await updateSessionCommandReceiptResult(db, reserved.receipt.id, {
     controlRevision: resumed.revision,
@@ -1739,6 +1740,7 @@ export async function submitHumanPromptInTransaction(
     sessionId: input.sessionId,
     temporalWorkflowId: workflowId,
     reason: input.delivery === "steer" ? "prompt_steer" : "prompt_send",
+    controlRequested: input.delivery === "steer",
   });
   await db.insert(schema.auditEvents).values({
     accountId: input.accountId,
@@ -2183,6 +2185,7 @@ export async function steerAgentSessionInTransaction(
     sessionId: input.targetSessionId,
     temporalWorkflowId: workflowId,
     reason: "agent_steer",
+    controlRequested: true,
   });
   await db
     .update(schema.sessions)

package/src/session-realtime-context.ts

@@ -3,7 +3,7 @@ import { createHash } from "node:crypto";
 import { ReasoningEffort } from "@opengeni/contracts";
 import { and, asc, desc, eq, inArray, isNull, sql } from "drizzle-orm";
 
-import type { Database } from "./index";
+import type { Database } from "./database";
 import * as schema from "./schema";
 import { submitHumanPromptInTransaction } from "./session-queue-commands";
 

package/src/session-realtime-ledger.ts

@@ -4,7 +4,7 @@ import { LatencyMode, ReasoningEffort, type SessionRealtimeMode } from "@opengen
 import { and, asc, desc, eq, inArray, isNull, or, sql } from "drizzle-orm";
 
 import { sanitizeEventPayload } from "./event-payload-sanitizer";
-import type { Database } from "./index";
+import type { Database } from "./database";
 import * as schema from "./schema";
 import {
   assertSessionRealtimeOwnerInTransaction,

package/src/session-realtime-mirror.ts

@@ -4,7 +4,7 @@ import type { HumanInputQuestion, HumanInputResponse } from "@opengeni/contracts
 import { and, desc, eq, gt, sql } from "drizzle-orm";
 
 import { sanitizeEventPayload } from "./event-payload-sanitizer";
-import type { Database } from "./index";
+import type { Database } from "./database";
 import * as schema from "./schema";
 
 const SESSION_REALTIME_MIRROR_MAX_TEXT_BYTES = 131_072;

package/src/session-realtime-state.ts

@@ -1,6 +1,6 @@
 import { and, eq, gt } from "drizzle-orm";
 
-import type { Database } from "./index";
+import type { Database } from "./database";
 import * as schema from "./schema";
 
 export async function sessionRealtimeIsActiveInTransaction(

package/src/session-realtime-terminal.ts

@@ -3,7 +3,7 @@ import { createHash } from "node:crypto";
 import { and, eq, inArray, sql } from "drizzle-orm";
 
 import { sanitizeEventPayload } from "./event-payload-sanitizer";
-import type { Database } from "./index";
+import type { Database } from "./database";
 import { mirrorSessionRealtimeContextInTransaction } from "./session-realtime-mirror";
 import * as schema from "./schema";
 

package/src/session-realtime.ts

@@ -9,7 +9,7 @@ import type {
 import { and, asc, eq, gt, inArray, isNull, or } from "drizzle-orm";
 
 import { sanitizeEventPayload } from "./event-payload-sanitizer";
-import type { Database } from "./index";
+import type { Database } from "./database";
 import { flushSessionRealtimeTranscriptTailInTransaction } from "./session-realtime-context";
 import {
   evaluateSessionControl,

package/src/session-tool-call-settlement.ts

@@ -1,7 +1,7 @@
 import type { SessionEvent } from "@opengeni/contracts";
 import { boundModelToolOutputItem } from "@opengeni/codex";
 import { and, asc, eq, sql } from "drizzle-orm";
-import type { Database } from "./index";
+import type { Database } from "./database";
 import { sanitizeEventPayload, sanitizeModelPayload } from "./event-payload-sanitizer";
 import * as schema from "./schema";
 

package/src/turn-initiator.ts

@@ -4,7 +4,7 @@ import {
   type TurnInitiatorContext,
 } from "@opengeni/contracts";
 import { and, eq } from "drizzle-orm";
-import type { Database } from "./index";
+import type { Database } from "./database";
 import type { SessionCommandActor } from "./session-control";
 import * as schema from "./schema";
 

package/src/workspace-artifacts.ts

@@ -6,8 +6,8 @@ import type {
   WorkspaceArtifactVersion,
 } from "@opengeni/contracts";
 import { and, desc, eq, lt, or, sql } from "drizzle-orm";
-import type { Database } from "./index";
-import { withRlsContext, withWorkspaceRls } from "./index";
+import type { Database } from "./database";
+import { withRlsContext, withWorkspaceRls } from "./database";
 import * as schema from "./schema";
 
 type ArtifactRow = typeof schema.workspaceArtifacts.$inferSelect;

package/src/workspace-instruction-policies.ts

@@ -25,8 +25,8 @@ import type {
   WorkspaceInstructionPolicyTarget,
 } from "@opengeni/contracts";
 import { and, asc, desc, eq, inArray, isNull, lt, or, sql, type SQL } from "drizzle-orm";
-import type { Database } from "./index";
-import { withRlsContext, withWorkspaceRls, withWorkspaceSubjectRls } from "./index";
+import type { Database } from "./database";
+import { withRlsContext, withWorkspaceRls, withWorkspaceSubjectRls } from "./database";
 import { nestedPostgresSqlState } from "./persistence-errors";
 import * as schema from "./schema";