studiograph 1.3.48-next.2 → 1.3.48-next.201

package/dist/services/chat-session-service.js

@@ -0,0 +1,316 @@
+import Database from 'better-sqlite3';
+import { join } from 'node:path';
+import { existsSync, mkdirSync, chmodSync } from 'node:fs';
+/**
+ * Server-side persistence for the agent rail's chat sessions (the conversation
+ * history users see in the launcher pills / Chats view). The server is the SOLE
+ * store — the web client keeps no localStorage copy.
+ *
+ * Storage is two tables (mirrors the CommentService layout: a dedicated
+ * `chat-sessions.db` SQLite file under `.studiograph/`, WAL mode, a `meta`
+ * schema-version scaffold, per-user scoping via a `user_id` column):
+ *
+ *   chat_sessions   — one row per chat, metadata only (NO messages column).
+ *   chat_messages   — one row per message, ordered by `seq` within a session.
+ *
+ * Rows (not a JSON blob) mean a single chat has no size limit, the full history
+ * is retained, and title/pin updates (the session row) are decoupled from
+ * message appends (message rows) so a late stream completion can't clobber a
+ * rename. `meta` carries the web `ChatMessage` extras (segments/suggestions/
+ * error) as opaque JSON so the message schema can evolve without a migration.
+ */
+const SCHEMA_VERSION = 2;
+/**
+ * Abuse safety bound (DoS insurance, NOT a UX cap). A runaway client can't grow
+ * the DB without limit: appends are rejected once a user is at the ceiling.
+ * Generous enough to never bite real use — a heavy user with hundreds of long
+ * chats sits orders of magnitude below this.
+ */
+const MAX_MESSAGES_PER_USER = 100_000;
+/** Per-append byte budget across the appended rows' content + meta. Fastify's
+ *  bodyLimit (10MB) is the outer wall; this is the inner, per-turn guard. */
+const MAX_APPEND_BYTES = 1024 * 1024;
+export class ChatSessionTooLargeError extends Error {
+}
+export class ChatSessionForbiddenError extends Error {
+}
+const PREVIEW_LEN = 80;
+export class ChatSessionService {
+    db;
+    constructor(workspacePath) {
+        const sgDir = join(workspacePath, '.studiograph');
+        if (!existsSync(sgDir))
+            mkdirSync(sgDir, { recursive: true });
+        const dbPath = join(sgDir, 'chat-sessions.db');
+        this.db = new Database(dbPath);
+        if (process.platform !== 'win32') {
+            try {
+                chmodSync(dbPath, 0o600);
+            }
+            catch { /* non-fatal */ }
+        }
+        this.db.pragma('journal_mode = WAL');
+        this.db.exec(`
+      CREATE TABLE IF NOT EXISTS chat_sessions (
+        id TEXT PRIMARY KEY,
+        user_id INTEGER NOT NULL,
+        title TEXT NOT NULL,
+        title_generated INTEGER NOT NULL DEFAULT 0 CHECK(title_generated IN (0, 1)),
+        pinned INTEGER NOT NULL DEFAULT 0 CHECK(pinned IN (0, 1)),
+        pinned_at TEXT,
+        created_at TEXT NOT NULL DEFAULT (datetime('now')),
+        updated_at TEXT NOT NULL DEFAULT (datetime('now'))
+      )
+    `);
+        this.db.exec(`CREATE INDEX IF NOT EXISTS idx_chat_sessions_user ON chat_sessions(user_id, updated_at)`);
+        this.db.exec(`
+      CREATE TABLE IF NOT EXISTS chat_messages (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        session_id TEXT NOT NULL,
+        user_id INTEGER NOT NULL,
+        seq INTEGER NOT NULL,
+        role TEXT NOT NULL,
+        content TEXT NOT NULL DEFAULT '',
+        meta TEXT,
+        created_at TEXT NOT NULL DEFAULT (datetime('now'))
+      )
+    `);
+        this.db.exec(`CREATE INDEX IF NOT EXISTS idx_chat_messages_session ON chat_messages(session_id, seq)`);
+        this.db.exec(`
+      CREATE TABLE IF NOT EXISTS meta (
+        key TEXT PRIMARY KEY,
+        value TEXT NOT NULL
+      );
+    `);
+        this.runMigrations();
+    }
+    /**
+     * Idempotent: guarded by the recorded `schema_version`. The undeployed blob
+     * model (`a2ba23b6`, schema 1) never shipped, so there's no data to migrate —
+     * a local dev DB that touched it simply keeps its orphaned `messages` column
+     * (the new chat_sessions inserts don't reference it; it defaults). Add a new
+     * branch when bumping `SCHEMA_VERSION`; never edit existing branches.
+     */
+    runMigrations() {
+        const row = this.db.prepare("SELECT value FROM meta WHERE key='schema_version'").get();
+        const current = row ? parseInt(row.value, 10) || 0 : 0;
+        if (current >= SCHEMA_VERSION)
+            return;
+        // No data migration: schema 1 (blob) never deployed. Future migrations live here.
+        this.db.prepare('INSERT OR REPLACE INTO meta (key, value) VALUES (?, ?)').run('schema_version', String(SCHEMA_VERSION));
+    }
+    rowToMeta(row, messageCount, preview) {
+        return {
+            id: row.id,
+            title: row.title,
+            titleGenerated: row.title_generated === 1,
+            pinned: row.pinned === 1,
+            pinnedAt: row.pinned_at,
+            createdAt: row.created_at,
+            updatedAt: row.updated_at,
+            messageCount,
+            preview,
+        };
+    }
+    /**
+     * A user's sessions as metadata (no messages), most-recently-updated first.
+     * `messageCount` + `preview` (first user message) come from correlated
+     * subqueries so the list stays a single query and never ships message bodies.
+     */
+    listMeta(userId) {
+        const rows = this.db
+            .prepare(`
+        SELECT s.*,
+          (SELECT COUNT(*) FROM chat_messages m WHERE m.session_id = s.id) AS message_count,
+          (SELECT m.content FROM chat_messages m
+             WHERE m.session_id = s.id AND m.role = 'user'
+             ORDER BY m.seq LIMIT 1) AS preview
+        FROM chat_sessions s
+        WHERE s.user_id = ?
+        ORDER BY s.updated_at DESC
+      `)
+            .all(userId);
+        return rows.map(r => this.rowToMeta(r, r.message_count, (r.preview ?? '').slice(0, PREVIEW_LEN)));
+    }
+    /** A single session WITH its messages (seq order), or null if not owned/found. */
+    get(userId, id) {
+        const row = this.db
+            .prepare('SELECT * FROM chat_sessions WHERE id = ? AND user_id = ?')
+            .get(id, userId);
+        if (!row)
+            return null;
+        const msgRows = this.db
+            .prepare('SELECT role, content, meta FROM chat_messages WHERE session_id = ? ORDER BY seq')
+            .all(id);
+        const messages = msgRows.map(m => ({
+            role: m.role === 'user' ? 'user' : 'agent',
+            content: m.content,
+            meta: m.meta ? this.safeParse(m.meta) : null,
+        }));
+        return {
+            id: row.id,
+            title: row.title,
+            titleGenerated: row.title_generated === 1,
+            pinned: row.pinned === 1,
+            pinnedAt: row.pinned_at,
+            createdAt: row.created_at,
+            updatedAt: row.updated_at,
+            messages,
+        };
+    }
+    safeParse(s) {
+        try {
+            const v = JSON.parse(s);
+            return v && typeof v === 'object' ? v : null;
+        }
+        catch {
+            return null;
+        }
+    }
+    ownerOf(id) {
+        const r = this.db.prepare('SELECT user_id FROM chat_sessions WHERE id = ?').get(id);
+        return r?.user_id;
+    }
+    /**
+     * Create a session, or update its metadata in place. A session id is owned by
+     * the first user who writes it; a write from a different user is rejected so
+     * one account can't overwrite another's history. Only the provided fields are
+     * updated (PATCH semantics) — messages are NEVER touched here, so a metadata
+     * write (rename/pin) can't clobber an in-flight stream's appended messages.
+     * `createdAt`/`updatedAt` are honored on create; on update, `createdAt` is
+     * preserved and `updatedAt` only advances when supplied.
+     */
+    upsertMeta(userId, input) {
+        const owner = this.ownerOf(input.id);
+        if (owner !== undefined && owner !== userId) {
+            throw new ChatSessionForbiddenError('Session belongs to another user');
+        }
+        const now = new Date().toISOString();
+        if (owner === undefined) {
+            // Create.
+            this.db
+                .prepare(`
+          INSERT INTO chat_sessions (id, user_id, title, title_generated, pinned, pinned_at, created_at, updated_at)
+          VALUES (@id, @user_id, @title, @title_generated, @pinned, @pinned_at, @created_at, @updated_at)
+        `)
+                .run({
+                id: input.id,
+                user_id: userId,
+                title: input.title ?? 'New chat',
+                title_generated: input.titleGenerated ? 1 : 0,
+                pinned: input.pinned ? 1 : 0,
+                pinned_at: input.pinnedAt ?? null,
+                created_at: input.createdAt ?? now,
+                updated_at: input.updatedAt ?? now,
+            });
+        }
+        else {
+            // Update only the provided fields (COALESCE keeps the stored value when null).
+            this.db
+                .prepare(`
+          UPDATE chat_sessions SET
+            title = COALESCE(@title, title),
+            title_generated = COALESCE(@title_generated, title_generated),
+            pinned = COALESCE(@pinned, pinned),
+            pinned_at = CASE WHEN @pinned_at_set = 1 THEN @pinned_at ELSE pinned_at END,
+            updated_at = COALESCE(@updated_at, updated_at)
+          WHERE id = @id AND user_id = @user_id
+        `)
+                .run({
+                id: input.id,
+                user_id: userId,
+                title: input.title ?? null,
+                title_generated: input.titleGenerated === undefined ? null : input.titleGenerated ? 1 : 0,
+                pinned: input.pinned === undefined ? null : input.pinned ? 1 : 0,
+                // pinnedAt is meaningfully nullable (cleared on unpin), so a dedicated
+                // "was it provided" flag distinguishes "set to null" from "leave alone".
+                pinned_at_set: input.pinned === undefined ? 0 : 1,
+                pinned_at: input.pinnedAt ?? null,
+                updated_at: input.updatedAt ?? null,
+            });
+        }
+        const row = this.db.prepare('SELECT * FROM chat_sessions WHERE id = ?').get(input.id);
+        const count = this.db.prepare('SELECT COUNT(*) AS n FROM chat_messages WHERE session_id = ?').get(input.id).n;
+        const preview = this.db
+            .prepare("SELECT content FROM chat_messages WHERE session_id = ? AND role = 'user' ORDER BY seq LIMIT 1")
+            .get(input.id)?.content ?? '';
+        return this.rowToMeta(row, count, preview.slice(0, PREVIEW_LEN));
+    }
+    /**
+     * Append messages to a session, in order, after the current max `seq`.
+     * Create-on-demand: if the session row doesn't exist it's inserted (owned by
+     * the caller) so a turn-completion append can't lose data to a create race.
+     * Bumps the session's `updated_at` so active chats float to the top of the
+     * list. Rejects another user's session id, an over-budget append, or one that
+     * would push the user past the abuse ceiling.
+     */
+    appendMessages(userId, sessionId, messages, opts) {
+        if (messages.length === 0)
+            return { appended: 0 };
+        let bytes = 0;
+        for (const m of messages) {
+            bytes += Buffer.byteLength(m.content ?? '', 'utf8');
+            if (m.meta)
+                bytes += Buffer.byteLength(JSON.stringify(m.meta), 'utf8');
+        }
+        if (bytes > MAX_APPEND_BYTES) {
+            throw new ChatSessionTooLargeError(`Append exceeds ${MAX_APPEND_BYTES} bytes`);
+        }
+        const owner = this.ownerOf(sessionId);
+        if (owner !== undefined && owner !== userId) {
+            throw new ChatSessionForbiddenError('Session belongs to another user');
+        }
+        const total = this.db.prepare('SELECT COUNT(*) AS n FROM chat_messages WHERE user_id = ?').get(userId).n;
+        if (total + messages.length > MAX_MESSAGES_PER_USER) {
+            throw new ChatSessionTooLargeError(`User exceeds the ${MAX_MESSAGES_PER_USER}-message ceiling`);
+        }
+        const now = new Date().toISOString();
+        const tx = this.db.transaction(() => {
+            if (owner === undefined) {
+                this.db
+                    .prepare(`
+            INSERT INTO chat_sessions (id, user_id, title, created_at, updated_at)
+            VALUES (?, ?, ?, ?, ?)
+          `)
+                    .run(sessionId, userId, opts?.title ?? 'New chat', now, now);
+            }
+            const maxSeq = this.db.prepare('SELECT COALESCE(MAX(seq), -1) AS s FROM chat_messages WHERE session_id = ?').get(sessionId).s;
+            const insert = this.db.prepare(`
+        INSERT INTO chat_messages (session_id, user_id, seq, role, content, meta, created_at)
+        VALUES (?, ?, ?, ?, ?, ?, ?)
+      `);
+            messages.forEach((m, i) => {
+                insert.run(sessionId, userId, maxSeq + 1 + i, m.role === 'user' ? 'user' : 'agent', m.content ?? '', m.meta ? JSON.stringify(m.meta) : null, now);
+            });
+            this.db.prepare('UPDATE chat_sessions SET updated_at = ? WHERE id = ?').run(now, sessionId);
+        });
+        tx();
+        return { appended: messages.length };
+    }
+    /** Delete a session and cascade its messages, scoped to its owner (no-op if not owned). */
+    delete(userId, id) {
+        const owner = this.ownerOf(id);
+        if (owner !== userId)
+            return;
+        const tx = this.db.transaction(() => {
+            this.db.prepare('DELETE FROM chat_messages WHERE session_id = ?').run(id);
+            this.db.prepare('DELETE FROM chat_sessions WHERE id = ? AND user_id = ?').run(id, userId);
+        });
+        tx();
+    }
+    /** Delete every session + message for a user (the /api/reset fresh-start path). */
+    deleteAllForUser(userId) {
+        const tx = this.db.transaction(() => {
+            this.db.prepare('DELETE FROM chat_messages WHERE user_id = ?').run(userId);
+            this.db.prepare('DELETE FROM chat_sessions WHERE user_id = ?').run(userId);
+        });
+        tx();
+    }
+    close() {
+        try {
+            this.db.close();
+        }
+        catch { /* already closed */ }
+    }
+}
+//# sourceMappingURL=chat-session-service.js.map

