@rudderhq/db 0.7.1 → 0.7.2

package/dist/migrations/0143_goal_contract.sql

@@ -0,0 +1,97 @@
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "outcome_statement" text;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "objective_mode" text DEFAULT 'target' NOT NULL;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "lifecycle" text DEFAULT 'draft' NOT NULL;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "contract_revision" integer DEFAULT 1 NOT NULL;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "criteria" jsonb DEFAULT '[]'::jsonb NOT NULL;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "autonomy_envelope" jsonb DEFAULT '{}'::jsonb NOT NULL;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "human_authorities" jsonb DEFAULT '{}'::jsonb NOT NULL;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "evaluation_policy" jsonb DEFAULT '{}'::jsonb NOT NULL;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "action_deadline" timestamp with time zone;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "evaluation_deadline" timestamp with time zone;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "evaluation_result" jsonb;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "close_reason" text;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "result_payload" jsonb;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "focus" boolean DEFAULT false NOT NULL;
+ALTER TABLE "goals" ADD COLUMN IF NOT EXISTS "plan_revision" integer DEFAULT 0 NOT NULL;
+
+UPDATE "goals"
+SET
+  "lifecycle" = CASE
+    -- Legacy active rows have no canonical Contract/Owner/Plan proof. They
+    -- must be explicitly activated after migration instead of bypassing Draft.
+    WHEN "status" = 'active' THEN 'draft'
+    WHEN "status" IN ('achieved', 'cancelled') THEN 'closed'
+    ELSE 'draft'
+  END,
+  "close_reason" = CASE
+    WHEN "status" = 'achieved' THEN 'evaluated'
+    WHEN "status" = 'cancelled' THEN 'cancelled'
+    ELSE "close_reason"
+  END;
+
+CREATE UNIQUE INDEX IF NOT EXISTS "goals_org_focus_uq"
+  ON "goals" USING btree ("org_id")
+  WHERE "focus" = true;
+
+CREATE TABLE IF NOT EXISTS "goal_owner_assignments" (
+  "id" uuid PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
+  "org_id" uuid NOT NULL REFERENCES "organizations"("id") ON DELETE cascade,
+  "goal_id" uuid NOT NULL REFERENCES "goals"("id") ON DELETE cascade,
+  "agent_id" uuid NOT NULL REFERENCES "agents"("id"),
+  "assignment_revision" integer DEFAULT 1 NOT NULL,
+  "assigned_by_authority_ref" text,
+  "starts_at" timestamp with time zone DEFAULT now() NOT NULL,
+  "ends_at" timestamp with time zone,
+  "created_at" timestamp with time zone DEFAULT now() NOT NULL
+);
+CREATE INDEX IF NOT EXISTS "goal_owner_assignments_goal_idx"
+  ON "goal_owner_assignments" USING btree ("goal_id", "starts_at");
+CREATE UNIQUE INDEX IF NOT EXISTS "goal_owner_assignments_current_goal_uq"
+  ON "goal_owner_assignments" USING btree ("goal_id")
+  WHERE "ends_at" IS NULL;
+
+CREATE TABLE IF NOT EXISTS "goal_plans" (
+  "id" uuid PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
+  "org_id" uuid NOT NULL REFERENCES "organizations"("id") ON DELETE cascade,
+  "goal_id" uuid NOT NULL REFERENCES "goals"("id") ON DELETE cascade,
+  "revision" integer NOT NULL,
+  "summary" text NOT NULL,
+  "hypotheses" jsonb DEFAULT '[]'::jsonb NOT NULL,
+  "selected_paths" jsonb DEFAULT '[]'::jsonb NOT NULL,
+  "rejected_paths" jsonb DEFAULT '[]'::jsonb NOT NULL,
+  "sequencing" jsonb DEFAULT '[]'::jsonb NOT NULL,
+  "budget_allocations" jsonb DEFAULT '{}'::jsonb NOT NULL,
+  "invalidation_conditions" jsonb DEFAULT '[]'::jsonb NOT NULL,
+  "created_by_agent_id" uuid REFERENCES "agents"("id") ON DELETE set null,
+  "created_at" timestamp with time zone DEFAULT now() NOT NULL,
+  "updated_at" timestamp with time zone DEFAULT now() NOT NULL
+);
+CREATE UNIQUE INDEX IF NOT EXISTS "goal_plans_goal_revision_uq"
+  ON "goal_plans" USING btree ("goal_id", "revision");
+CREATE INDEX IF NOT EXISTS "goal_plans_goal_revision_idx"
+  ON "goal_plans" USING btree ("goal_id", "revision");
+
+CREATE TABLE IF NOT EXISTS "goal_activities" (
+  "id" uuid PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
+  "org_id" uuid NOT NULL REFERENCES "organizations"("id") ON DELETE cascade,
+  "goal_id" uuid NOT NULL REFERENCES "goals"("id") ON DELETE cascade,
+  "contract_revision" integer NOT NULL,
+  "submitted_by_agent_id" uuid REFERENCES "agents"("id") ON DELETE set null,
+  "agent_owner_ref_at_time" uuid REFERENCES "agents"("id") ON DELETE set null,
+  "commitment_ref" text,
+  "run_ref" uuid,
+  "activity_kind" text,
+  "summary" text NOT NULL,
+  "evidence_refs" jsonb DEFAULT '[]'::jsonb NOT NULL,
+  "idempotency_key" text,
+  "occurred_at" timestamp with time zone DEFAULT now() NOT NULL,
+  "created_at" timestamp with time zone DEFAULT now() NOT NULL
+);
+CREATE INDEX IF NOT EXISTS "goal_activities_goal_occurred_idx"
+  ON "goal_activities" USING btree ("goal_id", "occurred_at");
+CREATE UNIQUE INDEX IF NOT EXISTS "goal_activities_closeout_run_uq"
+  ON "goal_activities" USING btree ("goal_id", "run_ref", "activity_kind")
+  WHERE "activity_kind" = 'closeout' AND "run_ref" IS NOT NULL;
+CREATE UNIQUE INDEX IF NOT EXISTS "goal_activities_idempotency_uq"
+  ON "goal_activities" USING btree ("goal_id", "idempotency_key")
+  WHERE "idempotency_key" IS NOT NULL;

