@piecemaker-legal/piecemaker 1.0.0

package/dist-server/server/modules/providers/tests/provider-token-usage.service.test.js

@@ -0,0 +1,329 @@
+import assert from 'node:assert/strict';
+import { mkdtemp, rm, writeFile } from 'node:fs/promises';
+import { tmpdir } from 'node:os';
+import path from 'node:path';
+import test from 'node:test';
+import Database from 'better-sqlite3';
+import { createProviderTokenUsageService, summarizeClaudeTokenUsage, } from '../../../modules/providers/services/provider-token-usage.service.js';
+import { AppError } from '../../../shared/utils.js';
+function createSessionRow(overrides = {}) {
+    return {
+        session_id: 'app-session',
+        provider: 'claude',
+        provider_session_id: 'provider-session',
+        project_path: null,
+        jsonl_path: null,
+        custom_name: null,
+        model: null,
+        effort: null,
+        forked_from_session_id: null,
+        isArchived: 0,
+        created_at: '2026-01-01T00:00:00.000Z',
+        updated_at: '2026-01-01T00:00:00.000Z',
+        ...overrides,
+    };
+}
+test('token usage lookup requires only the app-facing session id for Claude', async () => {
+    const tempDirectory = await mkdtemp(path.join(tmpdir(), 'provider-token-usage-claude-'));
+    const sessionFilePath = path.join(tempDirectory, 'provider-session.jsonl');
+    try {
+        await writeFile(sessionFilePath, [
+            JSON.stringify({
+                type: 'assistant',
+                message: {
+                    usage: {
+                        input_tokens: 100,
+                        cache_read_input_tokens: 20,
+                        cache_creation_input_tokens: 5,
+                        output_tokens: 30,
+                    },
+                },
+            }),
+            '{incomplete',
+        ].join('\n'));
+        const service = createProviderTokenUsageService({
+            getSessionById: () => createSessionRow({ jsonl_path: sessionFilePath }),
+            getClaudeContextWindow: () => '180000',
+        });
+        assert.deepEqual(await service.getSessionTokenUsage('app-session'), {
+            used: 155,
+            total: 180_000,
+            inputTokens: 125,
+            outputTokens: 30,
+            cacheReadTokens: 20,
+            cacheCreationTokens: 5,
+            cacheTokens: 25,
+            breakdown: { input: 125, output: 30 },
+        });
+    }
+    finally {
+        await rm(tempDirectory, { recursive: true, force: true });
+    }
+});
+test('Codex token usage uses the latest token_count snapshot', async () => {
+    const tempDirectory = await mkdtemp(path.join(tmpdir(), 'provider-token-usage-codex-'));
+    const sessionFilePath = path.join(tempDirectory, 'rollout-provider-session.jsonl');
+    try {
+        await writeFile(sessionFilePath, [
+            JSON.stringify({
+                type: 'event_msg',
+                payload: {
+                    type: 'token_count',
+                    info: {
+                        total_token_usage: { input_tokens: 10, output_tokens: 4, total_tokens: 14 },
+                        model_context_window: 100_000,
+                    },
+                },
+            }),
+            JSON.stringify({
+                type: 'event_msg',
+                payload: {
+                    type: 'token_count',
+                    info: {
+                        total_token_usage: { input_tokens: 40, output_tokens: 9, total_tokens: 49 },
+                        model_context_window: 250_000,
+                    },
+                },
+            }),
+        ].join('\n'));
+        const service = createProviderTokenUsageService({
+            getSessionById: () => createSessionRow({
+                provider: 'codex',
+                jsonl_path: sessionFilePath,
+            }),
+        });
+        assert.deepEqual(await service.getSessionTokenUsage('app-session'), {
+            used: 49,
+            total: 250_000,
+            inputTokens: 40,
+            outputTokens: 9,
+            breakdown: { input: 40, output: 9 },
+        });
+    }
+    finally {
+        await rm(tempDirectory, { recursive: true, force: true });
+    }
+});
+test('OpenCode token usage resolves its provider-native id from the session row', async () => {
+    const tempDirectory = await mkdtemp(path.join(tmpdir(), 'provider-token-usage-opencode-'));
+    const databasePath = path.join(tempDirectory, 'opencode.db');
+    const database = new Database(databasePath);
+    try {
+        database.exec(`
+      CREATE TABLE session (
+        id TEXT PRIMARY KEY,
+        tokens_input INTEGER,
+        tokens_output INTEGER,
+        tokens_reasoning INTEGER,
+        tokens_cache_read INTEGER,
+        tokens_cache_write INTEGER
+      )
+    `);
+        database.prepare(`
+      INSERT INTO session (
+        id,
+        tokens_input,
+        tokens_output,
+        tokens_reasoning,
+        tokens_cache_read,
+        tokens_cache_write
+      ) VALUES (?, ?, ?, ?, ?, ?)
+    `).run('provider-session', 12, 7, 3, 5, 2);
+    }
+    finally {
+        database.close();
+    }
+    try {
+        const service = createProviderTokenUsageService({
+            getSessionById: () => createSessionRow({ provider: 'opencode' }),
+            getOpenCodeDatabasePath: () => databasePath,
+        });
+        assert.deepEqual(await service.getSessionTokenUsage('app-session'), {
+            used: 29,
+            inputTokens: 17,
+            outputTokens: 7,
+            breakdown: { input: 17, output: 7 },
+        });
+    }
+    finally {
+        await rm(tempDirectory, { recursive: true, force: true });
+    }
+});
+test('Cursor returns an explicit unsupported token usage result', async () => {
+    const service = createProviderTokenUsageService({
+        getSessionById: () => createSessionRow({ provider: 'cursor' }),
+    });
+    const result = await service.getSessionTokenUsage('app-session');
+    assert.equal(result.unsupported, true);
+    assert.equal(result.used, 0);
+    assert.equal(result.total, 0);
+});
+test('token usage reports SESSION_NOT_FOUND for an unknown app session id', async () => {
+    const service = createProviderTokenUsageService({ getSessionById: () => null });
+    await assert.rejects(() => service.getSessionTokenUsage('missing-session'), (error) => (error instanceof AppError
+        && error.code === 'SESSION_NOT_FOUND'
+        && error.statusCode === 404));
+});
+test('the Claude summarizer reads the newest assistant turn, not the whole conversation', () => {
+    const entries = [
+        { type: 'assistant', message: { usage: { input_tokens: 5, cache_read_input_tokens: 1000, output_tokens: 50 } } },
+        { type: 'user', message: { role: 'user', content: 'next' } },
+        // The newest turn's prompt is the whole context, so its cache_read already
+        // includes everything before it. Summing turns would double-count.
+        { type: 'assistant', message: { usage: { input_tokens: 3, cache_read_input_tokens: 4000, cache_creation_input_tokens: 100, output_tokens: 80 } } },
+    ];
+    assert.deepEqual(summarizeClaudeTokenUsage(entries, '200000'), {
+        used: 4183,
+        total: 200_000,
+        inputTokens: 4103,
+        outputTokens: 80,
+        cacheReadTokens: 4000,
+        cacheCreationTokens: 100,
+        cacheTokens: 4100,
+        breakdown: { input: 4103, output: 80 },
+    });
+});
+test('the Claude summarizer skips synthetic rows that carry an all-zero usage block', () => {
+    // Interrupts, API errors and "No response requested." are written as
+    // assistant rows with a fully zeroed usage block. Reading one as the newest
+    // turn dropped the composer counter to 0 until the next turn pushed it back.
+    const entries = [
+        { type: 'assistant', message: { usage: { input_tokens: 3, cache_read_input_tokens: 4000, output_tokens: 80 } } },
+        {
+            type: 'assistant',
+            message: {
+                model: '<synthetic>',
+                usage: {
+                    input_tokens: 0,
+                    output_tokens: 0,
+                    cache_creation_input_tokens: 0,
+                    cache_read_input_tokens: 0,
+                },
+            },
+        },
+    ];
+    assert.equal(summarizeClaudeTokenUsage(entries, '200000').used, 4083);
+});
+test('the Claude summarizer skips a subagent sidechain turn', () => {
+    // A sidechain turn reports the subagent's own context window. Reading it
+    // made the counter drop to the subagent's number mid-run.
+    const entries = [
+        { type: 'assistant', message: { usage: { input_tokens: 3, cache_read_input_tokens: 4000, output_tokens: 80 } } },
+        {
+            type: 'assistant',
+            isSidechain: true,
+            message: { usage: { input_tokens: 10, cache_read_input_tokens: 900, output_tokens: 5 } },
+        },
+    ];
+    assert.equal(summarizeClaudeTokenUsage(entries, '200000').used, 4083);
+});
+test('the Claude summarizer reports zero for a transcript with no assistant turn yet', () => {
+    const usage = summarizeClaudeTokenUsage([{ type: 'user', message: { role: 'user', content: 'hi' } }], '200000');
+    assert.equal(usage.used, 0);
+    assert.equal(usage.total, 200_000);
+});
+/** Padding rows large enough to push earlier rows out of the 4MB tail window. */
+function paddingLines(totalBytes) {
+    const line = JSON.stringify({ type: 'attachment', filler: 'x'.repeat(4096) });
+    return Array.from({ length: Math.ceil(totalBytes / line.length) }, () => line).join('\n');
+}
+test('Claude token usage reads only the tail of a large transcript', async () => {
+    const tempDirectory = await mkdtemp(path.join(tmpdir(), 'provider-token-usage-claude-tail-'));
+    const sessionFilePath = path.join(tempDirectory, 'provider-session.jsonl');
+    try {
+        await writeFile(sessionFilePath, [
+            paddingLines(5 * 1024 * 1024),
+            JSON.stringify({ type: 'assistant', message: { usage: { input_tokens: 7, output_tokens: 2 } } }),
+        ].join('\n'));
+        const service = createProviderTokenUsageService({
+            getSessionById: () => createSessionRow({ jsonl_path: sessionFilePath }),
+            getClaudeContextWindow: () => '180000',
+            // Reading the whole file here would defeat the tail read; fail loudly.
+            readTextFile: () => { throw new Error('full read must not happen when the tail has usage'); },
+        });
+        const usage = await service.getSessionTokenUsage('app-session');
+        assert.equal(usage.inputTokens, 7);
+        assert.equal(usage.outputTokens, 2);
+    }
+    finally {
+        await rm(tempDirectory, { recursive: true, force: true });
+    }
+});
+test('Claude token usage falls back to the whole file when the tail has no usage row', async () => {
+    const tempDirectory = await mkdtemp(path.join(tmpdir(), 'provider-token-usage-claude-fallback-'));
+    const sessionFilePath = path.join(tempDirectory, 'provider-session.jsonl');
+    try {
+        await writeFile(sessionFilePath, [
+            JSON.stringify({ type: 'assistant', message: { usage: { input_tokens: 11, output_tokens: 3 } } }),
+            paddingLines(5 * 1024 * 1024),
+        ].join('\n'));
+        const service = createProviderTokenUsageService({
+            getSessionById: () => createSessionRow({ jsonl_path: sessionFilePath }),
+            getClaudeContextWindow: () => '180000',
+        });
+        const usage = await service.getSessionTokenUsage('app-session');
+        assert.equal(usage.inputTokens, 11);
+        assert.equal(usage.outputTokens, 3);
+    }
+    finally {
+        await rm(tempDirectory, { recursive: true, force: true });
+    }
+});
+test('Codex token usage reads only the tail of a large rollout', async () => {
+    const tempDirectory = await mkdtemp(path.join(tmpdir(), 'provider-token-usage-codex-tail-'));
+    const sessionFilePath = path.join(tempDirectory, 'rollout-provider-session.jsonl');
+    try {
+        await writeFile(sessionFilePath, [
+            paddingLines(5 * 1024 * 1024),
+            JSON.stringify({
+                type: 'event_msg',
+                payload: {
+                    type: 'token_count',
+                    info: {
+                        total_token_usage: { input_tokens: 21, output_tokens: 8, total_tokens: 29 },
+                        model_context_window: 150_000,
+                    },
+                },
+            }),
+        ].join('\n'));
+        const service = createProviderTokenUsageService({
+            getSessionById: () => createSessionRow({ provider: 'codex', jsonl_path: sessionFilePath }),
+            readTextFile: () => { throw new Error('full read must not happen when the tail has usage'); },
+        });
+        assert.deepEqual(await service.getSessionTokenUsage('app-session'), {
+            used: 29,
+            total: 150_000,
+            inputTokens: 21,
+            outputTokens: 8,
+            breakdown: { input: 21, output: 8 },
+        });
+    }
+    finally {
+        await rm(tempDirectory, { recursive: true, force: true });
+    }
+});
+test('Codex token usage falls back to the whole file when the tail has no token_count row', async () => {
+    const tempDirectory = await mkdtemp(path.join(tmpdir(), 'provider-token-usage-codex-fallback-'));
+    const sessionFilePath = path.join(tempDirectory, 'rollout-provider-session.jsonl');
+    try {
+        await writeFile(sessionFilePath, [
+            JSON.stringify({
+                type: 'event_msg',
+                payload: {
+                    type: 'token_count',
+                    info: { total_token_usage: { input_tokens: 5, output_tokens: 1, total_tokens: 6 } },
+                },
+            }),
+            paddingLines(5 * 1024 * 1024),
+        ].join('\n'));
+        const service = createProviderTokenUsageService({
+            getSessionById: () => createSessionRow({ provider: 'codex', jsonl_path: sessionFilePath }),
+        });
+        const usage = await service.getSessionTokenUsage('app-session');
+        assert.equal(usage.used, 6);
+    }
+    finally {
+        await rm(tempDirectory, { recursive: true, force: true });
+    }
+});
+//# sourceMappingURL=provider-token-usage.service.test.js.map