package/dist/services/chat-session-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-session-service.js","sourceRoot":"","sources":["../../src/services/chat-session-service.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAE3D;;;;;;;;;;;;;;;;;GAiBG;AAEH,MAAM,cAAc,GAAG,CAAC,CAAC;AAEzB;;;;;GAKG;AACH,MAAM,qBAAqB,GAAG,OAAO,CAAC;AAEtC;6EAC6E;AAC7E,MAAM,gBAAgB,GAAG,IAAI,GAAG,IAAI,CAAC;AAErC,MAAM,OAAO,wBAAyB,SAAQ,KAAK;CAAG;AACtD,MAAM,OAAO,yBAA0B,SAAQ,KAAK;CAAG;AAiEvD,MAAM,WAAW,GAAG,EAAE,CAAC;AAEvB,MAAM,OAAO,kBAAkB;IACrB,EAAE,CAAoB;IAE9B,YAAY,aAAqB;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC;QAClD,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;YAAE,SAAS,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,kBAAkB,CAAC,CAAC;QAC/C,IAAI,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC/B,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC;gBAAC,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC;QAC7D,CAAC;QACD,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;QAErC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;KAWZ,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,yFAAyF,CAAC,CAAC;QAExG,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;KAWZ,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,wFAAwF,CAAC,CAAC;QAEvG,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;;;;KAKZ,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC;IAED;;;;;;OAMG;IACK,aAAa;QACnB,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,mDAAmD,CAAC,CAAC,GAAG,EAAmC,CAAC;QACxH,MAAM,OAAO,GAAG,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACvD,IAAI,OAAO,IAAI,cAAc;YAAE,OAAO;QACtC,kFAAkF;QAClF,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,wDAAwD,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;IAC1H,CAAC;IAEO,SAAS,CAAC,GAAe,EAAE,YAAoB,EAAE,OAAe;QACtE,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,cAAc,EAAE,GAAG,CAAC,eAAe,KAAK,CAAC;YACzC,MAAM,EAAE,GAAG,CAAC,MAAM,KAAK,CAAC;YACxB,QAAQ,EAAE,GAAG,CAAC,SAAS;YACvB,SAAS,EAAE,GAAG,CAAC,UAAU;YACzB,SAAS,EAAE,GAAG,CAAC,UAAU;YACzB,YAAY;YACZ,OAAO;SACR,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,QAAQ,CAAC,MAAc;QACrB,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE;aACjB,OAAO,CAAC;;;;;;;;;OASR,CAAC;aACD,GAAG,CAAC,MAAM,CAA0E,CAAC;QACxF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAClB,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,CAC5E,CAAC;IACJ,CAAC;IAED,kFAAkF;IAClF,GAAG,CAAC,MAAc,EAAE,EAAU;QAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE;aAChB,OAAO,CAAC,0DAA0D,CAAC;aACnE,GAAG,CAAC,EAAE,EAAE,MAAM,CAA2B,CAAC;QAC7C,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;QACtB,MAAM,OAAO,GAAG,IAAI,CAAC,EAAE;aACpB,OAAO,CAAC,iFAAiF,CAAC;aAC1F,GAAG,CAAC,EAAE,CAAiB,CAAC;QAC3B,MAAM,QAAQ,GAAwB,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACtD,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO;YAC1C,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI;SAC7C,CAAC,CAAC,CAAC;QACJ,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,cAAc,EAAE,GAAG,CAAC,eAAe,KAAK,CAAC;YACzC,MAAM,EAAE,GAAG,CAAC,MAAM,KAAK,CAAC;YACxB,QAAQ,EAAE,GAAG,CAAC,SAAS;YACvB,SAAS,EAAE,GAAG,CAAC,UAAU;YACzB,SAAS,EAAE,GAAG,CAAC,UAAU;YACzB,QAAQ;SACT,CAAC;IACJ,CAAC;IAEO,SAAS,CAAC,CAAS;QACzB,IAAI,CAAC;YACH,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACxB,OAAO,CAAC,IAAI,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAE,CAA6B,CAAC,CAAC,CAAC,IAAI,CAAC;QAC5E,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEO,OAAO,CAAC,EAAU;QACxB,MAAM,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC,GAAG,CAAC,EAAE,CAAoC,CAAC;QACvH,OAAO,CAAC,EAAE,OAAO,CAAC;IACpB,CAAC;IAED;;;;;;;;OAQG;IACH,UAAU,CAAC,MAAc,EAAE,KAAiC;QAC1D,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACrC,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;YAC5C,MAAM,IAAI,yBAAyB,CAAC,iCAAiC,CAAC,CAAC;QACzE,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAErC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,UAAU;YACV,IAAI,CAAC,EAAE;iBACJ,OAAO,CAAC;;;SAGR,CAAC;iBACD,GAAG,CAAC;gBACH,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,OAAO,EAAE,MAAM;gBACf,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,UAAU;gBAChC,eAAe,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC7C,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC5B,SAAS,EAAE,KAAK,CAAC,QAAQ,IAAI,IAAI;gBACjC,UAAU,EAAE,KAAK,CAAC,SAAS,IAAI,GAAG;gBAClC,UAAU,EAAE,KAAK,CAAC,SAAS,IAAI,GAAG;aACnC,CAAC,CAAC;QACP,CAAC;aAAM,CAAC;YACN,+EAA+E;YAC/E,IAAI,CAAC,EAAE;iBACJ,OAAO,CAAC;;;;;;;;SAQR,CAAC;iBACD,GAAG,CAAC;gBACH,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,OAAO,EAAE,MAAM;gBACf,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,IAAI;gBAC1B,eAAe,EAAE,KAAK,CAAC,cAAc,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBACzF,MAAM,EAAE,KAAK,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAChE,uEAAuE;gBACvE,yEAAyE;gBACzE,aAAa,EAAE,KAAK,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBACjD,SAAS,EAAE,KAAK,CAAC,QAAQ,IAAI,IAAI;gBACjC,UAAU,EAAE,KAAK,CAAC,SAAS,IAAI,IAAI;aACpC,CAAC,CAAC;QACP,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,0CAA0C,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAe,CAAC;QACpG,MAAM,KAAK,GAAI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,8DAA8D,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAmB,CAAC,CAAC,CAAC;QACjI,MAAM,OAAO,GAAI,IAAI,CAAC,EAAE;aACrB,OAAO,CAAC,+FAA+F,CAAC;aACxG,GAAG,CAAC,KAAK,CAAC,EAAE,CAAqC,EAAE,OAAO,IAAI,EAAE,CAAC;QACpE,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC;IACnE,CAAC;IAED;;;;;;;OAOG;IACH,cAAc,CAAC,MAAc,EAAE,SAAiB,EAAE,QAA8B,EAAE,IAAyB;QACzG,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QAElD,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,KAAK,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC;YACpD,IAAI,CAAC,CAAC,IAAI;gBAAE,KAAK,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC;QACzE,CAAC;QACD,IAAI,KAAK,GAAG,gBAAgB,EAAE,CAAC;YAC7B,MAAM,IAAI,wBAAwB,CAAC,kBAAkB,gBAAgB,QAAQ,CAAC,CAAC;QACjF,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACtC,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;YAC5C,MAAM,IAAI,yBAAyB,CAAC,iCAAiC,CAAC,CAAC;QACzE,CAAC;QAED,MAAM,KAAK,GAAI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,2DAA2D,CAAC,CAAC,GAAG,CAAC,MAAM,CAAmB,CAAC,CAAC,CAAC;QAC5H,IAAI,KAAK,GAAG,QAAQ,CAAC,MAAM,GAAG,qBAAqB,EAAE,CAAC;YACpD,MAAM,IAAI,wBAAwB,CAAC,oBAAoB,qBAAqB,kBAAkB,CAAC,CAAC;QAClG,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YAClC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,IAAI,CAAC,EAAE;qBACJ,OAAO,CAAC;;;WAGR,CAAC;qBACD,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,IAAI,UAAU,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACjE,CAAC;YACD,MAAM,MAAM,GAAI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,4EAA4E,CAAC,CAAC,GAAG,CAAC,SAAS,CAAmB,CAAC,CAAC,CAAC;YACjJ,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;OAG9B,CAAC,CAAC;YACH,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;gBACxB,MAAM,CAAC,GAAG,CACR,SAAS,EACT,MAAM,EACN,MAAM,GAAG,CAAC,GAAG,CAAC,EACd,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EACpC,CAAC,CAAC,OAAO,IAAI,EAAE,EACf,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,EACtC,GAAG,CACJ,CAAC;YACJ,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,sDAAsD,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC9F,CAAC,CAAC,CAAC;QACH,EAAE,EAAE,CAAC;QACL,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC;IACvC,CAAC;IAED,2FAA2F;IAC3F,MAAM,CAAC,MAAc,EAAE,EAAU;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC/B,IAAI,KAAK,KAAK,MAAM;YAAE,OAAO;QAC7B,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YAClC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC1E,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,wDAAwD,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAC5F,CAAC,CAAC,CAAC;QACH,EAAE,EAAE,CAAC;IACP,CAAC;IAED,mFAAmF;IACnF,gBAAgB,CAAC,MAAc;QAC7B,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YAClC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC3E,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QACH,EAAE,EAAE,CAAC;IACP,CAAC;IAED,KAAK;QACH,IAAI,CAAC;YAAC,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,CAAC;IACzD,CAAC;CACF"}

