@jami-studio/core 0.92.26 → 0.92.28

package/dist/observability/store.js

@@ -9,6 +9,7 @@
 import { getDbExec, intType, isPostgres, retryOnDdlRace, } from "../db/client.js";
 import { ensureTableExists, ensureColumnExists, ensureIndexExists, } from "../db/ddl-guard.js";
 import { isDuplicateColumnError } from "../db/migrations.js";
+import { createInitMemo } from "../shared/init-memo.js";
 function safeJsonParse(value, fallback) {
     if (!value)
         return fallback;
@@ -43,244 +44,238 @@ function withUserFilter(baseWhere, baseArgs, userId) {
         args: [...baseArgs, userId],
     };
 }
-let _initPromise;
-export async function ensureObservabilityTables() {
-    if (!_initPromise) {
-        _initPromise = (async () => {
-            const client = getDbExec();
-            const traceSpansCreateSql = `
-        CREATE TABLE IF NOT EXISTS agent_trace_spans (
-          id TEXT PRIMARY KEY,
-          run_id TEXT NOT NULL,
-          thread_id TEXT,
-          user_id TEXT,
-          parent_span_id TEXT,
-          span_type TEXT NOT NULL,
-          name TEXT NOT NULL,
-          input_tokens ${intType()} NOT NULL DEFAULT 0,
-          output_tokens ${intType()} NOT NULL DEFAULT 0,
-          cache_read_tokens ${intType()} NOT NULL DEFAULT 0,
-          cache_write_tokens ${intType()} NOT NULL DEFAULT 0,
-          cost_cents_x100 ${intType()} NOT NULL DEFAULT 0,
-          duration_ms ${intType()} NOT NULL DEFAULT 0,
-          status TEXT NOT NULL DEFAULT 'success',
-          error_message TEXT,
-          metadata TEXT,
-          created_at ${intType()} NOT NULL
-        )
+// workerd-safe memo: the raw `let _initPromise` pattern wedged every agent
+// chat run on the unified Cloudflare runtime when an early-responding first
+// request created (and froze) the init promise — see shared/init-memo.ts.
+export const ensureObservabilityTables = createInitMemo(async () => {
+    const client = getDbExec();
+    const traceSpansCreateSql = `
+        CREATE TABLE IF NOT EXISTS agent_trace_spans (
+          id TEXT PRIMARY KEY,
+          run_id TEXT NOT NULL,
+          thread_id TEXT,
+          user_id TEXT,
+          parent_span_id TEXT,
+          span_type TEXT NOT NULL,
+          name TEXT NOT NULL,
+          input_tokens ${intType()} NOT NULL DEFAULT 0,
+          output_tokens ${intType()} NOT NULL DEFAULT 0,
+          cache_read_tokens ${intType()} NOT NULL DEFAULT 0,
+          cache_write_tokens ${intType()} NOT NULL DEFAULT 0,
+          cost_cents_x100 ${intType()} NOT NULL DEFAULT 0,
+          duration_ms ${intType()} NOT NULL DEFAULT 0,
+          status TEXT NOT NULL DEFAULT 'success',
+          error_message TEXT,
+          metadata TEXT,
+          created_at ${intType()} NOT NULL
+        )
       `;
-            const traceSummariesCreateSql = `
-        CREATE TABLE IF NOT EXISTS agent_trace_summaries (
-          run_id TEXT PRIMARY KEY,
-          thread_id TEXT,
-          user_id TEXT,
-          total_spans ${intType()} NOT NULL DEFAULT 0,
-          llm_calls ${intType()} NOT NULL DEFAULT 0,
-          tool_calls ${intType()} NOT NULL DEFAULT 0,
-          successful_tools ${intType()} NOT NULL DEFAULT 0,
-          failed_tools ${intType()} NOT NULL DEFAULT 0,
-          total_duration_ms ${intType()} NOT NULL DEFAULT 0,
-          total_cost_cents_x100 ${intType()} NOT NULL DEFAULT 0,
-          total_input_tokens ${intType()} NOT NULL DEFAULT 0,
-          total_output_tokens ${intType()} NOT NULL DEFAULT 0,
-          model TEXT NOT NULL DEFAULT '',
-          created_at ${intType()} NOT NULL
-        )
+    const traceSummariesCreateSql = `
+        CREATE TABLE IF NOT EXISTS agent_trace_summaries (
+          run_id TEXT PRIMARY KEY,
+          thread_id TEXT,
+          user_id TEXT,
+          total_spans ${intType()} NOT NULL DEFAULT 0,
+          llm_calls ${intType()} NOT NULL DEFAULT 0,
+          tool_calls ${intType()} NOT NULL DEFAULT 0,
+          successful_tools ${intType()} NOT NULL DEFAULT 0,
+          failed_tools ${intType()} NOT NULL DEFAULT 0,
+          total_duration_ms ${intType()} NOT NULL DEFAULT 0,
+          total_cost_cents_x100 ${intType()} NOT NULL DEFAULT 0,
+          total_input_tokens ${intType()} NOT NULL DEFAULT 0,
+          total_output_tokens ${intType()} NOT NULL DEFAULT 0,
+          model TEXT NOT NULL DEFAULT '',
+          created_at ${intType()} NOT NULL
+        )
       `;
-            const feedbackCreateSql = `
-        CREATE TABLE IF NOT EXISTS agent_feedback (
-          id TEXT PRIMARY KEY,
-          run_id TEXT,
-          thread_id TEXT,
-          message_seq ${intType()},
-          feedback_type TEXT NOT NULL,
-          value TEXT NOT NULL DEFAULT '',
-          user_id TEXT,
-          created_at ${intType()} NOT NULL
-        )
+    const feedbackCreateSql = `
+        CREATE TABLE IF NOT EXISTS agent_feedback (
+          id TEXT PRIMARY KEY,
+          run_id TEXT,
+          thread_id TEXT,
+          message_seq ${intType()},
+          feedback_type TEXT NOT NULL,
+          value TEXT NOT NULL DEFAULT '',
+          user_id TEXT,
+          created_at ${intType()} NOT NULL
+        )
       `;
-            const satisfactionScoresCreateSql = `
-        CREATE TABLE IF NOT EXISTS agent_satisfaction_scores (
-          id TEXT PRIMARY KEY,
-          thread_id TEXT NOT NULL,
-          user_id TEXT,
-          frustration_score REAL NOT NULL DEFAULT 0,
-          rephrasing_score REAL NOT NULL DEFAULT 0,
-          abandonment_score REAL NOT NULL DEFAULT 0,
-          sentiment_score REAL NOT NULL DEFAULT 0,
-          length_trend_score REAL NOT NULL DEFAULT 0,
-          computed_at ${intType()} NOT NULL
-        )
+    const satisfactionScoresCreateSql = `
+        CREATE TABLE IF NOT EXISTS agent_satisfaction_scores (
+          id TEXT PRIMARY KEY,
+          thread_id TEXT NOT NULL,
+          user_id TEXT,
+          frustration_score REAL NOT NULL DEFAULT 0,
+          rephrasing_score REAL NOT NULL DEFAULT 0,
+          abandonment_score REAL NOT NULL DEFAULT 0,
+          sentiment_score REAL NOT NULL DEFAULT 0,
+          length_trend_score REAL NOT NULL DEFAULT 0,
+          computed_at ${intType()} NOT NULL
+        )
       `;
-            const evalsCreateSql = `
-        CREATE TABLE IF NOT EXISTS agent_evals (
-          id TEXT PRIMARY KEY,
-          run_id TEXT NOT NULL,
-          thread_id TEXT,
-          user_id TEXT,
-          eval_type TEXT NOT NULL,
-          criteria TEXT NOT NULL,
-          score REAL NOT NULL DEFAULT 0,
-          reasoning TEXT,
-          metadata TEXT,
-          created_at ${intType()} NOT NULL
-        )
+    const evalsCreateSql = `
+        CREATE TABLE IF NOT EXISTS agent_evals (
+          id TEXT PRIMARY KEY,
+          run_id TEXT NOT NULL,
+          thread_id TEXT,
+          user_id TEXT,
+          eval_type TEXT NOT NULL,
+          criteria TEXT NOT NULL,
+          score REAL NOT NULL DEFAULT 0,
+          reasoning TEXT,
+          metadata TEXT,
+          created_at ${intType()} NOT NULL
+        )
       `;
-            const evalDatasetsCreateSql = `
-        CREATE TABLE IF NOT EXISTS agent_eval_datasets (
-          id TEXT PRIMARY KEY,
-          name TEXT NOT NULL,
-          description TEXT NOT NULL DEFAULT '',
-          entries TEXT NOT NULL DEFAULT '[]',
-          created_at ${intType()} NOT NULL,
-          updated_at ${intType()} NOT NULL
-        )
+    const evalDatasetsCreateSql = `
+        CREATE TABLE IF NOT EXISTS agent_eval_datasets (
+          id TEXT PRIMARY KEY,
+          name TEXT NOT NULL,
+          description TEXT NOT NULL DEFAULT '',
+          entries TEXT NOT NULL DEFAULT '[]',
+          created_at ${intType()} NOT NULL,
+          updated_at ${intType()} NOT NULL
+        )
       `;
-            const experimentsCreateSql = `
-        CREATE TABLE IF NOT EXISTS agent_experiments (
-          id TEXT PRIMARY KEY,
-          name TEXT NOT NULL,
-          status TEXT NOT NULL DEFAULT 'draft',
-          variants TEXT NOT NULL DEFAULT '[]',
-          metrics TEXT NOT NULL DEFAULT '[]',
-          assignment_level TEXT NOT NULL DEFAULT 'user',
-          started_at ${intType()},
-          ended_at ${intType()},
-          created_at ${intType()} NOT NULL,
-          owner_email TEXT
-        )
+    const experimentsCreateSql = `
+        CREATE TABLE IF NOT EXISTS agent_experiments (
+          id TEXT PRIMARY KEY,
+          name TEXT NOT NULL,
+          status TEXT NOT NULL DEFAULT 'draft',
+          variants TEXT NOT NULL DEFAULT '[]',
+          metrics TEXT NOT NULL DEFAULT '[]',
+          assignment_level TEXT NOT NULL DEFAULT 'user',
+          started_at ${intType()},
+          ended_at ${intType()},
+          created_at ${intType()} NOT NULL,
+          owner_email TEXT
+        )
       `;
-            const experimentAssignmentsCreateSql = `
-        CREATE TABLE IF NOT EXISTS agent_experiment_assignments (
-          experiment_id TEXT NOT NULL,
-          user_id TEXT NOT NULL,
-          variant_id TEXT NOT NULL,
-          assigned_at ${intType()} NOT NULL,
-          PRIMARY KEY (experiment_id, user_id)
-        )
+    const experimentAssignmentsCreateSql = `
+        CREATE TABLE IF NOT EXISTS agent_experiment_assignments (
+          experiment_id TEXT NOT NULL,
+          user_id TEXT NOT NULL,
+          variant_id TEXT NOT NULL,
+          assigned_at ${intType()} NOT NULL,
+          PRIMARY KEY (experiment_id, user_id)
+        )
       `;
-            const experimentResultsCreateSql = `
-        CREATE TABLE IF NOT EXISTS agent_experiment_results (
-          id TEXT PRIMARY KEY,
-          experiment_id TEXT NOT NULL,
-          variant_id TEXT NOT NULL,
-          metric TEXT NOT NULL,
-          value REAL NOT NULL DEFAULT 0,
-          sample_size ${intType()} NOT NULL DEFAULT 0,
-          confidence_low REAL NOT NULL DEFAULT 0,
-          confidence_high REAL NOT NULL DEFAULT 0,
-          computed_at ${intType()} NOT NULL
-        )
+    const experimentResultsCreateSql = `
+        CREATE TABLE IF NOT EXISTS agent_experiment_results (
+          id TEXT PRIMARY KEY,
+          experiment_id TEXT NOT NULL,
+          variant_id TEXT NOT NULL,
+          metric TEXT NOT NULL,
+          value REAL NOT NULL DEFAULT 0,
+          sample_size ${intType()} NOT NULL DEFAULT 0,
+          confidence_low REAL NOT NULL DEFAULT 0,
+          confidence_high REAL NOT NULL DEFAULT 0,
+          computed_at ${intType()} NOT NULL
+        )
       `;
-            if (isPostgres()) {
-                // PG guard: probe → guarded DDL → re-probe; skips lock on already-migrated path
-                await ensureTableExists("agent_trace_spans", traceSpansCreateSql);
-                await ensureTableExists("agent_trace_summaries", traceSummariesCreateSql);
-                await ensureTableExists("agent_feedback", feedbackCreateSql);
-                await ensureTableExists("agent_satisfaction_scores", satisfactionScoresCreateSql);
-                await ensureTableExists("agent_evals", evalsCreateSql);
-                await ensureTableExists("agent_eval_datasets", evalDatasetsCreateSql);
-                await ensureTableExists("agent_experiments", experimentsCreateSql);
-                await ensureTableExists("agent_experiment_assignments", experimentAssignmentsCreateSql);
-                await ensureTableExists("agent_experiment_results", experimentResultsCreateSql);
-                await ensureColumnExists("agent_experiments", "owner_email", `ALTER TABLE agent_experiments ADD COLUMN IF NOT EXISTS owner_email TEXT`);
-                for (const table of USER_SCOPED_TABLES) {
-                    await ensureColumnExists(table, "user_id", `ALTER TABLE ${table} ADD COLUMN IF NOT EXISTS user_id TEXT`);
-                }
-                await ensureIndexExists("idx_trace_spans_run", `CREATE INDEX IF NOT EXISTS idx_trace_spans_run ON agent_trace_spans (run_id)`);
-                await ensureIndexExists("idx_trace_spans_thread", `CREATE INDEX IF NOT EXISTS idx_trace_spans_thread ON agent_trace_spans (thread_id)`);
-                await ensureIndexExists("idx_trace_spans_created", `CREATE INDEX IF NOT EXISTS idx_trace_spans_created ON agent_trace_spans (created_at)`);
-                await ensureIndexExists("idx_trace_summaries_created", `CREATE INDEX IF NOT EXISTS idx_trace_summaries_created ON agent_trace_summaries (created_at)`);
-                await ensureIndexExists("idx_trace_summaries_user", `CREATE INDEX IF NOT EXISTS idx_trace_summaries_user ON agent_trace_summaries (user_id, created_at)`);
-                await ensureIndexExists("idx_trace_spans_user", `CREATE INDEX IF NOT EXISTS idx_trace_spans_user ON agent_trace_spans (user_id)`);
-                await ensureIndexExists("idx_feedback_thread", `CREATE INDEX IF NOT EXISTS idx_feedback_thread ON agent_feedback (thread_id)`);
-                await ensureIndexExists("idx_feedback_created", `CREATE INDEX IF NOT EXISTS idx_feedback_created ON agent_feedback (created_at)`);
-                await ensureIndexExists("idx_feedback_user", `CREATE INDEX IF NOT EXISTS idx_feedback_user ON agent_feedback (user_id, created_at)`);
-                await ensureIndexExists("idx_satisfaction_thread", `CREATE INDEX IF NOT EXISTS idx_satisfaction_thread ON agent_satisfaction_scores (thread_id)`);
-                await ensureIndexExists("idx_satisfaction_user", `CREATE INDEX IF NOT EXISTS idx_satisfaction_user ON agent_satisfaction_scores (user_id, computed_at)`);
-                await ensureIndexExists("idx_evals_run", `CREATE INDEX IF NOT EXISTS idx_evals_run ON agent_evals (run_id)`);
-                await ensureIndexExists("idx_evals_created", `CREATE INDEX IF NOT EXISTS idx_evals_created ON agent_evals (created_at)`);
-                await ensureIndexExists("idx_evals_user", `CREATE INDEX IF NOT EXISTS idx_evals_user ON agent_evals (user_id, created_at)`);
-                await ensureIndexExists("idx_experiment_results_exp", `CREATE INDEX IF NOT EXISTS idx_experiment_results_exp ON agent_experiment_results (experiment_id)`);
-                return;
-            }
-            // SQLite (local dev): no lock problem — keep the original behaviour.
-            await retryOnDdlRace(() => client.execute(traceSpansCreateSql));
-            await retryOnDdlRace(() => client.execute(traceSummariesCreateSql));
-            await retryOnDdlRace(() => client.execute(feedbackCreateSql));
-            await retryOnDdlRace(() => client.execute(satisfactionScoresCreateSql));
-            await retryOnDdlRace(() => client.execute(evalsCreateSql));
-            await retryOnDdlRace(() => client.execute(evalDatasetsCreateSql));
-            await retryOnDdlRace(() => client.execute(experimentsCreateSql));
-            // Additive migration for DBs created before the owner column shipped
-            // (any pre-existing rows have NULL owner — see `updateExperiment` for
-            // the migration semantics). Mutations on those rows fall back to the
-            // standard authentication gate but cannot enforce per-owner scoping
-            // until they're re-saved.
-            try {
-                await client.execute(`ALTER TABLE agent_experiments ADD COLUMN owner_email TEXT`);
-            }
-            catch {
-                // Column already exists — expected after first run.
-            }
-            await retryOnDdlRace(() => client.execute(experimentAssignmentsCreateSql));
-            await retryOnDdlRace(() => client.execute(experimentResultsCreateSql));
-            // Idempotent column upgrades for DBs created before per-user
-            // isolation. SQLite has no `ADD COLUMN IF NOT EXISTS`; Postgres
-            // surfaces "column ... already exists". `isDuplicateColumnError`
-            // (from db/migrations.ts) recognizes both shapes.
-            for (const table of USER_SCOPED_TABLES) {
-                try {
-                    await client.execute(`ALTER TABLE ${table} ADD COLUMN user_id TEXT`);
-                }
-                catch (err) {
-                    if (isDuplicateColumnError(err))
-                        continue;
-                    throw err;
-                }
-            }
-            // Indexes for common query patterns
-            const indexes = [
-                `CREATE INDEX IF NOT EXISTS idx_trace_spans_run ON agent_trace_spans (run_id)`,
-                `CREATE INDEX IF NOT EXISTS idx_trace_spans_thread ON agent_trace_spans (thread_id)`,
-                `CREATE INDEX IF NOT EXISTS idx_trace_spans_created ON agent_trace_spans (created_at)`,
-                `CREATE INDEX IF NOT EXISTS idx_trace_summaries_created ON agent_trace_summaries (created_at)`,
-                `CREATE INDEX IF NOT EXISTS idx_trace_summaries_user ON agent_trace_summaries (user_id, created_at)`,
-                `CREATE INDEX IF NOT EXISTS idx_trace_spans_user ON agent_trace_spans (user_id)`,
-                `CREATE INDEX IF NOT EXISTS idx_feedback_thread ON agent_feedback (thread_id)`,
-                `CREATE INDEX IF NOT EXISTS idx_feedback_created ON agent_feedback (created_at)`,
-                `CREATE INDEX IF NOT EXISTS idx_feedback_user ON agent_feedback (user_id, created_at)`,
-                `CREATE INDEX IF NOT EXISTS idx_satisfaction_thread ON agent_satisfaction_scores (thread_id)`,
-                `CREATE INDEX IF NOT EXISTS idx_satisfaction_user ON agent_satisfaction_scores (user_id, computed_at)`,
-                `CREATE INDEX IF NOT EXISTS idx_evals_run ON agent_evals (run_id)`,
-                `CREATE INDEX IF NOT EXISTS idx_evals_created ON agent_evals (created_at)`,
-                `CREATE INDEX IF NOT EXISTS idx_evals_user ON agent_evals (user_id, created_at)`,
-                `CREATE INDEX IF NOT EXISTS idx_experiment_results_exp ON agent_experiment_results (experiment_id)`,
-            ];
-            for (const sql of indexes) {
-                try {
-                    await client.execute(sql);
-                }
-                catch {
-                    // Index might already exist
-                }
-            }
-        })().catch((err) => {
-            _initPromise = undefined;
+    if (isPostgres()) {
+        // PG guard: probe → guarded DDL → re-probe; skips lock on already-migrated path
+        await ensureTableExists("agent_trace_spans", traceSpansCreateSql);
+        await ensureTableExists("agent_trace_summaries", traceSummariesCreateSql);
+        await ensureTableExists("agent_feedback", feedbackCreateSql);
+        await ensureTableExists("agent_satisfaction_scores", satisfactionScoresCreateSql);
+        await ensureTableExists("agent_evals", evalsCreateSql);
+        await ensureTableExists("agent_eval_datasets", evalDatasetsCreateSql);
+        await ensureTableExists("agent_experiments", experimentsCreateSql);
+        await ensureTableExists("agent_experiment_assignments", experimentAssignmentsCreateSql);
+        await ensureTableExists("agent_experiment_results", experimentResultsCreateSql);
+        await ensureColumnExists("agent_experiments", "owner_email", `ALTER TABLE agent_experiments ADD COLUMN IF NOT EXISTS owner_email TEXT`);
+        for (const table of USER_SCOPED_TABLES) {
+            await ensureColumnExists(table, "user_id", `ALTER TABLE ${table} ADD COLUMN IF NOT EXISTS user_id TEXT`);
+        }
+        await ensureIndexExists("idx_trace_spans_run", `CREATE INDEX IF NOT EXISTS idx_trace_spans_run ON agent_trace_spans (run_id)`);
+        await ensureIndexExists("idx_trace_spans_thread", `CREATE INDEX IF NOT EXISTS idx_trace_spans_thread ON agent_trace_spans (thread_id)`);
+        await ensureIndexExists("idx_trace_spans_created", `CREATE INDEX IF NOT EXISTS idx_trace_spans_created ON agent_trace_spans (created_at)`);
+        await ensureIndexExists("idx_trace_summaries_created", `CREATE INDEX IF NOT EXISTS idx_trace_summaries_created ON agent_trace_summaries (created_at)`);
+        await ensureIndexExists("idx_trace_summaries_user", `CREATE INDEX IF NOT EXISTS idx_trace_summaries_user ON agent_trace_summaries (user_id, created_at)`);
+        await ensureIndexExists("idx_trace_spans_user", `CREATE INDEX IF NOT EXISTS idx_trace_spans_user ON agent_trace_spans (user_id)`);
+        await ensureIndexExists("idx_feedback_thread", `CREATE INDEX IF NOT EXISTS idx_feedback_thread ON agent_feedback (thread_id)`);
+        await ensureIndexExists("idx_feedback_created", `CREATE INDEX IF NOT EXISTS idx_feedback_created ON agent_feedback (created_at)`);
+        await ensureIndexExists("idx_feedback_user", `CREATE INDEX IF NOT EXISTS idx_feedback_user ON agent_feedback (user_id, created_at)`);
+        await ensureIndexExists("idx_satisfaction_thread", `CREATE INDEX IF NOT EXISTS idx_satisfaction_thread ON agent_satisfaction_scores (thread_id)`);
+        await ensureIndexExists("idx_satisfaction_user", `CREATE INDEX IF NOT EXISTS idx_satisfaction_user ON agent_satisfaction_scores (user_id, computed_at)`);
+        await ensureIndexExists("idx_evals_run", `CREATE INDEX IF NOT EXISTS idx_evals_run ON agent_evals (run_id)`);
+        await ensureIndexExists("idx_evals_created", `CREATE INDEX IF NOT EXISTS idx_evals_created ON agent_evals (created_at)`);
+        await ensureIndexExists("idx_evals_user", `CREATE INDEX IF NOT EXISTS idx_evals_user ON agent_evals (user_id, created_at)`);
+        await ensureIndexExists("idx_experiment_results_exp", `CREATE INDEX IF NOT EXISTS idx_experiment_results_exp ON agent_experiment_results (experiment_id)`);
+        return;
+    }
+    // SQLite (local dev): no lock problem — keep the original behaviour.
+    await retryOnDdlRace(() => client.execute(traceSpansCreateSql));
+    await retryOnDdlRace(() => client.execute(traceSummariesCreateSql));
+    await retryOnDdlRace(() => client.execute(feedbackCreateSql));
+    await retryOnDdlRace(() => client.execute(satisfactionScoresCreateSql));
+    await retryOnDdlRace(() => client.execute(evalsCreateSql));
+    await retryOnDdlRace(() => client.execute(evalDatasetsCreateSql));
+    await retryOnDdlRace(() => client.execute(experimentsCreateSql));
+    // Additive migration for DBs created before the owner column shipped
+    // (any pre-existing rows have NULL owner — see `updateExperiment` for
+    // the migration semantics). Mutations on those rows fall back to the
+    // standard authentication gate but cannot enforce per-owner scoping
+    // until they're re-saved.
+    try {
+        await client.execute(`ALTER TABLE agent_experiments ADD COLUMN owner_email TEXT`);
+    }
+    catch {
+        // Column already exists — expected after first run.
+    }
+    await retryOnDdlRace(() => client.execute(experimentAssignmentsCreateSql));
+    await retryOnDdlRace(() => client.execute(experimentResultsCreateSql));
+    // Idempotent column upgrades for DBs created before per-user
+    // isolation. SQLite has no `ADD COLUMN IF NOT EXISTS`; Postgres
+    // surfaces "column ... already exists". `isDuplicateColumnError`
+    // (from db/migrations.ts) recognizes both shapes.
+    for (const table of USER_SCOPED_TABLES) {
+        try {
+            await client.execute(`ALTER TABLE ${table} ADD COLUMN user_id TEXT`);
+        }
+        catch (err) {
+            if (isDuplicateColumnError(err))
+                continue;
             throw err;
-        });
+        }
     }
-    return _initPromise;
-}
+    // Indexes for common query patterns
+    const indexes = [
+        `CREATE INDEX IF NOT EXISTS idx_trace_spans_run ON agent_trace_spans (run_id)`,
+        `CREATE INDEX IF NOT EXISTS idx_trace_spans_thread ON agent_trace_spans (thread_id)`,
+        `CREATE INDEX IF NOT EXISTS idx_trace_spans_created ON agent_trace_spans (created_at)`,
+        `CREATE INDEX IF NOT EXISTS idx_trace_summaries_created ON agent_trace_summaries (created_at)`,
+        `CREATE INDEX IF NOT EXISTS idx_trace_summaries_user ON agent_trace_summaries (user_id, created_at)`,
+        `CREATE INDEX IF NOT EXISTS idx_trace_spans_user ON agent_trace_spans (user_id)`,
+        `CREATE INDEX IF NOT EXISTS idx_feedback_thread ON agent_feedback (thread_id)`,
+        `CREATE INDEX IF NOT EXISTS idx_feedback_created ON agent_feedback (created_at)`,
+        `CREATE INDEX IF NOT EXISTS idx_feedback_user ON agent_feedback (user_id, created_at)`,
+        `CREATE INDEX IF NOT EXISTS idx_satisfaction_thread ON agent_satisfaction_scores (thread_id)`,
+        `CREATE INDEX IF NOT EXISTS idx_satisfaction_user ON agent_satisfaction_scores (user_id, computed_at)`,
+        `CREATE INDEX IF NOT EXISTS idx_evals_run ON agent_evals (run_id)`,
+        `CREATE INDEX IF NOT EXISTS idx_evals_created ON agent_evals (created_at)`,
+        `CREATE INDEX IF NOT EXISTS idx_evals_user ON agent_evals (user_id, created_at)`,
+        `CREATE INDEX IF NOT EXISTS idx_experiment_results_exp ON agent_experiment_results (experiment_id)`,
+    ];
+    for (const sql of indexes) {
+        try {
+            await client.execute(sql);
+        }
+        catch {
+            // Index might already exist
+        }
+    }
+}, { label: "observability" });
 // ─── Trace span CRUD ─────────────────────────────────────────────────
 export async function insertTraceSpan(span) {
     await ensureObservabilityTables();
     const client = getDbExec();
     await client.execute({
-        sql: `INSERT INTO agent_trace_spans
-      (id, run_id, thread_id, user_id, parent_span_id, span_type, name,
-       input_tokens, output_tokens, cache_read_tokens, cache_write_tokens,
-       cost_cents_x100, duration_ms, status, error_message, metadata, created_at)
+        sql: `INSERT INTO agent_trace_spans
+      (id, run_id, thread_id, user_id, parent_span_id, span_type, name,
+       input_tokens, output_tokens, cache_read_tokens, cache_write_tokens,
+       cost_cents_x100, duration_ms, status, error_message, metadata, created_at)
       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
         args: [
             span.id,
@@ -310,22 +305,22 @@ export async function upsertTraceSummary(summary) {
     // owner is recorded it shouldn't change under us.
     if (isPostgres()) {
         await client.execute({
-            sql: `INSERT INTO agent_trace_summaries
-        (run_id, thread_id, user_id, total_spans, llm_calls, tool_calls,
-         successful_tools, failed_tools, total_duration_ms,
-         total_cost_cents_x100, total_input_tokens, total_output_tokens,
-         model, created_at)
-        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-        ON CONFLICT (run_id) DO UPDATE SET
-          total_spans = EXCLUDED.total_spans,
-          llm_calls = EXCLUDED.llm_calls,
-          tool_calls = EXCLUDED.tool_calls,
-          successful_tools = EXCLUDED.successful_tools,
-          failed_tools = EXCLUDED.failed_tools,
-          total_duration_ms = EXCLUDED.total_duration_ms,
-          total_cost_cents_x100 = EXCLUDED.total_cost_cents_x100,
-          total_input_tokens = EXCLUDED.total_input_tokens,
-          total_output_tokens = EXCLUDED.total_output_tokens,
+            sql: `INSERT INTO agent_trace_summaries
+        (run_id, thread_id, user_id, total_spans, llm_calls, tool_calls,
+         successful_tools, failed_tools, total_duration_ms,
+         total_cost_cents_x100, total_input_tokens, total_output_tokens,
+         model, created_at)
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        ON CONFLICT (run_id) DO UPDATE SET
+          total_spans = EXCLUDED.total_spans,
+          llm_calls = EXCLUDED.llm_calls,
+          tool_calls = EXCLUDED.tool_calls,
+          successful_tools = EXCLUDED.successful_tools,
+          failed_tools = EXCLUDED.failed_tools,
+          total_duration_ms = EXCLUDED.total_duration_ms,
+          total_cost_cents_x100 = EXCLUDED.total_cost_cents_x100,
+          total_input_tokens = EXCLUDED.total_input_tokens,
+          total_output_tokens = EXCLUDED.total_output_tokens,
           model = EXCLUDED.model`,
             args: [
                 summary.runId,
@@ -347,22 +342,22 @@ export async function upsertTraceSummary(summary) {
     }
     else {
         await client.execute({
-            sql: `INSERT INTO agent_trace_summaries
-        (run_id, thread_id, user_id, total_spans, llm_calls, tool_calls,
-         successful_tools, failed_tools, total_duration_ms,
-         total_cost_cents_x100, total_input_tokens, total_output_tokens,
-         model, created_at)
-        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-        ON CONFLICT (run_id) DO UPDATE SET
-          total_spans = EXCLUDED.total_spans,
-          llm_calls = EXCLUDED.llm_calls,
-          tool_calls = EXCLUDED.tool_calls,
-          successful_tools = EXCLUDED.successful_tools,
-          failed_tools = EXCLUDED.failed_tools,
-          total_duration_ms = EXCLUDED.total_duration_ms,
-          total_cost_cents_x100 = EXCLUDED.total_cost_cents_x100,
-          total_input_tokens = EXCLUDED.total_input_tokens,
-          total_output_tokens = EXCLUDED.total_output_tokens,
+            sql: `INSERT INTO agent_trace_summaries
+        (run_id, thread_id, user_id, total_spans, llm_calls, tool_calls,
+         successful_tools, failed_tools, total_duration_ms,
+         total_cost_cents_x100, total_input_tokens, total_output_tokens,
+         model, created_at)
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        ON CONFLICT (run_id) DO UPDATE SET
+          total_spans = EXCLUDED.total_spans,
+          llm_calls = EXCLUDED.llm_calls,
+          tool_calls = EXCLUDED.tool_calls,
+          successful_tools = EXCLUDED.successful_tools,
+          failed_tools = EXCLUDED.failed_tools,
+          total_duration_ms = EXCLUDED.total_duration_ms,
+          total_cost_cents_x100 = EXCLUDED.total_cost_cents_x100,
+          total_input_tokens = EXCLUDED.total_input_tokens,
+          total_output_tokens = EXCLUDED.total_output_tokens,
           model = EXCLUDED.model`,
             args: [
                 summary.runId,
@@ -435,9 +430,9 @@ export async function getTraceSummaries(opts) {
     const limit = opts.limit ?? 100;
     const { where, args } = withUserFilter("created_at >= ?", [sinceMs], opts.userId);
     const { rows } = await client.execute({
-        sql: `SELECT * FROM agent_trace_summaries
-      WHERE ${where}
-      ORDER BY created_at DESC
+        sql: `SELECT * FROM agent_trace_summaries
+      WHERE ${where}
+      ORDER BY created_at DESC
       LIMIT ?`,
         args: [...args, limit],
     });
@@ -460,8 +455,8 @@ export async function insertFeedback(entry) {
     await ensureObservabilityTables();
     const client = getDbExec();
     await client.execute({
-        sql: `INSERT INTO agent_feedback
-      (id, run_id, thread_id, message_seq, feedback_type, value, user_id, created_at)
+        sql: `INSERT INTO agent_feedback
+      (id, run_id, thread_id, message_seq, feedback_type, value, user_id, created_at)
       VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
         args: [
             entry.id,
@@ -509,8 +504,8 @@ export async function getFeedbackStats(sinceMs, opts = {}) {
     const client = getDbExec();
     const { where, args } = withUserFilter("created_at >= ?", [sinceMs], opts.userId);
     const { rows } = await client.execute({
-        sql: `SELECT feedback_type, value, COUNT(*) as cnt
-      FROM agent_feedback WHERE ${where}
+        sql: `SELECT feedback_type, value, COUNT(*) as cnt
+      FROM agent_feedback WHERE ${where}
       GROUP BY feedback_type, value`,
         args,
     });
@@ -536,16 +531,16 @@ export async function upsertSatisfactionScore(score) {
     const client = getDbExec();
     if (isPostgres()) {
         await client.execute({
-            sql: `INSERT INTO agent_satisfaction_scores
-        (id, thread_id, user_id, frustration_score, rephrasing_score,
-         abandonment_score, sentiment_score, length_trend_score, computed_at)
-        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
-        ON CONFLICT (id) DO UPDATE SET
-          frustration_score = EXCLUDED.frustration_score,
-          rephrasing_score = EXCLUDED.rephrasing_score,
-          abandonment_score = EXCLUDED.abandonment_score,
-          sentiment_score = EXCLUDED.sentiment_score,
-          length_trend_score = EXCLUDED.length_trend_score,
+            sql: `INSERT INTO agent_satisfaction_scores
+        (id, thread_id, user_id, frustration_score, rephrasing_score,
+         abandonment_score, sentiment_score, length_trend_score, computed_at)
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+        ON CONFLICT (id) DO UPDATE SET
+          frustration_score = EXCLUDED.frustration_score,
+          rephrasing_score = EXCLUDED.rephrasing_score,
+          abandonment_score = EXCLUDED.abandonment_score,
+          sentiment_score = EXCLUDED.sentiment_score,
+          length_trend_score = EXCLUDED.length_trend_score,
           computed_at = EXCLUDED.computed_at`,
             args: [
                 score.id,
@@ -562,16 +557,16 @@ export async function upsertSatisfactionScore(score) {
     }
     else {
         await client.execute({
-            sql: `INSERT INTO agent_satisfaction_scores
-        (id, thread_id, user_id, frustration_score, rephrasing_score,
-         abandonment_score, sentiment_score, length_trend_score, computed_at)
-        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
-        ON CONFLICT (id) DO UPDATE SET
-          frustration_score = EXCLUDED.frustration_score,
-          rephrasing_score = EXCLUDED.rephrasing_score,
-          abandonment_score = EXCLUDED.abandonment_score,
-          sentiment_score = EXCLUDED.sentiment_score,
-          length_trend_score = EXCLUDED.length_trend_score,
+            sql: `INSERT INTO agent_satisfaction_scores
+        (id, thread_id, user_id, frustration_score, rephrasing_score,
+         abandonment_score, sentiment_score, length_trend_score, computed_at)
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+        ON CONFLICT (id) DO UPDATE SET
+          frustration_score = EXCLUDED.frustration_score,
+          rephrasing_score = EXCLUDED.rephrasing_score,
+          abandonment_score = EXCLUDED.abandonment_score,
+          sentiment_score = EXCLUDED.sentiment_score,
+          length_trend_score = EXCLUDED.length_trend_score,
           computed_at = EXCLUDED.computed_at`,
             args: [
                 score.id,
@@ -606,7 +601,7 @@ export async function getSatisfactionScores(opts) {
     }
     const where = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
     const { rows } = await client.execute({
-        sql: `SELECT * FROM agent_satisfaction_scores ${where}
+        sql: `SELECT * FROM agent_satisfaction_scores ${where}
       ORDER BY computed_at DESC LIMIT ?`,
         args: [...args, opts.limit ?? 100],
     });
@@ -617,8 +612,8 @@ export async function insertEvalResult(result) {
     await ensureObservabilityTables();
     const client = getDbExec();
     await client.execute({
-        sql: `INSERT INTO agent_evals
-      (id, run_id, thread_id, user_id, eval_type, criteria, score, reasoning, metadata, created_at)
+        sql: `INSERT INTO agent_evals
+      (id, run_id, thread_id, user_id, eval_type, criteria, score, reasoning, metadata, created_at)
       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
         args: [
             result.id,
@@ -649,14 +644,14 @@ export async function getEvalStats(sinceMs, opts = {}) {
     const client = getDbExec();
     const { where, args } = withUserFilter("created_at >= ?", [sinceMs], opts.userId);
     const { rows: totalRows } = await client.execute({
-        sql: `SELECT COUNT(*) as cnt, AVG(score) as avg_score
+        sql: `SELECT COUNT(*) as cnt, AVG(score) as avg_score
       FROM agent_evals WHERE ${where}`,
         args,
     });
     const t = (totalRows[0] ?? {});
     const { rows: criteriaRows } = await client.execute({
-        sql: `SELECT criteria, AVG(score) as avg_score, COUNT(*) as cnt
-      FROM agent_evals WHERE ${where}
+        sql: `SELECT criteria, AVG(score) as avg_score, COUNT(*) as cnt
+      FROM agent_evals WHERE ${where}
       GROUP BY criteria ORDER BY cnt DESC`,
         args,
     });
@@ -675,8 +670,8 @@ export async function insertEvalDataset(dataset) {
     await ensureObservabilityTables();
     const client = getDbExec();
     await client.execute({
-        sql: `INSERT INTO agent_eval_datasets
-      (id, name, description, entries, created_at, updated_at)
+        sql: `INSERT INTO agent_eval_datasets
+      (id, name, description, entries, created_at, updated_at)
       VALUES (?, ?, ?, ?, ?, ?)`,
         args: [
             dataset.id,
@@ -735,9 +730,9 @@ export async function insertExperiment(exp) {
     await ensureObservabilityTables();
     const client = getDbExec();
     await client.execute({
-        sql: `INSERT INTO agent_experiments
-      (id, name, status, variants, metrics, assignment_level,
-       started_at, ended_at, created_at, owner_email)
+        sql: `INSERT INTO agent_experiments
+      (id, name, status, variants, metrics, assignment_level,
+       started_at, ended_at, created_at, owner_email)
       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
         args: [
             exp.id,
@@ -813,11 +808,11 @@ export async function upsertAssignment(assignment) {
     const client = getDbExec();
     if (isPostgres()) {
         await client.execute({
-            sql: `INSERT INTO agent_experiment_assignments
-        (experiment_id, user_id, variant_id, assigned_at)
-        VALUES (?, ?, ?, ?)
-        ON CONFLICT (experiment_id, user_id) DO UPDATE SET
-          variant_id = EXCLUDED.variant_id,
+            sql: `INSERT INTO agent_experiment_assignments
+        (experiment_id, user_id, variant_id, assigned_at)
+        VALUES (?, ?, ?, ?)
+        ON CONFLICT (experiment_id, user_id) DO UPDATE SET
+          variant_id = EXCLUDED.variant_id,
           assigned_at = EXCLUDED.assigned_at`,
             args: [
                 assignment.experimentId,
@@ -829,8 +824,8 @@ export async function upsertAssignment(assignment) {
     }
     else {
         await client.execute({
-            sql: `INSERT OR REPLACE INTO agent_experiment_assignments
-        (experiment_id, user_id, variant_id, assigned_at)
+            sql: `INSERT OR REPLACE INTO agent_experiment_assignments
+        (experiment_id, user_id, variant_id, assigned_at)
         VALUES (?, ?, ?, ?)`,
             args: [
                 assignment.experimentId,
@@ -845,7 +840,7 @@ export async function getAssignment(experimentId, userId) {
     await ensureObservabilityTables();
     const client = getDbExec();
     const { rows } = await client.execute({
-        sql: `SELECT * FROM agent_experiment_assignments
+        sql: `SELECT * FROM agent_experiment_assignments
       WHERE experiment_id = ? AND user_id = ?`,
         args: [experimentId, userId],
     });
@@ -864,9 +859,9 @@ export async function insertExperimentResult(result) {
     await ensureObservabilityTables();
     const client = getDbExec();
     await client.execute({
-        sql: `INSERT INTO agent_experiment_results
-      (id, experiment_id, variant_id, metric, value,
-       sample_size, confidence_low, confidence_high, computed_at)
+        sql: `INSERT INTO agent_experiment_results
+      (id, experiment_id, variant_id, metric, value,
+       sample_size, confidence_low, confidence_high, computed_at)
       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`,
         args: [
             result.id,
@@ -885,8 +880,8 @@ export async function getExperimentResults(experimentId) {
     await ensureObservabilityTables();
     const client = getDbExec();
     const { rows } = await client.execute({
-        sql: `SELECT * FROM agent_experiment_results
-      WHERE experiment_id = ?
+        sql: `SELECT * FROM agent_experiment_results
+      WHERE experiment_id = ?
       ORDER BY computed_at DESC`,
         args: [experimentId],
     });
@@ -904,29 +899,29 @@ export async function getObservabilityOverview(sinceMs, opts = {}) {
     const computed = withUserFilter("computed_at >= ?", [sinceMs], opts.userId);
     const [tracesResult, satisfactionResult, feedbackResult, evalsResult] = await Promise.all([
         client.execute({
-            sql: `SELECT
-          COUNT(*) as total_runs,
-          COALESCE(SUM(total_cost_cents_x100), 0) as total_cost,
-          COALESCE(AVG(total_duration_ms), 0) as avg_duration,
-          COALESCE(SUM(successful_tools), 0) as success_tools,
-          COALESCE(SUM(tool_calls), 0) as total_tools
+            sql: `SELECT
+          COUNT(*) as total_runs,
+          COALESCE(SUM(total_cost_cents_x100), 0) as total_cost,
+          COALESCE(AVG(total_duration_ms), 0) as avg_duration,
+          COALESCE(SUM(successful_tools), 0) as success_tools,
+          COALESCE(SUM(tool_calls), 0) as total_tools
           FROM agent_trace_summaries WHERE ${created.where}`,
             args: created.args,
         }),
         client.execute({
-            sql: `SELECT COALESCE(AVG(frustration_score), 0) as avg_frustration
+            sql: `SELECT COALESCE(AVG(frustration_score), 0) as avg_frustration
           FROM agent_satisfaction_scores WHERE ${computed.where}`,
             args: computed.args,
         }),
         client.execute({
-            sql: `SELECT
-          COALESCE(SUM(CASE WHEN feedback_type = 'thumbs_up' THEN 1 ELSE 0 END), 0) as up,
-          COALESCE(SUM(CASE WHEN feedback_type IN ('thumbs_up', 'thumbs_down') THEN 1 ELSE 0 END), 0) as total
+            sql: `SELECT
+          COALESCE(SUM(CASE WHEN feedback_type = 'thumbs_up' THEN 1 ELSE 0 END), 0) as up,
+          COALESCE(SUM(CASE WHEN feedback_type IN ('thumbs_up', 'thumbs_down') THEN 1 ELSE 0 END), 0) as total
           FROM agent_feedback WHERE ${created.where}`,
             args: created.args,
         }),
         client.execute({
-            sql: `SELECT COALESCE(AVG(score), 0) as avg_score
+            sql: `SELECT COALESCE(AVG(score), 0) as avg_score
           FROM agent_evals WHERE ${created.where}`,
             args: created.args,
         }),