package/dist-server/server/modules/providers/tests/provider-token-usage.service.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-token-usage.service.test.js","sourceRoot":"","sources":["../../../../../server/modules/providers/tests/provider-token-usage.service.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAEtC,OAAO,EACL,+BAA+B,EAC/B,yBAAyB,GAC1B,MAAM,8DAA8D,CAAC;AACtE,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,SAAS,gBAAgB,CAAC,YAAqC,EAAE;IAC/D,OAAO;QACL,UAAU,EAAE,aAAa;QACzB,QAAQ,EAAE,QAAQ;QAClB,mBAAmB,EAAE,kBAAkB;QACvC,YAAY,EAAE,IAAI;QAClB,UAAU,EAAE,IAAI;QAChB,WAAW,EAAE,IAAI;QACjB,KAAK,EAAE,IAAI;QACX,MAAM,EAAE,IAAI;QACZ,sBAAsB,EAAE,IAAI;QAC5B,UAAU,EAAE,CAAC;QACb,UAAU,EAAE,0BAA0B;QACtC,UAAU,EAAE,0BAA0B;QACtC,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,IAAI,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;IACvF,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,8BAA8B,CAAC,CAAC,CAAC;IACzF,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,wBAAwB,CAAC,CAAC;IAE3E,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,eAAe,EAAE;YAC/B,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP,KAAK,EAAE;wBACL,YAAY,EAAE,GAAG;wBACjB,uBAAuB,EAAE,EAAE;wBAC3B,2BAA2B,EAAE,CAAC;wBAC9B,aAAa,EAAE,EAAE;qBAClB;iBACF;aACF,CAAC;YACF,aAAa;SACd,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAEd,MAAM,OAAO,GAAG,+BAA+B,CAAC;YAC9C,cAAc,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC;YACvE,sBAAsB,EAAE,GAAG,EAAE,CAAC,QAAQ;SACvC,CAAC,CAAC;QAEH,MAAM,CAAC,SAAS,CAAC,MAAM,OAAO,CAAC,oBAAoB,CAAC,aAAa,CAAC,EAAE;YAClE,IAAI,EAAE,GAAG;YACT,KAAK,EAAE,OAAO;YACd,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,EAAE;YAChB,eAAe,EAAE,EAAE;YACnB,mBAAmB,EAAE,CAAC;YACtB,WAAW,EAAE,EAAE;YACf,SAAS,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE;SACtC,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;IACxE,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,6BAA6B,CAAC,CAAC,CAAC;IACxF,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,gCAAgC,CAAC,CAAC;IAEnF,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,eAAe,EAAE;YAC/B,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP,IAAI,EAAE,aAAa;oBACnB,IAAI,EAAE;wBACJ,iBAAiB,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE;wBAC3E,oBAAoB,EAAE,OAAO;qBAC9B;iBACF;aACF,CAAC;YACF,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP,IAAI,EAAE,aAAa;oBACnB,IAAI,EAAE;wBACJ,iBAAiB,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE;wBAC3E,oBAAoB,EAAE,OAAO;qBAC9B;iBACF;aACF,CAAC;SACH,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAEd,MAAM,OAAO,GAAG,+BAA+B,CAAC;YAC9C,cAAc,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC;gBACrC,QAAQ,EAAE,OAAO;gBACjB,UAAU,EAAE,eAAe;aAC5B,CAAC;SACH,CAAC,CAAC;QAEH,MAAM,CAAC,SAAS,CAAC,MAAM,OAAO,CAAC,oBAAoB,CAAC,aAAa,CAAC,EAAE;YAClE,IAAI,EAAE,EAAE;YACR,KAAK,EAAE,OAAO;YACd,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,CAAC;YACf,SAAS,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE;SACpC,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;IAC3F,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,gCAAgC,CAAC,CAAC,CAAC;IAC3F,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;IAC7D,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,YAAY,CAAC,CAAC;IAE5C,IAAI,CAAC;QACH,QAAQ,CAAC,IAAI,CAAC;;;;;;;;;KASb,CAAC,CAAC;QACH,QAAQ,CAAC,OAAO,CAAC;;;;;;;;;KAShB,CAAC,CAAC,GAAG,CAAC,kBAAkB,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7C,CAAC;YAAS,CAAC;QACT,QAAQ,CAAC,KAAK,EAAE,CAAC;IACnB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,+BAA+B,CAAC;YAC9C,cAAc,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;YAChE,uBAAuB,EAAE,GAAG,EAAE,CAAC,YAAY;SAC5C,CAAC,CAAC;QAEH,MAAM,CAAC,SAAS,CAAC,MAAM,OAAO,CAAC,oBAAoB,CAAC,aAAa,CAAC,EAAE;YAClE,IAAI,EAAE,EAAE;YACR,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,CAAC;YACf,SAAS,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE;SACpC,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;IAC3E,MAAM,OAAO,GAAG,+BAA+B,CAAC;QAC9C,cAAc,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;KAC/D,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;IAEjE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IACvC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC7B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;AAChC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;IACrF,MAAM,OAAO,GAAG,+BAA+B,CAAC,EAAE,cAAc,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;IAEhF,MAAM,MAAM,CAAC,OAAO,CAClB,GAAG,EAAE,CAAC,OAAO,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,EACrD,CAAC,KAAc,EAAE,EAAE,CAAC,CAClB,KAAK,YAAY,QAAQ;WACtB,KAAK,CAAC,IAAI,KAAK,mBAAmB;WAClC,KAAK,CAAC,UAAU,KAAK,GAAG,CAC5B,CACF,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mFAAmF,EAAE,GAAG,EAAE;IAC7F,MAAM,OAAO,GAAG;QACd,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,uBAAuB,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,EAAE,EAAE;QAChH,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;QAC5D,2EAA2E;QAC3E,mEAAmE;QACnE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,uBAAuB,EAAE,IAAI,EAAE,2BAA2B,EAAE,GAAG,EAAE,aAAa,EAAE,EAAE,EAAE,EAAE,EAAE;KACnJ,CAAC;IAEF,MAAM,CAAC,SAAS,CAAC,yBAAyB,CAAC,OAAO,EAAE,QAAQ,CAAC,EAAE;QAC7D,IAAI,EAAE,IAAI;QACV,KAAK,EAAE,OAAO;QACd,WAAW,EAAE,IAAI;QACjB,YAAY,EAAE,EAAE;QAChB,eAAe,EAAE,IAAI;QACrB,mBAAmB,EAAE,GAAG;QACxB,WAAW,EAAE,IAAI;QACjB,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE;KACvC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+EAA+E,EAAE,GAAG,EAAE;IACzF,qEAAqE;IACrE,4EAA4E;IAC5E,6EAA6E;IAC7E,MAAM,OAAO,GAAG;QACd,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,uBAAuB,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,EAAE,EAAE;QAChH;YACE,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP,KAAK,EAAE,aAAa;gBACpB,KAAK,EAAE;oBACL,YAAY,EAAE,CAAC;oBACf,aAAa,EAAE,CAAC;oBAChB,2BAA2B,EAAE,CAAC;oBAC9B,uBAAuB,EAAE,CAAC;iBAC3B;aACF;SACF;KACF,CAAC;IAEF,MAAM,CAAC,KAAK,CAAC,yBAAyB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;AACxE,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,uDAAuD,EAAE,GAAG,EAAE;IACjE,yEAAyE;IACzE,0DAA0D;IAC1D,MAAM,OAAO,GAAG;QACd,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,uBAAuB,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,EAAE,EAAE;QAChH;YACE,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,IAAI;YACjB,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,uBAAuB,EAAE,GAAG,EAAE,aAAa,EAAE,CAAC,EAAE,EAAE;SACzF;KACF,CAAC;IAEF,MAAM,CAAC,KAAK,CAAC,yBAAyB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;AACxE,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,gFAAgF,EAAE,GAAG,EAAE;IAC1F,MAAM,KAAK,GAAG,yBAAyB,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;IAEhH,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC5B,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;AACrC,CAAC,CAAC,CAAC;AAEH,iFAAiF;AACjF,SAAS,YAAY,CAAC,UAAkB;IACtC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9E,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC5F,CAAC;AAED,IAAI,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;IAC9E,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,mCAAmC,CAAC,CAAC,CAAC;IAC9F,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,wBAAwB,CAAC,CAAC;IAE3E,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,eAAe,EAAE;YAC/B,YAAY,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;YAC7B,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;SACjG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAEd,MAAM,OAAO,GAAG,+BAA+B,CAAC;YAC9C,cAAc,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC;YACvE,sBAAsB,EAAE,GAAG,EAAE,CAAC,QAAQ;YACtC,uEAAuE;YACvE,YAAY,EAAE,GAAG,EAAE,GAAG,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC,CAAC,CAAC;SAC9F,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;QAChE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IACtC,CAAC;YAAS,CAAC;QACT,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;IAChG,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uCAAuC,CAAC,CAAC,CAAC;IAClG,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,wBAAwB,CAAC,CAAC;IAE3E,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,eAAe,EAAE;YAC/B,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;YACjG,YAAY,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;SAC9B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAEd,MAAM,OAAO,GAAG,+BAA+B,CAAC;YAC9C,cAAc,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC;YACvE,sBAAsB,EAAE,GAAG,EAAE,CAAC,QAAQ;SACvC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;QAChE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IACtC,CAAC;YAAS,CAAC;QACT,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;IAC1E,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,kCAAkC,CAAC,CAAC,CAAC;IAC7F,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,gCAAgC,CAAC,CAAC;IAEnF,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,eAAe,EAAE;YAC/B,YAAY,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;YAC7B,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP,IAAI,EAAE,aAAa;oBACnB,IAAI,EAAE;wBACJ,iBAAiB,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE;wBAC3E,oBAAoB,EAAE,OAAO;qBAC9B;iBACF;aACF,CAAC;SACH,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAEd,MAAM,OAAO,GAAG,+BAA+B,CAAC;YAC9C,cAAc,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC;YAC1F,YAAY,EAAE,GAAG,EAAE,GAAG,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC,CAAC,CAAC;SAC9F,CAAC,CAAC;QAEH,MAAM,CAAC,SAAS,CAAC,MAAM,OAAO,CAAC,oBAAoB,CAAC,aAAa,CAAC,EAAE;YAClE,IAAI,EAAE,EAAE;YACR,KAAK,EAAE,OAAO;YACd,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,CAAC;YACf,SAAS,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE;SACpC,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qFAAqF,EAAE,KAAK,IAAI,EAAE;IACrG,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,sCAAsC,CAAC,CAAC,CAAC;IACjG,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,gCAAgC,CAAC,CAAC;IAEnF,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,eAAe,EAAE;YAC/B,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP,IAAI,EAAE,aAAa;oBACnB,IAAI,EAAE,EAAE,iBAAiB,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,EAAE;iBACpF;aACF,CAAC;YACF,YAAY,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;SAC9B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAEd,MAAM,OAAO,GAAG,+BAA+B,CAAC;YAC9C,cAAc,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC;SAC3F,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;QAChE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC9B,CAAC;YAAS,CAAC;QACT,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC,CAAC,CAAC"}

