pilotswarm-sdk 0.1.19 → 0.1.21

package/dist/cms-migrations.js

@@ -32,6 +32,31 @@ export function CMS_MIGRATIONS(schema) {
             name: "stored_procedures",
             sql: migration_0004_stored_procedures(schema),
         },
+        {
+            version: "0005",
+            name: "skill_usage_procs",
+            sql: migration_0005_skill_usage_procs(schema),
+        },
+        {
+            version: "0006",
+            name: "fleet_cache_columns",
+            sql: migration_0006_fleet_cache_columns(schema),
+        },
+        {
+            version: "0007",
+            name: "session_tree_stats_by_model",
+            sql: migration_0007_session_tree_stats_by_model(schema),
+        },
+        {
+            version: "0008",
+            name: "session_owner_users",
+            sql: migration_0008_session_owner_users(schema),
+        },
+        {
+            version: "0009",
+            name: "user_stats_by_model",
+            sql: migration_0009_user_stats_by_model(schema),
+        },
     ];
 }
 // ─── Migration 0001: Baseline ────────────────────────────────────
@@ -552,4 +577,607 @@ END;
 $$ LANGUAGE plpgsql;
 `;
 }
+// ─── Migration 0005: Skill Usage Procs ───────────────────────────
+function migration_0005_skill_usage_procs(schema) {
+    const s = `"${schema}"`;
+    return `
+-- 0005_skill_usage_procs: per-session, tree, and fleet skill-usage queries.
+-- Two source event types, both rare relative to assistant.delta /
+-- tool.execution_*:
+--   * 'skill.invoked'      — Copilot SDK fires this when the model expands
+--                             a static skill from a plugin's skills/ dir.
+--                             Payload: { name, pluginName?, pluginVersion?, ... }
+--   * 'learned_skill.read' — emitted by the read_facts tool wrapper when
+--                             the call touches the 'skills/' fact namespace.
+--                             Payload: { name (key|keyPattern), scope, matchCount, ... }
+--
+-- Each row carries a 'kind' discriminator so callers can distinguish the
+-- two flavors without inspecting event_type. 'name' is the static skill
+-- name OR the requested learned-skill key/keyPattern. Plugin metadata is
+-- only meaningful for static skills.
+
+-- ── Unified partial index for skill-signal rows ──────────────────
+CREATE INDEX IF NOT EXISTS idx_${schema}_events_skill_signals
+    ON ${s}.session_events (session_id, created_at DESC)
+    WHERE event_type IN ('skill.invoked', 'learned_skill.read');
+
+-- ── cms_get_session_skill_usage ──────────────────────────────────
+CREATE OR REPLACE FUNCTION ${s}.cms_get_session_skill_usage(
+    p_session_id TEXT,
+    p_since      TIMESTAMPTZ
+) RETURNS TABLE (
+    kind           TEXT,
+    name           TEXT,
+    plugin_name    TEXT,
+    plugin_version TEXT,
+    invocations    BIGINT,
+    first_used_at  TIMESTAMPTZ,
+    last_used_at   TIMESTAMPTZ
+) AS $$
+BEGIN
+    RETURN QUERY
+    SELECT
+        CASE WHEN e.event_type = 'skill.invoked'
+             THEN 'static' ELSE 'learned' END::TEXT    AS kind,
+        COALESCE(e.data->>'name', '')::TEXT            AS name,
+        NULLIF(e.data->>'pluginName', '')::TEXT        AS plugin_name,
+        NULLIF(e.data->>'pluginVersion', '')::TEXT     AS plugin_version,
+        COUNT(*)::BIGINT                               AS invocations,
+        MIN(e.created_at)                              AS first_used_at,
+        MAX(e.created_at)                              AS last_used_at
+    FROM ${s}.session_events e
+    WHERE e.session_id = p_session_id
+      AND e.event_type IN ('skill.invoked', 'learned_skill.read')
+      AND (p_since IS NULL OR e.created_at >= p_since)
+    GROUP BY 1, 2, 3, 4
+    ORDER BY invocations DESC, last_used_at DESC;
+END;
+$$ LANGUAGE plpgsql;
+
+-- ── cms_get_session_tree_skill_usage ─────────────────────────────
+CREATE OR REPLACE FUNCTION ${s}.cms_get_session_tree_skill_usage(
+    p_session_id TEXT,
+    p_since      TIMESTAMPTZ
+) RETURNS TABLE (
+    session_id     TEXT,
+    agent_id       TEXT,
+    kind           TEXT,
+    name           TEXT,
+    plugin_name    TEXT,
+    plugin_version TEXT,
+    invocations    BIGINT,
+    first_used_at  TIMESTAMPTZ,
+    last_used_at   TIMESTAMPTZ
+) AS $$
+BEGIN
+    RETURN QUERY
+    WITH RECURSIVE tree AS (
+        SELECT s0.session_id, s0.agent_id FROM ${s}.sessions s0 WHERE s0.session_id = p_session_id
+        UNION ALL
+        SELECT s1.session_id, s1.agent_id FROM ${s}.sessions s1
+        INNER JOIN tree t ON s1.parent_session_id = t.session_id
+    )
+    SELECT
+        e.session_id                                   AS session_id,
+        t.agent_id                                     AS agent_id,
+        CASE WHEN e.event_type = 'skill.invoked'
+             THEN 'static' ELSE 'learned' END::TEXT    AS kind,
+        COALESCE(e.data->>'name', '')::TEXT            AS name,
+        NULLIF(e.data->>'pluginName', '')::TEXT        AS plugin_name,
+        NULLIF(e.data->>'pluginVersion', '')::TEXT     AS plugin_version,
+        COUNT(*)::BIGINT                               AS invocations,
+        MIN(e.created_at)                              AS first_used_at,
+        MAX(e.created_at)                              AS last_used_at
+    FROM ${s}.session_events e
+    INNER JOIN tree t ON e.session_id = t.session_id
+    WHERE e.event_type IN ('skill.invoked', 'learned_skill.read')
+      AND (p_since IS NULL OR e.created_at >= p_since)
+    GROUP BY e.session_id, t.agent_id, kind, name, plugin_name, plugin_version
+    ORDER BY e.session_id, invocations DESC;
+END;
+$$ LANGUAGE plpgsql;
+
+-- ── cms_get_fleet_skill_usage ────────────────────────────────────
+-- Joined to the sessions row for agent_id. p_include_deleted controls
+-- whether soft-deleted sessions contribute. p_since bounds the scan.
+CREATE OR REPLACE FUNCTION ${s}.cms_get_fleet_skill_usage(
+    p_since           TIMESTAMPTZ,
+    p_include_deleted BOOLEAN
+) RETURNS TABLE (
+    agent_id       TEXT,
+    kind           TEXT,
+    name           TEXT,
+    plugin_name    TEXT,
+    plugin_version TEXT,
+    session_count  BIGINT,
+    invocations    BIGINT,
+    last_used_at   TIMESTAMPTZ
+) AS $$
+BEGIN
+    RETURN QUERY
+    SELECT
+        s.agent_id                                     AS agent_id,
+        CASE WHEN e.event_type = 'skill.invoked'
+             THEN 'static' ELSE 'learned' END::TEXT    AS kind,
+        COALESCE(e.data->>'name', '')::TEXT            AS name,
+        NULLIF(e.data->>'pluginName', '')::TEXT        AS plugin_name,
+        NULLIF(e.data->>'pluginVersion', '')::TEXT     AS plugin_version,
+        COUNT(DISTINCT e.session_id)::BIGINT           AS session_count,
+        COUNT(*)::BIGINT                               AS invocations,
+        MAX(e.created_at)                              AS last_used_at
+    FROM ${s}.session_events e
+    INNER JOIN ${s}.sessions s ON s.session_id = e.session_id
+    WHERE e.event_type IN ('skill.invoked', 'learned_skill.read')
+      AND (p_include_deleted OR s.deleted_at IS NULL)
+      AND (p_since IS NULL OR e.created_at >= p_since)
+    GROUP BY s.agent_id, kind, name, plugin_name, plugin_version
+    ORDER BY invocations DESC, last_used_at DESC;
+END;
+$$ LANGUAGE plpgsql;
+`;
+}
+// ─── Migration 0006: Fleet Cache Columns ─────────────────────────
+function migration_0006_fleet_cache_columns(schema) {
+    const s = `"${schema}"`;
+    return `
+-- 0006_fleet_cache_columns: surface prompt-cache token counts at the fleet
+-- aggregation level. Data is already collected per session in
+-- session_metric_summaries.tokens_cache_read / tokens_cache_write; the prior
+-- fleet procs simply ignored those columns. This migration adds them to the
+-- two fleet read paths.
+--
+-- PostgreSQL refuses CREATE OR REPLACE FUNCTION when the RETURNS TABLE shape
+-- changes. We DROP-then-CREATE for both procs. Idempotent via IF EXISTS.
+
+-- ── cms_get_fleet_stats_by_agent (drop + recreate) ───────────────
+DROP FUNCTION IF EXISTS ${s}.cms_get_fleet_stats_by_agent(BOOLEAN, TIMESTAMPTZ);
+CREATE FUNCTION ${s}.cms_get_fleet_stats_by_agent(
+    p_include_deleted BOOLEAN,
+    p_since           TIMESTAMPTZ
+) RETURNS TABLE (
+    agent_id                    TEXT,
+    model                       TEXT,
+    session_count               INT,
+    total_snapshot_size_bytes    BIGINT,
+    total_dehydration_count     INT,
+    total_hydration_count       INT,
+    total_lossy_handoff_count   INT,
+    total_tokens_input          BIGINT,
+    total_tokens_output         BIGINT,
+    total_tokens_cache_read     BIGINT,
+    total_tokens_cache_write    BIGINT
+) AS $$
+BEGIN
+    RETURN QUERY
+    SELECT
+        m.agent_id,
+        m.model,
+        COUNT(*)::int                                          AS session_count,
+        COALESCE(SUM(m.snapshot_size_bytes), 0)::bigint        AS total_snapshot_size_bytes,
+        COALESCE(SUM(m.dehydration_count), 0)::int             AS total_dehydration_count,
+        COALESCE(SUM(m.hydration_count), 0)::int               AS total_hydration_count,
+        COALESCE(SUM(m.lossy_handoff_count), 0)::int           AS total_lossy_handoff_count,
+        COALESCE(SUM(m.tokens_input), 0)::bigint               AS total_tokens_input,
+        COALESCE(SUM(m.tokens_output), 0)::bigint              AS total_tokens_output,
+        COALESCE(SUM(m.tokens_cache_read), 0)::bigint          AS total_tokens_cache_read,
+        COALESCE(SUM(m.tokens_cache_write), 0)::bigint         AS total_tokens_cache_write
+    FROM ${s}.session_metric_summaries m
+    WHERE (p_include_deleted OR m.deleted_at IS NULL)
+      AND (p_since IS NULL OR m.created_at >= p_since)
+    GROUP BY m.agent_id, m.model;
+END;
+$$ LANGUAGE plpgsql;
+
+-- ── cms_get_fleet_stats_totals (drop + recreate) ─────────────────
+DROP FUNCTION IF EXISTS ${s}.cms_get_fleet_stats_totals(BOOLEAN, TIMESTAMPTZ);
+CREATE FUNCTION ${s}.cms_get_fleet_stats_totals(
+    p_include_deleted BOOLEAN,
+    p_since           TIMESTAMPTZ
+) RETURNS TABLE (
+    session_count                INT,
+    total_snapshot_size_bytes     BIGINT,
+    total_tokens_input           BIGINT,
+    total_tokens_output          BIGINT,
+    total_tokens_cache_read      BIGINT,
+    total_tokens_cache_write     BIGINT,
+    earliest_session_created_at  TIMESTAMPTZ
+) AS $$
+BEGIN
+    RETURN QUERY
+    SELECT
+        COUNT(*)::int                                          AS session_count,
+        COALESCE(SUM(m.snapshot_size_bytes), 0)::bigint        AS total_snapshot_size_bytes,
+        COALESCE(SUM(m.tokens_input), 0)::bigint               AS total_tokens_input,
+        COALESCE(SUM(m.tokens_output), 0)::bigint              AS total_tokens_output,
+        COALESCE(SUM(m.tokens_cache_read), 0)::bigint          AS total_tokens_cache_read,
+        COALESCE(SUM(m.tokens_cache_write), 0)::bigint         AS total_tokens_cache_write,
+        MIN(m.created_at)                                      AS earliest_session_created_at
+    FROM ${s}.session_metric_summaries m
+    WHERE (p_include_deleted OR m.deleted_at IS NULL)
+      AND (p_since IS NULL OR m.created_at >= p_since);
+END;
+$$ LANGUAGE plpgsql;
+`;
+}
+// ─── Migration 0007: Session-Tree Stats By Model ─────────────────
+function migration_0007_session_tree_stats_by_model(schema) {
+    const s = `"${schema}"`;
+    return `
+-- 0007_session_tree_stats_by_model: per-model breakdown across the
+-- spawn tree rooted at a session. Mirrors the shape of
+-- cms_get_fleet_stats_by_agent so the TUI/portal "By Model" card can
+-- render uniformly for both the fleet view and the per-session tree
+-- view. Uses the same recursive-descendant CTE pattern as
+-- cms_get_session_tree_stats so they stay in sync.
+
+CREATE OR REPLACE FUNCTION ${s}.cms_get_session_tree_stats_by_model(
+    p_session_id TEXT
+) RETURNS TABLE (
+    model                       TEXT,
+    session_count               INT,
+    total_tokens_input          BIGINT,
+    total_tokens_output         BIGINT,
+    total_tokens_cache_read     BIGINT,
+    total_tokens_cache_write    BIGINT,
+    total_snapshot_size_bytes   BIGINT
+) AS $$
+BEGIN
+    RETURN QUERY
+    WITH RECURSIVE tree AS (
+        SELECT m.session_id FROM ${s}.session_metric_summaries m
+        WHERE m.session_id = p_session_id
+        UNION ALL
+        SELECT m.session_id FROM ${s}.session_metric_summaries m
+        INNER JOIN tree t ON m.parent_session_id = t.session_id
+    )
+    SELECT
+        COALESCE(m.model, '(unknown)')                  AS model,
+        COUNT(*)::int                                    AS session_count,
+        COALESCE(SUM(m.tokens_input), 0)::bigint        AS total_tokens_input,
+        COALESCE(SUM(m.tokens_output), 0)::bigint       AS total_tokens_output,
+        COALESCE(SUM(m.tokens_cache_read), 0)::bigint   AS total_tokens_cache_read,
+        COALESCE(SUM(m.tokens_cache_write), 0)::bigint  AS total_tokens_cache_write,
+        COALESCE(SUM(m.snapshot_size_bytes), 0)::bigint AS total_snapshot_size_bytes
+    FROM ${s}.session_metric_summaries m
+    WHERE m.session_id IN (SELECT tree.session_id FROM tree)
+    GROUP BY m.model
+    ORDER BY total_tokens_input DESC, model;
+END;
+$$ LANGUAGE plpgsql;
+`;
+}
+// ─── Migration 0008: Session Owner Users ─────────────────────────
+function migration_0008_session_owner_users(schema) {
+    const s = `"${schema}"`;
+    return `
+-- 0008_session_owner_users: lazily catalog authenticated users and link
+-- non-system sessions to their first-seen owner. CMS access remains behind
+-- stored procedures; callers do not read or mutate these tables directly.
+
+CREATE TABLE IF NOT EXISTS ${s}.users (
+    user_id      BIGSERIAL PRIMARY KEY,
+    provider     TEXT NOT NULL,
+    subject      TEXT NOT NULL,
+    email        TEXT,
+    display_name TEXT,
+    created_at   TIMESTAMPTZ NOT NULL DEFAULT now(),
+    updated_at   TIMESTAMPTZ NOT NULL DEFAULT now()
+);
+
+CREATE UNIQUE INDEX IF NOT EXISTS idx_${schema}_users_provider_subject
+    ON ${s}.users(provider, subject);
+
+CREATE TABLE IF NOT EXISTS ${s}.session_owners (
+    session_id  TEXT PRIMARY KEY REFERENCES ${s}.sessions(session_id) ON DELETE CASCADE,
+    user_id     BIGINT NOT NULL REFERENCES ${s}.users(user_id),
+    assigned_at TIMESTAMPTZ NOT NULL DEFAULT now()
+);
+
+CREATE INDEX IF NOT EXISTS idx_${schema}_session_owners_user
+    ON ${s}.session_owners(user_id);
+
+-- ── cms_register_user ────────────────────────────────────────────
+CREATE OR REPLACE FUNCTION ${s}.cms_register_user(
+    p_provider     TEXT,
+    p_subject      TEXT,
+    p_email        TEXT,
+    p_display_name TEXT
+) RETURNS BIGINT AS $$
+DECLARE
+    v_provider TEXT := NULLIF(BTRIM(p_provider), '');
+    v_subject  TEXT := NULLIF(BTRIM(p_subject), '');
+    v_user_id  BIGINT;
+BEGIN
+    IF v_provider IS NULL OR v_subject IS NULL THEN
+        RAISE EXCEPTION 'User provider and subject are required';
+    END IF;
+
+    -- First-seen-write-wins: do not refresh profile fields on later sightings.
+    INSERT INTO ${s}.users (provider, subject, email, display_name)
+    VALUES (
+        v_provider,
+        v_subject,
+        NULLIF(BTRIM(p_email), ''),
+        NULLIF(BTRIM(p_display_name), '')
+    )
+    ON CONFLICT (provider, subject) DO NOTHING;
+
+    SELECT user_id INTO v_user_id
+    FROM ${s}.users
+    WHERE provider = v_provider AND subject = v_subject;
+
+    RETURN v_user_id;
+END;
+$$ LANGUAGE plpgsql;
+
+-- ── cms_set_session_owner ────────────────────────────────────────
+CREATE OR REPLACE FUNCTION ${s}.cms_set_session_owner(
+    p_session_id    TEXT,
+    p_provider      TEXT,
+    p_subject       TEXT,
+    p_email         TEXT,
+    p_display_name  TEXT
+) RETURNS VOID AS $$
+DECLARE
+    v_user_id   BIGINT;
+    v_is_system BOOLEAN;
+BEGIN
+    SELECT is_system INTO v_is_system
+    FROM ${s}.sessions
+    WHERE session_id = p_session_id AND deleted_at IS NULL;
+
+    IF NOT FOUND OR v_is_system THEN
+        RETURN;
+    END IF;
+
+    v_user_id := ${s}.cms_register_user(p_provider, p_subject, p_email, p_display_name);
+
+    -- First assignment wins for a session.
+    INSERT INTO ${s}.session_owners (session_id, user_id)
+    VALUES (p_session_id, v_user_id)
+    ON CONFLICT (session_id) DO NOTHING;
+END;
+$$ LANGUAGE plpgsql;
+
+-- ── cms_inherit_session_owner ────────────────────────────────────
+CREATE OR REPLACE FUNCTION ${s}.cms_inherit_session_owner(
+    p_session_id        TEXT,
+    p_parent_session_id TEXT
+) RETURNS VOID AS $$
+DECLARE
+    v_is_system BOOLEAN;
+BEGIN
+    SELECT is_system INTO v_is_system
+    FROM ${s}.sessions
+    WHERE session_id = p_session_id AND deleted_at IS NULL;
+
+    IF NOT FOUND OR v_is_system THEN
+        RETURN;
+    END IF;
+
+    INSERT INTO ${s}.session_owners (session_id, user_id)
+    SELECT p_session_id, so.user_id
+    FROM ${s}.session_owners so
+    WHERE so.session_id = p_parent_session_id
+    ON CONFLICT (session_id) DO NOTHING;
+END;
+$$ LANGUAGE plpgsql;
+
+-- PostgreSQL refuses CREATE OR REPLACE FUNCTION when the return row shape
+-- changes, so the read functions are drop-then-create.
+
+-- ── cms_list_sessions (drop + recreate with owner join) ──────────
+DROP FUNCTION IF EXISTS ${s}.cms_list_sessions();
+CREATE FUNCTION ${s}.cms_list_sessions()
+RETURNS TABLE (
+    session_id         TEXT,
+    orchestration_id   TEXT,
+    title              TEXT,
+    title_locked       BOOLEAN,
+    state              TEXT,
+    model              TEXT,
+    created_at         TIMESTAMPTZ,
+    updated_at         TIMESTAMPTZ,
+    last_active_at     TIMESTAMPTZ,
+    deleted_at         TIMESTAMPTZ,
+    current_iteration  INTEGER,
+    last_error         TEXT,
+    parent_session_id  TEXT,
+    wait_reason        TEXT,
+    is_system          BOOLEAN,
+    agent_id           TEXT,
+    splash             TEXT,
+    owner_provider     TEXT,
+    owner_subject      TEXT,
+    owner_email        TEXT,
+    owner_display_name TEXT
+) AS $$
+BEGIN
+    RETURN QUERY
+    SELECT
+        sess.session_id,
+        sess.orchestration_id,
+        sess.title,
+        sess.title_locked,
+        sess.state,
+        sess.model,
+        sess.created_at,
+        sess.updated_at,
+        sess.last_active_at,
+        sess.deleted_at,
+        sess.current_iteration,
+        sess.last_error,
+        sess.parent_session_id,
+        sess.wait_reason,
+        sess.is_system,
+        sess.agent_id,
+        sess.splash,
+        u.provider AS owner_provider,
+        u.subject AS owner_subject,
+        u.email AS owner_email,
+        u.display_name AS owner_display_name
+    FROM ${s}.sessions sess
+    LEFT JOIN ${s}.session_owners so ON so.session_id = sess.session_id
+    LEFT JOIN ${s}.users u ON u.user_id = so.user_id
+    WHERE sess.deleted_at IS NULL
+    ORDER BY sess.updated_at DESC;
+END;
+$$ LANGUAGE plpgsql;
+
+-- ── cms_get_session (drop + recreate with owner join) ────────────
+DROP FUNCTION IF EXISTS ${s}.cms_get_session(TEXT);
+CREATE FUNCTION ${s}.cms_get_session(
+    p_session_id TEXT
+) RETURNS TABLE (
+    session_id         TEXT,
+    orchestration_id   TEXT,
+    title              TEXT,
+    title_locked       BOOLEAN,
+    state              TEXT,
+    model              TEXT,
+    created_at         TIMESTAMPTZ,
+    updated_at         TIMESTAMPTZ,
+    last_active_at     TIMESTAMPTZ,
+    deleted_at         TIMESTAMPTZ,
+    current_iteration  INTEGER,
+    last_error         TEXT,
+    parent_session_id  TEXT,
+    wait_reason        TEXT,
+    is_system          BOOLEAN,
+    agent_id           TEXT,
+    splash             TEXT,
+    owner_provider     TEXT,
+    owner_subject      TEXT,
+    owner_email        TEXT,
+    owner_display_name TEXT
+) AS $$
+BEGIN
+    RETURN QUERY
+    SELECT
+        sess.session_id,
+        sess.orchestration_id,
+        sess.title,
+        sess.title_locked,
+        sess.state,
+        sess.model,
+        sess.created_at,
+        sess.updated_at,
+        sess.last_active_at,
+        sess.deleted_at,
+        sess.current_iteration,
+        sess.last_error,
+        sess.parent_session_id,
+        sess.wait_reason,
+        sess.is_system,
+        sess.agent_id,
+        sess.splash,
+        u.provider AS owner_provider,
+        u.subject AS owner_subject,
+        u.email AS owner_email,
+        u.display_name AS owner_display_name
+    FROM ${s}.sessions sess
+    LEFT JOIN ${s}.session_owners so ON so.session_id = sess.session_id
+    LEFT JOIN ${s}.users u ON u.user_id = so.user_id
+    WHERE sess.session_id = p_session_id AND sess.deleted_at IS NULL;
+END;
+$$ LANGUAGE plpgsql;
+`;
+}
+// ─── Migration 0009: User Stats By Model ─────────────────────────
+function migration_0009_user_stats_by_model(schema) {
+    const s = `"${schema}"`;
+    return `
+-- 0009_user_stats_by_model: user/session-owner aggregate for the stats pane.
+-- Runtime orchestration history bytes are enriched by management code because
+-- they live in the orchestration provider, not in CMS tables.
+
+CREATE OR REPLACE FUNCTION ${s}.cms_get_user_stats_by_model(
+    p_include_deleted BOOLEAN,
+    p_since           TIMESTAMPTZ
+) RETURNS TABLE (
+    owner_kind                  TEXT,
+    owner_provider              TEXT,
+    owner_subject               TEXT,
+    owner_email                 TEXT,
+    owner_display_name          TEXT,
+    model                       TEXT,
+    session_ids                 TEXT[],
+    session_count               INT,
+    total_snapshot_size_bytes    BIGINT,
+    total_dehydration_count     INT,
+    total_hydration_count       INT,
+    total_lossy_handoff_count   INT,
+    total_tokens_input          BIGINT,
+    total_tokens_output         BIGINT,
+    total_tokens_cache_read     BIGINT,
+    total_tokens_cache_write    BIGINT,
+    earliest_session_created_at TIMESTAMPTZ
+) AS $$
+BEGIN
+    RETURN QUERY
+    WITH base AS (
+        SELECT
+            CASE
+                WHEN sess.is_system THEN 'system'
+                WHEN u.user_id IS NULL THEN 'unowned'
+                ELSE 'user'
+            END::text      AS owner_kind,
+            u.provider     AS owner_provider,
+            u.subject      AS owner_subject,
+            u.email        AS owner_email,
+            u.display_name AS owner_display_name,
+            m.model,
+            m.session_id,
+            m.created_at,
+            m.snapshot_size_bytes,
+            m.dehydration_count,
+            m.hydration_count,
+            m.lossy_handoff_count,
+            m.tokens_input,
+            m.tokens_output,
+            m.tokens_cache_read,
+            m.tokens_cache_write
+        FROM ${s}.session_metric_summaries m
+        INNER JOIN ${s}.sessions sess ON sess.session_id = m.session_id
+        LEFT JOIN ${s}.session_owners so ON so.session_id = m.session_id
+        LEFT JOIN ${s}.users u ON u.user_id = so.user_id
+        WHERE (p_include_deleted OR m.deleted_at IS NULL)
+          AND (p_since IS NULL OR m.created_at >= p_since)
+    )
+    SELECT
+        b.owner_kind                                           AS owner_kind,
+        b.owner_provider                                       AS owner_provider,
+        b.owner_subject                                        AS owner_subject,
+        b.owner_email                                          AS owner_email,
+        b.owner_display_name                                   AS owner_display_name,
+        b.model                                                AS model,
+        ARRAY_AGG(b.session_id ORDER BY b.created_at DESC)     AS session_ids,
+        COUNT(*)::int                                          AS session_count,
+        COALESCE(SUM(b.snapshot_size_bytes), 0)::bigint        AS total_snapshot_size_bytes,
+        COALESCE(SUM(b.dehydration_count), 0)::int             AS total_dehydration_count,
+        COALESCE(SUM(b.hydration_count), 0)::int               AS total_hydration_count,
+        COALESCE(SUM(b.lossy_handoff_count), 0)::int           AS total_lossy_handoff_count,
+        COALESCE(SUM(b.tokens_input), 0)::bigint               AS total_tokens_input,
+        COALESCE(SUM(b.tokens_output), 0)::bigint              AS total_tokens_output,
+        COALESCE(SUM(b.tokens_cache_read), 0)::bigint          AS total_tokens_cache_read,
+        COALESCE(SUM(b.tokens_cache_write), 0)::bigint         AS total_tokens_cache_write,
+        MIN(b.created_at)                                      AS earliest_session_created_at
+    FROM base b
+    GROUP BY
+        b.owner_kind,
+        b.owner_provider,
+        b.owner_subject,
+        b.owner_email,
+        b.owner_display_name,
+        b.model
+    ORDER BY
+        COALESCE(SUM(b.tokens_input), 0)::bigint DESC,
+        b.owner_kind,
+        b.owner_display_name,
+        b.owner_email,
+        b.model;
+END;
+$$ LANGUAGE plpgsql;
+`;
+}
 //# sourceMappingURL=cms-migrations.js.map

package/dist/cms-migrations.js.map

@@ -1 +1 @@
-{"version":3,"file":"cms-migrations.js","sourceRoot":"","sources":["../src/cms-migrations.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAIH;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,MAAc;IACzC,OAAO;QACH;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,UAAU;YAChB,GAAG,EAAE,uBAAuB,CAAC,MAAM,CAAC;SACvC;QACD;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,0BAA0B;YAChC,GAAG,EAAE,uCAAuC,CAAC,MAAM,CAAC;SACvD;QACD;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,+CAA+C;YACrD,GAAG,EAAE,4DAA4D,CAAC,MAAM,CAAC;SAC5E;QACD;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,mBAAmB;YACzB,GAAG,EAAE,gCAAgC,CAAC,MAAM,CAAC;SAChD;KACJ,CAAC;AACN,CAAC;AAED,oEAAoE;AAEpE,SAAS,uBAAuB,CAAC,MAAc;IAC3C,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;;6BAIkB,CAAC;;;;;;;;;;;;;;;;;6BAiBD,CAAC;;;;;;;;iCAQG,MAAM;SAC9B,CAAC;iCACuB,MAAM;SAC9B,CAAC;iCACuB,MAAM;SAC9B,CAAC;;;cAGI,CAAC;cACD,CAAC;cACD,CAAC;cACD,CAAC;cACD,CAAC;cACD,CAAC;cACD,CAAC;CACd,CAAC;AACF,CAAC;AAED,oEAAoE;AAEpE,SAAS,uCAAuC,CAAC,MAAc;IAC3D,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;6BAGkB,CAAC;;;;;;;;;;;;;;;;;;;;;iCAqBG,MAAM;SAC9B,CAAC;iCACuB,MAAM;SAC9B,CAAC;iCACuB,MAAM;SAC9B,CAAC;;;cAGI,CAAC;;OAER,CAAC;;CAEP,CAAC;AACF,CAAC;AAED,SAAS,4DAA4D,CAAC,MAAc;IAChF,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WA+BA,CAAC;;;SAGH,CAAC;;;;;;;;;;;;;;CAcT,CAAC;AACF,CAAC;AAED,mEAAmE;AAEnE,SAAS,gCAAgC,CAAC,MAAc;IACpD,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;;6BAIkB,CAAC;;;;;;;;;kBASZ,CAAC;;;;;;;;;;;;;;;;;;;YAmBP,CAAC;;;kBAGK,CAAC;;;;;;;;;6BASU,CAAC;;;;;aAKjB,CAAC;;;;;;;;;;;;;;;;;;;6BAmBe,CAAC;;;;;;;WAOnB,CAAC;;;;;;;aAOC,CAAC;;;;aAID,CAAC;;;;;;;6BAOe,CAAC;gBACd,CAAC;;;oBAGG,CAAC;;;;;;;6BAOQ,CAAC;;kBAEZ,CAAC;;;oBAGC,CAAC;;;;;;6BAMQ,CAAC;;;;;;mCAMK,CAAC;;;mCAGD,CAAC;;;;;;;;;6BASP,CAAC;;;;;;WAMnB,CAAC;;;;;;;;;6BASiB,CAAC;;;;;;kBAMZ,CAAC;;;;;;;;;;;6BAWU,CAAC;;;;kBAIZ,CAAC;;;;wBAIK,CAAC;;;;;;4BAMG,CAAC;;;;;;;;;6BASA,CAAC;;;;kBAIZ,CAAC;;;;wBAIK,CAAC;;;;;;;;6BAQI,CAAC;;kBAEZ,CAAC;;;oBAGC,CAAC;;;;;;6BAMQ,CAAC;;;;;;;;;;;;;;;;mCAgBK,CAAC;;;mCAGD,CAAC;;;;;;;;;;;;;WAazB,CAAC;;;;;;6BAMiB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;WA0BnB,CAAC;;;;;;;;6BAQiB,CAAC;;;;;;;;;;;;;;;;;;WAkBnB,CAAC;;;;;;;6BAOiB,CAAC;;;;;;;;;;;;;;;;;kBAiBZ,CAAC;;;;;;;;;;;;;mBAaA,CAAC;;gCAEY,CAAC;gCACD,CAAC;gCACD,CAAC;gCACD,CAAC;gCACD,CAAC;gCACD,CAAC;gCACD,CAAC;2EAC0C,CAAC;2EACD,CAAC;4EACA,CAAC;;;;;;6BAMhD,CAAC;;;;;;kBAMZ,CAAC;;;;;;CAMlB,CAAC;AACF,CAAC"}
+{"version":3,"file":"cms-migrations.js","sourceRoot":"","sources":["../src/cms-migrations.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAIH;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,MAAc;IACzC,OAAO;QACH;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,UAAU;YAChB,GAAG,EAAE,uBAAuB,CAAC,MAAM,CAAC;SACvC;QACD;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,0BAA0B;YAChC,GAAG,EAAE,uCAAuC,CAAC,MAAM,CAAC;SACvD;QACD;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,+CAA+C;YACrD,GAAG,EAAE,4DAA4D,CAAC,MAAM,CAAC;SAC5E;QACD;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,mBAAmB;YACzB,GAAG,EAAE,gCAAgC,CAAC,MAAM,CAAC;SAChD;QACD;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,mBAAmB;YACzB,GAAG,EAAE,gCAAgC,CAAC,MAAM,CAAC;SAChD;QACD;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,qBAAqB;YAC3B,GAAG,EAAE,kCAAkC,CAAC,MAAM,CAAC;SAClD;QACD;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,6BAA6B;YACnC,GAAG,EAAE,0CAA0C,CAAC,MAAM,CAAC;SAC1D;QACD;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,qBAAqB;YAC3B,GAAG,EAAE,kCAAkC,CAAC,MAAM,CAAC;SAClD;QACD;YACI,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,qBAAqB;YAC3B,GAAG,EAAE,kCAAkC,CAAC,MAAM,CAAC;SAClD;KACJ,CAAC;AACN,CAAC;AAED,oEAAoE;AAEpE,SAAS,uBAAuB,CAAC,MAAc;IAC3C,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;;6BAIkB,CAAC;;;;;;;;;;;;;;;;;6BAiBD,CAAC;;;;;;;;iCAQG,MAAM;SAC9B,CAAC;iCACuB,MAAM;SAC9B,CAAC;iCACuB,MAAM;SAC9B,CAAC;;;cAGI,CAAC;cACD,CAAC;cACD,CAAC;cACD,CAAC;cACD,CAAC;cACD,CAAC;cACD,CAAC;CACd,CAAC;AACF,CAAC;AAED,oEAAoE;AAEpE,SAAS,uCAAuC,CAAC,MAAc;IAC3D,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;6BAGkB,CAAC;;;;;;;;;;;;;;;;;;;;;iCAqBG,MAAM;SAC9B,CAAC;iCACuB,MAAM;SAC9B,CAAC;iCACuB,MAAM;SAC9B,CAAC;;;cAGI,CAAC;;OAER,CAAC;;CAEP,CAAC;AACF,CAAC;AAED,SAAS,4DAA4D,CAAC,MAAc;IAChF,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WA+BA,CAAC;;;SAGH,CAAC;;;;;;;;;;;;;;CAcT,CAAC;AACF,CAAC;AAED,mEAAmE;AAEnE,SAAS,gCAAgC,CAAC,MAAc;IACpD,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;;6BAIkB,CAAC;;;;;;;;;kBASZ,CAAC;;;;;;;;;;;;;;;;;;;YAmBP,CAAC;;;kBAGK,CAAC;;;;;;;;;6BASU,CAAC;;;;;aAKjB,CAAC;;;;;;;;;;;;;;;;;;;6BAmBe,CAAC;;;;;;;WAOnB,CAAC;;;;;;;aAOC,CAAC;;;;aAID,CAAC;;;;;;;6BAOe,CAAC;gBACd,CAAC;;;oBAGG,CAAC;;;;;;;6BAOQ,CAAC;;kBAEZ,CAAC;;;oBAGC,CAAC;;;;;;6BAMQ,CAAC;;;;;;mCAMK,CAAC;;;mCAGD,CAAC;;;;;;;;;6BASP,CAAC;;;;;;WAMnB,CAAC;;;;;;;;;6BASiB,CAAC;;;;;;kBAMZ,CAAC;;;;;;;;;;;6BAWU,CAAC;;;;kBAIZ,CAAC;;;;wBAIK,CAAC;;;;;;4BAMG,CAAC;;;;;;;;;6BASA,CAAC;;;;kBAIZ,CAAC;;;;wBAIK,CAAC;;;;;;;;6BAQI,CAAC;;kBAEZ,CAAC;;;oBAGC,CAAC;;;;;;6BAMQ,CAAC;;;;;;;;;;;;;;;;mCAgBK,CAAC;;;mCAGD,CAAC;;;;;;;;;;;;;WAazB,CAAC;;;;;;6BAMiB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;WA0BnB,CAAC;;;;;;;;6BAQiB,CAAC;;;;;;;;;;;;;;;;;;WAkBnB,CAAC;;;;;;;6BAOiB,CAAC;;;;;;;;;;;;;;;;;kBAiBZ,CAAC;;;;;;;;;;;;;mBAaA,CAAC;;gCAEY,CAAC;gCACD,CAAC;gCACD,CAAC;gCACD,CAAC;gCACD,CAAC;gCACD,CAAC;gCACD,CAAC;2EAC0C,CAAC;2EACD,CAAC;4EACA,CAAC;;;;;;6BAMhD,CAAC;;;;;;kBAMZ,CAAC;;;;;;CAMlB,CAAC;AACF,CAAC;AAED,oEAAoE;AAEpE,SAAS,gCAAgC,CAAC,MAAc;IACpD,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;;;;;;;;;;;;;;;iCAiBsB,MAAM;SAC9B,CAAC;;;;6BAImB,CAAC;;;;;;;;;;;;;;;;;;;;;;;WAuBnB,CAAC;;;;;;;;;;6BAUiB,CAAC;;;;;;;;;;;;;;;;;iDAiBmB,CAAC;;iDAED,CAAC;;;;;;;;;;;;;;WAcvC,CAAC;;;;;;;;;;;;6BAYiB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;WAyBnB,CAAC;iBACK,CAAC;;;;;;;;CAQjB,CAAC;AACF,CAAC;AAED,oEAAoE;AAEpE,SAAS,kCAAkC,CAAC,MAAc;IACtD,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;;;;;;;;;0BAWe,CAAC;kBACT,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WA8BR,CAAC;;;;;;;;0BAQc,CAAC;kBACT,CAAC;;;;;;;;;;;;;;;;;;;;;;WAsBR,CAAC;;;;;CAKX,CAAC;AACF,CAAC;AAED,oEAAoE;AAEpE,SAAS,0CAA0C,CAAC,MAAc;IAC9D,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;;;;;;6BAQkB,CAAC;;;;;;;;;;;;;;mCAcK,CAAC;;;mCAGD,CAAC;;;;;;;;;;;WAWzB,CAAC;;;;;;CAMX,CAAC;AACF,CAAC;AAED,oEAAoE;AAEpE,SAAS,kCAAkC,CAAC,MAAc;IACtD,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;;;6BAKkB,CAAC;;;;;;;;;;wCAUU,MAAM;SACrC,CAAC;;6BAEmB,CAAC;8CACgB,CAAC;6CACF,CAAC;;;;iCAIb,MAAM;SAC9B,CAAC;;;6BAGmB,CAAC;;;;;;;;;;;;;;;;kBAgBZ,CAAC;;;;;;;;;;WAUR,CAAC;;;;;;;;6BAQiB,CAAC;;;;;;;;;;;;WAYnB,CAAC;;;;;;;mBAOO,CAAC;;;kBAGF,CAAC;;;;;;;6BAOU,CAAC;;;;;;;;WAQnB,CAAC;;;;;;;kBAOM,CAAC;;WAER,CAAC;;;;;;;;;;0BAUc,CAAC;kBACT,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WAgDR,CAAC;gBACI,CAAC;gBACD,CAAC;;;;;;;0BAOS,CAAC;kBACT,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WAiDR,CAAC;gBACI,CAAC;gBACD,CAAC;;;;CAIhB,CAAC;AACF,CAAC;AAED,oEAAoE;AAEpE,SAAS,kCAAkC,CAAC,MAAc;IACtD,MAAM,CAAC,GAAG,IAAI,MAAM,GAAG,CAAC;IACxB,OAAO;;;;;6BAKkB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;eA8Cf,CAAC;qBACK,CAAC;oBACF,CAAC;oBACD,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAsCpB,CAAC;AACF,CAAC"}