@amodalai/runtime 0.1.26 → 0.2.0

package/dist/src/session/session-builder.test.js

@@ -0,0 +1,352 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * Tests for Session Builder.
+ *
+ * Covers:
+ * 1. Build from fixture bundle → all tool types registered
+ * 2. Admin session → admin skills, user skills absent
+ * 3. Build with MCP → MCP tools registered
+ * 4. Build with pinned model → provider uses pinned model
+ * 5. stop_execution tool is registered
+ * 6. present tool is registered
+ * 7. shell_exec is NOT registered (G24 — dropped by design)
+ * 8. System prompt includes connection, skill, knowledge (G9)
+ */
+import { describe, it, expect, vi } from 'vitest';
+import { buildSessionComponents, PRESENT_TOOL_NAME, STOP_EXECUTION_TOOL_NAME } from './session-builder.js';
+import { createLogger } from '../logger.js';
+const logger = createLogger({ component: 'test:session-builder' });
+// ---------------------------------------------------------------------------
+// Mocks
+// ---------------------------------------------------------------------------
+// Mock the provider creation to avoid real API key requirements
+vi.mock('../providers/create-provider.js', () => ({
+    createProvider: vi.fn((config) => ({
+        model: config.model,
+        provider: config.provider,
+        languageModel: {},
+        streamText: () => { throw new Error('not implemented'); },
+        generateText: () => Promise.reject(new Error('not implemented')),
+    })),
+}));
+// ---------------------------------------------------------------------------
+// Fixtures
+// ---------------------------------------------------------------------------
+function makeConnection(name) {
+    return {
+        name,
+        spec: {
+            protocol: 'rest',
+            baseUrl: `https://api.${name}.com`,
+            auth: { type: 'bearer', token: 'env:API_KEY' },
+        },
+        access: {
+            endpoints: {
+                'GET /items': { returns: ['id', 'name'] },
+                'POST /items': { returns: ['id'], confirm: true },
+            },
+        },
+        surface: [
+            { method: 'GET', path: '/items', description: 'List items', included: true },
+            { method: 'POST', path: '/items', description: 'Create item', included: true },
+        ],
+        entities: 'Item — a thing',
+        rules: 'Always paginate',
+        location: '/tmp/connections/' + name,
+    };
+}
+function makeStore() {
+    return {
+        name: 'deals',
+        entity: {
+            name: 'Deal',
+            key: '{company}_{quarter}',
+            schema: {
+                company: { type: 'string' },
+                quarter: { type: 'string' },
+                amount: { type: 'number' },
+            },
+        },
+        location: '/tmp/stores/deals',
+    };
+}
+function makeSkill() {
+    return {
+        name: 'revenue-report',
+        description: 'Generate revenue report',
+        trigger: 'When user asks about revenue',
+        body: 'Step 1: Query charges. Step 2: Aggregate.',
+        location: '/tmp/skills/revenue-report',
+    };
+}
+function makeKnowledge() {
+    return {
+        name: 'style-guide',
+        title: 'Style Guide',
+        body: 'Always use markdown tables.',
+        location: '/tmp/knowledge/style-guide',
+    };
+}
+function makeCustomTool() {
+    return {
+        name: 'analyze_data',
+        description: 'Analyze data',
+        parameters: { type: 'object', properties: { query: { type: 'string' } }, required: ['query'] },
+        confirm: true,
+        timeout: 30000,
+        env: ['OPENAI_API_KEY'],
+        handlerPath: '/tmp/tools/analyze/handler.ts',
+        location: '/tmp/tools/analyze',
+        hasPackageJson: false,
+        hasSetupScript: false,
+        hasRequirementsTxt: false,
+        hasDockerfile: false,
+        sandboxLanguage: 'typescript',
+    };
+}
+function makeBundle(overrides) {
+    const connections = new Map();
+    connections.set('stripe', makeConnection('stripe'));
+    return {
+        source: 'local',
+        origin: '/tmp/agent',
+        config: {
+            name: 'test-agent',
+            version: '1.0.0',
+            description: 'A test agent',
+            userContext: 'Be helpful and concise.',
+            models: {
+                main: { provider: 'anthropic', model: 'claude-sonnet-4-20250514' },
+            },
+        },
+        connections,
+        skills: [makeSkill()],
+        agents: { main: undefined, simple: undefined, subagents: [] },
+        automations: [],
+        knowledge: [makeKnowledge()],
+        evals: [],
+        tools: [],
+        stores: [makeStore()],
+        ...overrides,
+    };
+}
+function makeStoreBackend() {
+    return {
+        put: vi.fn().mockResolvedValue({ version: 1 }),
+        get: vi.fn(),
+        list: vi.fn(),
+        delete: vi.fn(),
+        history: vi.fn(),
+        close: vi.fn(),
+        initialize: vi.fn(),
+        purgeExpired: vi.fn(),
+    };
+}
+function makeOpts(overrides) {
+    return {
+        bundle: makeBundle(),
+        storeBackend: makeStoreBackend(),
+        mcpManager: null,
+        logger,
+        ...overrides,
+    };
+}
+// ---------------------------------------------------------------------------
+// Tests
+// ---------------------------------------------------------------------------
+describe('buildSessionComponents', () => {
+    it('returns all required components', () => {
+        const components = buildSessionComponents(makeOpts());
+        expect(components.provider).toBeDefined();
+        expect(components.toolRegistry).toBeDefined();
+        expect(components.permissionChecker).toBeDefined();
+        expect(components.systemPrompt).toBeTruthy();
+        expect(components.toolContextFactory).toBeInstanceOf(Function);
+        expect(components.userRoles).toEqual([]);
+    });
+    it('registers store tools', () => {
+        const components = buildSessionComponents(makeOpts());
+        const names = components.toolRegistry.names();
+        expect(names).toContain('store_deals');
+        expect(names).toContain('store_deals_batch');
+        expect(names).toContain('query_store');
+    });
+    it('registers request tool when connections exist', () => {
+        const components = buildSessionComponents(makeOpts());
+        expect(components.toolRegistry.names()).toContain('request');
+    });
+    it('does not register request tool when no connections', () => {
+        const bundle = makeBundle({ connections: new Map() });
+        const components = buildSessionComponents(makeOpts({ bundle }));
+        expect(components.toolRegistry.names()).not.toContain('request');
+    });
+    it('registers present tool', () => {
+        const components = buildSessionComponents(makeOpts());
+        expect(components.toolRegistry.names()).toContain(PRESENT_TOOL_NAME);
+    });
+    it('registers stop_execution tool', () => {
+        const components = buildSessionComponents(makeOpts());
+        expect(components.toolRegistry.names()).toContain(STOP_EXECUTION_TOOL_NAME);
+    });
+    it('does NOT register shell_exec (dropped by design)', () => {
+        const components = buildSessionComponents(makeOpts());
+        expect(components.toolRegistry.names()).not.toContain('shell_exec');
+    });
+    it('does NOT register ask_user (dropped by design)', () => {
+        const components = buildSessionComponents(makeOpts());
+        expect(components.toolRegistry.names()).not.toContain('ask_user');
+    });
+    it('does NOT register activate_skill (dropped by design)', () => {
+        const components = buildSessionComponents(makeOpts());
+        expect(components.toolRegistry.names()).not.toContain('activate_skill');
+    });
+    it('does NOT register load_knowledge (dropped by design)', () => {
+        const components = buildSessionComponents(makeOpts());
+        expect(components.toolRegistry.names()).not.toContain('load_knowledge');
+    });
+    describe('custom tools', () => {
+        it('registers custom tools when executor provided', () => {
+            const bundle = makeBundle({ tools: [makeCustomTool()] });
+            const executor = { execute: vi.fn() };
+            const components = buildSessionComponents(makeOpts({ bundle, toolExecutor: executor }));
+            expect(components.toolRegistry.names()).toContain('analyze_data');
+        });
+        it('skips custom tools with confirm: never', () => {
+            const tool = makeCustomTool();
+            tool.confirm = 'never';
+            const bundle = makeBundle({ tools: [tool] });
+            const executor = { execute: vi.fn() };
+            const components = buildSessionComponents(makeOpts({ bundle, toolExecutor: executor }));
+            expect(components.toolRegistry.names()).not.toContain('analyze_data');
+        });
+        it('does not register custom tools without executor', () => {
+            const bundle = makeBundle({ tools: [makeCustomTool()] });
+            const components = buildSessionComponents(makeOpts({ bundle }));
+            expect(components.toolRegistry.names()).not.toContain('analyze_data');
+        });
+    });
+    describe('MCP tools', () => {
+        it('registers MCP tools when mcpManager provided', () => {
+            const mcpManager = {
+                getDiscoveredTools: vi.fn().mockReturnValue([
+                    {
+                        name: 'mcp-server__tool1',
+                        description: 'A tool',
+                        parameters: { type: 'object', properties: {} },
+                        serverName: 'mcp-server',
+                        originalName: 'tool1',
+                    },
+                ]),
+                callTool: vi.fn(),
+            };
+            const components = buildSessionComponents(makeOpts({ mcpManager: mcpManager }));
+            expect(components.toolRegistry.names()).toContain('mcp-server__tool1');
+        });
+    });
+    describe('admin sessions', () => {
+        it('uses admin skills/knowledge in system prompt', () => {
+            const adminContent = {
+                agentPrompt: 'You are the admin agent.',
+                skills: [{ name: 'manage-connections', description: 'Manage connections', body: 'Admin skill body', location: '/tmp' }],
+                knowledge: [{ name: 'admin-guide', title: 'Admin Guide', body: 'Admin knowledge body', location: '/tmp' }],
+            };
+            const components = buildSessionComponents(makeOpts({
+                sessionType: 'admin',
+                adminContent,
+                repoRoot: '/tmp/agent',
+            }));
+            // Admin skills should appear in prompt
+            expect(components.systemPrompt).toContain('manage-connections');
+            expect(components.systemPrompt).toContain('Admin skill body');
+            // User skills should NOT appear
+            expect(components.systemPrompt).not.toContain('revenue-report');
+        });
+        it('registers admin file tools', () => {
+            const adminContent = {
+                agentPrompt: null,
+                skills: [],
+                knowledge: [],
+            };
+            const components = buildSessionComponents(makeOpts({
+                sessionType: 'admin',
+                adminContent,
+                repoRoot: '/tmp/agent',
+            }));
+            const names = components.toolRegistry.names();
+            expect(names).toContain('read_repo_file');
+            expect(names).toContain('write_repo_file');
+            expect(names).toContain('delete_repo_file');
+            expect(names).toContain('internal_api');
+        });
+        it('does not register admin file tools for chat sessions', () => {
+            const components = buildSessionComponents(makeOpts());
+            const names = components.toolRegistry.names();
+            expect(names).not.toContain('read_repo_file');
+            expect(names).not.toContain('write_repo_file');
+            expect(names).not.toContain('delete_repo_file');
+        });
+    });
+    describe('pinned model', () => {
+        it('uses pinned model over bundle config', () => {
+            const components = buildSessionComponents(makeOpts({
+                pinnedModel: { provider: 'openai', model: 'gpt-4o' },
+            }));
+            expect(components.provider.model).toBe('gpt-4o');
+            expect(components.provider.provider).toBe('openai');
+        });
+    });
+    describe('system prompt (G9)', () => {
+        it('includes connection endpoints', () => {
+            const components = buildSessionComponents(makeOpts());
+            expect(components.systemPrompt).toContain('/items');
+        });
+        it('includes skill body', () => {
+            const components = buildSessionComponents(makeOpts());
+            expect(components.systemPrompt).toContain('Query charges');
+        });
+        it('includes knowledge body', () => {
+            const components = buildSessionComponents(makeOpts());
+            expect(components.systemPrompt).toContain('markdown tables');
+        });
+        it('includes userContext (G10)', () => {
+            const components = buildSessionComponents(makeOpts());
+            expect(components.systemPrompt).toContain('Be helpful and concise');
+        });
+    });
+    describe('stop_execution tool', () => {
+        it('returns __stop sentinel', async () => {
+            const components = buildSessionComponents(makeOpts());
+            const tool = components.toolRegistry.get(STOP_EXECUTION_TOOL_NAME);
+            expect(tool).toBeDefined();
+            const result = await tool.execute({ reason: 'Task complete' }, {});
+            expect(result).toEqual({ __stop: true, reason: 'Task complete' });
+        });
+    });
+    describe('present tool', () => {
+        it('returns rendered widget data', async () => {
+            const components = buildSessionComponents(makeOpts());
+            const tool = components.toolRegistry.get(PRESENT_TOOL_NAME);
+            expect(tool).toBeDefined();
+            const result = await tool.execute({ widget: 'entity-card', data: { name: 'Test' } }, {});
+            expect(result).toEqual({ widget: 'entity-card', data: { name: 'Test' }, rendered: true });
+        });
+    });
+    describe('tool context factory', () => {
+        it('factory produces ToolContext with correct session info', () => {
+            const components = buildSessionComponents(makeOpts({
+                sessionId: 'sess-123',
+                tenantId: 'tenant-456',
+                userRoles: ['analyst'],
+            }));
+            const ctx = components.toolContextFactory('call-1');
+            expect(ctx.sessionId).toBe('sess-123');
+            expect(ctx.tenantId).toBe('tenant-456');
+            expect(ctx.user.roles).toEqual(['analyst']);
+        });
+    });
+});
+//# sourceMappingURL=session-builder.test.js.map

