@amodalai/runtime 0.1.26 → 0.2.1

package/dist/src/session/manager.test.js

@@ -0,0 +1,309 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * Tests for StandaloneSessionManager.
+ *
+ * Tests session lifecycle: create, persist, resume, destroy, cleanup.
+ * Uses a real PGLite-backed store — no mocks for persistence.
+ * LLM provider is stubbed since we're testing the manager, not the loop.
+ */
+import { describe, it, expect, vi, beforeAll, afterAll } from 'vitest';
+import { StandaloneSessionManager } from './manager.js';
+import { createPGLiteSessionStore } from './pglite-session-store.js';
+import { SSEEventType } from '../types.js';
+import { createLogger } from '../logger.js';
+import { createToolRegistry } from '../tools/registry.js';
+const logger = createLogger({ component: 'test:session-manager' });
+// ---------------------------------------------------------------------------
+// Stubs
+// ---------------------------------------------------------------------------
+function stubProvider(model = 'test-model', provider = 'test-provider') {
+    return {
+        model,
+        provider,
+        languageModel: {},
+        streamText: () => {
+            throw new Error('streamText not implemented in stub');
+        },
+        generateText: () => Promise.reject(new Error('generateText not implemented in stub')),
+    };
+}
+/**
+ * Create a provider that streams a single text response and finishes.
+ */
+function streamingProvider(responseText) {
+    return {
+        model: 'test-model',
+        provider: 'test-provider',
+        languageModel: {},
+        streamText() {
+            const usage = { inputTokens: 10, outputTokens: 5, totalTokens: 15 };
+            const events = [
+                { type: 'text-delta', textDelta: responseText },
+                { type: 'finish', usage },
+            ];
+            async function* makeFullStream() {
+                for (const e of events)
+                    yield e;
+            }
+            async function* makeTextStream() {
+                yield responseText;
+            }
+            return {
+                fullStream: makeFullStream(),
+                textStream: makeTextStream(),
+                usage: Promise.resolve(usage),
+                text: Promise.resolve(responseText),
+            };
+        },
+        generateText: () => Promise.reject(new Error('not implemented')),
+    };
+}
+function stubPermissionChecker() {
+    return {
+        check: () => ({ allowed: true }),
+    };
+}
+function makeCreateOpts(overrides = {}) {
+    return {
+        provider: stubProvider(),
+        toolRegistry: createToolRegistry(),
+        permissionChecker: stubPermissionChecker(),
+        systemPrompt: 'You are a helpful assistant.',
+        ...overrides,
+    };
+}
+// ---------------------------------------------------------------------------
+// Tests
+// ---------------------------------------------------------------------------
+describe('StandaloneSessionManager', () => {
+    let store;
+    beforeAll(async () => {
+        store = await createPGLiteSessionStore({ logger });
+    });
+    afterAll(async () => {
+        await store.close();
+    });
+    describe('create', () => {
+        it('creates a session with a generated ID', () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts());
+            expect(session.id).toBeTruthy();
+            expect(session.model).toBe('test-model');
+            expect(session.providerName).toBe('test-provider');
+            expect(session.messages).toEqual([]);
+            expect(session.usage.inputTokens).toBe(0);
+        });
+        it('stores session in memory', () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts());
+            expect(mgr.has(session.id)).toBe(true);
+            expect(mgr.get(session.id)).toBe(session);
+            expect(mgr.size).toBe(1);
+        });
+        it('uses custom maxTurns and maxContextTokens', () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts({
+                maxTurns: 10,
+                maxContextTokens: 100_000,
+            }));
+            expect(session.maxTurns).toBe(10);
+            expect(session.maxContextTokens).toBe(100_000);
+        });
+        it('uses manager defaults when not specified', () => {
+            const mgr = new StandaloneSessionManager({
+                logger,
+                store,
+                defaultMaxTurns: 25,
+                defaultMaxContextTokens: 150_000,
+            });
+            const session = mgr.create(makeCreateOpts());
+            expect(session.maxTurns).toBe(25);
+            expect(session.maxContextTokens).toBe(150_000);
+        });
+    });
+    describe('persist and resume', () => {
+        it('persists a session and resumes it with preserved messages', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts());
+            // Simulate some messages
+            session.messages = [
+                { role: 'user', content: 'Hello' },
+                { role: 'assistant', content: 'Hi there!' },
+            ];
+            session.usage = { inputTokens: 100, outputTokens: 50, totalTokens: 150 };
+            await mgr.persist(session);
+            // Destroy and resume
+            await mgr.destroy(session.id);
+            expect(mgr.has(session.id)).toBe(false);
+            const resumed = await mgr.resume(session.id, makeCreateOpts({
+                systemPrompt: 'Updated system prompt',
+            }));
+            expect(resumed).not.toBeNull();
+            expect(resumed.id).toBe(session.id);
+            expect(resumed.messages).toHaveLength(2);
+            expect(resumed.usage.totalTokens).toBe(150);
+            // System prompt should be fresh, not stale
+            expect(resumed.systemPrompt).toBe('Updated system prompt');
+        });
+        it('resume returns null for missing session', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const result = await mgr.resume('nonexistent', makeCreateOpts());
+            expect(result).toBeNull();
+        });
+        it('version field is present on persisted sessions', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts());
+            await mgr.persist(session);
+            const persisted = await store.load(session.id);
+            expect(persisted.version).toBe(1);
+        });
+        it('deduplicates concurrent resume calls for the same session ID', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts());
+            session.messages = [{ role: 'user', content: 'test' }];
+            await mgr.persist(session);
+            await mgr.destroy(session.id);
+            // Spy on store.load to verify it's called once
+            const loadSpy = vi.spyOn(store, 'load');
+            // Fire two concurrent resumes for the same session
+            const opts = makeCreateOpts({ systemPrompt: 'fresh' });
+            const [result1, result2] = await Promise.all([
+                mgr.resume(session.id, opts),
+                mgr.resume(session.id, opts),
+            ]);
+            // Both should resolve to the same session (same ID)
+            expect(result1).not.toBeNull();
+            expect(result2).not.toBeNull();
+            expect(result1.id).toBe(session.id);
+            expect(result2.id).toBe(session.id);
+            // Store should only be hit once (deduplication)
+            expect(loadSpy).toHaveBeenCalledTimes(1);
+            loadSpy.mockRestore();
+        });
+    });
+    describe('list persisted', () => {
+        it('lists persisted sessions newest first', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const s1 = mgr.create(makeCreateOpts());
+            const s2 = mgr.create(makeCreateOpts());
+            await mgr.persist(s1);
+            await mgr.persist(s2);
+            const list = await mgr.listPersisted();
+            // Should include at least the two we just persisted
+            const ids = list.map((s) => s.id);
+            expect(ids).toContain(s1.id);
+            expect(ids).toContain(s2.id);
+        });
+    });
+    describe('destroy', () => {
+        it('removes session from memory', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts());
+            await mgr.destroy(session.id);
+            expect(mgr.has(session.id)).toBe(false);
+            expect(mgr.size).toBe(0);
+        });
+        it('optionally deletes from store', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts());
+            await mgr.persist(session);
+            await mgr.destroy(session.id, { deleteFromStore: true });
+            const loaded = await store.load(session.id);
+            expect(loaded).toBeNull();
+        });
+    });
+    describe('cleanup', () => {
+        it('destroys sessions idle beyond TTL', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store, ttlMs: 1 });
+            const session = mgr.create(makeCreateOpts());
+            // Force old lastAccessedAt
+            session.lastAccessedAt = Date.now() - 1000;
+            const destroyed = await mgr.cleanup();
+            expect(destroyed).toBeGreaterThanOrEqual(1);
+            expect(mgr.has(session.id)).toBe(false);
+        });
+        it('does not destroy recently accessed sessions', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store, ttlMs: 60_000 });
+            const session = mgr.create(makeCreateOpts());
+            const destroyed = await mgr.cleanup();
+            expect(destroyed).toBe(0);
+            expect(mgr.has(session.id)).toBe(true);
+        });
+    });
+    describe('shutdown', () => {
+        it('destroys all sessions and closes store', async () => {
+            const shutdownStore = await createPGLiteSessionStore({ logger });
+            const mgr = new StandaloneSessionManager({ logger, store: shutdownStore });
+            mgr.start();
+            mgr.create(makeCreateOpts());
+            mgr.create(makeCreateOpts());
+            expect(mgr.size).toBe(2);
+            await mgr.shutdown();
+            expect(mgr.size).toBe(0);
+        });
+    });
+    describe('runMessage', () => {
+        it('executes a message through the agent loop and yields SSE events', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts({
+                provider: streamingProvider('Hello from the agent!'),
+            }));
+            const events = [];
+            for await (const event of mgr.runMessage(session.id, 'Hi there')) {
+                events.push(event);
+            }
+            // Should have init, text delta, and done events
+            const types = events.map((e) => e.type);
+            expect(types).toContain(SSEEventType.Init);
+            expect(types).toContain(SSEEventType.TextDelta);
+            expect(types).toContain(SSEEventType.Done);
+        });
+        it('persists session after execution', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts({
+                provider: streamingProvider('Persisted response'),
+            }));
+            // Drain the generator
+            for await (const _event of mgr.runMessage(session.id, 'Save me')) {
+                // consume
+            }
+            // Session should be persisted
+            const persisted = await store.load(session.id);
+            expect(persisted).not.toBeNull();
+            expect(persisted.messages.length).toBeGreaterThan(0);
+        });
+        it('fires onUsage callback with token counts', async () => {
+            const usageEvents = [];
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts({
+                provider: streamingProvider('Usage test'),
+            }));
+            for await (const _event of mgr.runMessage(session.id, 'Count tokens', {
+                onUsage: (usage) => usageEvents.push(usage),
+            })) {
+                // consume
+            }
+            expect(usageEvents.length).toBe(1);
+            expect(usageEvents[0].inputTokens).toBe(10);
+            expect(usageEvents[0].outputTokens).toBe(5);
+            expect(usageEvents[0].turnNumber).toBe(1);
+        });
+        it('syncs messages back to session after execution', async () => {
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const session = mgr.create(makeCreateOpts({
+                provider: streamingProvider('Response text'),
+            }));
+            expect(session.messages).toHaveLength(0);
+            for await (const _event of mgr.runMessage(session.id, 'Hello')) {
+                // consume
+            }
+            // Should have user + assistant messages
+            expect(session.messages.length).toBeGreaterThanOrEqual(2);
+        });
+    });
+});
+//# sourceMappingURL=manager.test.js.map