package/dist-server/server/modules/providers/tests/provider.routes.test.js

@@ -0,0 +1,161 @@
+import assert from 'node:assert/strict';
+import { once } from 'node:events';
+import { mkdtemp, rm, writeFile } from 'node:fs/promises';
+import os from 'node:os';
+import path from 'node:path';
+import test from 'node:test';
+import express from 'express';
+import { closeConnection, initializeDatabase, sessionsDb } from '../../../modules/database/index.js';
+import providerRouter from '../../../modules/providers/provider.routes.js';
+import { AppError } from '../../../shared/utils.js';
+async function withProviderServer(run) {
+    const previousDatabasePath = process.env.DATABASE_PATH;
+    const tempDirectory = await mkdtemp(path.join(os.tmpdir(), 'provider-routes-'));
+    closeConnection();
+    process.env.DATABASE_PATH = path.join(tempDirectory, 'auth.db');
+    await writeFile(process.env.DATABASE_PATH, '');
+    await initializeDatabase();
+    const app = express().use(express.json()).use('/api/providers', providerRouter);
+    app.use((error, _req, res, _next) => {
+        if (error instanceof AppError) {
+            res.status(error.statusCode).json({
+                success: false,
+                error: { code: error.code, message: error.message },
+            });
+            return;
+        }
+        res.status(500).json({ success: false, error: { code: 'INTERNAL_ERROR' } });
+    });
+    const server = app.listen(0, '127.0.0.1');
+    await once(server, 'listening');
+    try {
+        const address = server.address();
+        await run(`http://127.0.0.1:${address.port}`, path.join(tempDirectory, 'workspace'));
+    }
+    finally {
+        await new Promise((resolve, reject) => {
+            server.close((error) => error ? reject(error) : resolve());
+        });
+        closeConnection();
+        if (previousDatabasePath === undefined) {
+            delete process.env.DATABASE_PATH;
+        }
+        else {
+            process.env.DATABASE_PATH = previousDatabasePath;
+        }
+        await rm(tempDirectory, { recursive: true, force: true });
+    }
+}
+test('session creation route names a CloudCLI session from the initial message', async () => {
+    await withProviderServer(async (baseUrl, workspacePath) => {
+        const response = await fetch(`${baseUrl}/api/providers/sessions`, {
+            method: 'POST',
+            headers: { 'content-type': 'application/json' },
+            body: JSON.stringify({
+                provider: 'codex',
+                projectPath: workspacePath,
+                initialMessage: 'abcd  efg\nhij klm nop',
+            }),
+        });
+        const payload = await response.json();
+        assert.equal(response.status, 201);
+        assert.equal(payload.data.sessionName, 'abcd efg hij klm');
+        assert.equal(sessionsDb.getSessionById(payload.data.sessionId)?.custom_name, 'abcd efg hij klm');
+    });
+});
+test('conversation search streams title matches before transcript results', async () => {
+    await withProviderServer(async (baseUrl, workspacePath) => {
+        sessionsDb.createAppSession('title-only-session', 'codex', workspacePath, 'Release planning notes');
+        const transcriptPath = path.join(path.dirname(workspacePath), 'codex-search.jsonl');
+        await writeFile(transcriptPath, `${JSON.stringify({
+            type: 'event_msg',
+            timestamp: '2026-08-12T09:00:00.000Z',
+            payload: {
+                type: 'user_message',
+                kind: 'plain',
+                message: 'Release planning also appears in this conversation.',
+            },
+        })}\n`);
+        sessionsDb.createSession('transcript-session', 'codex', workspacePath, 'Unrelated session', undefined, undefined, transcriptPath);
+        const response = await fetch(`${baseUrl}/api/providers/search/sessions?q=release%20planning&limit=50`);
+        const eventStream = await response.text();
+        const titleEventIndex = eventStream.indexOf('event: title-results');
+        const conversationEventIndex = eventStream.indexOf('event: result');
+        const doneEventIndex = eventStream.indexOf('event: done');
+        assert.equal(response.status, 200);
+        assert.ok(titleEventIndex >= 0);
+        assert.ok(conversationEventIndex > titleEventIndex);
+        assert.ok(doneEventIndex > titleEventIndex);
+        const titleDataLine = eventStream
+            .slice(titleEventIndex, conversationEventIndex)
+            .split('\n')
+            .find((line) => line.startsWith('data: '));
+        assert.ok(titleDataLine);
+        const titlePayload = JSON.parse(titleDataLine.slice('data: '.length));
+        assert.equal(titlePayload.titleResults.length, 1);
+        assert.equal(titlePayload.titleResults[0]?.sessionId, 'title-only-session');
+        assert.equal(titlePayload.titleResults[0]?.sessionTitle, 'Release planning notes');
+        assert.equal(titlePayload.titleResults[0]?.provider, 'codex');
+    });
+});
+test('reasoning effort is persisted and returned with the active session model', async () => {
+    await withProviderServer(async (baseUrl, workspacePath) => {
+        sessionsDb.createAppSession('effort-session', 'codex', workspacePath);
+        const updateResponse = await fetch(`${baseUrl}/api/providers/codex/sessions/effort-session/active-effort`, {
+            method: 'POST',
+            headers: { 'content-type': 'application/json' },
+            body: JSON.stringify({ effort: 'ultra' }),
+        });
+        const updatePayload = await updateResponse.json();
+        assert.equal(updateResponse.status, 200);
+        assert.equal(updatePayload.data.effort, 'ultra');
+        assert.equal(sessionsDb.getSessionById('effort-session')?.effort, 'ultra');
+        const readResponse = await fetch(`${baseUrl}/api/providers/codex/sessions/effort-session/active-model`);
+        const readPayload = await readResponse.json();
+        assert.equal(readResponse.status, 200);
+        assert.equal(readPayload.data.sessionId, 'effort-session');
+        assert.equal(readPayload.data.effort, 'ultra');
+    });
+});
+test('model routes expose immutable defaults and full custom model CRUD', async () => {
+    await withProviderServer(async (baseUrl) => {
+        const initialResponse = await fetch(`${baseUrl}/api/providers/codex/models`);
+        const initialPayload = await initialResponse.json();
+        assert.equal(initialResponse.status, 200);
+        assert.equal('cache' in initialPayload.data, false);
+        const predefined = initialPayload.data.models.OPTIONS[0];
+        assert.equal(predefined.isCustom, false);
+        assert.equal(predefined.recordId, undefined);
+        const createResponse = await fetch(`${baseUrl}/api/providers/codex/models`, {
+            method: 'POST',
+            headers: { 'content-type': 'application/json' },
+            body: JSON.stringify({ model: 'Gateway GPT', id: 'gateway/gpt' }),
+        });
+        const createPayload = await createResponse.json();
+        assert.equal(createResponse.status, 201);
+        assert.equal(createPayload.data.model.isCustom, true);
+        const customRecordId = createPayload.data.model.recordId;
+        const updateResponse = await fetch(`${baseUrl}/api/providers/codex/models/${customRecordId}`, {
+            method: 'PATCH',
+            headers: { 'content-type': 'application/json' },
+            body: JSON.stringify({ model: 'Gateway GPT Updated', id: 'gateway/gpt-v2' }),
+        });
+        const updatePayload = await updateResponse.json();
+        assert.equal(updateResponse.status, 200);
+        assert.equal(updatePayload.data.model.value, 'gateway/gpt-v2');
+        assert.equal(updatePayload.data.model.label, 'Gateway GPT Updated');
+        const immutableResponse = await fetch(`${baseUrl}/api/providers/codex/models/999999`, {
+            method: 'PATCH',
+            headers: { 'content-type': 'application/json' },
+            body: JSON.stringify({ model: 'Changed', id: 'changed' }),
+        });
+        const immutablePayload = await immutableResponse.json();
+        assert.equal(immutableResponse.status, 404);
+        assert.equal(immutablePayload.error.code, 'MODEL_NOT_FOUND');
+        const deleteResponse = await fetch(`${baseUrl}/api/providers/codex/models/${customRecordId}`, { method: 'DELETE' });
+        const deletePayload = await deleteResponse.json();
+        assert.equal(deleteResponse.status, 200);
+        assert.equal(deletePayload.data.models.OPTIONS.some((option) => option.recordId === customRecordId), false);
+    });
+});
+//# sourceMappingURL=provider.routes.test.js.map