package/dist/src/session/session-builder.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-builder.test.js","sourceRoot":"","sources":["../../../src/session/session-builder.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;;;;;GAYG;AAEH,OAAO,EAAC,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAC,MAAM,QAAQ,CAAC;AAChD,OAAO,EAAC,sBAAsB,EAAE,iBAAiB,EAAE,wBAAwB,EAAC,MAAM,sBAAsB,CAAC;AAIzG,OAAO,EAAC,YAAY,EAAC,MAAM,cAAc,CAAC;AAE1C,MAAM,MAAM,GAAG,YAAY,CAAC,EAAC,SAAS,EAAE,sBAAsB,EAAC,CAAC,CAAC;AAEjE,8EAA8E;AAC9E,QAAQ;AACR,8EAA8E;AAE9E,gEAAgE;AAChE,EAAE,CAAC,IAAI,CAAC,iCAAiC,EAAE,GAAG,EAAE,CAAC,CAAC;IAChD,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,MAAyC,EAAE,EAAE,CAAC,CAAC;QACpE,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,aAAa,EAAE,EAAE;QACjB,UAAU,EAAE,GAAG,EAAE,GAAG,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC;QACzD,YAAY,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;KACjE,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,8EAA8E;AAC9E,WAAW;AACX,8EAA8E;AAE9E,SAAS,cAAc,CAAC,IAAY;IAClC,OAAO;QACL,IAAI;QACJ,IAAI,EAAE;YACJ,QAAQ,EAAE,MAAe;YACzB,OAAO,EAAE,eAAe,IAAI,MAAM;YAClC,IAAI,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAC;SAC7C;QACD,MAAM,EAAE;YACN,SAAS,EAAE;gBACT,YAAY,EAAE,EAAC,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,EAAC;gBACvC,aAAa,EAAE,EAAC,OAAO,EAAE,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAC;aAChD;SACF;QACD,OAAO,EAAE;YACP,EAAC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,QAAQ,EAAE,IAAI,EAAC;YAC1E,EAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAC;SAC7E;QACD,QAAQ,EAAE,gBAAgB;QAC1B,KAAK,EAAE,iBAAiB;QACxB,QAAQ,EAAE,mBAAmB,GAAG,IAAI;KACrC,CAAC;AACJ,CAAC;AAED,SAAS,SAAS;IAChB,OAAO;QACL,IAAI,EAAE,OAAO;QACb,MAAM,EAAE;YACN,IAAI,EAAE,MAAM;YACZ,GAAG,EAAE,qBAAqB;YAC1B,MAAM,EAAE;gBACN,OAAO,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC;gBACzB,OAAO,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC;gBACzB,MAAM,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC;aACzB;SACF;QACD,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;AACJ,CAAC;AAED,SAAS,SAAS;IAChB,OAAO;QACL,IAAI,EAAE,gBAAgB;QACtB,WAAW,EAAE,yBAAyB;QACtC,OAAO,EAAE,8BAA8B;QACvC,IAAI,EAAE,2CAA2C;QACjD,QAAQ,EAAE,4BAA4B;KACvC,CAAC;AACJ,CAAC;AAED,SAAS,aAAa;IACpB,OAAO;QACL,IAAI,EAAE,aAAa;QACnB,KAAK,EAAE,aAAa;QACpB,IAAI,EAAE,6BAA6B;QACnC,QAAQ,EAAE,4BAA4B;KACvC,CAAC;AACJ,CAAC;AAED,SAAS,cAAc;IACrB,OAAO;QACL,IAAI,EAAE,cAAc;QACpB,WAAW,EAAE,cAAc;QAC3B,UAAU,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAC,KAAK,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC,EAAC,EAAE,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAC;QACxF,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,KAAK;QACd,GAAG,EAAE,CAAC,gBAAgB,CAAC;QACvB,WAAW,EAAE,+BAA+B;QAC5C,QAAQ,EAAE,oBAAoB;QAC9B,cAAc,EAAE,KAAK;QACrB,cAAc,EAAE,KAAK;QACrB,kBAAkB,EAAE,KAAK;QACzB,aAAa,EAAE,KAAK;QACpB,eAAe,EAAE,YAAY;KAC9B,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,SAAgC;IAClD,MAAM,WAAW,GAAG,IAAI,GAAG,EAA4B,CAAC;IACxD,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC;IAEpD,OAAO;QACL,MAAM,EAAE,OAAO;QACf,MAAM,EAAE,YAAY;QACpB,MAAM,EAAE;YACN,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,OAAO;YAChB,WAAW,EAAE,cAAc;YAC3B,WAAW,EAAE,yBAAyB;YACtC,MAAM,EAAE;gBACN,IAAI,EAAE,EAAC,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAC;aACjE;SACF;QACD,WAAW;QACX,MAAM,EAAE,CAAC,SAAS,EAAE,CAAC;QACrB,MAAM,EAAE,EAAC,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,EAAC;QAC3D,WAAW,EAAE,EAAE;QACf,SAAS,EAAE,CAAC,aAAa,EAAE,CAAC;QAC5B,KAAK,EAAE,EAAE;QACT,KAAK,EAAE,EAAE;QACT,MAAM,EAAE,CAAC,SAAS,EAAE,CAAC;QACrB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB;IACvB,OAAO;QACL,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAC,OAAO,EAAE,CAAC,EAAC,CAAC;QAC5C,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;QACZ,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;QACf,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;QAChB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;QACnB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;KACtB,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,SAAkD;IAClE,OAAO;QACL,MAAM,EAAE,UAAU,EAAE;QACpB,YAAY,EAAE,gBAAgB,EAAE;QAChC,UAAU,EAAE,IAAI;QAChB,MAAM;QACN,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,QAAQ;AACR,8EAA8E;AAE9E,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;QAEtD,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QAC1C,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9C,MAAM,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,WAAW,EAAE,CAAC;QACnD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,CAAC;QAC7C,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAC/D,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAC/B,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtD,MAAM,KAAK,GAAG,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAE9C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,MAAM,GAAG,UAAU,CAAC,EAAC,WAAW,EAAE,IAAI,GAAG,EAAE,EAAC,CAAC,CAAC;QACpD,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,CAAC,EAAC,MAAM,EAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;QAChC,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,MAAM,GAAG,UAAU,CAAC,EAAC,KAAK,EAAE,CAAC,cAAc,EAAE,CAAC,EAAC,CAAC,CAAC;YACvD,MAAM,QAAQ,GAAG,EAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAC,CAAC;YACpC,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,CAAC,EAAC,MAAM,EAAE,YAAY,EAAE,QAAQ,EAAC,CAAC,CAAC,CAAC;YAEtF,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,IAAI,GAAG,cAAc,EAAE,CAAC;YAC9B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;YACvB,MAAM,MAAM,GAAG,UAAU,CAAC,EAAC,KAAK,EAAE,CAAC,IAAI,CAAC,EAAC,CAAC,CAAC;YAC3C,MAAM,QAAQ,GAAG,EAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAC,CAAC;YACpC,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,CAAC,EAAC,MAAM,EAAE,YAAY,EAAE,QAAQ,EAAC,CAAC,CAAC,CAAC;YAEtF,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,MAAM,GAAG,UAAU,CAAC,EAAC,KAAK,EAAE,CAAC,cAAc,EAAE,CAAC,EAAC,CAAC,CAAC;YACvD,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,CAAC,EAAC,MAAM,EAAC,CAAC,CAAC,CAAC;YAE9D,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,UAAU,GAAG;gBACjB,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;oBAC1C;wBACE,IAAI,EAAE,mBAAmB;wBACzB,WAAW,EAAE,QAAQ;wBACrB,UAAU,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAC;wBAC5C,UAAU,EAAE,YAAY;wBACxB,YAAY,EAAE,OAAO;qBACtB;iBACF,CAAC;gBACF,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;aAClB,CAAC;YACF,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,CAAC,EAAC,UAAU,EAAE,UAAmB,EAAC,CAAC,CAAC,CAAC;YAEvF,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,YAAY,GAAsB;gBACtC,WAAW,EAAE,0BAA0B;gBACvC,MAAM,EAAE,CAAC,EAAC,IAAI,EAAE,oBAAoB,EAAE,WAAW,EAAE,oBAAoB,EAAE,IAAI,EAAE,kBAAkB,EAAE,QAAQ,EAAE,MAAM,EAAC,CAAC;gBACrH,SAAS,EAAE,CAAC,EAAC,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,sBAAsB,EAAE,QAAQ,EAAE,MAAM,EAAC,CAAC;aACzG,CAAC;YAEF,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,CAAC;gBACjD,WAAW,EAAE,OAAO;gBACpB,YAAY;gBACZ,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC,CAAC;YAEJ,uCAAuC;YACvC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAChE,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;YAC9D,gCAAgC;YAChC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,YAAY,GAAsB;gBACtC,WAAW,EAAE,IAAI;gBACjB,MAAM,EAAE,EAAE;gBACV,SAAS,EAAE,EAAE;aACd,CAAC;YAEF,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,CAAC;gBACjD,WAAW,EAAE,OAAO;gBACpB,YAAY;gBACZ,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC,CAAC;YAEJ,MAAM,KAAK,GAAG,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YAC9C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC1C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;YAC3C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;YAC5C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;YAEtD,MAAM,KAAK,GAAG,UAAU,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YAC9C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC9C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;YAC/C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,CAAC;gBACjD,WAAW,EAAE,EAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAC;aACnD,CAAC,CAAC,CAAC;YAEJ,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;YACtD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE;YAC7B,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;YACtD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;YACtD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;YACtD,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,UAAU,CAAC,YAAY,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;YACnE,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;YAE3B,MAAM,MAAM,GAAG,MAAM,IAAK,CAAC,OAAO,CAChC,EAAC,MAAM,EAAE,eAAe,EAAC,EACzB,EAAW,CACZ,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAC,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,eAAe,EAAC,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,UAAU,CAAC,YAAY,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;YAC5D,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;YAE3B,MAAM,MAAM,GAAG,MAAM,IAAK,CAAC,OAAO,CAChC,EAAC,MAAM,EAAE,aAAa,EAAE,IAAI,EAAE,EAAC,IAAI,EAAE,MAAM,EAAC,EAAC,EAC7C,EAAW,CACZ,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAC,MAAM,EAAE,aAAa,EAAE,IAAI,EAAE,EAAC,IAAI,EAAE,MAAM,EAAC,EAAE,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,UAAU,GAAG,sBAAsB,CAAC,QAAQ,CAAC;gBACjD,SAAS,EAAE,UAAU;gBACrB,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,CAAC,SAAS,CAAC;aACvB,CAAC,CAAC,CAAC;YAEJ,MAAM,GAAG,GAAG,UAAU,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YACpD,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACvC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACxC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/session/store.d.ts