package/dist/services/cloud-inference-billing.d.ts

@@ -0,0 +1,64 @@
+import { type Context, type Model, type SimpleStreamOptions, type StreamFunction } from '@mariozechner/pi-ai';
+export interface InferenceUsageSummary {
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+    totalCostUsd: number;
+}
+export interface CloudInferenceReservation {
+    id: string;
+    request_id: string;
+    status: string;
+    reserved_cents: number;
+}
+interface CloudInferenceBillingConfig {
+    baseUrl: string;
+    orgId: string;
+    workspaceId: string;
+    token: string;
+    required: boolean;
+    estimateCents: number;
+}
+interface LoadedCloudConfig {
+    config: CloudInferenceBillingConfig | null;
+    required: boolean;
+    disabledReason: string | null;
+}
+export declare function createInferenceUsageSummary(): InferenceUsageSummary;
+export declare function addInferenceUsage(summary: InferenceUsageSummary, usage: unknown): void;
+export declare function inferenceUsageToBillableCents(summary: InferenceUsageSummary): number | null;
+export declare function createCloudInferenceBillingClient(env?: NodeJS.ProcessEnv, fetchFn?: typeof fetch): CloudInferenceBillingClient;
+export declare function isStudiographCreditsMode(env?: NodeJS.ProcessEnv): boolean;
+export declare function studiographCreditsModel(env?: NodeJS.ProcessEnv): Model<'studiograph-cloud-inference'>;
+export declare function createCloudInferenceStreamFn(env?: NodeJS.ProcessEnv, fetchFn?: typeof fetch): StreamFunction<'studiograph-cloud-inference', SimpleStreamOptions>;
+export declare class CloudInferenceBillingClient {
+    private loaded;
+    private fetchFn;
+    constructor(loaded: LoadedCloudConfig, fetchFn: typeof fetch);
+    isConfigured(): boolean;
+    isRequired(): boolean;
+    disabledReason(): string | null;
+    estimateCents(): number;
+    nextRequestId(prefix?: string): string;
+    getAiSettings(): Promise<unknown | null>;
+    reserve(input: {
+        estimatedCents?: number;
+        provider?: string | null;
+        model?: string | null;
+        requestId?: string;
+        description?: string;
+    }): Promise<CloudInferenceReservation | null>;
+    settle(reservation: CloudInferenceReservation, input: {
+        actualCents: number;
+        inputTokens?: number;
+        outputTokens?: number;
+        description?: string;
+    }): Promise<void>;
+    release(reservation: CloudInferenceReservation, reason: string): Promise<void>;
+    streamChat(context: Context, options?: SimpleStreamOptions): import("@mariozechner/pi-ai").AssistantMessageEventStream;
+    private requireConfigOrNull;
+    private runtimePath;
+    private runStreamChat;
+    private requestJson;
+}
+export {};