package/dist-server/server/modules/providers/tests/provider.routes.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.routes.test.js","sourceRoot":"","sources":["../../../../../server/modules/providers/tests/provider.routes.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AACnC,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAE1D,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,OAA2D,MAAM,SAAS,CAAC;AAElF,OAAO,EAAE,eAAe,EAAE,kBAAkB,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AAC9F,OAAO,cAAc,MAAM,wCAAwC,CAAC;AACpE,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,KAAK,UAAU,kBAAkB,CAC/B,GAA8D;IAE9D,MAAM,oBAAoB,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;IACvD,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,kBAAkB,CAAC,CAAC,CAAC;IAEhF,eAAe,EAAE,CAAC;IAClB,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IAChE,MAAM,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;IAC/C,MAAM,kBAAkB,EAAE,CAAC;IAE3B,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,cAAc,CAAC,CAAC;IAChF,GAAG,CAAC,GAAG,CAAC,CAAC,KAAc,EAAE,IAAa,EAAE,GAAa,EAAE,KAAmB,EAAE,EAAE;QAC5E,IAAI,KAAK,YAAY,QAAQ,EAAE,CAAC;YAC9B,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC;gBAChC,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE;aACpD,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QACD,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;IAC1C,MAAM,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAEhC,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAiB,CAAC;QAChD,MAAM,GAAG,CAAC,oBAAoB,OAAO,CAAC,IAAI,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC,CAAC;IACvF,CAAC;YAAS,CAAC;QACT,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC1C,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QACH,eAAe,EAAE,CAAC;QAClB,IAAI,oBAAoB,KAAK,SAAS,EAAE,CAAC;YACvC,OAAO,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,oBAAoB,CAAC;QACnD,CAAC;QACD,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC;AAED,IAAI,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;IAC1F,MAAM,kBAAkB,CAAC,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE;QACxD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,yBAAyB,EAAE;YAChE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,QAAQ,EAAE,OAAO;gBACjB,WAAW,EAAE,aAAa;gBAC1B,cAAc,EAAE,wBAAwB;aACzC,CAAC;SACH,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,EAElC,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,kBAAkB,CAAC,CAAC;QAC3D,MAAM,CAAC,KAAK,CACV,UAAU,CAAC,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAC9D,kBAAkB,CACnB,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;IACrF,MAAM,kBAAkB,CAAC,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE;QACxD,UAAU,CAAC,gBAAgB,CACzB,oBAAoB,EACpB,OAAO,EACP,aAAa,EACb,wBAAwB,CACzB,CAAC;QACF,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,oBAAoB,CAAC,CAAC;QACpF,MAAM,SAAS,CAAC,cAAc,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;YAChD,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE,0BAA0B;YACrC,OAAO,EAAE;gBACP,IAAI,EAAE,cAAc;gBACpB,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,qDAAqD;aAC/D;SACF,CAAC,IAAI,CAAC,CAAC;QACR,UAAU,CAAC,aAAa,CACtB,oBAAoB,EACpB,OAAO,EACP,aAAa,EACb,mBAAmB,EACnB,SAAS,EACT,SAAS,EACT,cAAc,CACf,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,GAAG,OAAO,8DAA8D,CACzE,CAAC;QACF,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC1C,MAAM,eAAe,GAAG,WAAW,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;QACpE,MAAM,sBAAsB,GAAG,WAAW,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QACpE,MAAM,cAAc,GAAG,WAAW,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAE1D,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACnC,MAAM,CAAC,EAAE,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,EAAE,CAAC,sBAAsB,GAAG,eAAe,CAAC,CAAC;QACpD,MAAM,CAAC,EAAE,CAAC,cAAc,GAAG,eAAe,CAAC,CAAC;QAE5C,MAAM,aAAa,GAAG,WAAW;aAC9B,KAAK,CAAC,eAAe,EAAE,sBAAsB,CAAC;aAC9C,KAAK,CAAC,IAAI,CAAC;aACX,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC;QAEzB,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAMnE,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,oBAAoB,CAAC,CAAC;QAC5E,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,wBAAwB,CAAC,CAAC;QACnF,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;IAC1F,MAAM,kBAAkB,CAAC,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE;QACxD,UAAU,CAAC,gBAAgB,CAAC,gBAAgB,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;QAEtE,MAAM,cAAc,GAAG,MAAM,KAAK,CAChC,GAAG,OAAO,4DAA4D,EACtE;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;SAC1C,CACF,CAAC;QACF,MAAM,aAAa,GAAG,MAAM,cAAc,CAAC,IAAI,EAE9C,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACjD,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,cAAc,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QAE3E,MAAM,YAAY,GAAG,MAAM,KAAK,CAC9B,GAAG,OAAO,2DAA2D,CACtE,CAAC;QACF,MAAM,WAAW,GAAG,MAAM,YAAY,CAAC,IAAI,EAE1C,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;QAC3D,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;IACnF,MAAM,kBAAkB,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;QACzC,MAAM,eAAe,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,6BAA6B,CAAC,CAAC;QAC7E,MAAM,cAAc,GAAG,MAAM,eAAe,CAAC,IAAI,EAOhD,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,IAAI,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACpD,MAAM,UAAU,GAAG,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACzD,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAE7C,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,6BAA6B,EAAE;YAC1E,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC;SAClE,CAAC,CAAC;QACH,MAAM,aAAa,GAAG,MAAM,cAAc,CAAC,IAAI,EAE9C,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACtD,MAAM,cAAc,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;QAEzD,MAAM,cAAc,GAAG,MAAM,KAAK,CAChC,GAAG,OAAO,+BAA+B,cAAc,EAAE,EACzD;YACE,MAAM,EAAE,OAAO;YACf,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,qBAAqB,EAAE,EAAE,EAAE,gBAAgB,EAAE,CAAC;SAC7E,CACF,CAAC;QACF,MAAM,aAAa,GAAG,MAAM,cAAc,CAAC,IAAI,EAE9C,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC;QAC/D,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,qBAAqB,CAAC,CAAC;QAEpE,MAAM,iBAAiB,GAAG,MAAM,KAAK,CACnC,GAAG,OAAO,oCAAoC,EAC9C;YACE,MAAM,EAAE,OAAO;YACf,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;SAC1D,CACF,CAAC;QACF,MAAM,gBAAgB,GAAG,MAAM,iBAAiB,CAAC,IAAI,EAAiC,CAAC;QACvF,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC5C,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;QAE7D,MAAM,cAAc,GAAG,MAAM,KAAK,CAChC,GAAG,OAAO,+BAA+B,cAAc,EAAE,EACzD,EAAE,MAAM,EAAE,QAAQ,EAAE,CACrB,CAAC;QACF,MAAM,aAAa,GAAG,MAAM,cAAc,CAAC,IAAI,EAE9C,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CACV,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,KAAK,cAAc,CAAC,EACtF,KAAK,CACN,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist-server/server/modules/providers/tests/session-fork.test.js