@@ -0,0 +1,57 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { PersistedSession } from './types.js';
+import type { Logger } from '../logger.js';
+/**
+ * Interface for session persistence backends.
+ *
+ * Implementations: PGLiteSessionStore (local dev), future Postgres
+ * (hosted runtime via shared Drizzle schema).
+ */
+export interface SessionStore {
+    /** Initialize the backing store (create tables, run migrations). */
+    initialize(): Promise<void>;
+    /** Save or update a session. */
+    save(session: PersistedSession): Promise<void>;
+    /** Load a session by ID. Returns null if not found. */
+    load(sessionId: string): Promise<PersistedSession | null>;
+    /** List sessions for a tenant, newest first. */
+    list(tenantId: string, opts?: {
+        limit?: number;
+    }): Promise<PersistedSession[]>;
+    /** Delete a session by ID. Returns true if deleted. */
+    delete(sessionId: string): Promise<boolean>;
+    /** Delete sessions not updated since `before`. Returns count deleted. */
+    cleanup(before: Date): Promise<number>;
+    /** Close the backing store. */
+    close(): Promise<void>;
+}
+/**
+ * PGLite-backed session store using Drizzle ORM.
+ *
+ * Runs an in-process WASM Postgres instance. Data is persisted to a
+ * configurable directory (default: in-memory).
+ */
+export declare class PGLiteSessionStore implements SessionStore {
+    private db;
+    private pglite;
+    private readonly dataDir;
+    private readonly logger;
+    constructor(opts: {
+        dataDir?: string;
+        logger: Logger;
+    });
+    initialize(): Promise<void>;
+    save(session: PersistedSession): Promise<void>;
+    load(sessionId: string): Promise<PersistedSession | null>;
+    list(tenantId: string, opts?: {
+        limit?: number;
+    }): Promise<PersistedSession[]>;
+    delete(sessionId: string): Promise<boolean>;
+    cleanup(before: Date): Promise<number>;
+    close(): Promise<void>;
+    private ensureDb;
+}

