@dudousxd/nestjs-agent-store-drizzle 0.22.1 → 0.24.0

package/dist/index.js

@@ -16,6 +16,8 @@ var agentThread = sqliteTable("agent_thread", {
   defaultAgent: text("default_agent"),
   /** The model every turn on this thread runs on unless the send names one; `null` → default. */
   model: text("model"),
+  /** The persona a send on this thread runs under when it names none; `null` → the agent's default. */
+  persona: text("persona"),
   /** Why the thread's message queue stopped draining; `null` → it drains. */
   queuePause: text("queue_pause", {
     mode: "json"
@@ -55,6 +57,8 @@ var agentMessage = sqliteTable("agent_message", {
     mode: "json"
   }).$type(),
   agentName: text("agent_name"),
+  /** The persona the turn that wrote this message ran under; `null` → none. */
+  persona: text("persona"),
   runId: text("run_id"),
   /** The step's streamed thinking; `null` when the model produced none. */
   reasoning: text("reasoning"),
@@ -67,6 +71,12 @@ var agentMessage = sqliteTable("agent_message", {
   feedback: text("feedback", {
     mode: "json"
   }).$type(),
+  /**
+   * The message's place in its thread, 1-based, assigned on append — what orders a transcript.
+   * `created_at` cannot: two messages of one turn routinely share a timestamp, and the uuid that
+   * broke the tie is random. `null` on a row from before the column; those sort first.
+   */
+  seq: integer("seq"),
   createdAt: integer("created_at", {
     mode: "timestamp_ms"
   }).notNull()
@@ -127,6 +137,8 @@ var agentQueuedMessage = sqliteTable("agent_queued_message", {
     mode: "json"
   }).$type(),
   agentName: text("agent_name"),
+  /** The persona the send resolved — what the message starts under. */
+  persona: text("persona"),
   model: text("model"),
   pageContext: text("page_context", {
     mode: "json"
@@ -299,285 +311,721 @@ var agentSchema = {
   agentRun,
   agentMemory,
   ragIngestionLog,
-  agentConfirmToken
+  agentConfirmToken,
+  agentStreamFrame
 };
 
 // src/ensure-schema.ts
-import { sql } from "drizzle-orm";
-var statements = [
-  `CREATE TABLE IF NOT EXISTS agent_thread (
-    id TEXT PRIMARY KEY NOT NULL,
-    actor_ref TEXT NOT NULL,
-    tenant_ref TEXT,
-    title TEXT NOT NULL,
-    transient INTEGER NOT NULL DEFAULT 0,
-    active_stream_id TEXT,
-    default_agent TEXT,
-    model TEXT,
-    queue_pause TEXT,
-    created_at INTEGER NOT NULL,
-    updated_at INTEGER NOT NULL,
-    deleted_at INTEGER
-  )`,
-  `CREATE INDEX IF NOT EXISTS agent_thread_actor_updated_idx
-    ON agent_thread (actor_ref, updated_at)`,
-  `CREATE TABLE IF NOT EXISTS agent_message (
-    id TEXT PRIMARY KEY NOT NULL,
-    thread_id TEXT NOT NULL REFERENCES agent_thread(id) ON DELETE CASCADE,
-    role TEXT NOT NULL,
-    content TEXT NOT NULL,
-    tool_calls TEXT,
-    tool_results TEXT,
-    attachments TEXT,
-    follow_ups TEXT,
-    usage TEXT,
-    agent_name TEXT,
-    run_id TEXT,
-    reasoning TEXT,
-    reasoning_ms INTEGER,
-    ui TEXT,
-    feedback TEXT,
-    created_at INTEGER NOT NULL
-  )`,
-  `CREATE INDEX IF NOT EXISTS agent_message_thread_created_idx
-    ON agent_message (thread_id, created_at)`,
-  `CREATE TABLE IF NOT EXISTS agent_queued_message (
-    id TEXT PRIMARY KEY NOT NULL,
-    thread_id TEXT NOT NULL REFERENCES agent_thread(id) ON DELETE CASCADE,
-    actor TEXT NOT NULL,
-    content TEXT NOT NULL,
-    attachments TEXT,
-    agent_name TEXT,
-    model TEXT,
-    page_context TEXT,
-    interrupt INTEGER NOT NULL DEFAULT 0,
-    position INTEGER NOT NULL,
-    created_at INTEGER NOT NULL,
-    updated_at INTEGER NOT NULL
-  )`,
-  `CREATE INDEX IF NOT EXISTS agent_queued_message_thread_position_idx
-    ON agent_queued_message (thread_id, position)`,
-  `CREATE TABLE IF NOT EXISTS agent_tool_call (
-    id TEXT PRIMARY KEY NOT NULL,
-    message_id TEXT NOT NULL REFERENCES agent_message(id) ON DELETE CASCADE,
-    tool_name TEXT NOT NULL,
-    tool_type TEXT NOT NULL,
-    input TEXT,
-    output TEXT,
-    status TEXT NOT NULL,
-    executed_by_ref TEXT,
-    execution_ms INTEGER,
-    error TEXT,
-    created_at INTEGER NOT NULL,
-    executed_at INTEGER,
-    run_id TEXT,
-    approver TEXT,
-    expires_at INTEGER,
-    remember INTEGER,
-    decided_via TEXT
-  )`,
-  `CREATE INDEX IF NOT EXISTS agent_tool_call_message_idx
-    ON agent_tool_call (message_id)`,
-  `CREATE TABLE IF NOT EXISTS agent_token_usage (
-    id TEXT PRIMARY KEY NOT NULL,
-    thread_id TEXT NOT NULL REFERENCES agent_thread(id) ON DELETE CASCADE,
-    actor_ref TEXT NOT NULL,
-    message_id TEXT,
-    model_id TEXT NOT NULL,
-    purpose TEXT NOT NULL,
-    input_tokens INTEGER NOT NULL,
-    output_tokens INTEGER NOT NULL,
-    cache_write_tokens INTEGER,
-    cache_read_tokens INTEGER,
-    cost_usd REAL,
-    created_at INTEGER NOT NULL
-  )`,
-  `CREATE INDEX IF NOT EXISTS agent_token_usage_actor_created_idx
-    ON agent_token_usage (actor_ref, created_at)`,
-  `CREATE TABLE IF NOT EXISTS agent_model_pricing (
-    id TEXT PRIMARY KEY NOT NULL,
-    model_id TEXT NOT NULL,
-    input_price_per_1m REAL NOT NULL,
-    output_price_per_1m REAL NOT NULL,
-    cache_write_price_per_1m REAL,
-    cache_read_price_per_1m REAL,
-    effective_from INTEGER NOT NULL,
-    is_current INTEGER NOT NULL
-  )`,
-  `CREATE TABLE IF NOT EXISTS agent_run (
-    id TEXT PRIMARY KEY NOT NULL,
-    thread_id TEXT NOT NULL REFERENCES agent_thread(id) ON DELETE CASCADE,
-    actor_ref TEXT NOT NULL,
-    agent_name TEXT,
-    status TEXT NOT NULL,
-    duration_ms INTEGER,
-    error_code TEXT,
-    error_message TEXT,
-    retries INTEGER NOT NULL DEFAULT 0,
-    started_at INTEGER NOT NULL,
-    settled_at INTEGER,
-    prompt_hash TEXT,
-    parent_run_id TEXT
-  )`,
-  `CREATE INDEX IF NOT EXISTS agent_run_started_idx
-    ON agent_run (started_at)`,
-  // `key` and `text` are quoted: both are keywords in at least one engine this DDL is read by, and
-  // they are the SPI's own field names, which is worth more here than dodging the quoting.
-  `CREATE TABLE IF NOT EXISTS agent_memory (
-    id TEXT PRIMARY KEY NOT NULL,
-    scope TEXT NOT NULL,
-    "key" TEXT NOT NULL,
-    "text" TEXT NOT NULL,
-    origin_author TEXT NOT NULL,
-    origin_thread_id TEXT,
-    origin_run_id TEXT,
-    origin_actor_ref TEXT,
-    pinned INTEGER NOT NULL DEFAULT 0,
-    created_at INTEGER NOT NULL,
-    updated_at INTEGER NOT NULL
-  )`,
-  `CREATE UNIQUE INDEX IF NOT EXISTS agent_memory_scope_key_uq
-    ON agent_memory (scope, "key")`,
-  `CREATE TABLE IF NOT EXISTS rag_ingestion_log (
-    document_id TEXT PRIMARY KEY NOT NULL,
-    status TEXT NOT NULL,
-    collection TEXT,
-    owner_type TEXT,
-    owner_id TEXT,
-    source TEXT,
-    mime_type TEXT,
-    size INTEGER,
-    chunks INTEGER,
-    reason TEXT,
-    error TEXT,
-    created_at INTEGER NOT NULL,
-    updated_at INTEGER NOT NULL
-  )`,
-  `CREATE INDEX IF NOT EXISTS rag_ingestion_log_collection_idx
-    ON rag_ingestion_log (collection, updated_at)`,
-  `CREATE TABLE IF NOT EXISTS agent_stream_frame (
-    run_id TEXT NOT NULL,
-    seq INTEGER NOT NULL,
-    frame TEXT,
-    error TEXT,
-    created_at INTEGER NOT NULL,
-    PRIMARY KEY (run_id, seq)
-  )`,
-  `CREATE TABLE IF NOT EXISTS agent_confirm_token (
-    hash TEXT PRIMARY KEY NOT NULL,
-    actor_ref TEXT NOT NULL,
-    tool TEXT NOT NULL,
-    expires_at INTEGER NOT NULL,
-    created_at INTEGER NOT NULL
-  )`,
-  `CREATE INDEX IF NOT EXISTS agent_confirm_token_expires_idx
-    ON agent_confirm_token (expires_at)`
-];
-var additiveColumns = [
-  {
-    table: "agent_message",
-    column: "run_id",
-    ddl: "ALTER TABLE agent_message ADD COLUMN run_id TEXT"
-  },
-  {
-    table: "agent_message",
-    column: "attachments",
-    ddl: "ALTER TABLE agent_message ADD COLUMN attachments TEXT"
-  },
-  {
-    table: "agent_thread",
-    column: "default_agent",
-    ddl: "ALTER TABLE agent_thread ADD COLUMN default_agent TEXT"
-  },
-  {
-    table: "agent_run",
-    column: "parent_run_id",
-    ddl: "ALTER TABLE agent_run ADD COLUMN parent_run_id TEXT"
-  },
-  {
-    table: "agent_message",
-    column: "reasoning",
-    ddl: "ALTER TABLE agent_message ADD COLUMN reasoning TEXT"
-  },
-  {
-    table: "agent_message",
-    column: "reasoning_ms",
-    ddl: "ALTER TABLE agent_message ADD COLUMN reasoning_ms INTEGER"
-  },
-  {
-    table: "agent_message",
-    column: "ui",
-    ddl: "ALTER TABLE agent_message ADD COLUMN ui TEXT"
-  },
-  {
-    table: "agent_message",
-    column: "feedback",
-    ddl: "ALTER TABLE agent_message ADD COLUMN feedback TEXT"
-  },
-  {
-    table: "agent_thread",
-    column: "model",
-    ddl: "ALTER TABLE agent_thread ADD COLUMN model TEXT"
-  },
-  {
-    table: "agent_tool_call",
-    column: "approver",
-    ddl: "ALTER TABLE agent_tool_call ADD COLUMN approver TEXT"
-  },
-  {
-    table: "agent_tool_call",
-    column: "expires_at",
-    ddl: "ALTER TABLE agent_tool_call ADD COLUMN expires_at INTEGER"
-  },
-  {
-    table: "agent_tool_call",
-    column: "remember",
-    ddl: "ALTER TABLE agent_tool_call ADD COLUMN remember INTEGER"
-  },
-  {
-    table: "agent_thread",
-    column: "queue_pause",
-    ddl: "ALTER TABLE agent_thread ADD COLUMN queue_pause TEXT"
-  },
-  {
-    table: "agent_tool_call",
-    column: "decided_via",
-    ddl: "ALTER TABLE agent_tool_call ADD COLUMN decided_via TEXT"
+import { getTableName, sql } from "drizzle-orm";
+import { getTableConfig as mysqlTableConfig } from "drizzle-orm/mysql-core";
+import { getTableConfig as pgTableConfig } from "drizzle-orm/pg-core";
+import { getTableConfig as sqliteTableConfig } from "drizzle-orm/sqlite-core";
+
+// src/dialect.ts
+import { is } from "drizzle-orm";
+import { MySqlDatabase } from "drizzle-orm/mysql-core";
+import { PgDatabase } from "drizzle-orm/pg-core";
+
+// src/schema-mysql.ts
+import { bigint, boolean, datetime, double, index as index2, int, json, longtext, mysqlTable, primaryKey as primaryKey2, uniqueIndex as uniqueIndex2, varchar } from "drizzle-orm/mysql-core";
+var ts = /* @__PURE__ */ __name((name) => datetime(name, {
+  mode: "date",
+  fsp: 3
+}), "ts");
+var key = /* @__PURE__ */ __name((name) => varchar(name, {
+  length: 255
+}), "key");
+var agentThread2 = mysqlTable("agent_thread", {
+  id: key("id").primaryKey(),
+  actorRef: key("actor_ref").notNull(),
+  tenantRef: key("tenant_ref"),
+  title: key("title").notNull(),
+  transient: boolean("transient").notNull().default(false),
+  activeStreamId: key("active_stream_id"),
+  defaultAgent: key("default_agent"),
+  model: key("model"),
+  /** The persona a send on this thread runs under when it names none; `null` → the agent's default. */
+  persona: key("persona"),
+  queuePause: json("queue_pause").$type(),
+  createdAt: ts("created_at").notNull(),
+  updatedAt: ts("updated_at").notNull(),
+  deletedAt: ts("deleted_at")
+}, (table) => [
+  index2("agent_thread_actor_updated_idx").on(table.actorRef, table.updatedAt)
+]);
+var agentMessage2 = mysqlTable("agent_message", {
+  id: key("id").primaryKey(),
+  threadId: key("thread_id").notNull().references(() => agentThread2.id, {
+    onDelete: "cascade"
+  }),
+  role: key("role").$type().notNull(),
+  content: longtext("content").notNull(),
+  toolCalls: json("tool_calls").$type(),
+  toolResults: json("tool_results").$type(),
+  attachments: json("attachments").$type(),
+  followUps: json("follow_ups").$type(),
+  usage: json("usage").$type(),
+  agentName: key("agent_name"),
+  /** The persona the turn that wrote this message ran under; `null` → none. */
+  persona: key("persona"),
+  runId: key("run_id"),
+  reasoning: longtext("reasoning"),
+  reasoningMs: int("reasoning_ms"),
+  ui: json("ui").$type(),
+  feedback: json("feedback").$type(),
+  seq: int("seq"),
+  createdAt: ts("created_at").notNull()
+}, (table) => [
+  index2("agent_message_thread_created_idx").on(table.threadId, table.createdAt)
+]);
+var agentToolCall2 = mysqlTable("agent_tool_call", {
+  id: key("id").primaryKey(),
+  messageId: key("message_id").notNull().references(() => agentMessage2.id, {
+    onDelete: "cascade"
+  }),
+  toolName: key("tool_name").notNull(),
+  toolType: key("tool_type").$type().notNull(),
+  input: json("input"),
+  output: json("output"),
+  status: key("status").$type().notNull(),
+  executedByRef: key("executed_by_ref"),
+  executionMs: int("execution_ms"),
+  error: longtext("error"),
+  createdAt: ts("created_at").notNull(),
+  executedAt: ts("executed_at"),
+  runId: key("run_id"),
+  approver: key("approver"),
+  expiresAt: ts("expires_at"),
+  remember: boolean("remember"),
+  decidedVia: key("decided_via")
+}, (table) => [
+  index2("agent_tool_call_message_idx").on(table.messageId)
+]);
+var agentQueuedMessage2 = mysqlTable("agent_queued_message", {
+  id: key("id").primaryKey(),
+  threadId: key("thread_id").notNull().references(() => agentThread2.id, {
+    onDelete: "cascade"
+  }),
+  actor: json("actor").$type().notNull(),
+  content: longtext("content").notNull(),
+  attachments: json("attachments").$type(),
+  agentName: key("agent_name"),
+  /** The persona the send resolved — what the message starts under. */
+  persona: key("persona"),
+  model: key("model"),
+  pageContext: json("page_context").$type(),
+  interrupt: boolean("interrupt").notNull().default(false),
+  position: int("position").notNull(),
+  createdAt: ts("created_at").notNull(),
+  updatedAt: ts("updated_at").notNull()
+}, (table) => [
+  index2("agent_queued_message_thread_position_idx").on(table.threadId, table.position)
+]);
+var agentTokenUsage2 = mysqlTable("agent_token_usage", {
+  id: key("id").primaryKey(),
+  threadId: key("thread_id").notNull().references(() => agentThread2.id, {
+    onDelete: "cascade"
+  }),
+  actorRef: key("actor_ref").notNull(),
+  messageId: key("message_id"),
+  modelId: key("model_id").notNull(),
+  purpose: key("purpose").$type().notNull(),
+  inputTokens: int("input_tokens").notNull(),
+  outputTokens: int("output_tokens").notNull(),
+  cacheWriteTokens: int("cache_write_tokens"),
+  cacheReadTokens: int("cache_read_tokens"),
+  costUsd: double("cost_usd"),
+  createdAt: ts("created_at").notNull()
+}, (table) => [
+  index2("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt)
+]);
+var agentModelPricing2 = mysqlTable("agent_model_pricing", {
+  id: key("id").primaryKey(),
+  modelId: key("model_id").notNull(),
+  inputPricePer1m: double("input_price_per_1m").notNull(),
+  outputPricePer1m: double("output_price_per_1m").notNull(),
+  cacheWritePricePer1m: double("cache_write_price_per_1m"),
+  cacheReadPricePer1m: double("cache_read_price_per_1m"),
+  effectiveFrom: ts("effective_from").notNull(),
+  isCurrent: boolean("is_current").notNull()
+});
+var agentRun2 = mysqlTable("agent_run", {
+  id: key("id").primaryKey(),
+  threadId: key("thread_id").notNull().references(() => agentThread2.id, {
+    onDelete: "cascade"
+  }),
+  actorRef: key("actor_ref").notNull(),
+  agentName: key("agent_name"),
+  status: key("status").$type().notNull(),
+  durationMs: int("duration_ms"),
+  errorCode: key("error_code"),
+  errorMessage: longtext("error_message"),
+  retries: int("retries").notNull().default(0),
+  startedAt: ts("started_at").notNull(),
+  settledAt: ts("settled_at"),
+  promptHash: key("prompt_hash"),
+  parentRunId: key("parent_run_id")
+}, (table) => [
+  index2("agent_run_started_idx").on(table.startedAt)
+]);
+var agentMemory2 = mysqlTable("agent_memory", {
+  id: key("id").primaryKey(),
+  scope: key("scope").notNull(),
+  key: key("key").notNull(),
+  text: longtext("text").notNull(),
+  originAuthor: key("origin_author").$type().notNull(),
+  originThreadId: key("origin_thread_id"),
+  originRunId: key("origin_run_id"),
+  originActorRef: key("origin_actor_ref"),
+  pinned: boolean("pinned").notNull().default(false),
+  createdAt: ts("created_at").notNull(),
+  updatedAt: ts("updated_at").notNull()
+}, (table) => [
+  uniqueIndex2("agent_memory_scope_key_uq").on(table.scope, table.key)
+]);
+var ragIngestionLog2 = mysqlTable("rag_ingestion_log", {
+  documentId: key("document_id").primaryKey(),
+  status: key("status").$type().notNull(),
+  collection: key("collection"),
+  ownerType: key("owner_type"),
+  ownerId: key("owner_id"),
+  source: longtext("source"),
+  mimeType: key("mime_type"),
+  size: bigint("size", {
+    mode: "number"
+  }),
+  chunks: int("chunks"),
+  reason: key("reason"),
+  error: longtext("error"),
+  createdAt: ts("created_at").notNull(),
+  updatedAt: ts("updated_at").notNull()
+}, (table) => [
+  index2("rag_ingestion_log_collection_idx").on(table.collection, table.updatedAt)
+]);
+var agentConfirmToken2 = mysqlTable("agent_confirm_token", {
+  hash: key("hash").primaryKey(),
+  actorRef: key("actor_ref").notNull(),
+  tool: key("tool").notNull(),
+  expiresAt: bigint("expires_at", {
+    mode: "number"
+  }).notNull(),
+  createdAt: bigint("created_at", {
+    mode: "number"
+  }).notNull()
+}, (table) => [
+  index2("agent_confirm_token_expires_idx").on(table.expiresAt)
+]);
+var agentStreamFrame2 = mysqlTable("agent_stream_frame", {
+  runId: key("run_id").notNull(),
+  seq: int("seq").notNull(),
+  frame: longtext("frame"),
+  error: longtext("error"),
+  createdAt: bigint("created_at", {
+    mode: "number"
+  }).notNull()
+}, (table) => [
+  primaryKey2({
+    columns: [
+      table.runId,
+      table.seq
+    ]
+  })
+]);
+var mysqlAgentSchema = {
+  agentThread: agentThread2,
+  agentMessage: agentMessage2,
+  agentQueuedMessage: agentQueuedMessage2,
+  agentToolCall: agentToolCall2,
+  agentTokenUsage: agentTokenUsage2,
+  agentModelPricing: agentModelPricing2,
+  agentRun: agentRun2,
+  agentMemory: agentMemory2,
+  ragIngestionLog: ragIngestionLog2,
+  agentConfirmToken: agentConfirmToken2,
+  agentStreamFrame: agentStreamFrame2
+};
+
+// src/schema-pg.ts
+import { bigint as bigint2, boolean as boolean2, doublePrecision, index as index3, integer as integer2, jsonb, pgTable, primaryKey as primaryKey3, text as text2, timestamp, uniqueIndex as uniqueIndex3 } from "drizzle-orm/pg-core";
+var ts2 = /* @__PURE__ */ __name((name) => timestamp(name, {
+  withTimezone: true,
+  precision: 3,
+  mode: "date"
+}), "ts");
+var agentThread3 = pgTable("agent_thread", {
+  id: text2("id").primaryKey(),
+  actorRef: text2("actor_ref").notNull(),
+  tenantRef: text2("tenant_ref"),
+  title: text2("title").notNull(),
+  transient: boolean2("transient").notNull().default(false),
+  activeStreamId: text2("active_stream_id"),
+  defaultAgent: text2("default_agent"),
+  model: text2("model"),
+  /** The persona a send on this thread runs under when it names none; `null` → the agent's default. */
+  persona: text2("persona"),
+  queuePause: jsonb("queue_pause").$type(),
+  createdAt: ts2("created_at").notNull(),
+  updatedAt: ts2("updated_at").notNull(),
+  deletedAt: ts2("deleted_at")
+}, (table) => [
+  index3("agent_thread_actor_updated_idx").on(table.actorRef, table.updatedAt)
+]);
+var agentMessage3 = pgTable("agent_message", {
+  id: text2("id").primaryKey(),
+  threadId: text2("thread_id").notNull().references(() => agentThread3.id, {
+    onDelete: "cascade"
+  }),
+  role: text2("role").$type().notNull(),
+  content: text2("content").notNull(),
+  toolCalls: jsonb("tool_calls").$type(),
+  toolResults: jsonb("tool_results").$type(),
+  attachments: jsonb("attachments").$type(),
+  followUps: jsonb("follow_ups").$type(),
+  usage: jsonb("usage").$type(),
+  agentName: text2("agent_name"),
+  /** The persona the turn that wrote this message ran under; `null` → none. */
+  persona: text2("persona"),
+  runId: text2("run_id"),
+  reasoning: text2("reasoning"),
+  reasoningMs: integer2("reasoning_ms"),
+  ui: jsonb("ui").$type(),
+  feedback: jsonb("feedback").$type(),
+  seq: integer2("seq"),
+  createdAt: ts2("created_at").notNull()
+}, (table) => [
+  index3("agent_message_thread_created_idx").on(table.threadId, table.createdAt)
+]);
+var agentToolCall3 = pgTable("agent_tool_call", {
+  id: text2("id").primaryKey(),
+  messageId: text2("message_id").notNull().references(() => agentMessage3.id, {
+    onDelete: "cascade"
+  }),
+  toolName: text2("tool_name").notNull(),
+  toolType: text2("tool_type").$type().notNull(),
+  input: jsonb("input"),
+  output: jsonb("output"),
+  status: text2("status").$type().notNull(),
+  executedByRef: text2("executed_by_ref"),
+  executionMs: integer2("execution_ms"),
+  error: text2("error"),
+  createdAt: ts2("created_at").notNull(),
+  executedAt: ts2("executed_at"),
+  runId: text2("run_id"),
+  approver: text2("approver"),
+  expiresAt: ts2("expires_at"),
+  remember: boolean2("remember"),
+  decidedVia: text2("decided_via")
+}, (table) => [
+  index3("agent_tool_call_message_idx").on(table.messageId)
+]);
+var agentQueuedMessage3 = pgTable("agent_queued_message", {
+  id: text2("id").primaryKey(),
+  threadId: text2("thread_id").notNull().references(() => agentThread3.id, {
+    onDelete: "cascade"
+  }),
+  actor: jsonb("actor").$type().notNull(),
+  content: text2("content").notNull(),
+  attachments: jsonb("attachments").$type(),
+  agentName: text2("agent_name"),
+  /** The persona the send resolved — what the message starts under. */
+  persona: text2("persona"),
+  model: text2("model"),
+  pageContext: jsonb("page_context").$type(),
+  interrupt: boolean2("interrupt").notNull().default(false),
+  position: integer2("position").notNull(),
+  createdAt: ts2("created_at").notNull(),
+  updatedAt: ts2("updated_at").notNull()
+}, (table) => [
+  index3("agent_queued_message_thread_position_idx").on(table.threadId, table.position)
+]);
+var agentTokenUsage3 = pgTable("agent_token_usage", {
+  id: text2("id").primaryKey(),
+  threadId: text2("thread_id").notNull().references(() => agentThread3.id, {
+    onDelete: "cascade"
+  }),
+  actorRef: text2("actor_ref").notNull(),
+  messageId: text2("message_id"),
+  modelId: text2("model_id").notNull(),
+  purpose: text2("purpose").$type().notNull(),
+  inputTokens: integer2("input_tokens").notNull(),
+  outputTokens: integer2("output_tokens").notNull(),
+  cacheWriteTokens: integer2("cache_write_tokens"),
+  cacheReadTokens: integer2("cache_read_tokens"),
+  costUsd: doublePrecision("cost_usd"),
+  createdAt: ts2("created_at").notNull()
+}, (table) => [
+  index3("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt)
+]);
+var agentModelPricing3 = pgTable("agent_model_pricing", {
+  id: text2("id").primaryKey(),
+  modelId: text2("model_id").notNull(),
+  inputPricePer1m: doublePrecision("input_price_per_1m").notNull(),
+  outputPricePer1m: doublePrecision("output_price_per_1m").notNull(),
+  cacheWritePricePer1m: doublePrecision("cache_write_price_per_1m"),
+  cacheReadPricePer1m: doublePrecision("cache_read_price_per_1m"),
+  effectiveFrom: ts2("effective_from").notNull(),
+  isCurrent: boolean2("is_current").notNull()
+});
+var agentRun3 = pgTable("agent_run", {
+  id: text2("id").primaryKey(),
+  threadId: text2("thread_id").notNull().references(() => agentThread3.id, {
+    onDelete: "cascade"
+  }),
+  actorRef: text2("actor_ref").notNull(),
+  agentName: text2("agent_name"),
+  status: text2("status").$type().notNull(),
+  durationMs: integer2("duration_ms"),
+  errorCode: text2("error_code"),
+  errorMessage: text2("error_message"),
+  retries: integer2("retries").notNull().default(0),
+  startedAt: ts2("started_at").notNull(),
+  settledAt: ts2("settled_at"),
+  promptHash: text2("prompt_hash"),
+  parentRunId: text2("parent_run_id")
+}, (table) => [
+  index3("agent_run_started_idx").on(table.startedAt)
+]);
+var agentMemory3 = pgTable("agent_memory", {
+  id: text2("id").primaryKey(),
+  scope: text2("scope").notNull(),
+  key: text2("key").notNull(),
+  text: text2("text").notNull(),
+  originAuthor: text2("origin_author").$type().notNull(),
+  originThreadId: text2("origin_thread_id"),
+  originRunId: text2("origin_run_id"),
+  originActorRef: text2("origin_actor_ref"),
+  pinned: boolean2("pinned").notNull().default(false),
+  createdAt: ts2("created_at").notNull(),
+  updatedAt: ts2("updated_at").notNull()
+}, (table) => [
+  uniqueIndex3("agent_memory_scope_key_uq").on(table.scope, table.key)
+]);
+var ragIngestionLog3 = pgTable("rag_ingestion_log", {
+  documentId: text2("document_id").primaryKey(),
+  status: text2("status").$type().notNull(),
+  collection: text2("collection"),
+  ownerType: text2("owner_type"),
+  ownerId: text2("owner_id"),
+  source: text2("source"),
+  mimeType: text2("mime_type"),
+  size: bigint2("size", {
+    mode: "number"
+  }),
+  chunks: integer2("chunks"),
+  reason: text2("reason"),
+  error: text2("error"),
+  createdAt: ts2("created_at").notNull(),
+  updatedAt: ts2("updated_at").notNull()
+}, (table) => [
+  index3("rag_ingestion_log_collection_idx").on(table.collection, table.updatedAt)
+]);
+var agentConfirmToken3 = pgTable("agent_confirm_token", {
+  hash: text2("hash").primaryKey(),
+  actorRef: text2("actor_ref").notNull(),
+  tool: text2("tool").notNull(),
+  expiresAt: bigint2("expires_at", {
+    mode: "number"
+  }).notNull(),
+  createdAt: bigint2("created_at", {
+    mode: "number"
+  }).notNull()
+}, (table) => [
+  index3("agent_confirm_token_expires_idx").on(table.expiresAt)
+]);
+var agentStreamFrame3 = pgTable("agent_stream_frame", {
+  runId: text2("run_id").notNull(),
+  seq: integer2("seq").notNull(),
+  frame: text2("frame"),
+  error: text2("error"),
+  createdAt: bigint2("created_at", {
+    mode: "number"
+  }).notNull()
+}, (table) => [
+  primaryKey3({
+    columns: [
+      table.runId,
+      table.seq
+    ]
+  })
+]);
+var pgAgentSchema = {
+  agentThread: agentThread3,
+  agentMessage: agentMessage3,
+  agentQueuedMessage: agentQueuedMessage3,
+  agentToolCall: agentToolCall3,
+  agentTokenUsage: agentTokenUsage3,
+  agentModelPricing: agentModelPricing3,
+  agentRun: agentRun3,
+  agentMemory: agentMemory3,
+  ragIngestionLog: ragIngestionLog3,
+  agentConfirmToken: agentConfirmToken3,
+  agentStreamFrame: agentStreamFrame3
+};
+
+// src/dialect.ts
+function agentDialectOf(db) {
+  if (is(db, PgDatabase)) return "postgres";
+  if (is(db, MySqlDatabase)) return "mysql";
+  return "sqlite";
+}
+__name(agentDialectOf, "agentDialectOf");
+function agentTablesFor(dialect) {
+  if (dialect === "postgres") return pgAgentSchema;
+  if (dialect === "mysql") return mysqlAgentSchema;
+  return agentSchema;
+}
+__name(agentTablesFor, "agentTablesFor");
+function asBuilder(db) {
+  return db;
+}
+__name(asBuilder, "asBuilder");
+async function affectedRows(dialect, statement, column) {
+  if (dialect !== "mysql") {
+    return (await statement.returning({
+      affected: column
+    })).length;
+  }
+  return mysqlAffectedRows(await statement);
+}
+__name(affectedRows, "affectedRows");
+function mysqlAffectedRows(result) {
+  const header = Array.isArray(result) ? result[0] : result;
+  const counts = header;
+  const count3 = counts?.affectedRows ?? counts?.rowsAffected;
+  return typeof count3 === "number" ? count3 : 0;
+}
+__name(mysqlAffectedRows, "mysqlAffectedRows");
+async function runSql(db, dialect, statement) {
+  if (dialect === "sqlite") {
+    await db.run(statement);
+    return;
   }
-];
-async function ensureAgentSchema(db) {
-  for (const statement of statements) {
-    await db.run(sql.raw(statement));
+  if (dialect === "postgres") {
+    await db.execute(statement);
+    return;
   }
-  for (const { table, column, ddl } of additiveColumns) {
-    const existing = await db.all(sql.raw(`PRAGMA table_info(${table})`));
-    if (!existing.some((row) => row.name === column)) {
-      await db.run(sql.raw(ddl));
+  await db.execute(statement);
+}
+__name(runSql, "runSql");
+async function querySql(db, dialect, statement) {
+  if (dialect === "sqlite") {
+    return db.all(statement);
+  }
+  const result = dialect === "postgres" ? await db.execute(statement) : await db.execute(statement);
+  if (dialect === "mysql") {
+    return Array.isArray(result) ? result[0] : [];
+  }
+  return Array.isArray(result) ? result : result.rows ?? [];
+}
+__name(querySql, "querySql");
+function isUniqueViolation(error) {
+  for (let current = error; current !== null && typeof current === "object"; ) {
+    const { code, errno } = current;
+    if (code === "23505" || code === "ER_DUP_ENTRY" || errno === 1062 || code === "ER_LOCK_DEADLOCK" || errno === 1213 || typeof code === "string" && code.startsWith("SQLITE_CONSTRAINT")) {
+      return true;
     }
+    current = current.cause;
   }
+  return false;
+}
+__name(isUniqueViolation, "isUniqueViolation");
+function upsert(dialect, insert, target, set) {
+  if (dialect === "mysql") {
+    return insert.onDuplicateKeyUpdate({
+      set
+    });
+  }
+  return insert.onConflictDoUpdate({
+    target,
+    set
+  });
+}
+__name(upsert, "upsert");
+
+// src/ensure-schema.ts
+async function ensureAgentSchema(db) {
+  const dialect = agentDialectOf(db);
+  if (dialect === "postgres") {
+    await db.transaction(async (tx) => {
+      await tx.execute(sql.raw(`select pg_advisory_xact_lock(hashtext('${LOCK_NAME}'))`));
+      await heal(tx, dialect);
+    });
+    return;
+  }
+  if (dialect === "mysql") {
+    await db.transaction(async (tx) => {
+      await tx.execute(sql.raw(`select get_lock('${LOCK_NAME}', 30)`));
+      try {
+        await heal(tx, dialect);
+      } finally {
+        await tx.execute(sql.raw(`select release_lock('${LOCK_NAME}')`));
+      }
+    });
+    return;
+  }
+  await heal(db, dialect);
 }
 __name(ensureAgentSchema, "ensureAgentSchema");
+var LOCK_NAME = "nestjs_agent_schema";
+var MYSQL_TABLE_OPTIONS = "ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin";
+function tableSpec(dialect, table) {
+  const config = dialect === "postgres" ? pgTableConfig(table) : dialect === "mysql" ? mysqlTableConfig(table) : sqliteTableConfig(table);
+  const columns = config.columns;
+  const compositeKey = config.primaryKeys[0]?.columns.map((column) => column.name);
+  return {
+    name: config.name,
+    columns,
+    primaryKey: compositeKey ?? columns.filter((column) => column.primary).map((c) => c.name),
+    foreignKeys: config.foreignKeys.map((foreignKey) => {
+      const reference = foreignKey.reference();
+      return {
+        columns: reference.columns.map((column) => column.name),
+        table: getTableName(reference.foreignTable),
+        foreignColumns: reference.foreignColumns.map((column) => column.name),
+        ...foreignKey.onDelete !== void 0 ? {
+          onDelete: foreignKey.onDelete
+        } : {}
+      };
+    }),
+    indexes: config.indexes.map((index4) => ({
+      name: String(index4.config.name),
+      unique: index4.config.unique === true,
+      columns: index4.config.columns.map((column) => String(column.name))
+    }))
+  };
+}
+__name(tableSpec, "tableSpec");
+function quote(dialect, identifier) {
+  return dialect === "mysql" ? `\`${identifier}\`` : `"${identifier}"`;
+}
+__name(quote, "quote");
+function literal(dialect, value) {
+  if (typeof value === "boolean") {
+    return dialect === "sqlite" ? value ? "1" : "0" : value ? "true" : "false";
+  }
+  if (typeof value === "number") return String(value);
+  return `'${String(value).replaceAll("'", "''")}'`;
+}
+__name(literal, "literal");
+function columnDefinition(dialect, column) {
+  const parts = [
+    quote(dialect, column.name),
+    column.getSQLType()
+  ];
+  if (column.notNull) parts.push("NOT NULL");
+  if (column.hasDefault && column.default !== void 0) {
+    parts.push(`DEFAULT ${literal(dialect, column.default)}`);
+  }
+  return parts.join(" ");
+}
+__name(columnDefinition, "columnDefinition");
+function createTableSql(dialect, spec) {
+  const q = /* @__PURE__ */ __name((identifier) => quote(dialect, identifier), "q");
+  const lines = spec.columns.map((column) => columnDefinition(dialect, column));
+  lines.push(`PRIMARY KEY (${spec.primaryKey.map(q).join(", ")})`);
+  for (const foreignKey of spec.foreignKeys) {
+    lines.push(`FOREIGN KEY (${foreignKey.columns.map(q).join(", ")}) REFERENCES ${q(foreignKey.table)} (${foreignKey.foreignColumns.map(q).join(", ")})${foreignKey.onDelete !== void 0 ? ` ON DELETE ${foreignKey.onDelete.toUpperCase()}` : ""}`);
+  }
+  const options = dialect === "mysql" ? ` ${MYSQL_TABLE_OPTIONS}` : "";
+  return `CREATE TABLE IF NOT EXISTS ${q(spec.name)} (
+  ${lines.join(",\n  ")}
+)${options}`;
+}
+__name(createTableSql, "createTableSql");
+function agentSchemaDdl(dialect) {
+  const statements = [];
+  for (const spec of specs(dialect)) {
+    statements.push(createTableSql(dialect, spec));
+    for (const index4 of spec.indexes) {
+      statements.push(createIndexSql(dialect, spec.name, index4));
+    }
+  }
+  return statements;
+}
+__name(agentSchemaDdl, "agentSchemaDdl");
+function createIndexSql(dialect, table, index4) {
+  const q = /* @__PURE__ */ __name((identifier) => quote(dialect, identifier), "q");
+  const guard = dialect === "mysql" ? "" : "IF NOT EXISTS ";
+  return `CREATE ${index4.unique ? "UNIQUE " : ""}INDEX ${guard}${q(index4.name)} ON ${q(table)} (${index4.columns.map(q).join(", ")})`;
+}
+__name(createIndexSql, "createIndexSql");
+function specs(dialect) {
+  const all = Object.values(agentTablesFor(dialect)).map((table) => tableSpec(dialect, table));
+  const ordered = [];
+  const placed = /* @__PURE__ */ new Set();
+  while (ordered.length < all.length) {
+    const ready = all.filter((spec) => !placed.has(spec.name) && spec.foreignKeys.every((key2) => key2.table === spec.name || placed.has(key2.table)));
+    if (ready.length === 0) throw new Error("agent schema: circular foreign keys");
+    for (const spec of ready) {
+      ordered.push(spec);
+      placed.add(spec.name);
+    }
+  }
+  return ordered;
+}
+__name(specs, "specs");
+async function existingColumns(db, dialect, table) {
+  const rows = dialect === "sqlite" ? await querySql(db, dialect, sql.raw(`PRAGMA table_info("${table}")`)) : await querySql(db, dialect, sql`select column_name as name from information_schema.columns where table_schema = ${sql.raw(dialect === "mysql" ? "database()" : "current_schema()")} and table_name = ${table}`);
+  return new Set(rows.map((row) => String(row.name)));
+}
+__name(existingColumns, "existingColumns");
+async function indexExists(db, dialect, table, index4) {
+  let query;
+  if (dialect === "sqlite") {
+    query = sql`select name from sqlite_master where type = 'index' and name = ${index4}`;
+  } else if (dialect === "postgres") {
+    query = sql`select indexname as name from pg_indexes where schemaname = current_schema() and indexname = ${index4}`;
+  } else {
+    query = sql`select index_name as name from information_schema.statistics where table_schema = database() and table_name = ${table} and index_name = ${index4}`;
+  }
+  return (await querySql(db, dialect, query)).length > 0;
+}
+__name(indexExists, "indexExists");
+async function heal(db, dialect) {
+  for (const spec of specs(dialect)) {
+    await runSql(db, dialect, sql.raw(createTableSql(dialect, spec)));
+    const present = await existingColumns(db, dialect, spec.name);
+    for (const column of spec.columns) {
+      if (!present.has(column.name)) {
+        await runSql(db, dialect, sql.raw(`ALTER TABLE ${quote(dialect, spec.name)} ADD COLUMN ${columnDefinition(dialect, column)}`));
+      }
+    }
+    for (const index4 of spec.indexes) {
+      if (!await indexExists(db, dialect, spec.name, index4.name)) {
+        await runSql(db, dialect, sql.raw(createIndexSql(dialect, spec.name, index4)));
+      }
+    }
+  }
+}
+__name(heal, "heal");
 
 // src/drizzle-agent-store.ts
 import { toolCallApprovalFromRow } from "@dudousxd/nestjs-agent-core";
 import { and, asc, desc, eq, gte, inArray, isNotNull, isNull, lte, max, min, or, sql as sql2 } from "drizzle-orm";
-var TURN_MESSAGE_COLUMNS = {
-  id: agentMessage.id,
-  role: agentMessage.role,
-  content: agentMessage.content,
-  agentName: agentMessage.agentName,
-  toolCalls: agentMessage.toolCalls,
-  toolResults: agentMessage.toolResults,
-  attachments: agentMessage.attachments,
-  createdAt: agentMessage.createdAt
-};
+function turnMessageColumns(message) {
+  return {
+    id: message.id,
+    role: message.role,
+    content: message.content,
+    agentName: message.agentName,
+    toolCalls: message.toolCalls,
+    toolResults: message.toolResults,
+    attachments: message.attachments,
+    createdAt: message.createdAt
+  };
+}
+__name(turnMessageColumns, "turnMessageColumns");
 var DrizzleAgentStore = class {
   static {
     __name(this, "DrizzleAgentStore");
   }
   db;
+  dialect;
+  t;
   constructor(db) {
-    this.db = db;
+    this.dialect = agentDialectOf(db);
+    this.t = agentTablesFor(this.dialect);
+    this.db = asBuilder(db);
   }
   async createThread(input) {
     const now = /* @__PURE__ */ new Date();
@@ -590,20 +1038,21 @@ var DrizzleAgentStore = class {
       activeStreamId: null,
       defaultAgent: null,
       model: null,
+      persona: input.persona ?? null,
       queuePause: null,
       createdAt: now,
       updatedAt: now,
       deletedAt: null
     };
-    await this.db.insert(agentThread).values(thread);
+    await this.db.insert(this.t.agentThread).values(thread);
     return this.toSummary(thread);
   }
   async getThread(threadId) {
-    const [thread] = await this.db.select().from(agentThread).where(and(eq(agentThread.id, threadId), isNull(agentThread.deletedAt)));
+    const [thread] = await this.db.select().from(this.t.agentThread).where(and(eq(this.t.agentThread.id, threadId), isNull(this.t.agentThread.deletedAt)));
     if (thread === void 0) {
       return null;
     }
-    const messages = await this.db.select().from(agentMessage).where(eq(agentMessage.threadId, threadId)).orderBy(asc(agentMessage.createdAt), asc(agentMessage.id));
+    const messages = await this.db.select().from(this.t.agentMessage).where(eq(this.t.agentMessage.threadId, threadId)).orderBy(...this.messageOrder());
     const last = messages[messages.length - 1];
     const approvals = await this.approvalsFor(messages.map((message) => message.id));
     return {
@@ -636,16 +1085,16 @@ var DrizzleAgentStore = class {
   */
   async loadThreadForTurn(query) {
     const [thread] = await this.db.select({
-      title: agentThread.title,
-      defaultAgent: agentThread.defaultAgent
-    }).from(agentThread).where(and(eq(agentThread.id, query.threadId), isNull(agentThread.deletedAt)));
+      title: this.t.agentThread.title,
+      defaultAgent: this.t.agentThread.defaultAgent
+    }).from(this.t.agentThread).where(and(eq(this.t.agentThread.id, query.threadId), isNull(this.t.agentThread.deletedAt)));
     if (thread === void 0) {
       return null;
     }
     const messages = await this.loadTurnWindow(query.threadId, query.messageLimit);
     const [answered] = await this.db.select({
-      id: agentMessage.id
-    }).from(agentMessage).where(and(eq(agentMessage.threadId, query.threadId), eq(agentMessage.role, "assistant"))).limit(1);
+      id: this.t.agentMessage.id
+    }).from(this.t.agentMessage).where(and(eq(this.t.agentMessage.threadId, query.threadId), eq(this.t.agentMessage.role, "assistant"))).limit(1);
     return {
       title: thread.title,
       defaultAgent: thread.defaultAgent,
@@ -661,30 +1110,30 @@ var DrizzleAgentStore = class {
     if (limit !== void 0 && limit <= 0) {
       return [];
     }
-    const window = this.db.select(TURN_MESSAGE_COLUMNS).from(agentMessage).where(eq(agentMessage.threadId, threadId)).orderBy(desc(agentMessage.createdAt), desc(agentMessage.id));
+    const window = this.db.select(turnMessageColumns(this.t.agentMessage)).from(this.t.agentMessage).where(eq(this.t.agentMessage.threadId, threadId)).orderBy(...this.messageOrderNewestFirst());
     const rows = limit === void 0 ? await window : await window.limit(limit);
     return rows.reverse().map((row) => this.toStoredMessage(row));
   }
   async listThreads(actorRef, limit = 50) {
-    const threads = await this.db.select().from(agentThread).where(and(eq(agentThread.actorRef, actorRef), eq(agentThread.transient, false), isNull(agentThread.deletedAt))).orderBy(desc(agentThread.updatedAt)).limit(limit);
+    const threads = await this.db.select().from(this.t.agentThread).where(and(eq(this.t.agentThread.actorRef, actorRef), eq(this.t.agentThread.transient, false), isNull(this.t.agentThread.deletedAt))).orderBy(desc(this.t.agentThread.updatedAt)).limit(limit);
     const summaries = [];
     for (const thread of threads) {
-      const [last] = await this.db.select().from(agentMessage).where(eq(agentMessage.threadId, thread.id)).orderBy(desc(agentMessage.createdAt), desc(agentMessage.id)).limit(1);
+      const [last] = await this.db.select().from(this.t.agentMessage).where(eq(this.t.agentMessage.threadId, thread.id)).orderBy(...this.messageOrderNewestFirst()).limit(1);
       summaries.push(this.toSummary(thread, last?.content));
     }
     return summaries;
   }
   async softDeleteThread(threadId) {
-    await this.db.update(agentThread).set({
+    await this.db.update(this.t.agentThread).set({
       deletedAt: /* @__PURE__ */ new Date()
-    }).where(eq(agentThread.id, threadId));
+    }).where(eq(this.t.agentThread.id, threadId));
   }
   async forkThread(threadId, fromMessageId) {
-    const [source] = await this.db.select().from(agentThread).where(eq(agentThread.id, threadId));
+    const [source] = await this.db.select().from(this.t.agentThread).where(eq(this.t.agentThread.id, threadId));
     if (source === void 0) {
       throw new Error(`thread ${threadId} not found`);
     }
-    const messages = await this.db.select().from(agentMessage).where(eq(agentMessage.threadId, threadId)).orderBy(asc(agentMessage.createdAt), asc(agentMessage.id));
+    const messages = await this.db.select().from(this.t.agentMessage).where(eq(this.t.agentMessage.threadId, threadId)).orderBy(...this.messageOrder());
     const cutoff = messages.findIndex((message) => message.id === fromMessageId);
     const kept = cutoff >= 0 ? messages.slice(0, cutoff + 1) : messages;
     const now = /* @__PURE__ */ new Date();
@@ -697,16 +1146,19 @@ var DrizzleAgentStore = class {
       activeStreamId: null,
       defaultAgent: source.defaultAgent,
       model: source.model,
+      persona: source.persona,
       // A fork is a new conversation: nothing is waiting on it, and nothing is paused.
       queuePause: null,
       createdAt: now,
       updatedAt: now,
       deletedAt: null
     };
-    await this.db.insert(agentThread).values(fork);
+    await this.db.insert(this.t.agentThread).values(fork);
     if (kept.length > 0) {
-      await this.db.insert(agentMessage).values(kept.map((message) => ({
+      await this.db.insert(this.t.agentMessage).values(kept.map((message, index4) => ({
         id: crypto.randomUUID(),
+        // Numbered afresh in the order just read, so the copy reads back in the original's order.
+        seq: index4 + 1,
         threadId: fork.id,
         role: message.role,
         content: message.content,
@@ -716,6 +1168,7 @@ var DrizzleAgentStore = class {
         followUps: message.followUps,
         usage: message.usage,
         agentName: message.agentName,
+        persona: message.persona,
         // The copy is the same message, so it keeps the run that wrote it. A reader asking which
         // turn produced this text gets the truthful answer; the run's own thread is still the
         // original, so a run-scoped read never picks the fork's rows up.
@@ -730,21 +1183,21 @@ var DrizzleAgentStore = class {
   }
   async ownerOfThread(threadId) {
     const [thread] = await this.db.select({
-      actorRef: agentThread.actorRef
-    }).from(agentThread).where(and(eq(agentThread.id, threadId), isNull(agentThread.deletedAt)));
+      actorRef: this.t.agentThread.actorRef
+    }).from(this.t.agentThread).where(and(eq(this.t.agentThread.id, threadId), isNull(this.t.agentThread.deletedAt)));
     return thread?.actorRef ?? null;
   }
   async ownerOfToolCall(toolCallId) {
     const [row] = await this.db.select({
-      actorRef: agentThread.actorRef
-    }).from(agentToolCall).innerJoin(agentMessage, eq(agentToolCall.messageId, agentMessage.id)).innerJoin(agentThread, eq(agentMessage.threadId, agentThread.id)).where(eq(agentToolCall.id, toolCallId));
+      actorRef: this.t.agentThread.actorRef
+    }).from(this.t.agentToolCall).innerJoin(this.t.agentMessage, eq(this.t.agentToolCall.messageId, this.t.agentMessage.id)).innerJoin(this.t.agentThread, eq(this.t.agentMessage.threadId, this.t.agentThread.id)).where(eq(this.t.agentToolCall.id, toolCallId));
     return row?.actorRef ?? null;
   }
   async runForToolCall(toolCallId) {
     const [row] = await this.db.select({
-      runId: agentToolCall.runId,
-      activeStreamId: agentThread.activeStreamId
-    }).from(agentToolCall).innerJoin(agentMessage, eq(agentToolCall.messageId, agentMessage.id)).innerJoin(agentThread, eq(agentMessage.threadId, agentThread.id)).where(eq(agentToolCall.id, toolCallId));
+      runId: this.t.agentToolCall.runId,
+      activeStreamId: this.t.agentThread.activeStreamId
+    }).from(this.t.agentToolCall).innerJoin(this.t.agentMessage, eq(this.t.agentToolCall.messageId, this.t.agentMessage.id)).innerJoin(this.t.agentThread, eq(this.t.agentMessage.threadId, this.t.agentThread.id)).where(eq(this.t.agentToolCall.id, toolCallId));
     return row?.runId ?? row?.activeStreamId ?? null;
   }
   /**
@@ -758,16 +1211,16 @@ var DrizzleAgentStore = class {
       return byMessage;
     }
     const rows = await this.db.select({
-      id: agentToolCall.id,
-      messageId: agentToolCall.messageId,
-      status: agentToolCall.status,
-      approver: agentToolCall.approver,
-      expiresAt: agentToolCall.expiresAt,
-      remember: agentToolCall.remember,
-      executedByRef: agentToolCall.executedByRef,
-      decidedVia: agentToolCall.decidedVia,
-      error: agentToolCall.error
-    }).from(agentToolCall).where(and(inArray(agentToolCall.messageId, messageIds), isNotNull(agentToolCall.approver))).orderBy(asc(agentToolCall.createdAt), asc(agentToolCall.id));
+      id: this.t.agentToolCall.id,
+      messageId: this.t.agentToolCall.messageId,
+      status: this.t.agentToolCall.status,
+      approver: this.t.agentToolCall.approver,
+      expiresAt: this.t.agentToolCall.expiresAt,
+      remember: this.t.agentToolCall.remember,
+      executedByRef: this.t.agentToolCall.executedByRef,
+      decidedVia: this.t.agentToolCall.decidedVia,
+      error: this.t.agentToolCall.error
+    }).from(this.t.agentToolCall).where(and(inArray(this.t.agentToolCall.messageId, messageIds), isNotNull(this.t.agentToolCall.approver))).orderBy(asc(this.t.agentToolCall.createdAt), asc(this.t.agentToolCall.id));
     for (const row of rows) {
       const approval = toolCallApprovalFromRow({
         ...row,
@@ -785,22 +1238,22 @@ var DrizzleAgentStore = class {
   /** Tools whose approval someone asked to remember in this thread. */
   async rememberedApprovals(threadId) {
     const rows = await this.db.selectDistinct({
-      toolName: agentToolCall.toolName
-    }).from(agentToolCall).innerJoin(agentMessage, eq(agentToolCall.messageId, agentMessage.id)).where(and(eq(agentMessage.threadId, threadId), eq(agentToolCall.remember, true)));
+      toolName: this.t.agentToolCall.toolName
+    }).from(this.t.agentToolCall).innerJoin(this.t.agentMessage, eq(this.t.agentToolCall.messageId, this.t.agentMessage.id)).where(and(eq(this.t.agentMessage.threadId, threadId), eq(this.t.agentToolCall.remember, true)));
     return rows.map((row) => row.toolName);
   }
   async toolCallInput(toolCallId) {
     const [row] = await this.db.select({
-      input: agentToolCall.input
-    }).from(agentToolCall).where(eq(agentToolCall.id, toolCallId));
+      input: this.t.agentToolCall.input
+    }).from(this.t.agentToolCall).where(eq(this.t.agentToolCall.id, toolCallId));
     return row?.input ?? null;
   }
   async toolCallApproval(toolCallId) {
     const [row] = await this.db.select({
-      status: agentToolCall.status,
-      approver: agentToolCall.approver,
-      expiresAt: agentToolCall.expiresAt
-    }).from(agentToolCall).where(eq(agentToolCall.id, toolCallId));
+      status: this.t.agentToolCall.status,
+      approver: this.t.agentToolCall.approver,
+      expiresAt: this.t.agentToolCall.expiresAt
+    }).from(this.t.agentToolCall).where(eq(this.t.agentToolCall.id, toolCallId));
     if (row === void 0) {
       return null;
     }
@@ -812,27 +1265,67 @@ var DrizzleAgentStore = class {
   }
   async ownerOfActiveStream(runId) {
     const [thread] = await this.db.select({
-      actorRef: agentThread.actorRef
-    }).from(agentThread).where(and(eq(agentThread.activeStreamId, runId), isNull(agentThread.deletedAt)));
+      actorRef: this.t.agentThread.actorRef
+    }).from(this.t.agentThread).where(and(eq(this.t.agentThread.activeStreamId, runId), isNull(this.t.agentThread.deletedAt)));
     return thread?.actorRef ?? null;
   }
   async setTitle(threadId, title) {
-    await this.db.update(agentThread).set({
+    await this.db.update(this.t.agentThread).set({
       title,
       updatedAt: /* @__PURE__ */ new Date()
-    }).where(eq(agentThread.id, threadId));
+    }).where(eq(this.t.agentThread.id, threadId));
   }
   async promoteThread(threadId) {
-    await this.db.update(agentThread).set({
+    await this.db.update(this.t.agentThread).set({
       transient: false,
       updatedAt: /* @__PURE__ */ new Date()
-    }).where(and(eq(agentThread.id, threadId), eq(agentThread.transient, true)));
+    }).where(and(eq(this.t.agentThread.id, threadId), eq(this.t.agentThread.transient, true)));
+  }
+  /** Who holds the thread's stream right now, deleted or not. */
+  async activeHolder(threadId) {
+    const [thread] = await this.db.select({
+      activeStreamId: this.t.agentThread.activeStreamId
+    }).from(this.t.agentThread).where(eq(this.t.agentThread.id, threadId));
+    return thread?.activeStreamId ?? null;
+  }
+  /**
+  * A thread's messages in the order they were appended: by `seq`, which append assigns. A row from
+  * before `seq` existed has none and counts as 0 — first, since every such row is older than every
+  * numbered one — and those order among themselves by `created_at`, then `id`, as they always did.
+  * `coalesce` rather than `nulls first`, which MySQL cannot spell and Postgres defaults the other way.
+  */
+  messageOrder() {
+    const message = this.t.agentMessage;
+    return [
+      asc(sql2`coalesce(${message.seq}, 0)`),
+      asc(message.createdAt),
+      asc(message.id)
+    ];
+  }
+  /** {@link messageOrder}, newest first. */
+  messageOrderNewestFirst() {
+    const message = this.t.agentMessage;
+    return [
+      desc(sql2`coalesce(${message.seq}, 0)`),
+      desc(message.createdAt),
+      desc(message.id)
+    ];
+  }
+  /**
+  * The next `seq` in a thread: one past its highest. Two appends racing on ONE thread can draw the
+  * same number — the turn loop never does that, and the tie then falls back to `created_at`, `id`.
+  */
+  async nextMessageSeq(threadId) {
+    const [row] = await this.db.select({
+      last: max(this.t.agentMessage.seq)
+    }).from(this.t.agentMessage).where(eq(this.t.agentMessage.threadId, threadId));
+    return Number(row?.last ?? 0) + 1;
   }
   /** The run streaming the thread, or `null` — a projection of the one column. */
   async activeRunForThread(threadId) {
     const [thread] = await this.db.select({
-      activeStreamId: agentThread.activeStreamId
-    }).from(agentThread).where(and(eq(agentThread.id, threadId), isNull(agentThread.deletedAt)));
+      activeStreamId: this.t.agentThread.activeStreamId
+    }).from(this.t.agentThread).where(and(eq(this.t.agentThread.id, threadId), isNull(this.t.agentThread.deletedAt)));
     return thread?.activeStreamId ?? null;
   }
   /**
@@ -841,33 +1334,32 @@ var DrizzleAgentStore = class {
   */
   async claimActiveStream(threadId, runId, options = {}) {
     const holders = [
-      isNull(agentThread.activeStreamId),
-      eq(agentThread.activeStreamId, runId),
+      isNull(this.t.agentThread.activeStreamId),
+      eq(this.t.agentThread.activeStreamId, runId),
       ...options.replacing !== void 0 ? [
-        eq(agentThread.activeStreamId, options.replacing)
+        eq(this.t.agentThread.activeStreamId, options.replacing)
       ] : []
     ];
-    const claimed = await this.db.update(agentThread).set({
+    const claimed = await affectedRows(this.dialect, this.db.update(this.t.agentThread).set({
       activeStreamId: runId
-    }).where(and(eq(agentThread.id, threadId), or(...holders))).returning({
-      id: agentThread.id
-    });
-    return claimed.length > 0;
+    }).where(and(eq(this.t.agentThread.id, threadId), or(...holders))), this.t.agentThread.id);
+    if (claimed > 0) {
+      return true;
+    }
+    return await this.activeHolder(threadId) === runId;
   }
   async releaseActiveStream(threadId, runId) {
-    const released = await this.db.update(agentThread).set({
+    const released = await affectedRows(this.dialect, this.db.update(this.t.agentThread).set({
       activeStreamId: null
-    }).where(and(eq(agentThread.id, threadId), eq(agentThread.activeStreamId, runId))).returning({
-      id: agentThread.id
-    });
-    return released.length > 0;
+    }).where(and(eq(this.t.agentThread.id, threadId), eq(this.t.agentThread.activeStreamId, runId))), this.t.agentThread.id);
+    return released > 0;
   }
   async enqueueMessage(input) {
     const now = /* @__PURE__ */ new Date();
     const [edge] = await this.db.select({
-      first: min(agentQueuedMessage.position),
-      last: max(agentQueuedMessage.position)
-    }).from(agentQueuedMessage).where(eq(agentQueuedMessage.threadId, input.threadId));
+      first: min(this.t.agentQueuedMessage.position),
+      last: max(this.t.agentQueuedMessage.position)
+    }).from(this.t.agentQueuedMessage).where(eq(this.t.agentQueuedMessage.threadId, input.threadId));
     const position = input.at === "head" ? (edge?.first ?? 1) - 1 : (edge?.last ?? -1) + 1;
     const row = {
       id: crypto.randomUUID(),
@@ -876,6 +1368,7 @@ var DrizzleAgentStore = class {
       content: input.content,
       attachments: input.attachments !== void 0 && input.attachments.length > 0 ? input.attachments : null,
       agentName: input.agentName ?? null,
+      persona: input.persona ?? null,
       model: input.model ?? null,
       pageContext: input.pageContext ?? null,
       interrupt: input.interrupt === true,
@@ -883,15 +1376,15 @@ var DrizzleAgentStore = class {
       createdAt: now,
       updatedAt: now
     };
-    await this.db.insert(agentQueuedMessage).values(row);
+    await this.db.insert(this.t.agentQueuedMessage).values(row);
     return toQueuedMessage(row);
   }
   async listQueue(threadId) {
-    const rows = await this.db.select().from(agentQueuedMessage).where(eq(agentQueuedMessage.threadId, threadId)).orderBy(asc(agentQueuedMessage.position), asc(agentQueuedMessage.createdAt));
+    const rows = await this.db.select().from(this.t.agentQueuedMessage).where(eq(this.t.agentQueuedMessage.threadId, threadId)).orderBy(asc(this.t.agentQueuedMessage.position), asc(this.t.agentQueuedMessage.createdAt));
     return rows.map(toQueuedMessage);
   }
   async getQueuedMessage(id) {
-    const [row] = await this.db.select().from(agentQueuedMessage).where(eq(agentQueuedMessage.id, id));
+    const [row] = await this.db.select().from(this.t.agentQueuedMessage).where(eq(this.t.agentQueuedMessage.id, id));
     return row === void 0 ? null : toQueuedMessage(row);
   }
   async updateQueuedMessage(id, patch) {
@@ -907,52 +1400,47 @@ var DrizzleAgentStore = class {
     if (patch.interrupt !== void 0) {
       updates.interrupt = patch.interrupt;
     }
-    await this.db.update(agentQueuedMessage).set(updates).where(eq(agentQueuedMessage.id, id));
+    await this.db.update(this.t.agentQueuedMessage).set(updates).where(eq(this.t.agentQueuedMessage.id, id));
     return this.getQueuedMessage(id);
   }
   /** Rewrites the thread's positions as 0..n-1 in the new order — a queue is a handful of rows. */
-  async moveQueuedMessage(id, index2) {
+  async moveQueuedMessage(id, index4) {
     const moving = await this.getQueuedMessage(id);
     if (moving === null) {
       return false;
     }
     const order = (await this.listQueue(moving.threadId)).filter((message) => message.id !== id);
-    const target = Math.max(0, Math.min(order.length, Math.trunc(index2)));
+    const target = Math.max(0, Math.min(order.length, Math.trunc(index4)));
     order.splice(target, 0, moving);
     for (const [position, message] of order.entries()) {
-      await this.db.update(agentQueuedMessage).set({
+      await this.db.update(this.t.agentQueuedMessage).set({
         position
-      }).where(eq(agentQueuedMessage.id, message.id));
+      }).where(eq(this.t.agentQueuedMessage.id, message.id));
     }
     return true;
   }
   async removeQueuedMessage(id) {
-    const removed = await this.db.delete(agentQueuedMessage).where(eq(agentQueuedMessage.id, id)).returning({
-      id: agentQueuedMessage.id
-    });
-    return removed.length > 0;
+    const removed = await affectedRows(this.dialect, this.db.delete(this.t.agentQueuedMessage).where(eq(this.t.agentQueuedMessage.id, id)), this.t.agentQueuedMessage.id);
+    return removed > 0;
   }
   async clearQueue(threadId) {
-    const removed = await this.db.delete(agentQueuedMessage).where(eq(agentQueuedMessage.threadId, threadId)).returning({
-      id: agentQueuedMessage.id
-    });
-    return removed.length;
+    return affectedRows(this.dialect, this.db.delete(this.t.agentQueuedMessage).where(eq(this.t.agentQueuedMessage.threadId, threadId)), this.t.agentQueuedMessage.id);
   }
   async queuePause(threadId) {
     const [thread] = await this.db.select({
-      queuePause: agentThread.queuePause
-    }).from(agentThread).where(eq(agentThread.id, threadId));
+      queuePause: this.t.agentThread.queuePause
+    }).from(this.t.agentThread).where(eq(this.t.agentThread.id, threadId));
     return thread?.queuePause ?? null;
   }
   async setQueuePause(threadId, pause) {
-    await this.db.update(agentThread).set({
+    await this.db.update(this.t.agentThread).set({
       queuePause: pause
-    }).where(eq(agentThread.id, threadId));
+    }).where(eq(this.t.agentThread.id, threadId));
   }
   async setActiveStream(threadId, runId) {
-    await this.db.update(agentThread).set({
+    await this.db.update(this.t.agentThread).set({
       activeStreamId: runId
-    }).where(eq(agentThread.id, threadId));
+    }).where(eq(this.t.agentThread.id, threadId));
   }
   /**
   * Patch a thread's `title` and/or `defaultAgent`. Each field is only touched when PRESENT in
@@ -971,11 +1459,14 @@ var DrizzleAgentStore = class {
     if (patch.model !== void 0) {
       updates.model = patch.model;
     }
+    if (patch.persona !== void 0) {
+      updates.persona = patch.persona;
+    }
     if (Object.keys(updates).length === 0) {
       return;
     }
     updates.updatedAt = /* @__PURE__ */ new Date();
-    await this.db.update(agentThread).set(updates).where(eq(agentThread.id, threadId));
+    await this.db.update(this.t.agentThread).set(updates).where(eq(this.t.agentThread.id, threadId));
   }
   /**
   * The thread's default agent, projected — the caller wants one nullable scalar to decide which
@@ -985,14 +1476,21 @@ var DrizzleAgentStore = class {
   /** The thread's pinned model, projected like {@link defaultAgentForThread}. */
   async modelForThread(threadId) {
     const [thread] = await this.db.select({
-      model: agentThread.model
-    }).from(agentThread).where(and(eq(agentThread.id, threadId), isNull(agentThread.deletedAt)));
+      model: this.t.agentThread.model
+    }).from(this.t.agentThread).where(and(eq(this.t.agentThread.id, threadId), isNull(this.t.agentThread.deletedAt)));
     return thread?.model ?? null;
   }
+  /** The thread's pinned persona, projected like {@link defaultAgentForThread}. */
+  async personaForThread(threadId) {
+    const [thread] = await this.db.select({
+      persona: this.t.agentThread.persona
+    }).from(agentThread).where(and(eq(this.t.agentThread.id, threadId), isNull(this.t.agentThread.deletedAt)));
+    return thread?.persona ?? null;
+  }
   async defaultAgentForThread(threadId) {
     const [thread] = await this.db.select({
-      defaultAgent: agentThread.defaultAgent
-    }).from(agentThread).where(and(eq(agentThread.id, threadId), isNull(agentThread.deletedAt)));
+      defaultAgent: this.t.agentThread.defaultAgent
+    }).from(this.t.agentThread).where(and(eq(this.t.agentThread.id, threadId), isNull(this.t.agentThread.deletedAt)));
     return thread?.defaultAgent ?? null;
   }
   /**
@@ -1017,7 +1515,7 @@ var DrizzleAgentStore = class {
       promptHash: run.promptHash ?? null,
       parentRunId: run.parentRunId ?? null
     };
-    await this.db.insert(agentRun).values(runRow);
+    await this.db.insert(this.t.agentRun).values(runRow);
   }
   /** Settle a run's outcome. A no-op when the run is unknown (mirrors `setTitle`/`updateThread`). */
   async recordRunEnd(end) {
@@ -1034,16 +1532,16 @@ var DrizzleAgentStore = class {
     if (end.errorMessage !== void 0) {
       updates.errorMessage = end.errorMessage;
     }
-    await this.db.update(agentRun).set(updates).where(eq(agentRun.id, end.runId));
+    await this.db.update(this.t.agentRun).set(updates).where(eq(this.t.agentRun.id, end.runId));
   }
   /** Bump the run's llm-step retry counter. Atomic `retries = retries + 1`, no read-modify-write. */
   async bumpRunRetries(runId) {
-    await this.db.update(agentRun).set({
-      retries: sql2`${agentRun.retries} + 1`
-    }).where(eq(agentRun.id, runId));
+    await this.db.update(this.t.agentRun).set({
+      retries: sql2`${this.t.agentRun.retries} + 1`
+    }).where(eq(this.t.agentRun.id, runId));
   }
   async appendMessage(input) {
-    const [thread] = await this.db.select().from(agentThread).where(eq(agentThread.id, input.threadId));
+    const [thread] = await this.db.select().from(this.t.agentThread).where(eq(this.t.agentThread.id, input.threadId));
     if (thread === void 0) {
       throw new Error(`thread ${input.threadId} not found`);
     }
@@ -1059,52 +1557,54 @@ var DrizzleAgentStore = class {
       followUps: input.followUps ?? null,
       usage: input.usage ?? null,
       agentName: input.agentName ?? null,
+      persona: input.persona ?? null,
       runId: input.runId ?? null,
       reasoning: input.reasoning ?? null,
       reasoningMs: input.reasoningMs ?? null,
       ui: input.ui ?? null,
       feedback: null,
+      seq: await this.nextMessageSeq(input.threadId),
       createdAt: now
     };
-    await this.db.insert(agentMessage).values(message);
-    await this.db.update(agentThread).set({
+    await this.db.insert(this.t.agentMessage).values(message);
+    await this.db.update(this.t.agentThread).set({
       updatedAt: now
-    }).where(eq(agentThread.id, input.threadId));
+    }).where(eq(this.t.agentThread.id, input.threadId));
     return this.toStoredMessage(message);
   }
   async setMessageToolResults(messageId, results) {
-    await this.db.update(agentMessage).set({
+    await this.db.update(this.t.agentMessage).set({
       toolResults: results
-    }).where(eq(agentMessage.id, messageId));
+    }).where(eq(this.t.agentMessage.id, messageId));
   }
   async setMessageUi(messageId, ui) {
-    await this.db.update(agentMessage).set({
+    await this.db.update(this.t.agentMessage).set({
       ui: ui.length > 0 ? ui : null
-    }).where(eq(agentMessage.id, messageId));
+    }).where(eq(this.t.agentMessage.id, messageId));
   }
   async threadOfMessage(messageId) {
     const [row] = await this.db.select({
-      threadId: agentMessage.threadId
-    }).from(agentMessage).where(eq(agentMessage.id, messageId)).limit(1);
+      threadId: this.t.agentMessage.threadId
+    }).from(this.t.agentMessage).where(eq(this.t.agentMessage.id, messageId)).limit(1);
     return row?.threadId ?? null;
   }
   async setMessageFeedback(messageId, feedback) {
-    await this.db.update(agentMessage).set({
+    await this.db.update(this.t.agentMessage).set({
       feedback
-    }).where(eq(agentMessage.id, messageId));
+    }).where(eq(this.t.agentMessage.id, messageId));
   }
   async truncateFrom(threadId, messageId) {
-    const messages = await this.db.select().from(agentMessage).where(eq(agentMessage.threadId, threadId)).orderBy(asc(agentMessage.createdAt), asc(agentMessage.id));
+    const messages = await this.db.select().from(this.t.agentMessage).where(eq(this.t.agentMessage.threadId, threadId)).orderBy(...this.messageOrder());
     const cutoff = messages.findIndex((message) => message.id === messageId);
     if (cutoff < 0) {
       return;
     }
     const doomedIds = messages.slice(cutoff).map((message) => message.id);
-    await this.db.delete(agentToolCall).where(inArray(agentToolCall.messageId, doomedIds));
-    await this.db.delete(agentMessage).where(inArray(agentMessage.id, doomedIds));
+    await this.db.delete(this.t.agentToolCall).where(inArray(this.t.agentToolCall.messageId, doomedIds));
+    await this.db.delete(this.t.agentMessage).where(inArray(this.t.agentMessage.id, doomedIds));
   }
   async recordToolCall(input) {
-    await this.db.insert(agentToolCall).values({
+    await this.db.insert(this.t.agentToolCall).values({
       id: input.toolCallId,
       messageId: input.messageId,
       toolName: input.toolName,
@@ -1149,18 +1649,18 @@ var DrizzleAgentStore = class {
     if (input.status === "executed" || input.status === "auto_executed") {
       updates.executedAt = /* @__PURE__ */ new Date();
     }
-    await this.db.update(agentToolCall).set(updates).where(eq(agentToolCall.id, input.toolCallId));
+    await this.db.update(this.t.agentToolCall).set(updates).where(eq(this.t.agentToolCall.id, input.toolCallId));
   }
   async toolCallOutcomes(toolCallIds) {
     if (toolCallIds.length === 0) {
       return [];
     }
     const rows = await this.db.select({
-      id: agentToolCall.id,
-      status: agentToolCall.status,
-      output: agentToolCall.output,
-      error: agentToolCall.error
-    }).from(agentToolCall).where(inArray(agentToolCall.id, [
+      id: this.t.agentToolCall.id,
+      status: this.t.agentToolCall.status,
+      output: this.t.agentToolCall.output,
+      error: this.t.agentToolCall.error
+    }).from(this.t.agentToolCall).where(inArray(this.t.agentToolCall.id, [
       ...toolCallIds
     ]));
     return rows.map((row) => ({
@@ -1175,14 +1675,14 @@ var DrizzleAgentStore = class {
     }));
   }
   async failUnsettledToolCalls(runId, error) {
-    const pending = and(eq(agentToolCall.runId, runId), eq(agentToolCall.status, "pending_approval"));
+    const pending = and(eq(this.t.agentToolCall.runId, runId), eq(this.t.agentToolCall.status, "pending_approval"));
     const rows = await this.db.select({
-      id: agentToolCall.id
-    }).from(agentToolCall).where(pending);
+      id: this.t.agentToolCall.id
+    }).from(this.t.agentToolCall).where(pending);
     if (rows.length === 0) {
       return 0;
     }
-    await this.db.update(agentToolCall).set({
+    await this.db.update(this.t.agentToolCall).set({
       status: "failed",
       error
     }).where(pending);
@@ -1207,11 +1707,11 @@ var DrizzleAgentStore = class {
       return [];
     }
     const rows = await this.db.select({
-      attachments: agentMessage.attachments
-    }).from(agentMessage).innerJoin(agentThread, eq(agentMessage.threadId, agentThread.id)).where(and(eq(agentThread.actorRef, actorRef), isNotNull(agentMessage.attachments)));
+      attachments: this.t.agentMessage.attachments
+    }).from(this.t.agentMessage).innerJoin(this.t.agentThread, eq(this.t.agentMessage.threadId, this.t.agentThread.id)).where(and(eq(this.t.agentThread.actorRef, actorRef), isNotNull(this.t.agentMessage.attachments)));
     const queued = await this.db.select({
-      attachments: agentQueuedMessage.attachments
-    }).from(agentQueuedMessage).innerJoin(agentThread, eq(agentQueuedMessage.threadId, agentThread.id)).where(and(eq(agentThread.actorRef, actorRef), isNotNull(agentQueuedMessage.attachments)));
+      attachments: this.t.agentQueuedMessage.attachments
+    }).from(this.t.agentQueuedMessage).innerJoin(this.t.agentThread, eq(this.t.agentQueuedMessage.threadId, this.t.agentThread.id)).where(and(eq(this.t.agentThread.actorRef, actorRef), isNotNull(this.t.agentQueuedMessage.attachments)));
     const wanted = new Set(mediaIds);
     const found = /* @__PURE__ */ new Set();
     for (const row of [
@@ -1229,7 +1729,7 @@ var DrizzleAgentStore = class {
     ].filter((mediaId) => found.has(mediaId));
   }
   async recordUsage(input) {
-    await this.db.insert(agentTokenUsage).values({
+    await this.db.insert(this.t.agentTokenUsage).values({
       id: crypto.randomUUID(),
       threadId: input.threadId,
       actorRef: input.actorRef,
@@ -1250,7 +1750,7 @@ var DrizzleAgentStore = class {
   async usageBetween(actorRef, fromDay, toDay) {
     const start = /* @__PURE__ */ new Date(`${fromDay}T00:00:00.000Z`);
     const end = /* @__PURE__ */ new Date(`${toDay}T23:59:59.999Z`);
-    const rows = await this.db.select().from(agentTokenUsage).where(and(eq(agentTokenUsage.actorRef, actorRef), gte(agentTokenUsage.createdAt, start), lte(agentTokenUsage.createdAt, end)));
+    const rows = await this.db.select().from(this.t.agentTokenUsage).where(and(eq(this.t.agentTokenUsage.actorRef, actorRef), gte(this.t.agentTokenUsage.createdAt, start), lte(this.t.agentTokenUsage.createdAt, end)));
     const usedTokens = rows.reduce((sum2, row) => sum2 + row.inputTokens + row.outputTokens, 0);
     const costUsd = rows.reduce((sum2, row) => sum2 + (row.costUsd ?? 0), 0);
     return {
@@ -1269,6 +1769,7 @@ var DrizzleAgentStore = class {
       ...thread.model != null ? {
         model: thread.model
       } : {},
+      persona: thread.persona ?? null,
       ...lastContent !== void 0 ? {
         lastMessagePreview: lastContent.slice(0, 120)
       } : {}
@@ -1298,6 +1799,9 @@ var DrizzleAgentStore = class {
       ...message.agentName != null ? {
         agentName: message.agentName
       } : {},
+      ...message.persona != null ? {
+        persona: message.persona
+      } : {},
       ...message.runId != null ? {
         runId: message.runId
       } : {},
@@ -1328,6 +1832,9 @@ function toQueuedMessage(row) {
     ...row.agentName !== null ? {
       agentName: row.agentName
     } : {},
+    ...row.persona !== null ? {
+      persona: row.persona
+    } : {},
     ...row.model !== null ? {
       model: row.model
     } : {},
@@ -1458,11 +1965,15 @@ var DrizzleGovernanceQueries = class {
   static {
     __name(this, "DrizzleGovernanceQueries");
   }
-  db;
   pricingStore;
+  db;
+  dialect;
+  t;
   constructor(db, pricingStore) {
-    this.db = db;
     this.pricingStore = pricingStore;
+    this.dialect = agentDialectOf(db);
+    this.t = agentTablesFor(this.dialect);
+    this.db = asBuilder(db);
   }
   async loadPricing() {
     const prices = await this.pricingStore.listCurrentPrices();
@@ -1479,7 +1990,7 @@ var DrizzleGovernanceQueries = class {
   }
   async usageInRange(range) {
     const { start, end } = dayBoundsUtc(range);
-    const rows = await this.db.select().from(agentTokenUsage).where(and2(gte2(agentTokenUsage.createdAt, start), lte2(agentTokenUsage.createdAt, end)));
+    const rows = await this.db.select().from(this.t.agentTokenUsage).where(and2(gte2(this.t.agentTokenUsage.createdAt, start), lte2(this.t.agentTokenUsage.createdAt, end)));
     return rows.map(toUsageInput);
   }
   async spendByModel(range) {
@@ -1504,7 +2015,7 @@ var DrizzleGovernanceQueries = class {
     if (threadIds.length === 0) {
       return [];
     }
-    const threads = await this.db.select().from(agentThread).where(and2(inArray2(agentThread.id, threadIds), isNull2(agentThread.deletedAt)));
+    const threads = await this.db.select().from(this.t.agentThread).where(and2(inArray2(this.t.agentThread.id, threadIds), isNull2(this.t.agentThread.deletedAt)));
     const threadsById = new Map(threads.map((thread) => [
       thread.id,
       {
@@ -1523,14 +2034,14 @@ var DrizzleGovernanceQueries = class {
   }
   async recentToolCalls(limit) {
     const rows = await this.db.select({
-      toolCallId: agentToolCall.id,
-      toolName: agentToolCall.toolName,
-      toolType: agentToolCall.toolType,
-      status: agentToolCall.status,
-      threadId: agentMessage.threadId,
-      createdAt: agentToolCall.createdAt,
-      runId: agentToolCall.runId
-    }).from(agentToolCall).innerJoin(agentMessage, eq2(agentToolCall.messageId, agentMessage.id)).orderBy(desc2(agentToolCall.createdAt), desc2(agentToolCall.id)).limit(limit);
+      toolCallId: this.t.agentToolCall.id,
+      toolName: this.t.agentToolCall.toolName,
+      toolType: this.t.agentToolCall.toolType,
+      status: this.t.agentToolCall.status,
+      threadId: this.t.agentMessage.threadId,
+      createdAt: this.t.agentToolCall.createdAt,
+      runId: this.t.agentToolCall.runId
+    }).from(this.t.agentToolCall).innerJoin(this.t.agentMessage, eq2(this.t.agentToolCall.messageId, this.t.agentMessage.id)).orderBy(desc2(this.t.agentToolCall.createdAt), desc2(this.t.agentToolCall.id)).limit(limit);
     return rows.map((row) => ({
       toolCallId: row.toolCallId,
       toolName: row.toolName,
@@ -1558,9 +2069,9 @@ var DrizzleGovernanceQueries = class {
       return rollups;
     }
     const messageCounts = await this.db.select({
-      threadId: agentMessage.threadId,
+      threadId: this.t.agentMessage.threadId,
       value: count()
-    }).from(agentMessage).where(inArray2(agentMessage.threadId, threadIds)).groupBy(agentMessage.threadId);
+    }).from(this.t.agentMessage).where(inArray2(this.t.agentMessage.threadId, threadIds)).groupBy(this.t.agentMessage.threadId);
     for (const row of messageCounts) {
       const bucket = rollups.get(row.threadId);
       if (bucket !== void 0) {
@@ -1568,10 +2079,10 @@ var DrizzleGovernanceQueries = class {
       }
     }
     const usageTotals = await this.db.select({
-      threadId: agentTokenUsage.threadId,
-      inputTokens: sum(agentTokenUsage.inputTokens),
-      outputTokens: sum(agentTokenUsage.outputTokens)
-    }).from(agentTokenUsage).where(inArray2(agentTokenUsage.threadId, threadIds)).groupBy(agentTokenUsage.threadId);
+      threadId: this.t.agentTokenUsage.threadId,
+      inputTokens: sum(this.t.agentTokenUsage.inputTokens),
+      outputTokens: sum(this.t.agentTokenUsage.outputTokens)
+    }).from(this.t.agentTokenUsage).where(inArray2(this.t.agentTokenUsage.threadId, threadIds)).groupBy(this.t.agentTokenUsage.threadId);
     for (const row of usageTotals) {
       const bucket = rollups.get(row.threadId);
       if (bucket !== void 0) {
@@ -1596,12 +2107,12 @@ var DrizzleGovernanceQueries = class {
     });
   }
   async recentThreads(limit) {
-    const threads = await this.db.select().from(agentThread).where(isNull2(agentThread.deletedAt)).orderBy(desc2(agentThread.updatedAt), desc2(agentThread.id)).limit(limit);
+    const threads = await this.db.select().from(this.t.agentThread).where(isNull2(this.t.agentThread.deletedAt)).orderBy(desc2(this.t.agentThread.updatedAt), desc2(this.t.agentThread.id)).limit(limit);
     return this.toThreadActivityRows(threads);
   }
   async runsInRange(range) {
     const { start, end } = dayBoundsUtc(range);
-    return this.db.select().from(agentRun).where(and2(gte2(agentRun.startedAt, start), lte2(agentRun.startedAt, end)));
+    return this.db.select().from(this.t.agentRun).where(and2(gte2(this.t.agentRun.startedAt, start), lte2(this.t.agentRun.startedAt, end)));
   }
   async runMetrics(range) {
     const runs = await this.runsInRange(range);
@@ -1658,7 +2169,7 @@ var DrizzleGovernanceQueries = class {
   }
   async runErrors(range) {
     const { start, end } = dayBoundsUtc(range);
-    const runs = await this.db.select().from(agentRun).where(and2(gte2(agentRun.startedAt, start), lte2(agentRun.startedAt, end), eq2(agentRun.status, "failed")));
+    const runs = await this.db.select().from(this.t.agentRun).where(and2(gte2(this.t.agentRun.startedAt, start), lte2(this.t.agentRun.startedAt, end), eq2(this.t.agentRun.status, "failed")));
     const byError = /* @__PURE__ */ new Map();
     for (const run of runs) {
       const errorCode = run.errorCode ?? UNCLASSIFIED_ERROR_CODE;
@@ -1698,7 +2209,7 @@ var DrizzleGovernanceQueries = class {
     return result;
   }
   async recentRuns(limit) {
-    const runs = await this.db.select().from(agentRun).orderBy(desc2(agentRun.startedAt), desc2(agentRun.id)).limit(limit);
+    const runs = await this.db.select().from(this.t.agentRun).orderBy(desc2(this.t.agentRun.startedAt), desc2(this.t.agentRun.id)).limit(limit);
     return runs.map(toRecentRunRow);
   }
   /**
@@ -1708,16 +2219,16 @@ var DrizzleGovernanceQueries = class {
   */
   async pendingApprovals(limit) {
     const rows = await this.db.select({
-      toolCallId: agentToolCall.id,
-      toolName: agentToolCall.toolName,
-      input: agentToolCall.input,
-      threadId: agentThread.id,
-      threadTitle: agentThread.title,
-      actorRef: agentThread.actorRef,
-      agentName: agentMessage.agentName,
-      requestedAt: agentToolCall.createdAt,
-      runId: agentToolCall.runId
-    }).from(agentToolCall).innerJoin(agentMessage, eq2(agentToolCall.messageId, agentMessage.id)).innerJoin(agentThread, eq2(agentMessage.threadId, agentThread.id)).where(eq2(agentToolCall.status, "pending_approval")).orderBy(agentToolCall.createdAt, agentToolCall.id).limit(limit);
+      toolCallId: this.t.agentToolCall.id,
+      toolName: this.t.agentToolCall.toolName,
+      input: this.t.agentToolCall.input,
+      threadId: this.t.agentThread.id,
+      threadTitle: this.t.agentThread.title,
+      actorRef: this.t.agentThread.actorRef,
+      agentName: this.t.agentMessage.agentName,
+      requestedAt: this.t.agentToolCall.createdAt,
+      runId: this.t.agentToolCall.runId
+    }).from(this.t.agentToolCall).innerJoin(this.t.agentMessage, eq2(this.t.agentToolCall.messageId, this.t.agentMessage.id)).innerJoin(this.t.agentThread, eq2(this.t.agentMessage.threadId, this.t.agentThread.id)).where(eq2(this.t.agentToolCall.status, "pending_approval")).orderBy(this.t.agentToolCall.createdAt, this.t.agentToolCall.id).limit(limit);
     return rows.map((row) => ({
       toolCallId: row.toolCallId,
       toolName: row.toolName,
@@ -1739,11 +2250,11 @@ var DrizzleGovernanceQueries = class {
   */
   async toolStats(range) {
     const { start, end } = dayBoundsUtc(range);
-    const calls = await this.db.select().from(agentToolCall).where(and2(gte2(agentToolCall.createdAt, start), lte2(agentToolCall.createdAt, end)));
+    const calls = await this.db.select().from(this.t.agentToolCall).where(and2(gte2(this.t.agentToolCall.createdAt, start), lte2(this.t.agentToolCall.createdAt, end)));
     const byTool = /* @__PURE__ */ new Map();
     for (const call of calls) {
-      const key = `${call.toolName} ${call.toolType}`;
-      const bucket = byTool.get(key) ?? {
+      const key2 = `${call.toolName} ${call.toolType}`;
+      const bucket = byTool.get(key2) ?? {
         toolName: call.toolName,
         toolType: call.toolType,
         calls: 0,
@@ -1757,7 +2268,7 @@ var DrizzleGovernanceQueries = class {
       if (call.executionMs != null) {
         bucket.executionMs.push(call.executionMs);
       }
-      byTool.set(key, bucket);
+      byTool.set(key2, bucket);
     }
     const result = [];
     for (const bucket of byTool.values()) {
@@ -1789,19 +2300,19 @@ var DrizzleGovernanceQueries = class {
     if (filters?.toolType !== void 0 && !isToolKind(filters.toolType)) {
       return emptyPage(query);
     }
-    const whereClause = and2(filters?.toolName !== void 0 ? eq2(agentToolCall.toolName, filters.toolName) : void 0, filters?.toolType !== void 0 && isToolKind(filters.toolType) ? eq2(agentToolCall.toolType, filters.toolType) : void 0, filters?.status !== void 0 && isToolCallStatus(filters.status) ? eq2(agentToolCall.status, filters.status) : void 0, filters?.threadId !== void 0 ? eq2(agentMessage.threadId, filters.threadId) : void 0, filters?.fromDay !== void 0 ? gte2(agentToolCall.createdAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? lte2(agentToolCall.createdAt, dayEndUtc(filters.toDay)) : void 0);
+    const whereClause = and2(filters?.toolName !== void 0 ? eq2(this.t.agentToolCall.toolName, filters.toolName) : void 0, filters?.toolType !== void 0 && isToolKind(filters.toolType) ? eq2(this.t.agentToolCall.toolType, filters.toolType) : void 0, filters?.status !== void 0 && isToolCallStatus(filters.status) ? eq2(this.t.agentToolCall.status, filters.status) : void 0, filters?.threadId !== void 0 ? eq2(this.t.agentMessage.threadId, filters.threadId) : void 0, filters?.fromDay !== void 0 ? gte2(this.t.agentToolCall.createdAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? lte2(this.t.agentToolCall.createdAt, dayEndUtc(filters.toDay)) : void 0);
     const [totalRow] = await this.db.select({
       value: count()
-    }).from(agentToolCall).innerJoin(agentMessage, eq2(agentToolCall.messageId, agentMessage.id)).where(whereClause);
+    }).from(this.t.agentToolCall).innerJoin(this.t.agentMessage, eq2(this.t.agentToolCall.messageId, this.t.agentMessage.id)).where(whereClause);
     const rows = await this.db.select({
-      toolCallId: agentToolCall.id,
-      toolName: agentToolCall.toolName,
-      toolType: agentToolCall.toolType,
-      status: agentToolCall.status,
-      threadId: agentMessage.threadId,
-      createdAt: agentToolCall.createdAt,
-      runId: agentToolCall.runId
-    }).from(agentToolCall).innerJoin(agentMessage, eq2(agentToolCall.messageId, agentMessage.id)).where(whereClause).orderBy(desc2(agentToolCall.createdAt), desc2(agentToolCall.id)).limit(query.pageSize).offset((query.page - 1) * query.pageSize);
+      toolCallId: this.t.agentToolCall.id,
+      toolName: this.t.agentToolCall.toolName,
+      toolType: this.t.agentToolCall.toolType,
+      status: this.t.agentToolCall.status,
+      threadId: this.t.agentMessage.threadId,
+      createdAt: this.t.agentToolCall.createdAt,
+      runId: this.t.agentToolCall.runId
+    }).from(this.t.agentToolCall).innerJoin(this.t.agentMessage, eq2(this.t.agentToolCall.messageId, this.t.agentMessage.id)).where(whereClause).orderBy(desc2(this.t.agentToolCall.createdAt), desc2(this.t.agentToolCall.id)).limit(query.pageSize).offset((query.page - 1) * query.pageSize);
     return {
       rows: rows.map((row) => ({
         toolCallId: row.toolCallId,
@@ -1825,11 +2336,11 @@ var DrizzleGovernanceQueries = class {
   */
   async threadsPage(query) {
     const filters = query.where;
-    const whereClause = and2(isNull2(agentThread.deletedAt), filters?.actorRef !== void 0 ? eq2(agentThread.actorRef, filters.actorRef) : void 0, filters?.title !== void 0 ? like(sql3`lower(${agentThread.title})`, `%${filters.title.toLowerCase()}%`) : void 0, filters?.fromDay !== void 0 ? gte2(agentThread.updatedAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? lte2(agentThread.updatedAt, dayEndUtc(filters.toDay)) : void 0);
+    const whereClause = and2(isNull2(this.t.agentThread.deletedAt), filters?.actorRef !== void 0 ? eq2(this.t.agentThread.actorRef, filters.actorRef) : void 0, filters?.title !== void 0 ? like(sql3`lower(${this.t.agentThread.title})`, `%${filters.title.toLowerCase()}%`) : void 0, filters?.fromDay !== void 0 ? gte2(this.t.agentThread.updatedAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? lte2(this.t.agentThread.updatedAt, dayEndUtc(filters.toDay)) : void 0);
     const [totalRow] = await this.db.select({
       value: count()
-    }).from(agentThread).where(whereClause);
-    const threads = await this.db.select().from(agentThread).where(whereClause).orderBy(desc2(agentThread.updatedAt), desc2(agentThread.id)).limit(query.pageSize).offset((query.page - 1) * query.pageSize);
+    }).from(this.t.agentThread).where(whereClause);
+    const threads = await this.db.select().from(this.t.agentThread).where(whereClause).orderBy(desc2(this.t.agentThread.updatedAt), desc2(this.t.agentThread.id)).limit(query.pageSize).offset((query.page - 1) * query.pageSize);
     const rows = await this.toThreadActivityRows(threads);
     return {
       rows,
@@ -1847,11 +2358,11 @@ var DrizzleGovernanceQueries = class {
     if (filters?.status !== void 0 && !isRunStatus(filters.status)) {
       return emptyPage(query);
     }
-    const whereClause = and2(filters?.agentName !== void 0 ? eq2(agentRun.agentName, filters.agentName) : void 0, filters?.status !== void 0 && isRunStatus(filters.status) ? eq2(agentRun.status, filters.status) : void 0, filters?.errorCode !== void 0 ? eq2(agentRun.errorCode, filters.errorCode) : void 0, filters?.threadId !== void 0 ? eq2(agentRun.threadId, filters.threadId) : void 0, filters?.fromDay !== void 0 ? gte2(agentRun.startedAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? lte2(agentRun.startedAt, dayEndUtc(filters.toDay)) : void 0);
+    const whereClause = and2(filters?.agentName !== void 0 ? eq2(this.t.agentRun.agentName, filters.agentName) : void 0, filters?.status !== void 0 && isRunStatus(filters.status) ? eq2(this.t.agentRun.status, filters.status) : void 0, filters?.errorCode !== void 0 ? eq2(this.t.agentRun.errorCode, filters.errorCode) : void 0, filters?.threadId !== void 0 ? eq2(this.t.agentRun.threadId, filters.threadId) : void 0, filters?.fromDay !== void 0 ? gte2(this.t.agentRun.startedAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? lte2(this.t.agentRun.startedAt, dayEndUtc(filters.toDay)) : void 0);
     const [totalRow] = await this.db.select({
       value: count()
-    }).from(agentRun).where(whereClause);
-    const runs = await this.db.select().from(agentRun).where(whereClause).orderBy(desc2(agentRun.startedAt), desc2(agentRun.id)).limit(query.pageSize).offset((query.page - 1) * query.pageSize);
+    }).from(this.t.agentRun).where(whereClause);
+    const runs = await this.db.select().from(this.t.agentRun).where(whereClause).orderBy(desc2(this.t.agentRun.startedAt), desc2(this.t.agentRun.id)).limit(query.pageSize).offset((query.page - 1) * query.pageSize);
     return {
       rows: runs.map(toRecentRunRow),
       total: totalRow?.value ?? 0,
@@ -1871,21 +2382,21 @@ var DrizzleGovernanceQueries = class {
   */
   async approvalsPage(query) {
     const filters = query.where;
-    const whereClause = and2(eq2(agentToolCall.status, "pending_approval"), filters?.toolName !== void 0 ? eq2(agentToolCall.toolName, filters.toolName) : void 0, filters?.threadId !== void 0 ? eq2(agentThread.id, filters.threadId) : void 0, filters?.actorRef !== void 0 ? eq2(agentThread.actorRef, filters.actorRef) : void 0, filters?.agentName !== void 0 ? eq2(agentMessage.agentName, filters.agentName) : void 0, filters?.fromDay !== void 0 ? gte2(agentToolCall.createdAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? lte2(agentToolCall.createdAt, dayEndUtc(filters.toDay)) : void 0);
+    const whereClause = and2(eq2(this.t.agentToolCall.status, "pending_approval"), filters?.toolName !== void 0 ? eq2(this.t.agentToolCall.toolName, filters.toolName) : void 0, filters?.threadId !== void 0 ? eq2(this.t.agentThread.id, filters.threadId) : void 0, filters?.actorRef !== void 0 ? eq2(this.t.agentThread.actorRef, filters.actorRef) : void 0, filters?.agentName !== void 0 ? eq2(this.t.agentMessage.agentName, filters.agentName) : void 0, filters?.fromDay !== void 0 ? gte2(this.t.agentToolCall.createdAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? lte2(this.t.agentToolCall.createdAt, dayEndUtc(filters.toDay)) : void 0);
     const [totalRow] = await this.db.select({
       value: count()
-    }).from(agentToolCall).innerJoin(agentMessage, eq2(agentToolCall.messageId, agentMessage.id)).innerJoin(agentThread, eq2(agentMessage.threadId, agentThread.id)).where(whereClause);
+    }).from(this.t.agentToolCall).innerJoin(this.t.agentMessage, eq2(this.t.agentToolCall.messageId, this.t.agentMessage.id)).innerJoin(this.t.agentThread, eq2(this.t.agentMessage.threadId, this.t.agentThread.id)).where(whereClause);
     const rows = await this.db.select({
-      toolCallId: agentToolCall.id,
-      toolName: agentToolCall.toolName,
-      input: agentToolCall.input,
-      threadId: agentThread.id,
-      threadTitle: agentThread.title,
-      actorRef: agentThread.actorRef,
-      agentName: agentMessage.agentName,
-      requestedAt: agentToolCall.createdAt,
-      runId: agentToolCall.runId
-    }).from(agentToolCall).innerJoin(agentMessage, eq2(agentToolCall.messageId, agentMessage.id)).innerJoin(agentThread, eq2(agentMessage.threadId, agentThread.id)).where(whereClause).orderBy(asc2(agentToolCall.createdAt), asc2(agentToolCall.id)).limit(query.pageSize).offset((query.page - 1) * query.pageSize);
+      toolCallId: this.t.agentToolCall.id,
+      toolName: this.t.agentToolCall.toolName,
+      input: this.t.agentToolCall.input,
+      threadId: this.t.agentThread.id,
+      threadTitle: this.t.agentThread.title,
+      actorRef: this.t.agentThread.actorRef,
+      agentName: this.t.agentMessage.agentName,
+      requestedAt: this.t.agentToolCall.createdAt,
+      runId: this.t.agentToolCall.runId
+    }).from(this.t.agentToolCall).innerJoin(this.t.agentMessage, eq2(this.t.agentToolCall.messageId, this.t.agentMessage.id)).innerJoin(this.t.agentThread, eq2(this.t.agentMessage.threadId, this.t.agentThread.id)).where(whereClause).orderBy(asc2(this.t.agentToolCall.createdAt), asc2(this.t.agentToolCall.id)).limit(query.pageSize).offset((query.page - 1) * query.pageSize);
     return {
       rows: rows.map((row) => ({
         toolCallId: row.toolCallId,
@@ -1910,13 +2421,13 @@ var DrizzleGovernanceQueries = class {
   */
   async runDetail(runId) {
     const [row] = await this.db.select({
-      run: agentRun,
-      thread: agentThread
-    }).from(agentRun).innerJoin(agentThread, eq2(agentRun.threadId, agentThread.id)).where(eq2(agentRun.id, runId)).limit(1);
+      run: this.t.agentRun,
+      thread: this.t.agentThread
+    }).from(this.t.agentRun).innerJoin(this.t.agentThread, eq2(this.t.agentRun.threadId, this.t.agentThread.id)).where(eq2(this.t.agentRun.id, runId)).limit(1);
     if (row === void 0) {
       return null;
     }
-    const toolCalls = await this.db.select().from(agentToolCall).where(eq2(agentToolCall.runId, runId)).orderBy(asc2(agentToolCall.createdAt), asc2(agentToolCall.id));
+    const toolCalls = await this.db.select().from(this.t.agentToolCall).where(eq2(this.t.agentToolCall.runId, runId)).orderBy(asc2(this.t.agentToolCall.createdAt), asc2(this.t.agentToolCall.id));
     return {
       run: toRecentRunRow(row.run),
       thread: {
@@ -1935,22 +2446,27 @@ var DrizzleGovernanceQueries = class {
   * message. Soft-deleted threads are returned with `deleted: true`; an audit still needs them.
   */
   async threadDetail(query) {
-    const [thread] = await this.db.select().from(agentThread).where(eq2(agentThread.id, query.threadId)).limit(1);
+    const [thread] = await this.db.select().from(this.t.agentThread).where(eq2(this.t.agentThread.id, query.threadId)).limit(1);
     if (thread === void 0) {
       return null;
     }
     const pricing = await this.loadPricing();
-    const usageRows = await this.db.select().from(agentTokenUsage).where(eq2(agentTokenUsage.threadId, thread.id));
+    const usageRows = await this.db.select().from(this.t.agentTokenUsage).where(eq2(this.t.agentTokenUsage.threadId, thread.id));
     const usage = rollupThreadUsage(usageRows.map(toUsageInput), pricing);
     const [messageCountRow] = await this.db.select({
       value: count()
-    }).from(agentMessage).where(eq2(agentMessage.threadId, thread.id));
-    const messages = await this.db.select().from(agentMessage).where(eq2(agentMessage.threadId, thread.id)).orderBy(desc2(agentMessage.createdAt), desc2(agentMessage.id)).limit(query.messageLimit);
+    }).from(this.t.agentMessage).where(eq2(this.t.agentMessage.threadId, thread.id));
+    const messages = await this.db.select().from(this.t.agentMessage).where(eq2(this.t.agentMessage.threadId, thread.id)).orderBy(
+      // Append order (see DrizzleAgentStore's message order), newest first.
+      desc2(sql3`coalesce(${this.t.agentMessage.seq}, 0)`),
+      desc2(this.t.agentMessage.createdAt),
+      desc2(this.t.agentMessage.id)
+    ).limit(query.messageLimit);
     const toolCallCounts = await this.toolCallCountsByMessage(messages.map((message) => message.id));
     const [runTotalRow] = await this.db.select({
       value: count()
-    }).from(agentRun).where(eq2(agentRun.threadId, thread.id));
-    const runs = await this.db.select().from(agentRun).where(eq2(agentRun.threadId, thread.id)).orderBy(desc2(agentRun.startedAt), desc2(agentRun.id)).limit(query.runLimit);
+    }).from(this.t.agentRun).where(eq2(this.t.agentRun.threadId, thread.id));
+    const runs = await this.db.select().from(this.t.agentRun).where(eq2(this.t.agentRun.threadId, thread.id)).orderBy(desc2(this.t.agentRun.startedAt), desc2(this.t.agentRun.id)).limit(query.runLimit);
     return {
       thread: {
         threadId: thread.id,
@@ -1985,9 +2501,9 @@ var DrizzleGovernanceQueries = class {
       return counts;
     }
     const rows = await this.db.select({
-      messageId: agentToolCall.messageId,
+      messageId: this.t.agentToolCall.messageId,
       value: count()
-    }).from(agentToolCall).where(inArray2(agentToolCall.messageId, messageIds)).groupBy(agentToolCall.messageId);
+    }).from(this.t.agentToolCall).where(inArray2(this.t.agentToolCall.messageId, messageIds)).groupBy(this.t.agentToolCall.messageId);
     for (const row of rows) {
       counts.set(row.messageId, row.value);
     }
@@ -2026,8 +2542,12 @@ var DrizzleMemoryProvider = class {
     __name(this, "DrizzleMemoryProvider");
   }
   db;
+  dialect;
+  t;
   constructor(db) {
-    this.db = db;
+    this.dialect = agentDialectOf(db);
+    this.t = agentTablesFor(this.dialect);
+    this.db = asBuilder(db);
   }
   /**
   * Every memory visible at `scopes` — asked on EVERY turn, so it is one indexed `scope in (…)`
@@ -2041,7 +2561,7 @@ var DrizzleMemoryProvider = class {
     if (scopes.length === 0) {
       return [];
     }
-    const rows = await this.db.select().from(agentMemory).where(inArray3(agentMemory.scope, [
+    const rows = await this.db.select().from(this.t.agentMemory).where(inArray3(this.t.agentMemory.scope, [
       ...scopes
     ]));
     return rows.map(toRecord);
@@ -2057,10 +2577,8 @@ var DrizzleMemoryProvider = class {
   * answer cannot be used to find out which memories exist about other people.
   */
   async forget({ id, ctx }) {
-    const deleted = await this.db.delete(agentMemory).where(and3(eq3(agentMemory.id, id), eq3(agentMemory.scope, actorScope(ctx.actor)))).returning({
-      id: agentMemory.id
-    });
-    return deleted.length > 0;
+    const deleted = await affectedRows(this.dialect, this.db.delete(this.t.agentMemory).where(and3(eq3(this.t.agentMemory.id, id), eq3(this.t.agentMemory.scope, actorScope(ctx.actor)))), this.t.agentMemory.id);
+    return deleted > 0;
   }
   /**
   * Upsert on (`scope`, `key`) — one statement, against the unique index, rather than a read
@@ -2073,7 +2591,7 @@ var DrizzleMemoryProvider = class {
   * rewriting `createdAt` would lose when the belief was first formed; reassigning `id` would
   * invalidate the delete handle a person was already shown.
   */
-  async write({ key, text: text2, scope, origin, ctx }) {
+  async write({ key: key2, text: text3, scope, origin, ctx }) {
     assertAuthorMayWriteScope({
       scope,
       origin,
@@ -2086,28 +2604,40 @@ var DrizzleMemoryProvider = class {
       originRunId: origin.runId ?? null,
       originActorRef: origin.actorRef ?? null
     };
-    const [row] = await this.db.insert(agentMemory).values({
+    const memory = this.t.agentMemory;
+    const set = {
+      text: text3,
+      ...columns,
+      updatedAt: now
+    };
+    const insert = this.db.insert(memory).values({
       id: crypto.randomUUID(),
       scope,
-      key,
-      text: text2,
+      key: key2,
+      text: text3,
       ...columns,
       pinned: false,
       createdAt: now,
       updatedAt: now
-    }).onConflictDoUpdate({
-      target: [
-        agentMemory.scope,
-        agentMemory.key
-      ],
-      set: {
-        text: text2,
-        ...columns,
-        updatedAt: now
-      }
-    }).returning();
+    });
+    let row;
+    if (this.dialect === "mysql") {
+      await upsert(this.dialect, insert, [
+        memory.scope,
+        memory.key
+      ], set);
+      [row] = await this.db.select().from(memory).where(and3(eq3(memory.scope, scope), eq3(memory.key, key2)));
+    } else {
+      [row] = await insert.onConflictDoUpdate({
+        target: [
+          memory.scope,
+          memory.key
+        ],
+        set
+      }).returning();
+    }
     if (row === void 0) {
-      throw new Error(`DrizzleMemoryProvider: the upsert of "${key}" at "${scope}" returned no row; the driver does not support RETURNING on conflict.`);
+      throw new Error(`DrizzleMemoryProvider: the upsert of "${key2}" at "${scope}" returned no row; the driver does not support RETURNING on conflict.`);
     }
     return toRecord(row);
   }
@@ -2120,13 +2650,11 @@ var DrizzleMemoryProvider = class {
   * `false` where there is no such id.
   */
   async pin({ id, pinned }) {
-    const updated = await this.db.update(agentMemory).set({
+    const updated = await affectedRows(this.dialect, this.db.update(this.t.agentMemory).set({
       pinned,
       updatedAt: /* @__PURE__ */ new Date()
-    }).where(eq3(agentMemory.id, id)).returning({
-      id: agentMemory.id
-    });
-    return updated.length > 0;
+    }).where(eq3(this.t.agentMemory.id, id)), this.t.agentMemory.id);
+    return updated > 0;
   }
 };
 function assertAuthorMayWriteScope({ scope, origin, ctx }) {
@@ -2144,14 +2672,18 @@ var DrizzlePricingStore = class {
     __name(this, "DrizzlePricingStore");
   }
   db;
+  dialect;
+  t;
   constructor(db) {
-    this.db = db;
+    this.dialect = agentDialectOf(db);
+    this.t = agentTablesFor(this.dialect);
+    this.db = asBuilder(db);
   }
   async upsertModelPrice(input) {
-    await this.db.update(agentModelPricing).set({
+    await this.db.update(this.t.agentModelPricing).set({
       isCurrent: false
-    }).where(and4(eq4(agentModelPricing.modelId, input.modelId), eq4(agentModelPricing.isCurrent, true)));
-    await this.db.insert(agentModelPricing).values({
+    }).where(and4(eq4(this.t.agentModelPricing.modelId, input.modelId), eq4(this.t.agentModelPricing.isCurrent, true)));
+    await this.db.insert(this.t.agentModelPricing).values({
       id: crypto.randomUUID(),
       modelId: input.modelId,
       inputPricePer1m: input.inputPricePer1m,
@@ -2163,7 +2695,7 @@ var DrizzlePricingStore = class {
     });
   }
   async listCurrentPrices() {
-    const rows = await this.db.select().from(agentModelPricing).where(eq4(agentModelPricing.isCurrent, true));
+    const rows = await this.db.select().from(this.t.agentModelPricing).where(eq4(this.t.agentModelPricing.isCurrent, true));
     return rows.map((row) => ({
       modelId: row.modelId,
       inputPricePer1m: row.inputPricePer1m,
@@ -2182,51 +2714,46 @@ var DrizzlePricingStore = class {
 // src/drizzle-token-stream-sink.ts
 import { SqlTokenStreamSink } from "@dudousxd/nestjs-agent-core";
 import { and as and5, asc as asc3, eq as eq5, gt, inArray as inArray4, lt, max as max2, sql as sql4 } from "drizzle-orm";
-function isUniqueViolation(error) {
-  for (let current = error; current !== null && typeof current === "object"; ) {
-    const code = current.code;
-    if (typeof code === "string" && code.startsWith("SQLITE_CONSTRAINT")) return true;
-    current = current.cause;
-  }
-  return false;
-}
-__name(isUniqueViolation, "isUniqueViolation");
 var DrizzleStreamFrameTable = class {
   static {
     __name(this, "DrizzleStreamFrameTable");
   }
   db;
+  dialect;
+  t;
   constructor(db) {
-    this.db = db;
+    this.dialect = agentDialectOf(db);
+    this.t = agentTablesFor(this.dialect);
+    this.db = asBuilder(db);
   }
   async append(runId, row) {
-    await this.db.run(sql4`INSERT INTO agent_stream_frame (run_id, seq, frame, error, created_at)
+    await runSql(this.db, this.dialect, sql4`INSERT INTO agent_stream_frame (run_id, seq, frame, error, created_at)
           SELECT ${runId}, COALESCE(MAX(seq), 0) + 1, ${row.frame}, ${row.error}, ${row.createdAt}
           FROM agent_stream_frame WHERE run_id = ${runId}`);
   }
   async read(runId, after, limit) {
     return this.db.select({
-      seq: agentStreamFrame.seq,
-      frame: agentStreamFrame.frame,
-      error: agentStreamFrame.error
-    }).from(agentStreamFrame).where(and5(eq5(agentStreamFrame.runId, runId), gt(agentStreamFrame.seq, after))).orderBy(asc3(agentStreamFrame.seq)).limit(limit);
+      seq: this.t.agentStreamFrame.seq,
+      frame: this.t.agentStreamFrame.frame,
+      error: this.t.agentStreamFrame.error
+    }).from(this.t.agentStreamFrame).where(and5(eq5(this.t.agentStreamFrame.runId, runId), gt(this.t.agentStreamFrame.seq, after))).orderBy(asc3(this.t.agentStreamFrame.seq)).limit(limit);
   }
   async has(runId) {
     const rows = await this.db.select({
-      seq: agentStreamFrame.seq
-    }).from(agentStreamFrame).where(eq5(agentStreamFrame.runId, runId)).limit(1);
+      seq: this.t.agentStreamFrame.seq
+    }).from(this.t.agentStreamFrame).where(eq5(this.t.agentStreamFrame.runId, runId)).limit(1);
     return rows.length > 0;
   }
   async remove(runIds) {
     if (runIds.length === 0) return;
-    await this.db.delete(agentStreamFrame).where(inArray4(agentStreamFrame.runId, [
+    await this.db.delete(this.t.agentStreamFrame).where(inArray4(this.t.agentStreamFrame.runId, [
       ...runIds
     ]));
   }
   async lapsedRuns(cutoff) {
     const rows = await this.db.select({
-      runId: agentStreamFrame.runId
-    }).from(agentStreamFrame).groupBy(agentStreamFrame.runId).having(lt(max2(agentStreamFrame.createdAt), cutoff));
+      runId: this.t.agentStreamFrame.runId
+    }).from(this.t.agentStreamFrame).groupBy(this.t.agentStreamFrame.runId).having(lt(max2(this.t.agentStreamFrame.createdAt), cutoff));
     return rows.map((row) => row.runId);
   }
   isUniqueViolation(error) {
@@ -2249,29 +2776,35 @@ var DrizzleConfirmTokenStore = class {
     __name(this, "DrizzleConfirmTokenStore");
   }
   db;
+  dialect;
+  t;
   constructor(db) {
-    this.db = db;
+    this.dialect = agentDialectOf(db);
+    this.t = agentTablesFor(this.dialect);
+    this.db = asBuilder(db);
   }
   async claim(input) {
-    const inserted = await this.db.insert(agentConfirmToken).values({
+    const row = {
       hash: input.hash,
       actorRef: input.actorRef,
       tool: input.tool,
       expiresAt: input.expiresAt,
       createdAt: Date.now()
-    }).onConflictDoNothing().returning({
-      hash: agentConfirmToken.hash
-    });
-    return inserted.length === 1;
+    };
+    const table = this.t.agentConfirmToken;
+    if (this.dialect === "mysql") {
+      const insert = this.db.insert(table);
+      return mysqlAffectedRows(await insert.ignore().values(row)) === 1;
+    }
+    const inserted = await affectedRows(this.dialect, this.db.insert(table).values(row).onConflictDoNothing(), table.hash);
+    return inserted === 1;
   }
   async release(hash) {
-    await this.db.delete(agentConfirmToken).where(eq6(agentConfirmToken.hash, hash));
+    await this.db.delete(this.t.agentConfirmToken).where(eq6(this.t.agentConfirmToken.hash, hash));
   }
   async purgeExpired(now = Date.now()) {
-    const purged = await this.db.delete(agentConfirmToken).where(lt2(agentConfirmToken.expiresAt, now)).returning({
-      hash: agentConfirmToken.hash
-    });
-    return purged.length;
+    const table = this.t.agentConfirmToken;
+    return affectedRows(this.dialect, this.db.delete(table).where(lt2(table.expiresAt, now)), table.hash);
   }
 };
 
@@ -2279,18 +2812,18 @@ var DrizzleConfirmTokenStore = class {
 import { subscribe, unsubscribe } from "diagnostics_channel";
 import { Injectable, Logger } from "@nestjs/common";
 import { and as and6, asc as asc4, count as count2, desc as desc3, eq as eq7, gt as gt2, lt as lt3, or as or2 } from "drizzle-orm";
-function _ts_decorate(decorators, target, key, desc4) {
-  var c = arguments.length, r = c < 3 ? target : desc4 === null ? desc4 = Object.getOwnPropertyDescriptor(target, key) : desc4, d;
+function _ts_decorate(decorators, target, key2, desc4) {
+  var c = arguments.length, r = c < 3 ? target : desc4 === null ? desc4 = Object.getOwnPropertyDescriptor(target, key2) : desc4, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function") {
-    r = Reflect.decorate(decorators, target, key, desc4);
+    r = Reflect.decorate(decorators, target, key2, desc4);
   } else {
     for (var i = decorators.length - 1; i >= 0; i--) {
       if (d = decorators[i]) {
-        r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+        r = (c < 3 ? d(r) : c > 3 ? d(target, key2, r) : d(target, key2)) || r;
       }
     }
   }
-  return c > 3 && r && Object.defineProperty(target, key, r), r;
+  return c > 3 && r && Object.defineProperty(target, key2, r), r;
 }
 __name(_ts_decorate, "_ts_decorate");
 function _ts_metadata(metadataKey, metadataValue) {
@@ -2336,20 +2869,27 @@ function batches(batchSize) {
   return Number.isFinite(requested) && requested >= 1 ? requested : DEFAULT_BATCH_SIZE;
 }
 __name(batches, "batches");
-var CURSOR_COLUMNS = {
-  documentId: ragIngestionLog.documentId,
-  updatedAt: ragIngestionLog.updatedAt
-};
+function cursorColumns(log) {
+  return {
+    documentId: log.documentId,
+    updatedAt: log.updatedAt
+  };
+}
+__name(cursorColumns, "cursorColumns");
 var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
   static {
     __name(this, "DrizzleRagIngestionLog");
   }
-  db;
   logger = new Logger(_DrizzleRagIngestionLog.name);
   teardowns = [];
   inFlight = /* @__PURE__ */ new Set();
+  db;
+  dialect;
+  t;
   constructor(db) {
-    this.db = db;
+    this.dialect = agentDialectOf(db);
+    this.t = agentTablesFor(this.dialect);
+    this.db = asBuilder(db);
   }
   onModuleInit() {
     for (const channel of CHANNELS) {
@@ -2378,7 +2918,7 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
   }
   /** The latest recorded outcome per document, newest first. */
   async list(query = {}) {
-    const rows = this.db.select().from(ragIngestionLog).where(this.where(query)).orderBy(...RAG_INGESTION_LOG_PAGE_ORDER).limit(query.limit ?? DEFAULT_PAGE_SIZE);
+    const rows = this.db.select().from(this.t.ragIngestionLog).where(this.where(query)).orderBy(...RAG_INGESTION_LOG_PAGE_ORDER).limit(query.limit ?? DEFAULT_PAGE_SIZE);
     return query.offset === void 0 ? rows : rows.offset(query.offset);
   }
   /**
@@ -2392,7 +2932,7 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
   async listPage(query = {}) {
     const [totalRow] = await this.db.select({
       value: count2()
-    }).from(ragIngestionLog).where(this.where(query));
+    }).from(this.t.ragIngestionLog).where(this.where(query));
     return {
       rows: await this.list(query),
       total: totalRow?.value ?? 0
@@ -2445,7 +2985,7 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
   * sweep is run. A sweep that stops early stops by `break`ing out of the `for await`.
   */
   iterate(where = {}, options = {}) {
-    return this.sweep(where, batches(options.batchSize), options.after, (filter, limit) => this.db.select().from(ragIngestionLog).where(filter).orderBy(...RAG_INGESTION_LOG_PAGE_ORDER).limit(limit));
+    return this.sweep(where, batches(options.batchSize), options.after, (filter, limit) => this.db.select().from(this.t.ragIngestionLog).where(filter).orderBy(...RAG_INGESTION_LOG_PAGE_ORDER).limit(limit));
   }
   /**
   * Just the document ids of the matching rows, in {@link RAG_INGESTION_LOG_PAGE_ORDER}.
@@ -2462,7 +3002,7 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
   */
   async listDocumentIds(where = {}, options = {}) {
     const ids = [];
-    const rows = this.sweep(where, batches(options.batchSize), void 0, (filter, limit) => this.db.select(CURSOR_COLUMNS).from(ragIngestionLog).where(filter).orderBy(...RAG_INGESTION_LOG_PAGE_ORDER).limit(limit));
+    const rows = this.sweep(where, batches(options.batchSize), void 0, (filter, limit) => this.db.select(cursorColumns(this.t.ragIngestionLog)).from(this.t.ragIngestionLog).where(filter).orderBy(...RAG_INGESTION_LOG_PAGE_ORDER).limit(limit));
     for await (const row of rows) {
       ids.push(row.documentId);
     }
@@ -2470,22 +3010,18 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
   }
   /** The latest recorded outcome for one document, or null if it was never attempted. */
   async get(documentId) {
-    const [row] = await this.db.select().from(ragIngestionLog).where(eq7(ragIngestionLog.documentId, documentId));
+    const [row] = await this.db.select().from(this.t.ragIngestionLog).where(eq7(this.t.ragIngestionLog.documentId, documentId));
     return row ?? null;
   }
   /** Forget one document's record. Returns whether a row was actually removed. */
   async remove(documentId) {
-    const removed = await this.db.delete(ragIngestionLog).where(eq7(ragIngestionLog.documentId, documentId)).returning({
-      documentId: ragIngestionLog.documentId
-    });
-    return removed.length > 0;
+    const removed = await affectedRows(this.dialect, this.db.delete(this.t.ragIngestionLog).where(eq7(this.t.ragIngestionLog.documentId, documentId)), this.t.ragIngestionLog.documentId);
+    return removed > 0;
   }
   /** Forget every record for a collection — for when the collection itself is deleted. */
   async removeByCollection(collection) {
-    const removed = await this.db.delete(ragIngestionLog).where(eq7(ragIngestionLog.collection, collection)).returning({
-      documentId: ragIngestionLog.documentId
-    });
-    return removed.length;
+    const removed = await affectedRows(this.dialect, this.db.delete(this.t.ragIngestionLog).where(eq7(this.t.ragIngestionLog.collection, collection)), this.t.ragIngestionLog.documentId);
+    return removed;
   }
   /**
   * The keyset sweep both {@link iterate} and {@link listDocumentIds} are made of. `page` decides
@@ -2514,7 +3050,7 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
     }
   }
   where(query) {
-    return and6(query.collection !== void 0 ? eq7(ragIngestionLog.collection, query.collection) : void 0, query.status !== void 0 ? eq7(ragIngestionLog.status, query.status) : void 0);
+    return and6(query.collection !== void 0 ? eq7(this.t.ragIngestionLog.collection, query.collection) : void 0, query.status !== void 0 ? eq7(this.t.ragIngestionLog.status, query.status) : void 0);
   }
   /**
   * The coordinates filter AND'd with "strictly after `cursor` in
@@ -2529,7 +3065,7 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
     if (cursor === void 0) {
       return base;
     }
-    return and6(base, or2(lt3(ragIngestionLog.updatedAt, cursor.updatedAt), and6(eq7(ragIngestionLog.updatedAt, cursor.updatedAt), gt2(ragIngestionLog.documentId, cursor.documentId))));
+    return and6(base, or2(lt3(this.t.ragIngestionLog.updatedAt, cursor.updatedAt), and6(eq7(this.t.ragIngestionLog.updatedAt, cursor.updatedAt), gt2(this.t.ragIngestionLog.documentId, cursor.documentId))));
   }
   /**
   * Upsert one outcome. Keyed by document id so the row always reflects the *current* state — a
@@ -2577,14 +3113,11 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
           size
         } : {}
       };
-      await this.db.insert(ragIngestionLog).values({
+      await upsert(this.dialect, this.db.insert(this.t.ragIngestionLog).values({
         documentId,
         createdAt: now,
         ...outcome
-      }).onConflictDoUpdate({
-        target: ragIngestionLog.documentId,
-        set: outcome
-      });
+      }), this.t.ragIngestionLog.documentId, outcome);
     } catch (error) {
       this.logger.warn(`Could not record RAG ingestion outcome for ${documentId}: ${error instanceof Error ? error.message : String(error)}`);
     }
@@ -2607,18 +3140,18 @@ DrizzleRagIngestionLog = _ts_decorate([
 // src/drizzle-agent-store.module.ts
 import { AGENT_CONFIRM_TOKEN_STORE, AGENT_GOVERNANCE_QUERIES, AGENT_PRICING_STORE, AGENT_STORE } from "@dudousxd/nestjs-agent-core";
 import { Module } from "@nestjs/common";
-function _ts_decorate2(decorators, target, key, desc4) {
-  var c = arguments.length, r = c < 3 ? target : desc4 === null ? desc4 = Object.getOwnPropertyDescriptor(target, key) : desc4, d;
+function _ts_decorate2(decorators, target, key2, desc4) {
+  var c = arguments.length, r = c < 3 ? target : desc4 === null ? desc4 = Object.getOwnPropertyDescriptor(target, key2) : desc4, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function") {
-    r = Reflect.decorate(decorators, target, key, desc4);
+    r = Reflect.decorate(decorators, target, key2, desc4);
   } else {
     for (var i = decorators.length - 1; i >= 0; i--) {
       if (d = decorators[i]) {
-        r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+        r = (c < 3 ? d(r) : c > 3 ? d(target, key2, r) : d(target, key2)) || r;
       }
     }
   }
-  return c > 3 && r && Object.defineProperty(target, key, r), r;
+  return c > 3 && r && Object.defineProperty(target, key2, r), r;
 }
 __name(_ts_decorate2, "_ts_decorate");
 var DrizzleAgentStoreModule = class _DrizzleAgentStoreModule {
@@ -2718,17 +3251,21 @@ export {
   DrizzleTokenStreamSink,
   RAG_INGESTION_LOG_PAGE_ORDER,
   agentConfirmToken,
+  agentDialectOf,
   agentMemory,
   agentMessage,
   agentModelPricing,
   agentQueuedMessage,
   agentRun,
   agentSchema,
+  agentSchemaDdl,
   agentStreamFrame,
   agentThread,
   agentTokenUsage,
   agentToolCall,
   ensureAgentSchema,
+  mysqlAgentSchema,
+  pgAgentSchema,
   ragIngestionLog
 };
 //# sourceMappingURL=index.js.map