package/dist/migrations/0144_eminent_umar.sql

@@ -0,0 +1,16 @@
+ALTER TABLE "goals" ADD COLUMN "continuation_kind" text;--> statement-breakpoint
+ALTER TABLE "goals" ADD COLUMN "continuation_summary" text;--> statement-breakpoint
+ALTER TABLE "goals" ADD COLUMN "wake_condition" text;--> statement-breakpoint
+
+UPDATE "goals"
+SET
+  "continuation_kind" = COALESCE("continuation_kind", 'verification'),
+  "continuation_summary" = COALESCE(
+    "continuation_summary",
+    'Review the legacy Goal contract and define the next bounded verification'
+  )
+WHERE "status" = 'active'
+  AND "lifecycle" = 'draft'
+  AND "outcome_statement" IS NOT NULL
+  AND jsonb_array_length("criteria") > 0
+  AND "owner_agent_id" IS NOT NULL;--> statement-breakpoint

package/dist/migrations/0145_public_nehzno.sql

@@ -0,0 +1,96 @@
+CREATE TABLE "goal_change_proposals" (
+	"id" uuid PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
+	"org_id" uuid NOT NULL,
+	"goal_id" uuid NOT NULL,
+	"expected_contract_revision" integer NOT NULL,
+	"before_contract" jsonb NOT NULL,
+	"after_contract" jsonb NOT NULL,
+	"rationale" text NOT NULL,
+	"evidence_refs" jsonb DEFAULT '[]'::jsonb NOT NULL,
+	"approval_id" uuid NOT NULL,
+	"status" text DEFAULT 'pending' NOT NULL,
+	"idempotency_key" text NOT NULL,
+	"proposed_by_agent_id" uuid NOT NULL,
+	"applied_revision" integer,
+	"applied_at" timestamp with time zone,
+	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "goal_feedback_entries" (
+	"id" uuid PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
+	"org_id" uuid NOT NULL,
+	"goal_id" uuid NOT NULL,
+	"actor_type" text NOT NULL,
+	"actor_id" text NOT NULL,
+	"body" text NOT NULL,
+	"attachments" jsonb DEFAULT '[]'::jsonb NOT NULL,
+	"content_hash" text NOT NULL,
+	"feedback_kind" text DEFAULT 'ordinary' NOT NULL,
+	"idempotency_key" text NOT NULL,
+	"routed_wakeup_request_id" uuid,
+	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "goal_result_proposals" (
+	"id" uuid PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
+	"org_id" uuid NOT NULL,
+	"goal_id" uuid NOT NULL,
+	"contract_revision" integer NOT NULL,
+	"candidate" jsonb NOT NULL,
+	"candidate_hash" text NOT NULL,
+	"preflight" jsonb NOT NULL,
+	"risk_summary" text NOT NULL,
+	"status" text DEFAULT 'ready' NOT NULL,
+	"idempotency_key" text NOT NULL,
+	"proposed_by_agent_id" uuid NOT NULL,
+	"accepted_by_actor_type" text,
+	"accepted_by_actor_id" text,
+	"accepted_at" timestamp with time zone,
+	"rejected_by_actor_type" text,
+	"rejected_by_actor_id" text,
+	"rejected_at" timestamp with time zone,
+	"rejection_feedback" text,
+	"consumed_at" timestamp with time zone,
+	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "goal_start_requests" (
+	"id" uuid PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
+	"org_id" uuid NOT NULL,
+	"request_key" text NOT NULL,
+	"packet_hash" text NOT NULL,
+	"packet" jsonb NOT NULL,
+	"draft_goal_id" uuid,
+	"goal_id" uuid,
+	"status" text DEFAULT 'pending' NOT NULL,
+	"completed_at" timestamp with time zone,
+	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+ALTER TABLE "goals" ADD COLUMN "alignment_question" text;--> statement-breakpoint
+ALTER TABLE "goal_change_proposals" ADD CONSTRAINT "goal_change_proposals_org_id_organizations_id_fk" FOREIGN KEY ("org_id") REFERENCES "public"."organizations"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_change_proposals" ADD CONSTRAINT "goal_change_proposals_goal_id_goals_id_fk" FOREIGN KEY ("goal_id") REFERENCES "public"."goals"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_change_proposals" ADD CONSTRAINT "goal_change_proposals_approval_id_approvals_id_fk" FOREIGN KEY ("approval_id") REFERENCES "public"."approvals"("id") ON DELETE no action ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_change_proposals" ADD CONSTRAINT "goal_change_proposals_proposed_by_agent_id_agents_id_fk" FOREIGN KEY ("proposed_by_agent_id") REFERENCES "public"."agents"("id") ON DELETE no action ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_feedback_entries" ADD CONSTRAINT "goal_feedback_entries_org_id_organizations_id_fk" FOREIGN KEY ("org_id") REFERENCES "public"."organizations"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_feedback_entries" ADD CONSTRAINT "goal_feedback_entries_goal_id_goals_id_fk" FOREIGN KEY ("goal_id") REFERENCES "public"."goals"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_feedback_entries" ADD CONSTRAINT "goal_feedback_entries_routed_wakeup_request_id_agent_wakeup_requests_id_fk" FOREIGN KEY ("routed_wakeup_request_id") REFERENCES "public"."agent_wakeup_requests"("id") ON DELETE set null ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_result_proposals" ADD CONSTRAINT "goal_result_proposals_org_id_organizations_id_fk" FOREIGN KEY ("org_id") REFERENCES "public"."organizations"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_result_proposals" ADD CONSTRAINT "goal_result_proposals_goal_id_goals_id_fk" FOREIGN KEY ("goal_id") REFERENCES "public"."goals"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_result_proposals" ADD CONSTRAINT "goal_result_proposals_proposed_by_agent_id_agents_id_fk" FOREIGN KEY ("proposed_by_agent_id") REFERENCES "public"."agents"("id") ON DELETE no action ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_start_requests" ADD CONSTRAINT "goal_start_requests_org_id_organizations_id_fk" FOREIGN KEY ("org_id") REFERENCES "public"."organizations"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_start_requests" ADD CONSTRAINT "goal_start_requests_draft_goal_id_goals_id_fk" FOREIGN KEY ("draft_goal_id") REFERENCES "public"."goals"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "goal_start_requests" ADD CONSTRAINT "goal_start_requests_goal_id_goals_id_fk" FOREIGN KEY ("goal_id") REFERENCES "public"."goals"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+CREATE INDEX "goal_change_proposals_goal_status_created_idx" ON "goal_change_proposals" USING btree ("goal_id","status","created_at");--> statement-breakpoint
+CREATE UNIQUE INDEX "goal_change_proposals_goal_idempotency_uq" ON "goal_change_proposals" USING btree ("goal_id","idempotency_key");--> statement-breakpoint
+CREATE UNIQUE INDEX "goal_change_proposals_approval_uq" ON "goal_change_proposals" USING btree ("approval_id");--> statement-breakpoint
+CREATE INDEX "goal_feedback_entries_goal_created_idx" ON "goal_feedback_entries" USING btree ("goal_id","created_at");--> statement-breakpoint
+CREATE UNIQUE INDEX "goal_feedback_entries_goal_idempotency_uq" ON "goal_feedback_entries" USING btree ("goal_id","idempotency_key");--> statement-breakpoint
+CREATE INDEX "goal_result_proposals_goal_status_created_idx" ON "goal_result_proposals" USING btree ("goal_id","status","created_at");--> statement-breakpoint
+CREATE UNIQUE INDEX "goal_result_proposals_goal_idempotency_uq" ON "goal_result_proposals" USING btree ("goal_id","idempotency_key");--> statement-breakpoint
+CREATE UNIQUE INDEX "goal_start_requests_org_request_key_uq" ON "goal_start_requests" USING btree ("org_id","request_key");--> statement-breakpoint
+CREATE INDEX "goal_start_requests_org_status_updated_idx" ON "goal_start_requests" USING btree ("org_id","status","updated_at");

package/dist/migrations/0146_medical_roulette.sql

@@ -0,0 +1 @@
+ALTER TABLE "goal_result_proposals" ADD COLUMN "acceptance_idempotency_key" text;

package/dist/migrations/0147_github_mcp_provider.sql

@@ -0,0 +1,4 @@
+DROP INDEX "mcp_connections_org_official_canonical_uq";--> statement-breakpoint
+DROP INDEX "mcp_connections_agent_official_canonical_uq";--> statement-breakpoint
+CREATE UNIQUE INDEX "mcp_connections_org_official_canonical_uq" ON "mcp_connections" USING btree ("org_id","provider") WHERE "mcp_connections"."provider" in ('supabase', 'linear', 'notion', 'github') and "mcp_connections"."canonical_state" = 'canonical' and "mcp_connections"."scope" = 'organization';--> statement-breakpoint
+CREATE UNIQUE INDEX "mcp_connections_agent_official_canonical_uq" ON "mcp_connections" USING btree ("org_id","owner_agent_id","provider") WHERE "mcp_connections"."provider" in ('supabase', 'linear', 'notion', 'github') and "mcp_connections"."canonical_state" = 'canonical' and "mcp_connections"."scope" = 'agent';

package/dist/migrations/0148_chat_message_transcript_entries.sql

@@ -0,0 +1,20 @@
+CREATE TABLE "chat_message_transcript_entries" (
+  "org_id" uuid NOT NULL,
+  "message_id" uuid NOT NULL,
+  "entry_seq" integer NOT NULL,
+  "payload" jsonb NOT NULL,
+  CONSTRAINT "chat_message_transcript_entries_message_seq_pk" PRIMARY KEY("message_id","entry_seq")
+);
+--> statement-breakpoint
+ALTER TABLE "chat_message_transcript_entries" ADD CONSTRAINT "chat_message_transcript_entries_org_id_organizations_id_fk" FOREIGN KEY ("org_id") REFERENCES "public"."organizations"("id") ON DELETE cascade ON UPDATE no action;
+--> statement-breakpoint
+ALTER TABLE "chat_message_transcript_entries" ADD CONSTRAINT "chat_message_transcript_entries_message_id_chat_messages_id_fk" FOREIGN KEY ("message_id") REFERENCES "public"."chat_messages"("id") ON DELETE cascade ON UPDATE no action;
+--> statement-breakpoint
+CREATE INDEX "chat_message_transcript_entries_org_message_seq_idx" ON "chat_message_transcript_entries" USING btree ("org_id","message_id","entry_seq");
+--> statement-breakpoint
+ALTER TABLE "chat_messages" SET (
+  autovacuum_vacuum_scale_factor = 0.05,
+  autovacuum_vacuum_threshold = 500,
+  autovacuum_analyze_threshold = 500,
+  autovacuum_analyze_scale_factor = 0.02
+);

package/dist/migrations/chat-transcript-storage.test.ts

@@ -0,0 +1,211 @@
+import { spawn } from "node:child_process";
+import { randomUUID } from "node:crypto";
+import fs from "node:fs";
+import net from "node:net";
+import os from "node:os";
+import path from "node:path";
+import { fileURLToPath } from "node:url";
+import postgres from "postgres";
+import { describe, expect, it } from "vitest";
+import { applyPendingMigrations, ensurePostgresDatabase } from "../client.js";
+import { createLocalPostgresInstance } from "../local-postgres-provider.js";
+
+const repositoryRoot = path.resolve(path.dirname(fileURLToPath(import.meta.url)), "../../../..");
+
+type BackfillReport = {
+  legacyCandidates: number;
+  legacyEligible: number;
+  legacyMigrated: number;
+  legacySkipped: unknown[];
+  forksChecked: number;
+  forksEligible: number;
+  forksRepaired: number;
+};
+
+type JsonValue = null | boolean | number | string | JsonValue[] | { [key: string]: JsonValue };
+
+async function availablePort(): Promise<number> {
+  return await new Promise((resolve, reject) => {
+    const server = net.createServer();
+    server.unref();
+    server.once("error", reject);
+    server.listen(0, "127.0.0.1", () => {
+      const address = server.address();
+      if (!address || typeof address === "string") return reject(new Error("port allocation failed"));
+      server.close((error) => error ? reject(error) : resolve(address.port));
+    });
+  });
+}
+
+async function runBackfill(connectionString: string, args: string[]) {
+  return await new Promise<{ code: number | null; report: BackfillReport }>((resolve, reject) => {
+    const child = spawn(process.execPath, [
+      path.join(repositoryRoot, "cli/node_modules/tsx/dist/cli.mjs"),
+      "scripts/backfill-chat-transcripts.ts",
+      ...args,
+    ], {
+      cwd: repositoryRoot,
+      env: { ...process.env, DATABASE_URL: connectionString },
+      stdio: ["ignore", "pipe", "pipe"],
+    });
+    let stdout = "";
+    let stderr = "";
+    child.stdout.on("data", (chunk) => { stdout += String(chunk); });
+    child.stderr.on("data", (chunk) => { stderr += String(chunk); });
+    const timeout = setTimeout(() => child.kill("SIGTERM"), 30_000);
+    child.once("error", reject);
+    child.once("close", (code) => {
+      clearTimeout(timeout);
+      if (!stdout.trim()) return reject(new Error(stderr || "backfill produced no report"));
+      try {
+        resolve({ code, report: (JSON.parse(stdout) as { report: BackfillReport }).report });
+      } catch {
+        reject(new Error(`${stderr}\n${stdout}`));
+      }
+    });
+  });
+}
+
+describe("chat transcript storage migration", () => {
+  it("backfills safely, prefers generation events, and is repeatable", async () => {
+    const tempRoot = fs.mkdtempSync(path.join(os.tmpdir(), "rudder-chat-transcript-storage-"));
+    const port = await availablePort();
+    const { instance } = await createLocalPostgresInstance({
+      databaseDir: path.join(tempRoot, "postgres"),
+      user: "rudder",
+      password: "rudder",
+      port,
+      persistent: true,
+      initdbFlags: ["--encoding=UTF8", "--locale=C"],
+      onLog: () => {},
+      onError: () => {},
+    });
+    let sql: postgres.Sql | null = null;
+    try {
+      await instance.initialise();
+      await instance.start();
+      const adminUrl = `postgres://rudder:rudder@127.0.0.1:${port}/postgres`;
+      await ensurePostgresDatabase(adminUrl, "rudder");
+      const connectionString = `postgres://rudder:rudder@127.0.0.1:${port}/rudder`;
+      await applyPendingMigrations(connectionString);
+      sql = postgres(connectionString, { max: 1, onnotice: () => {} });
+      const db = sql;
+
+      const orgId = randomUUID();
+    const conversationId = randomUUID();
+    await db`
+      INSERT INTO organizations (id, url_key, name, issue_prefix)
+      VALUES (${orgId}, ${`transcript-${orgId}`}, 'Transcript migration', ${`T${orgId.replaceAll("-", "").slice(0, 8)}`})
+    `;
+    await db`
+      INSERT INTO chat_conversations (id, org_id, title)
+      VALUES (${conversationId}, ${orgId}, 'Transcript migration')
+    `;
+    const makeMessage = async (
+      targetConversationId: string,
+      role: string,
+      body: string,
+      payload: JsonValue,
+      createdAt?: Date,
+    ) => {
+      const id = randomUUID();
+      await db`
+        INSERT INTO chat_messages (id, org_id, conversation_id, role, kind, status, body, structured_payload)
+        VALUES (${id}, ${orgId}, ${targetConversationId}, ${role}, 'message', 'completed', ${body}, ${db.json(payload)})
+      `;
+      if (createdAt) {
+        await db`UPDATE chat_messages SET created_at = ${createdAt}, updated_at = ${createdAt} WHERE id = ${id}`;
+      }
+      return id;
+    };
+    await makeMessage(conversationId, "user", "user", { __chatTranscript: [{ kind: "stdout", ts: "2026-08-08T00:00:01.000Z", text: "one" }] });
+    await makeMessage(conversationId, "user", "user", { __chatTranscript: [{ kind: "stdout", ts: "2026-08-08T00:00:02.000Z", text: "two" }] });
+    await makeMessage(conversationId, "user", "user", { __chatTranscript: "malformed" });
+    const assistantMessageId = await makeMessage(conversationId, "assistant", "assistant", {
+      __chatTranscript: [{ kind: "thinking", ts: "2026-08-08T00:00:03.000Z", text: "legacy" }],
+    });
+    const generationId = randomUUID();
+    const generationAt = new Date("2026-08-08T00:00:00.000Z");
+    await db`
+      INSERT INTO chat_generations (id, org_id, conversation_id, status, started_at, created_at, updated_at)
+      VALUES (${generationId}, ${orgId}, ${conversationId}, 'completed', ${generationAt}, ${generationAt}, ${generationAt})
+    `;
+    await db`
+      INSERT INTO chat_generation_events (org_id, generation_id, generation_seq, attempt_epoch, event_kind, payload, assistant_message_id)
+      VALUES (${orgId}, ${generationId}, 1, 1, 'transcript', ${db.json({ entry: { kind: "thinking", ts: "2026-08-08T00:00:04.000Z", text: "ledger" } })}, ${assistantMessageId})
+    `;
+
+    const forkSourceConversationId = randomUUID();
+    const forkTargetConversationId = randomUUID();
+    const forkUserCreatedAt = new Date("2026-08-08T00:01:00.000Z");
+    const forkAssistantCreatedAt = new Date("2026-08-08T00:01:01.000Z");
+    const forkBoundaryCreatedAt = new Date("2026-08-08T00:01:02.000Z");
+    await db`
+      INSERT INTO chat_conversations (id, org_id, title)
+      VALUES (${forkSourceConversationId}, ${orgId}, 'Old fork source'), (${forkTargetConversationId}, ${orgId}, 'Old fork target')
+    `;
+    await makeMessage(
+      forkSourceConversationId,
+      "user",
+      "fork user",
+      { __chatTranscript: [{ kind: "stdout", ts: "2026-08-08T00:01:00.100Z", text: "fork user transcript" }] },
+      forkUserCreatedAt,
+    );
+    const forkSourceAssistantId = await makeMessage(
+      forkSourceConversationId,
+      "assistant",
+      "fork assistant",
+      { __chatTranscript: [{ kind: "thinking", ts: "2026-08-08T00:01:01.100Z", text: "fork assistant transcript" }] },
+      forkAssistantCreatedAt,
+    );
+    await db`
+      UPDATE chat_conversations
+      SET forked_from_conversation_id = ${forkSourceConversationId},
+          forked_from_message_id = ${forkSourceAssistantId}
+      WHERE id = ${forkTargetConversationId}
+    `;
+    await makeMessage(forkTargetConversationId, "user", "fork user", null, forkUserCreatedAt);
+    await makeMessage(forkTargetConversationId, "assistant", "fork assistant", null, forkAssistantCreatedAt);
+    await db`
+      INSERT INTO chat_messages (id, org_id, conversation_id, role, kind, status, body, structured_payload, created_at, updated_at)
+      VALUES (${randomUUID()}, ${orgId}, ${forkTargetConversationId}, 'system', 'system_event', 'completed', 'fork event', ${db.json({ eventType: "chat_fork" })}, ${forkBoundaryCreatedAt}, ${forkBoundaryCreatedAt})
+    `;
+
+    const dryRun = await runBackfill(connectionString, ["--dry-run", "--batch-size", "1"]);
+    expect(dryRun.code).toBe(2);
+    expect(dryRun.report).toMatchObject({ legacyCandidates: 6, legacyEligible: 5, legacyMigrated: 0, forksChecked: 1, forksEligible: 2 });
+    expect(dryRun.report.legacySkipped).toHaveLength(1);
+
+    const firstRun = await runBackfill(connectionString, ["--batch-size", "1"]);
+    expect(firstRun.code).toBe(2);
+    expect(firstRun.report).toMatchObject({ legacyCandidates: 6, legacyEligible: 5, legacyMigrated: 5, forksChecked: 1, forksRepaired: 1 });
+
+    const secondRun = await runBackfill(connectionString, ["--batch-size", "1"]);
+    expect(secondRun.code).toBe(2);
+    expect(secondRun.report).toMatchObject({ legacyCandidates: 1, legacyMigrated: 0, forksChecked: 1, forksRepaired: 0 });
+
+    const remaining = await db<{ count: number }[]>`SELECT count(*)::int AS count FROM chat_messages WHERE structured_payload ? '__chatTranscript'`;
+    const entries = await db<{ count: number }[]>`SELECT count(*)::int AS count FROM chat_message_transcript_entries`;
+    const assistantEntries = await db<{ count: number }[]>`
+      SELECT count(*)::int AS count FROM chat_message_transcript_entries WHERE message_id = ${assistantMessageId}
+    `;
+    expect(Number(remaining[0]?.count)).toBe(1);
+    expect(Number(entries[0]?.count)).toBe(6);
+    expect(Number(assistantEntries[0]?.count)).toBe(0);
+    const forkEntries = await db<{ text: string }[]>`
+      SELECT payload->>'text' AS text
+      FROM chat_message_transcript_entries
+      WHERE message_id = (
+        SELECT id FROM chat_messages
+        WHERE conversation_id = ${forkTargetConversationId} AND role = 'assistant'
+      )
+      ORDER BY entry_seq
+    `;
+      expect(forkEntries).toEqual([{ text: "fork assistant transcript" }]);
+    } finally {
+      await sql?.end();
+      await instance.stop().catch(() => undefined);
+      fs.rmSync(tempRoot, { recursive: true, force: true });
+    }
+  }, 60_000);
+});

package/dist/migrations/github-mcp-provider.test.ts

@@ -0,0 +1,20 @@
+import fs from "node:fs";
+import path from "node:path";
+import { describe, expect, it } from "vitest";
+
+const migrationSql = fs.readFileSync(
+  path.join(path.dirname(new URL(import.meta.url).pathname), "0147_github_mcp_provider.sql"),
+  "utf8",
+);
+
+describe("GitHub MCP provider migration", () => {
+  it("adds GitHub to organization and agent canonical uniqueness without widening custom MCP", () => {
+    expect(migrationSql).toMatch(
+      /create unique index\s+"mcp_connections_org_official_canonical_uq"[\s\S]*"provider"\s+in\s*\('supabase',\s*'linear',\s*'notion',\s*'github'\)[\s\S]*"scope"\s*=\s*'organization'/i,
+    );
+    expect(migrationSql).toMatch(
+      /create unique index\s+"mcp_connections_agent_official_canonical_uq"[\s\S]*"provider"\s+in\s*\('supabase',\s*'linear',\s*'notion',\s*'github'\)[\s\S]*"scope"\s*=\s*'agent'/i,
+    );
+    expect(migrationSql).not.toMatch(/provider"\s*=\s*'custom'/i);
+  });
+});