package/dist/src/session/store.js

@@ -0,0 +1,167 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * Session persistence layer.
+ *
+ * Defines the `SessionStore` interface and a PGLite implementation
+ * backed by Drizzle ORM. The Drizzle schema is in `stores/schema.ts`,
+ * shared with the store document tables.
+ */
+import { eq, lt, desc } from 'drizzle-orm';
+import { drizzle } from 'drizzle-orm/pglite';
+import { agentSessions } from '../stores/schema.js';
+import { ConfigError } from '../errors.js';
+// ---------------------------------------------------------------------------
+// PGLite implementation
+// ---------------------------------------------------------------------------
+/**
+ * PGLite-backed session store using Drizzle ORM.
+ *
+ * Runs an in-process WASM Postgres instance. Data is persisted to a
+ * configurable directory (default: in-memory).
+ */
+export class PGLiteSessionStore {
+    db = null;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any -- PGLite instance type is not exported cleanly
+    pglite = null;
+    dataDir;
+    logger;
+    constructor(opts) {
+        this.dataDir = opts.dataDir;
+        this.logger = opts.logger;
+    }
+    async initialize() {
+        if (this.db)
+            return;
+        if (this.dataDir) {
+            const { mkdirSync } = await import('node:fs');
+            mkdirSync(this.dataDir, { recursive: true });
+        }
+        const { PGlite } = await import('@electric-sql/pglite');
+        this.pglite = new PGlite(this.dataDir ?? undefined);
+        this.db = drizzle(this.pglite);
+        // Create table via raw SQL — Drizzle schema defines the shape,
+        // but we use raw DDL for initialization (no drizzle-kit in runtime).
+        await this.pglite.exec(`
+      CREATE TABLE IF NOT EXISTS agent_sessions (
+        id TEXT PRIMARY KEY,
+        tenant_id TEXT NOT NULL,
+        user_id TEXT NOT NULL,
+        messages JSONB NOT NULL,
+        token_usage JSONB NOT NULL,
+        metadata JSONB DEFAULT '{}',
+        version INTEGER NOT NULL DEFAULT 1,
+        created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+        updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
+      );
+
+      CREATE INDEX IF NOT EXISTS idx_agent_sessions_tenant
+        ON agent_sessions (tenant_id, updated_at DESC);
+    `);
+        this.logger.info('session_store_initialized', {
+            dataDir: this.dataDir ?? 'in-memory',
+        });
+    }
+    async save(session) {
+        this.ensureDb();
+        const values = {
+            id: session.id,
+            tenantId: session.tenantId,
+            userId: session.userId,
+            messages: session.messages,
+            tokenUsage: session.tokenUsage,
+            metadata: session.metadata,
+            version: session.version,
+            createdAt: session.createdAt,
+            updatedAt: session.updatedAt,
+        };
+        await this.db
+            .insert(agentSessions)
+            .values(values)
+            .onConflictDoUpdate({
+            target: agentSessions.id,
+            set: {
+                messages: values.messages,
+                tokenUsage: values.tokenUsage,
+                metadata: values.metadata,
+                updatedAt: values.updatedAt,
+            },
+        });
+    }
+    async load(sessionId) {
+        this.ensureDb();
+        const rows = await this.db
+            .select()
+            .from(agentSessions)
+            .where(eq(agentSessions.id, sessionId))
+            .limit(1);
+        if (rows.length === 0)
+            return null;
+        return rowToPersistedSession(rows[0]);
+    }
+    async list(tenantId, opts) {
+        this.ensureDb();
+        const rows = await this.db
+            .select()
+            .from(agentSessions)
+            .where(eq(agentSessions.tenantId, tenantId))
+            .orderBy(desc(agentSessions.updatedAt))
+            .limit(opts?.limit ?? 50);
+        return rows.map(rowToPersistedSession);
+    }
+    async delete(sessionId) {
+        this.ensureDb();
+        const result = await this.db
+            .delete(agentSessions)
+            .where(eq(agentSessions.id, sessionId))
+            .returning({ id: agentSessions.id });
+        return result.length > 0;
+    }
+    async cleanup(before) {
+        this.ensureDb();
+        const result = await this.db
+            .delete(agentSessions)
+            .where(lt(agentSessions.updatedAt, before))
+            .returning({ id: agentSessions.id });
+        if (result.length > 0) {
+            this.logger.info('session_store_cleanup', { deleted: result.length });
+        }
+        return result.length;
+    }
+    async close() {
+        if (this.pglite) {
+            await this.pglite.close();
+            this.pglite = null;
+            this.db = null;
+        }
+    }
+    ensureDb() {
+        if (!this.db) {
+            throw new ConfigError('PGLiteSessionStore not initialized — call initialize() first', {
+                key: 'sessionStore',
+                suggestion: 'Call initialize() before using the session store',
+            });
+        }
+    }
+}
+// ---------------------------------------------------------------------------
+// Row conversion
+// ---------------------------------------------------------------------------
+function rowToPersistedSession(row) {
+    return {
+        version: 1,
+        id: row.id,
+        tenantId: row.tenantId,
+        userId: row.userId,
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-type-assertion -- JSONB boundary: we wrote these values
+        messages: row.messages,
+        tokenUsage: row.tokenUsage,
+        metadata: (row.metadata ?? {}),
+        createdAt: row.createdAt,
+        updatedAt: row.updatedAt,
+    };
+}
+//# sourceMappingURL=store.js.map