package/dist/src/session/manager.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"manager.test.js","sourceRoot":"","sources":["../../../src/session/manager.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;GAMG;AAEH,OAAO,EAAC,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAC,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAC,wBAAwB,EAAC,MAAM,cAAc,CAAC;AACtD,OAAO,EAAC,wBAAwB,EAAC,MAAM,2BAA2B,CAAC;AAInE,OAAO,EAAC,YAAY,EAAC,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAC,YAAY,EAAC,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAC,kBAAkB,EAAC,MAAM,sBAAsB,CAAC;AAExD,MAAM,MAAM,GAAG,YAAY,CAAC,EAAC,SAAS,EAAE,sBAAsB,EAAC,CAAC,CAAC;AAEjE,8EAA8E;AAC9E,QAAQ;AACR,8EAA8E;AAE9E,SAAS,YAAY,CAAC,KAAK,GAAG,YAAY,EAAE,QAAQ,GAAG,eAAe;IACpE,OAAO;QACL,KAAK;QACL,QAAQ;QAER,aAAa,EAAE,EAAkC;QACjD,UAAU,EAAE,GAAG,EAAE;YACf,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACxD,CAAC;QACD,YAAY,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;KACtF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,YAAoB;IAC7C,OAAO;QACL,KAAK,EAAE,YAAY;QACnB,QAAQ,EAAE,eAAe;QAEzB,aAAa,EAAE,EAAkC;QACjD,UAAU;YACR,MAAM,KAAK,GAAe,EAAC,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAC,CAAC;YAC9E,MAAM,MAAM,GAAkB;gBAC5B,EAAC,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,YAAY,EAAC;gBAC7C,EAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAC;aACxB,CAAC;YAEF,KAAK,SAAS,CAAC,CAAC,cAAc;gBAC5B,KAAK,MAAM,CAAC,IAAI,MAAM;oBAAE,MAAM,CAAC,CAAC;YAClC,CAAC;YACD,KAAK,SAAS,CAAC,CAAC,cAAc;gBAC5B,MAAM,YAAY,CAAC;YACrB,CAAC;YAED,OAAO;gBACL,UAAU,EAAE,cAAc,EAAE;gBAC5B,UAAU,EAAE,cAAc,EAAE;gBAC5B,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC;gBAC7B,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC;aACpC,CAAC;QACJ,CAAC;QACD,YAAY,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;KACjE,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB;IAC5B,OAAO;QACL,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,EAAC,OAAO,EAAE,IAAa,EAAC,CAAC;KACxC,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,YAA2C,EAAE;IACnE,OAAO;QACL,QAAQ,EAAE,YAAY,EAAE;QACxB,YAAY,EAAE,kBAAkB,EAAE;QAClC,iBAAiB,EAAE,qBAAqB,EAAE;QAC1C,YAAY,EAAE,8BAA8B;QAC5C,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,QAAQ;AACR,8EAA8E;AAE9E,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,KAA0B,CAAC;IAE/B,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,KAAK,GAAG,MAAM,wBAAwB,CAAC,EAAC,MAAM,EAAC,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAE7C,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,UAAU,EAAE,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACzC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACrC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAE7C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC1C,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC;gBACxC,QAAQ,EAAE,EAAE;gBACZ,gBAAgB,EAAE,OAAO;aAC1B,CAAC,CAAC,CAAC;YAEJ,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAClC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC;gBACvC,MAAM;gBACN,KAAK;gBACL,eAAe,EAAE,EAAE;gBACnB,uBAAuB,EAAE,OAAO;aACjC,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAE7C,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAClC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAE7C,yBAAyB;YACzB,OAAO,CAAC,QAAQ,GAAG;gBACjB,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAC;gBAChC,EAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAC;aAC1C,CAAC;YACF,OAAO,CAAC,KAAK,GAAG,EAAC,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,EAAC,CAAC;YAEvE,MAAM,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE3B,qBAAqB;YACrB,MAAM,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC9B,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAExC,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,EAAE,cAAc,CAAC;gBAC1D,YAAY,EAAE,uBAAuB;aACtC,CAAC,CAAC,CAAC;YAEJ,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACrC,MAAM,CAAC,OAAQ,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC7C,2CAA2C;YAC3C,MAAM,CAAC,OAAQ,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,aAAa,EAAE,cAAc,EAAE,CAAC,CAAC;YACjE,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAC7C,MAAM,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE3B,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC/C,MAAM,CAAC,SAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAC7C,OAAO,CAAC,QAAQ,GAAG,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAC,CAAC,CAAC;YACrD,MAAM,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC3B,MAAM,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAE9B,+CAA+C;YAC/C,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAExC,mDAAmD;YACnD,MAAM,IAAI,GAAG,cAAc,CAAC,EAAC,YAAY,EAAE,OAAO,EAAC,CAAC,CAAC;YACrD,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC3C,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,CAAC;gBAC5B,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,CAAC;aAC7B,CAAC,CAAC;YAEH,oDAAoD;YACpD,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACrC,MAAM,CAAC,OAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAErC,gDAAgD;YAChD,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAEzC,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAE1D,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YACxC,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAExC,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACtB,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAEtB,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,aAAa,EAAE,CAAC;YACvC,oDAAoD;YACpD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAClC,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7B,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAE7C,MAAM,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC9B,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAC7C,MAAM,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE3B,MAAM,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,EAAC,eAAe,EAAE,IAAI,EAAC,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAC,CAAC,CAAC;YAEpE,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAC7C,2BAA2B;YAC3B,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YAE3C,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,CAAC,SAAS,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAC,CAAC,CAAC;YACzE,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAE7C,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1B,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,aAAa,GAAG,MAAM,wBAAwB,CAAC,EAAC,MAAM,EAAC,CAAC,CAAC;YAE/D,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAE,aAAa,EAAC,CAAC,CAAC;YACzE,GAAG,CAAC,KAAK,EAAE,CAAC;YAEZ,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAC7B,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;YAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAEzB,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;YACrB,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC/E,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC;gBACxC,QAAQ,EAAE,iBAAiB,CAAC,uBAAuB,CAAC;aACrD,CAAC,CAAC,CAAC;YAEJ,MAAM,MAAM,GAAe,EAAE,CAAC;YAC9B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,UAAU,CAAC,EAAE,CAAC;gBACjE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YAED,gDAAgD;YAChD,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACxC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;YAChD,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC;gBACxC,QAAQ,EAAE,iBAAiB,CAAC,oBAAoB,CAAC;aAClD,CAAC,CAAC,CAAC;YAEJ,sBAAsB;YACtB,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,CAAC,EAAE,CAAC;gBACjE,UAAU;YACZ,CAAC;YAED,8BAA8B;YAC9B,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC/C,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,CAAC,SAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,WAAW,GAAgB,EAAE,CAAC;YACpC,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC;gBACxC,QAAQ,EAAE,iBAAiB,CAAC,YAAY,CAAC;aAC1C,CAAC,CAAC,CAAC;YAEJ,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,cAAc,EAAE;gBACpE,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC;aAC5C,CAAC,EAAE,CAAC;gBACH,UAAU;YACZ,CAAC;YAED,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACnC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC5C,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC;gBACxC,QAAQ,EAAE,iBAAiB,CAAC,eAAe,CAAC;aAC7C,CAAC,CAAC,CAAC;YAEJ,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAEzC,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC/D,UAAU;YACZ,CAAC;YAED,wCAAwC;YACxC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { Logger } from '../logger.js';
+import { DrizzleSessionStore } from './drizzle-session-store.js';
+import type { SessionStoreHooks } from './store.js';
+export interface PGLiteSessionStoreOptions {
+    /** Data directory for PGLite. If unset, runs fully in-memory. */
+    dataDir?: string;
+    /** Logger. Defaults to the runtime's global logger. */
+    logger?: Logger;
+    /** Optional hooks for dual-write / observability. */
+    hooks?: SessionStoreHooks;
+}
+/**
+ * Create and initialize a PGLite-backed session store.
+ *
+ * Returns a `DrizzleSessionStore` whose `close()` will close the
+ * underlying PGLite instance.
+ */
+export declare function createPGLiteSessionStore(opts?: PGLiteSessionStoreOptions): Promise<DrizzleSessionStore>;

package/dist/src/session/pglite-session-store.js

@@ -0,0 +1,70 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * PGLite factory for the session store.
+ *
+ * Opens an in-process WASM Postgres (PGLite), runs idempotent DDL,
+ * and returns a `DrizzleSessionStore` wired to it. Thin factory —
+ * all query logic lives in DrizzleSessionStore.
+ *
+ * Default choice for `amodal dev`: zero config, no Docker, no DB to
+ * manage. Pair with `createPostgresSessionStore` for hosted runtime
+ * or ISV production deployments.
+ */
+import { drizzle } from 'drizzle-orm/pglite';
+import { agentSessions } from '../stores/schema.js';
+import { log as defaultLogger } from '../logger.js';
+import { DrizzleSessionStore } from './drizzle-session-store.js';
+const BACKEND_NAME = 'pglite';
+// Must match the Drizzle schema in ../stores/schema.ts exactly.
+const CREATE_TABLE_DDL = `
+  CREATE TABLE IF NOT EXISTS agent_sessions (
+    id TEXT PRIMARY KEY,
+    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_updated
+    ON agent_sessions (updated_at DESC);
+`;
+/**
+ * Create and initialize a PGLite-backed session store.
+ *
+ * Returns a `DrizzleSessionStore` whose `close()` will close the
+ * underlying PGLite instance.
+ */
+export async function createPGLiteSessionStore(opts = {}) {
+    const logger = opts.logger ?? defaultLogger;
+    if (opts.dataDir) {
+        const { mkdirSync } = await import('node:fs');
+        mkdirSync(opts.dataDir, { recursive: true });
+    }
+    // Dynamic import keeps PGLite lazy — Postgres-only deployments don't
+    // need to pay the WASM parsing cost.
+    const { PGlite } = await import('@electric-sql/pglite');
+    const pglite = new PGlite(opts.dataDir ?? undefined);
+    await pglite.exec(CREATE_TABLE_DDL);
+    const db = drizzle(pglite);
+    logger.info('session_store_initialized', {
+        backend: BACKEND_NAME,
+        dataDir: opts.dataDir ?? 'in-memory',
+    });
+    return new DrizzleSessionStore({
+        db,
+        table: agentSessions,
+        backendName: BACKEND_NAME,
+        logger,
+        hooks: opts.hooks,
+        onClose: async () => {
+            await pglite.close();
+        },
+    });
+}
+//# sourceMappingURL=pglite-session-store.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"pglite-session-store.js","sourceRoot":"","sources":["../../../src/session/pglite-session-store.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;;;GAUG;AAEH,OAAO,EAAC,OAAO,EAAC,MAAM,oBAAoB,CAAC;AAE3C,OAAO,EAAC,aAAa,EAAC,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAC,GAAG,IAAI,aAAa,EAAC,MAAM,cAAc,CAAC;AAElD,OAAO,EAAC,mBAAmB,EAAC,MAAM,4BAA4B,CAAC;AAG/D,MAAM,YAAY,GAAG,QAAQ,CAAC;AAE9B,gEAAgE;AAChE,MAAM,gBAAgB,GAAG;;;;;;;;;;;;;CAaxB,CAAC;AAWF;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,OAAkC,EAAE;IAEpC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,aAAa,CAAC;IAE5C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,EAAC,SAAS,EAAC,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;QAC5C,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,EAAC,SAAS,EAAE,IAAI,EAAC,CAAC,CAAC;IAC7C,CAAC;IAED,qEAAqE;IACrE,qCAAqC;IACrC,MAAM,EAAC,MAAM,EAAC,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;IACtD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,SAAS,CAAC,CAAC;IACrD,MAAM,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAEpC,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAE3B,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE;QACvC,OAAO,EAAE,YAAY;QACrB,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,WAAW;KACrC,CAAC,CAAC;IAEH,OAAO,IAAI,mBAAmB,CAAC;QAC7B,EAAE;QACF,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE,YAAY;QACzB,MAAM;QACN,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACvB,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}

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

@@ -0,0 +1,44 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { Pool } from 'pg';
+import type { Logger } from '../logger.js';
+import { DrizzleSessionStore } from './drizzle-session-store.js';
+import type { SessionStoreHooks } from './store.js';
+export interface PostgresSessionStoreOptions {
+    /** Postgres connection string. Required if `pool` is not provided. */
+    connectionString?: string;
+    /** Existing pg.Pool to reuse instead of opening a new one. */
+    pool?: Pool;
+    /** Pool size when opening a new pool (default 10). Ignored if `pool` is passed. */
+    max?: number;
+    /** Per-statement timeout in ms (default 30_000). Ignored if `pool` is passed. */
+    statementTimeoutMs?: number;
+    /**
+     * Custom table name (default `agent_sessions`). Lets consumers run
+     * this store in a DB that already uses that name for something else.
+     */
+    tableName?: string;
+    /** Logger. Defaults to the runtime's global logger. */
+    logger?: Logger;
+    /** Optional hooks for dual-write / observability. */
+    hooks?: SessionStoreHooks;
+    /**
+     * When true, `close()` will not `pool.end()` — use when the caller
+     * owns the pool lifecycle. Defaults to true iff `pool` was passed
+     * (caller-owned), false otherwise (store-owned).
+     */
+    ownsPool?: boolean;
+}
+/**
+ * Create and initialize a Postgres-backed session store.
+ *
+ * Accepts either a connection string (shorthand) or full options
+ * object. Validates `tableName` at construct time, opens (or adopts)
+ * a `pg.Pool`, runs idempotent DDL, and returns a DrizzleSessionStore.
+ * The returned store's `close()` will `pool.end()` unless
+ * `ownsPool: false` (set automatically when a pool is passed in).
+ */
+export declare function createPostgresSessionStore(optsOrUrl: PostgresSessionStoreOptions | string): Promise<DrizzleSessionStore>;

package/dist/src/session/postgres-session-store.js

@@ -0,0 +1,138 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * Postgres factory for the session store.
+ *
+ * Opens a `pg.Pool` (or reuses one passed in), runs idempotent DDL,
+ * and returns a `DrizzleSessionStore` wired to it. Thin factory —
+ * all query logic lives in DrizzleSessionStore.
+ *
+ * For hosted runtime and ISV production deployments. Pair with
+ * `createPGLiteSessionStore` for `amodal dev`.
+ */
+import { drizzle } from 'drizzle-orm/node-postgres';
+import { pgTable, text, integer, jsonb, timestamp } from 'drizzle-orm/pg-core';
+import { log as defaultLogger } from '../logger.js';
+import { SessionStoreError } from '../errors.js';
+import { DrizzleSessionStore } from './drizzle-session-store.js';
+const BACKEND_NAME = 'postgres';
+const DEFAULT_STATEMENT_TIMEOUT_MS = 30_000;
+const DEFAULT_POOL_MAX = 10;
+const DEFAULT_TABLE_NAME = 'agent_sessions';
+// Reject table names that could be used for SQL injection. The name
+// is interpolated into DDL so strict validation is mandatory.
+const SAFE_TABLE_NAME_RE = /^[a-zA-Z_][a-zA-Z0-9_]{0,62}$/;
+function validateTableName(name) {
+    if (!SAFE_TABLE_NAME_RE.test(name)) {
+        throw new SessionStoreError(`Invalid table name: ${JSON.stringify(name)}`, {
+            backend: BACKEND_NAME,
+            operation: 'construct',
+            context: { tableName: name, expected: 'identifier: letters, digits, underscore' },
+        });
+    }
+}
+/**
+ * Build a Drizzle table binding for a given table name. Columns match
+ * the static `agentSessions` schema exactly; only the SQL table name
+ * differs. The return is cast to `AgentSessionsTable` (the type alias
+ * for `typeof agentSessions`) because the column shape is identical —
+ * Drizzle's inferred generic differs only in the `name` literal.
+ */
+function makeAgentSessionsTable(name) {
+    const t = pgTable(name, {
+        id: text('id').primaryKey(),
+        messages: jsonb('messages').notNull().$type(),
+        tokenUsage: jsonb('token_usage').notNull().$type(),
+        metadata: jsonb('metadata').default({}).$type(),
+        version: integer('version').notNull().default(1),
+        createdAt: timestamp('created_at', { withTimezone: true }).defaultNow().notNull(),
+        updatedAt: timestamp('updated_at', { withTimezone: true }).defaultNow().notNull(),
+    });
+    // eslint-disable-next-line @typescript-eslint/no-unsafe-type-assertion -- dynamic-name table has identical column shape as the static `agentSessions`; only the `name` generic parameter differs
+    return t;
+}
+function buildDdl(tableName) {
+    return `
+    CREATE TABLE IF NOT EXISTS ${tableName} (
+      id TEXT PRIMARY KEY,
+      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_${tableName}_updated
+      ON ${tableName} (updated_at DESC);
+  `;
+}
+/**
+ * Create and initialize a Postgres-backed session store.
+ *
+ * Accepts either a connection string (shorthand) or full options
+ * object. Validates `tableName` at construct time, opens (or adopts)
+ * a `pg.Pool`, runs idempotent DDL, and returns a DrizzleSessionStore.
+ * The returned store's `close()` will `pool.end()` unless
+ * `ownsPool: false` (set automatically when a pool is passed in).
+ */
+export async function createPostgresSessionStore(optsOrUrl) {
+    const opts = typeof optsOrUrl === 'string' ? { connectionString: optsOrUrl } : optsOrUrl;
+    if (!opts.connectionString && !opts.pool) {
+        throw new SessionStoreError('createPostgresSessionStore requires either connectionString or pool', { backend: BACKEND_NAME, operation: 'construct' });
+    }
+    const logger = opts.logger ?? defaultLogger;
+    const tableName = opts.tableName ?? DEFAULT_TABLE_NAME;
+    validateTableName(tableName);
+    const table = makeAgentSessionsTable(tableName);
+    const ownsPool = opts.ownsPool ?? !opts.pool;
+    let pool;
+    if (opts.pool) {
+        pool = opts.pool;
+    }
+    else {
+        // Dynamic import so `pg` stays an optional peer at the package
+        // level — PGLite users don't need it installed.
+        const pg = await import('pg');
+        const { Pool: PgPool } = pg.default ?? pg;
+        pool = new PgPool({
+            connectionString: opts.connectionString,
+            max: opts.max ?? DEFAULT_POOL_MAX,
+            statement_timeout: opts.statementTimeoutMs ?? DEFAULT_STATEMENT_TIMEOUT_MS,
+        });
+    }
+    try {
+        await pool.query(buildDdl(tableName));
+    }
+    catch (cause) {
+        throw new SessionStoreError('Failed to create session table', {
+            backend: BACKEND_NAME,
+            operation: 'initialize',
+            cause,
+            context: { tableName },
+        });
+    }
+    const db = drizzle(pool);
+    logger.info('session_store_initialized', {
+        backend: BACKEND_NAME,
+        tableName,
+        poolMax: opts.max ?? DEFAULT_POOL_MAX,
+        statementTimeoutMs: opts.statementTimeoutMs ?? DEFAULT_STATEMENT_TIMEOUT_MS,
+        ownsPool,
+    });
+    return new DrizzleSessionStore({
+        db,
+        table,
+        backendName: BACKEND_NAME,
+        logger,
+        hooks: opts.hooks,
+        onClose: async () => {
+            if (ownsPool)
+                await pool.end();
+        },
+    });
+}
+//# sourceMappingURL=postgres-session-store.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"postgres-session-store.js","sourceRoot":"","sources":["../../../src/session/postgres-session-store.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;;GASG;AAEH,OAAO,EAAC,OAAO,EAAC,MAAM,2BAA2B,CAAC;AAClD,OAAO,EAAC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAC,MAAM,qBAAqB,CAAC;AAG7E,OAAO,EAAC,GAAG,IAAI,aAAa,EAAC,MAAM,cAAc,CAAC;AAElD,OAAO,EAAC,iBAAiB,EAAC,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAC,mBAAmB,EAAC,MAAM,4BAA4B,CAAC;AAI/D,MAAM,YAAY,GAAG,UAAU,CAAC;AAChC,MAAM,4BAA4B,GAAG,MAAM,CAAC;AAC5C,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAC5B,MAAM,kBAAkB,GAAG,gBAAgB,CAAC;AA4B5C,oEAAoE;AACpE,8DAA8D;AAC9D,MAAM,kBAAkB,GAAG,+BAA+B,CAAC;AAE3D,SAAS,iBAAiB,CAAC,IAAY;IACrC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,iBAAiB,CAAC,uBAAuB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE;YACzE,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,WAAW;YACtB,OAAO,EAAE,EAAC,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,yCAAyC,EAAC;SAChF,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,SAAS,sBAAsB,CAAC,IAAY;IAC1C,MAAM,CAAC,GAAG,OAAO,CAAC,IAAI,EAAE;QACtB,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE;QAC3B,QAAQ,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC,KAAK,EAAa;QACxD,UAAU,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE,CAAC,KAAK,EAI5C;QACJ,QAAQ,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,KAAK,EAA2B;QACxE,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;QAChD,SAAS,EAAE,SAAS,CAAC,YAAY,EAAE,EAAC,YAAY,EAAE,IAAI,EAAC,CAAC,CAAC,UAAU,EAAE,CAAC,OAAO,EAAE;QAC/E,SAAS,EAAE,SAAS,CAAC,YAAY,EAAE,EAAC,YAAY,EAAE,IAAI,EAAC,CAAC,CAAC,UAAU,EAAE,CAAC,OAAO,EAAE;KAChF,CAAC,CAAC;IACH,iMAAiM;IACjM,OAAO,CAAkC,CAAC;AAC5C,CAAC;AAED,SAAS,QAAQ,CAAC,SAAiB;IACjC,OAAO;iCACwB,SAAS;;;;;;;;;;qCAUL,SAAS;WACnC,SAAS;GACjB,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,SAA+C;IAE/C,MAAM,IAAI,GACR,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAC,gBAAgB,EAAE,SAAS,EAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAE5E,IAAI,CAAC,IAAI,CAAC,gBAAgB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACzC,MAAM,IAAI,iBAAiB,CACzB,qEAAqE,EACrE,EAAC,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,WAAW,EAAC,CAChD,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,aAAa,CAAC;IAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,kBAAkB,CAAC;IACvD,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,sBAAsB,CAAC,SAAS,CAAC,CAAC;IAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;IAE7C,IAAI,IAAU,CAAC;IACf,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;SAAM,CAAC;QACN,+DAA+D;QAC/D,gDAAgD;QAChD,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,EAAC,IAAI,EAAE,MAAM,EAAC,GAAG,EAAE,CAAC,OAAO,IAAI,EAAE,CAAC;QACxC,IAAI,GAAG,IAAI,MAAM,CAAC;YAChB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,GAAG,EAAE,IAAI,CAAC,GAAG,IAAI,gBAAgB;YACjC,iBAAiB,EAAE,IAAI,CAAC,kBAAkB,IAAI,4BAA4B;SAC3E,CAAC,CAAC;IACL,CAAC;IAED,IAAI,CAAC;QACH,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IACxC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,iBAAiB,CAAC,gCAAgC,EAAE;YAC5D,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,KAAK;YACL,OAAO,EAAE,EAAC,SAAS,EAAC;SACrB,CAAC,CAAC;IACL,CAAC;IAED,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEzB,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE;QACvC,OAAO,EAAE,YAAY;QACrB,SAAS;QACT,OAAO,EAAE,IAAI,CAAC,GAAG,IAAI,gBAAgB;QACrC,kBAAkB,EAAE,IAAI,CAAC,kBAAkB,IAAI,4BAA4B;QAC3E,QAAQ;KACT,CAAC,CAAC;IAEH,OAAO,IAAI,mBAAmB,CAAC;QAC7B,EAAE;QACF,KAAK;QACL,WAAW,EAAE,YAAY;QACzB,MAAM;QACN,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,QAAQ;gBAAE,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}