@amodalai/runtime 0.1.25 → 0.2.0

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

@@ -0,0 +1,315 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * Tests for StandaloneSessionManager.
+ *
+ * Tests session lifecycle: create, persist, resume, destroy, cleanup.
+ * Uses a real PGLiteSessionStore — 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 { PGLiteSessionStore } from './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 {
+        tenantId: 'tenant-1',
+        userId: 'user-1',
+        provider: stubProvider(),
+        toolRegistry: createToolRegistry(),
+        permissionChecker: stubPermissionChecker(),
+        systemPrompt: 'You are a helpful assistant.',
+        ...overrides,
+    };
+}
+// ---------------------------------------------------------------------------
+// Tests
+// ---------------------------------------------------------------------------
+describe('StandaloneSessionManager', () => {
+    let store;
+    beforeAll(async () => {
+        store = new PGLiteSessionStore({ logger });
+        await store.initialize();
+    });
+    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.tenantId).toBe('tenant-1');
+            expect(session.userId).toBe('user-1');
+            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 sessions filtered by tenant', async () => {
+            const tenantId = `tenant-list-mgr-${Date.now()}`;
+            const mgr = new StandaloneSessionManager({ logger, store });
+            const s1 = mgr.create(makeCreateOpts({ tenantId }));
+            const s2 = mgr.create(makeCreateOpts({ tenantId }));
+            mgr.create(makeCreateOpts({ tenantId: 'other-tenant' }));
+            await mgr.persist(s1);
+            await mgr.persist(s2);
+            const list = await mgr.listPersisted(tenantId);
+            expect(list.length).toBeGreaterThanOrEqual(2);
+            expect(list.every((s) => s.tenantId === tenantId)).toBe(true);
+        });
+    });
+    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 = new PGLiteSessionStore({ logger });
+            await shutdownStore.initialize();
+            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,kBAAkB,EAAC,MAAM,YAAY,CAAC;AAG9C,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,UAAU;QACpB,MAAM,EAAE,QAAQ;QAChB,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,KAAyB,CAAC;IAE9B,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,KAAK,GAAG,IAAI,kBAAkB,CAAC,EAAC,MAAM,EAAC,CAAC,CAAC;QACzC,MAAM,KAAK,CAAC,UAAU,EAAE,CAAC;IAC3B,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,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACtC,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,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,QAAQ,GAAG,mBAAmB,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YACjD,MAAM,GAAG,GAAG,IAAI,wBAAwB,CAAC,EAAC,MAAM,EAAE,KAAK,EAAC,CAAC,CAAC;YAE1D,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,EAAC,QAAQ,EAAC,CAAC,CAAC,CAAC;YAClD,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,EAAC,QAAQ,EAAC,CAAC,CAAC,CAAC;YAClD,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,EAAC,QAAQ,EAAE,cAAc,EAAC,CAAC,CAAC,CAAC;YAEvD,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,CAAC,QAAQ,CAAC,CAAC;YAC/C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChE,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,IAAI,kBAAkB,CAAC,EAAC,MAAM,EAAC,CAAC,CAAC;YACvD,MAAM,aAAa,CAAC,UAAU,EAAE,CAAC;YAEjC,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/session-builder.d.ts

@@ -0,0 +1,71 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { AgentBundle, CustomToolExecutor, StoreBackend } from '@amodalai/types';
+import type { McpManager, AdminAgentContent, FieldScrubber } from '@amodalai/core';
+import type { LLMProvider } from '../providers/types.js';
+import type { ToolRegistry, ToolContext } from '../tools/types.js';
+import type { PermissionChecker } from '../security/permission-checker.js';
+import { DISPATCH_TOOL_NAME } from '../tools/dispatch-tool.js';
+import type { Logger } from '../logger.js';
+/** Session type determines which content and tools are available. */
+export type SessionType = 'chat' | 'admin' | 'automation';
+/** Components needed to create a session via StandaloneSessionManager. */
+export interface SessionComponents {
+    provider: LLMProvider;
+    toolRegistry: ToolRegistry;
+    permissionChecker: PermissionChecker;
+    systemPrompt: string;
+    toolContextFactory: (callId: string) => ToolContext;
+    userRoles: string[];
+}
+/** Options for building session components. */
+export interface BuildSessionComponentsOptions {
+    /** The agent bundle (repo config). */
+    bundle: AgentBundle;
+    /** Shared store backend (lifecycle managed by the server). */
+    storeBackend: StoreBackend | null;
+    /** Shared MCP manager (lifecycle managed by the server). */
+    mcpManager: McpManager | null;
+    /** Session type — determines which content and tools are available. */
+    sessionType?: SessionType;
+    /** User roles for permission checks and field guidance. */
+    userRoles?: string[];
+    /** Pinned model override (takes precedence over bundle config). */
+    pinnedModel?: {
+        provider: string;
+        model: string;
+    };
+    /** Logger instance. */
+    logger: Logger;
+    /** App ID for store isolation (default: 'local'). */
+    appId?: string;
+    /** Custom tool executor for running compiled tool handlers. */
+    toolExecutor?: CustomToolExecutor;
+    /** Admin agent content (required when sessionType is 'admin'). */
+    adminContent?: AdminAgentContent;
+    /** Repo root path (required when sessionType is 'admin' for file tools). */
+    repoRoot?: string;
+    /** Callback to get the server port (for admin internal_api tool). */
+    getPort?: () => number | null;
+    /** Session ID for correlation in tool context (default: generated). */
+    sessionId?: string;
+    /** Tenant ID for multi-tenant isolation. */
+    tenantId?: string;
+    /** Optional field scrubber for response sanitization on ctx.request() */
+    fieldScrubber?: FieldScrubber;
+}
+declare const PRESENT_TOOL_NAME = "present";
+declare const STOP_EXECUTION_TOOL_NAME = "stop_execution";
+/**
+ * Build all session components from a bundle.
+ *
+ * This is the single entry point for translating repo config into runtime
+ * components. The caller (route handler or session manager) provides the
+ * bundle and shared resources; the builder returns everything needed to
+ * create a session.
+ */
+export declare function buildSessionComponents(opts: BuildSessionComponentsOptions): SessionComponents;
+export { PRESENT_TOOL_NAME, STOP_EXECUTION_TOOL_NAME, DISPATCH_TOOL_NAME };