package/dist/src/session/store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/session/store.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;GAMG;AAEH,OAAO,EAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAC,MAAM,aAAa,CAAC;AACzC,OAAO,EAAC,OAAO,EAAC,MAAM,oBAAoB,CAAC;AAC3C,OAAO,EAAC,aAAa,EAAC,MAAM,qBAAqB,CAAC;AAKlD,OAAO,EAAC,WAAW,EAAC,MAAM,cAAc,CAAC;AAmCzC,8EAA8E;AAC9E,wBAAwB;AACxB,8EAA8E;AAE9E;;;;;GAKG;AACH,MAAM,OAAO,kBAAkB;IACrB,EAAE,GAAsC,IAAI,CAAC;IACrD,8GAA8G;IACtG,MAAM,GAAQ,IAAI,CAAC;IACV,OAAO,CAAqB;IAC5B,MAAM,CAAS;IAEhC,YAAY,IAAwC;QAClD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,EAAE;YAAE,OAAO;QAEpB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,EAAC,SAAS,EAAC,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;YAC5C,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,EAAC,SAAS,EAAE,IAAI,EAAC,CAAC,CAAC;QAC7C,CAAC;QAED,MAAM,EAAC,MAAM,EAAC,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;QACtD,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,SAAS,CAAC,CAAC;QACpD,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE/B,+DAA+D;QAC/D,qEAAqE;QACrE,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;KAetB,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE;YAC5C,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,WAAW;SACrC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAyB;QAClC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEhB,MAAM,MAAM,GAAG;YACb,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,QAAQ,EAAE,OAAO,CAAC,QAAqB;YACvC,UAAU,EAAE,OAAO,CAAC,UAA8E;YAClG,QAAQ,EAAE,OAAO,CAAC,QAAmC;YACrD,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,SAAS,EAAE,OAAO,CAAC,SAAS;SAC7B,CAAC;QAEF,MAAM,IAAI,CAAC,EAAG;aACX,MAAM,CAAC,aAAa,CAAC;aACrB,MAAM,CAAC,MAAM,CAAC;aACd,kBAAkB,CAAC;YAClB,MAAM,EAAE,aAAa,CAAC,EAAE;YACxB,GAAG,EAAE;gBACH,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,SAAS,EAAE,MAAM,CAAC,SAAS;aAC5B;SACF,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAiB;QAC1B,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEhB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAG;aACxB,MAAM,EAAE;aACR,IAAI,CAAC,aAAa,CAAC;aACnB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;aACtC,KAAK,CAAC,CAAC,CAAC,CAAC;QAEZ,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACnC,OAAO,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE,IAAuB;QAClD,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEhB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAG;aACxB,MAAM,EAAE;aACR,IAAI,CAAC,aAAa,CAAC;aACnB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;aAC3C,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;aACtC,KAAK,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAE5B,OAAO,IAAI,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB;QAC5B,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEhB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAG;aAC1B,MAAM,CAAC,aAAa,CAAC;aACrB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;aACtC,SAAS,CAAC,EAAC,EAAE,EAAE,aAAa,CAAC,EAAE,EAAC,CAAC,CAAC;QAErC,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAY;QACxB,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEhB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAG;aAC1B,MAAM,CAAC,aAAa,CAAC;aACrB,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;aAC1C,SAAS,CAAC,EAAC,EAAE,EAAE,aAAa,CAAC,EAAE,EAAC,CAAC,CAAC;QAErC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAC,CAAC,CAAC;QACtE,CAAC;QAED,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;YACnB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,QAAQ;QACd,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YACb,MAAM,IAAI,WAAW,CAAC,8DAA8D,EAAE;gBACpF,GAAG,EAAE,cAAc;gBACnB,UAAU,EAAE,kDAAkD;aAC/D,CAAC,CAAC;QACL,CAAC;IACH,CAAC;CACF;AAED,8EAA8E;AAC9E,iBAAiB;AACjB,8EAA8E;AAE9E,SAAS,qBAAqB,CAAC,GAAsC;IACnE,OAAO;QACL,OAAO,EAAE,CAAC;QACV,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,MAAM,EAAE,GAAG,CAAC,MAAM;QAClB,gHAAgH;QAChH,QAAQ,EAAE,GAAG,CAAC,QAA0B;QAExC,UAAU,EAAE,GAAG,CAAC,UAAwB;QAExC,QAAQ,EAAE,CAAC,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAoB;QACjD,SAAS,EAAE,GAAG,CAAC,SAAS;QACxB,SAAS,EAAE,GAAG,CAAC,SAAS;KACzB,CAAC;AACJ,CAAC"}

package/dist/src/session/store.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+export {};