@@ -0,0 +1,113 @@
+import assert from 'node:assert/strict';
+import { mkdtemp, rm, writeFile } from 'node:fs/promises';
+import os from 'node:os';
+import path from 'node:path';
+import test from 'node:test';
+import { closeConnection, initializeDatabase, sessionsDb } from '../../../modules/database/index.js';
+import { providerRegistry } from '../../../modules/providers/provider.registry.js';
+import { sessionsService } from '../../../modules/providers/services/sessions.service.js';
+const SOURCE_ID = 'fork-source';
+async function withForkableClaude(runTest, options = {}) {
+    const previousDatabasePath = process.env.DATABASE_PATH;
+    const directory = await mkdtemp(path.join(os.tmpdir(), 'session-fork-'));
+    const sourcePath = path.join(directory, 'native-source.jsonl');
+    const forkedPath = path.join(directory, 'native-fork.jsonl');
+    await writeFile(sourcePath, '{}\n', 'utf8');
+    await writeFile(forkedPath, '{}\n', 'utf8');
+    closeConnection();
+    process.env.DATABASE_PATH = path.join(directory, 'auth.db');
+    await initializeDatabase();
+    const calls = [];
+    const claude = providerRegistry.resolveProvider('claude');
+    const realFork = claude.fork;
+    const replacement = options.disableFork
+        ? undefined
+        : {
+            forkSession: async (input) => {
+                calls.push(input);
+                return { providerSessionId: 'native-fork', jsonlPath: forkedPath };
+            },
+        };
+    Object.defineProperty(claude, 'fork', { value: replacement, configurable: true, writable: true });
+    try {
+        await runTest({ calls, directory });
+    }
+    finally {
+        Object.defineProperty(claude, 'fork', { value: realFork, configurable: true, writable: true });
+        closeConnection();
+        if (previousDatabasePath === undefined) {
+            delete process.env.DATABASE_PATH;
+        }
+        else {
+            process.env.DATABASE_PATH = previousDatabasePath;
+        }
+        await rm(directory, { recursive: true, force: true });
+    }
+}
+function seedSource(directory) {
+    const now = new Date().toISOString();
+    sessionsDb.createSession(SOURCE_ID, 'claude', directory, 'Original session', now, now, path.join(directory, 'native-source.jsonl'));
+    sessionsDb.assignProviderSessionId(SOURCE_ID, 'native-source');
+}
+test('a fork becomes an independent session that points back at its source', async () => {
+    await withForkableClaude(async ({ calls, directory }) => {
+        seedSource(directory);
+        const result = await sessionsService.forkSessionById(SOURCE_ID, { upToAnchorId: 'uuid-3' });
+        assert.equal(calls.length, 1);
+        assert.equal(calls[0].providerSessionId, 'native-source');
+        assert.equal(calls[0].upToAnchorId, 'uuid-3');
+        const forked = sessionsDb.getSessionById(result.sessionId);
+        assert.ok(forked);
+        // Written before the watcher can see the new file, so it is never indexed
+        // a second time under its provider-native id.
+        assert.equal(forked?.provider_session_id, 'native-fork');
+        assert.equal(forked?.jsonl_path, path.join(directory, 'native-fork.jsonl'));
+        assert.equal(forked?.forked_from_session_id, SOURCE_ID);
+        assert.equal(forked?.custom_name, 'Original session (fork)');
+        // The source is untouched: this is "try two approaches", not a move.
+        const source = sessionsDb.getSessionById(SOURCE_ID);
+        assert.equal(source?.provider_session_id, 'native-source');
+        assert.equal(source?.custom_name, 'Original session');
+    });
+});
+test('a fork inherits the model and effort of the conversation it branched from', async () => {
+    await withForkableClaude(async ({ directory }) => {
+        seedSource(directory);
+        sessionsDb.setSessionModel(SOURCE_ID, 'claude-opus-5');
+        sessionsDb.setSessionEffort(SOURCE_ID, 'xhigh');
+        const result = await sessionsService.forkSessionById(SOURCE_ID);
+        const forked = sessionsDb.getSessionById(result.sessionId);
+        assert.equal(forked?.model, 'claude-opus-5');
+        assert.equal(forked?.effort, 'xhigh');
+    });
+});
+test('forking replaces a row the watcher already made for the new transcript', async () => {
+    await withForkableClaude(async ({ directory }) => {
+        seedSource(directory);
+        // The watcher wins the race and indexes the fork under its native id.
+        const now = new Date().toISOString();
+        sessionsDb.createSession('native-fork', 'claude', directory, 'Indexed by the watcher', now, now, path.join(directory, 'native-fork.jsonl'));
+        const result = await sessionsService.forkSessionById(SOURCE_ID);
+        assert.equal(sessionsDb.getSessionById('native-fork'), null);
+        assert.equal(sessionsDb.getSessionById(result.sessionId)?.provider_session_id, 'native-fork');
+    });
+});
+test('a session with no transcript yet cannot be forked', async () => {
+    await withForkableClaude(async ({ directory }) => {
+        // An app-created session that has never run: no provider id, no transcript.
+        sessionsDb.createAppSession('never-ran', 'claude', directory, 'Never ran');
+        await assert.rejects(() => sessionsService.forkSessionById('never-ran'), (error) => error.code === 'FORK_SOURCE_NOT_READY');
+    });
+});
+test('a provider without the capability is refused rather than silently ignored', async () => {
+    await withForkableClaude(async ({ directory }) => {
+        seedSource(directory);
+        await assert.rejects(() => sessionsService.forkSessionById(SOURCE_ID), (error) => error.code === 'FORK_NOT_SUPPORTED');
+    }, { disableFork: true });
+});
+test('forking a session that does not exist is a 404', async () => {
+    await withForkableClaude(async () => {
+        await assert.rejects(() => sessionsService.forkSessionById('no-such-session'), (error) => error.code === 'SESSION_NOT_FOUND');
+    });
+});
+//# sourceMappingURL=session-fork.test.js.map