@amodalai/runtime 0.1.26 → 0.2.1

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

@@ -0,0 +1,203 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * Shared Drizzle-ORM SessionStore implementation.
+ *
+ * One query layer used by both the PGLite factory (local dev,
+ * in-process WASM Postgres) and the Postgres factory (hosted runtime,
+ * real Postgres). Each factory constructs the underlying db client,
+ * runs DDL, and hands the result here along with a table binding and
+ * an `onClose` callback for teardown.
+ *
+ * Mirrors the `DrizzleStoreBackend` pattern introduced in amodal#146.
+ *
+ * Errors bubble up as SessionStoreError — callers at module edges
+ * (session manager, routes) are responsible for handling.
+ */
+import { eq, lt, desc } from 'drizzle-orm';
+import { SessionStoreError } from '../errors.js';
+import { buildListConditions, encodeCursor, rowToPersistedSession, sessionToRow, } from './store.js';
+const DEFAULT_LIST_LIMIT = 50;
+export class DrizzleSessionStore {
+    db;
+    table;
+    backendName;
+    logger;
+    hooks;
+    onClose;
+    closed = false;
+    constructor(opts) {
+        this.db = opts.db;
+        this.table = opts.table;
+        this.backendName = opts.backendName;
+        this.logger = opts.logger;
+        this.hooks = opts.hooks ?? {};
+        this.onClose = opts.onClose;
+    }
+    /**
+     * `initialize()` on this class is a no-op — the factory does the
+     * DDL and client setup before constructing the store. Kept for
+     * interface compliance.
+     */
+    async initialize() {
+        // Intentionally empty — initialization happens in the factories.
+    }
+    async save(session) {
+        this.ensureOpen('save');
+        const values = sessionToRow(session);
+        try {
+            await this.db
+                .insert(this.table)
+                .values(values)
+                .onConflictDoUpdate({
+                target: this.table.id,
+                set: {
+                    messages: values.messages,
+                    tokenUsage: values.tokenUsage,
+                    metadata: values.metadata,
+                    updatedAt: values.updatedAt,
+                },
+            });
+        }
+        catch (cause) {
+            throw new SessionStoreError('Failed to save session', {
+                backend: this.backendName,
+                operation: 'save',
+                cause,
+                context: { sessionId: session.id },
+            });
+        }
+        if (this.hooks.onAfterSave)
+            await this.hooks.onAfterSave(session);
+    }
+    async load(sessionId) {
+        this.ensureOpen('load');
+        try {
+            const rows = await this.db
+                .select()
+                .from(this.table)
+                .where(eq(this.table.id, sessionId))
+                .limit(1);
+            if (rows.length === 0)
+                return null;
+            return rowToPersistedSession(this.backendName, rows[0]);
+        }
+        catch (cause) {
+            if (cause instanceof SessionStoreError)
+                throw cause;
+            throw new SessionStoreError('Failed to load session', {
+                backend: this.backendName,
+                operation: 'load',
+                cause,
+                context: { sessionId },
+            });
+        }
+    }
+    async list(opts) {
+        this.ensureOpen('list');
+        const limit = opts?.limit ?? DEFAULT_LIST_LIMIT;
+        const where = buildListConditions(this.backendName, opts, this.table);
+        try {
+            const rows = await this.db
+                .select()
+                .from(this.table)
+                .where(where)
+                .orderBy(desc(this.table.updatedAt), desc(this.table.id))
+                .limit(limit + 1); // +1 to detect "is there a next page"
+            const hasMore = rows.length > limit;
+            const page = hasMore ? rows.slice(0, limit) : rows;
+            const sessions = page.map((r) => rowToPersistedSession(this.backendName, r));
+            const nextCursor = hasMore
+                ? encodeCursor(page[page.length - 1].updatedAt, page[page.length - 1].id)
+                : null;
+            return { sessions, nextCursor };
+        }
+        catch (cause) {
+            if (cause instanceof SessionStoreError)
+                throw cause;
+            throw new SessionStoreError('Failed to list sessions', {
+                backend: this.backendName,
+                operation: 'list',
+                cause,
+            });
+        }
+    }
+    async delete(sessionId) {
+        this.ensureOpen('delete');
+        let result;
+        try {
+            result = await this.db
+                .delete(this.table)
+                .where(eq(this.table.id, sessionId))
+                .returning({ id: this.table.id });
+        }
+        catch (cause) {
+            throw new SessionStoreError('Failed to delete session', {
+                backend: this.backendName,
+                operation: 'delete',
+                cause,
+                context: { sessionId },
+            });
+        }
+        const deleted = result.length > 0;
+        if (deleted && this.hooks.onAfterDelete) {
+            await this.hooks.onAfterDelete(sessionId);
+        }
+        return deleted;
+    }
+    async cleanup(before) {
+        this.ensureOpen('cleanup');
+        let result;
+        try {
+            result = await this.db
+                .delete(this.table)
+                .where(lt(this.table.updatedAt, before))
+                .returning({ id: this.table.id });
+        }
+        catch (cause) {
+            throw new SessionStoreError('Failed to cleanup sessions', {
+                backend: this.backendName,
+                operation: 'cleanup',
+                cause,
+                context: { before: before.toISOString() },
+            });
+        }
+        if (result.length > 0) {
+            this.logger.info('session_store_cleanup', {
+                backend: this.backendName,
+                deleted: result.length,
+            });
+        }
+        if (this.hooks.onAfterCleanup) {
+            await this.hooks.onAfterCleanup({ deleted: result.length, before });
+        }
+        return result.length;
+    }
+    async close() {
+        if (this.closed)
+            return;
+        this.closed = true;
+        try {
+            await this.onClose();
+        }
+        catch (cause) {
+            throw new SessionStoreError('Failed to close session store', {
+                backend: this.backendName,
+                operation: 'close',
+                cause,
+            });
+        }
+    }
+    ensureOpen(operation) {
+        if (this.closed) {
+            throw new SessionStoreError('Session store is closed', {
+                backend: this.backendName,
+                operation,
+            });
+        }
+    }
+}
+//# sourceMappingURL=drizzle-session-store.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"drizzle-session-store.js","sourceRoot":"","sources":["../../../src/session/drizzle-session-store.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;;;;;;GAaG;AAEH,OAAO,EAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAC,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAC,iBAAiB,EAAC,MAAM,cAAc,CAAC;AAI/C,OAAO,EACL,mBAAmB,EACnB,YAAY,EACZ,qBAAqB,EACrB,YAAY,GACb,MAAM,YAAY,CAAC;AAkBpB,MAAM,kBAAkB,GAAG,EAAE,CAAC;AAiB9B,MAAM,OAAO,mBAAmB;IACb,EAAE,CAAgB;IAClB,KAAK,CAAqB;IAClC,WAAW,CAAS;IACZ,MAAM,CAAS;IACf,KAAK,CAAoB;IACzB,OAAO,CAAsB;IACtC,MAAM,GAAG,KAAK,CAAC;IAEvB,YAAY,IAAgC;QAC1C,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC;QAClB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACxB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACpC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;QAC9B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;IAC9B,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,UAAU;QACd,iEAAiE;IACnE,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAyB;QAClC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACxB,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;QAErC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,EAAE;iBACV,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;iBAClB,MAAM,CAAC,MAAM,CAAC;iBACd,kBAAkB,CAAC;gBAClB,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE;gBACrB,GAAG,EAAE;oBACH,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,SAAS,EAAE,MAAM,CAAC,SAAS;iBAC5B;aACF,CAAC,CAAC;QACP,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,iBAAiB,CAAC,wBAAwB,EAAE;gBACpD,OAAO,EAAE,IAAI,CAAC,WAAW;gBACzB,SAAS,EAAE,MAAM;gBACjB,KAAK;gBACL,OAAO,EAAE,EAAC,SAAS,EAAE,OAAO,CAAC,EAAE,EAAC;aACjC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,CAAC,WAAW;YAAE,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAiB;QAC1B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAExB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;iBACvB,MAAM,EAAE;iBACR,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;iBAChB,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;iBACnC,KAAK,CAAC,CAAC,CAAC,CAAC;YAEZ,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC;YACnC,OAAO,qBAAqB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,iBAAiB;gBAAE,MAAM,KAAK,CAAC;YACpD,MAAM,IAAI,iBAAiB,CAAC,wBAAwB,EAAE;gBACpD,OAAO,EAAE,IAAI,CAAC,WAAW;gBACzB,SAAS,EAAE,MAAM;gBACjB,KAAK;gBACL,OAAO,EAAE,EAAC,SAAS,EAAC;aACrB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,IAAyB;QAClC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAExB,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK,IAAI,kBAAkB,CAAC;QAChD,MAAM,KAAK,GAAG,mBAAmB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAEtE,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;iBACvB,MAAM,EAAE;iBACR,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;iBAChB,KAAK,CAAC,KAAK,CAAC;iBACZ,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;iBACxD,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,sCAAsC;YAE3D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;YACpC,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;YAE7E,MAAM,UAAU,GAAG,OAAO;gBACxB,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzE,CAAC,CAAC,IAAI,CAAC;YAET,OAAO,EAAC,QAAQ,EAAE,UAAU,EAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,iBAAiB;gBAAE,MAAM,KAAK,CAAC;YACpD,MAAM,IAAI,iBAAiB,CAAC,yBAAyB,EAAE;gBACrD,OAAO,EAAE,IAAI,CAAC,WAAW;gBACzB,SAAS,EAAE,MAAM;gBACjB,KAAK;aACN,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB;QAC5B,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAE1B,IAAI,MAA2B,CAAC;QAChC,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE;iBACnB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;iBAClB,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;iBACnC,SAAS,CAAC,EAAC,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,EAAC,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,iBAAiB,CAAC,0BAA0B,EAAE;gBACtD,OAAO,EAAE,IAAI,CAAC,WAAW;gBACzB,SAAS,EAAE,QAAQ;gBACnB,KAAK;gBACL,OAAO,EAAE,EAAC,SAAS,EAAC;aACrB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;QAClC,IAAI,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAY;QACxB,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAE3B,IAAI,MAA2B,CAAC;QAChC,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE;iBACnB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;iBAClB,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;iBACvC,SAAS,CAAC,EAAC,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,EAAC,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,iBAAiB,CAAC,4BAA4B,EAAE;gBACxD,OAAO,EAAE,IAAI,CAAC,WAAW;gBACzB,SAAS,EAAE,SAAS;gBACpB,KAAK;gBACL,OAAO,EAAE,EAAC,MAAM,EAAE,MAAM,CAAC,WAAW,EAAE,EAAC;aACxC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE;gBACxC,OAAO,EAAE,IAAI,CAAC,WAAW;gBACzB,OAAO,EAAE,MAAM,CAAC,MAAM;aACvB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC;YAC9B,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,EAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAC,CAAC,CAAC;QACpE,CAAC;QACD,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,iBAAiB,CAAC,+BAA+B,EAAE;gBAC3D,OAAO,EAAE,IAAI,CAAC,WAAW;gBACzB,SAAS,EAAE,OAAO;gBAClB,KAAK;aACN,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,SAAiB;QAClC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,iBAAiB,CAAC,yBAAyB,EAAE;gBACrD,OAAO,EAAE,IAAI,CAAC,WAAW;gBACzB,SAAS;aACV,CAAC,CAAC;QACL,CAAC;IACH,CAAC;CACF"}

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

@@ -0,0 +1,101 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { AgentContext, AgentLoopConfig } from '../agent/loop-types.js';
+import type { SSEEvent } from '../types.js';
+import type { ToolContext } from '../tools/types.js';
+import type { SessionStore } from './store.js';
+import type { Session, SessionManagerOptions, CreateSessionOptions, PersistedSession, TurnUsage, AutomationResult } from './types.js';
+/**
+ * Standalone session manager.
+ *
+ * Manages the full session lifecycle: create, execute messages, persist,
+ * resume, and cleanup. Does not depend on gemini-cli-core.
+ */
+export declare class StandaloneSessionManager {
+    private readonly sessions;
+    private readonly pendingResumes;
+    private readonly store;
+    private readonly logger;
+    private readonly ttlMs;
+    private readonly cleanupIntervalMs;
+    private readonly defaultMaxTurns;
+    private readonly defaultMaxContextTokens;
+    private readonly defaultMaxSessionTokens;
+    private readonly eventBus;
+    private cleanupTimer;
+    constructor(opts: SessionManagerOptions & {
+        store?: SessionStore;
+    });
+    /**
+     * Start the cleanup timer. Call once at startup.
+     */
+    start(): void;
+    /**
+     * Gracefully shut down: destroy all sessions and stop cleanup.
+     */
+    shutdown(): Promise<void>;
+    /**
+     * Create a new session.
+     *
+     * The caller is responsible for building the provider, tool registry,
+     * permission checker, and system prompt (via the context compiler).
+     * This keeps the session manager focused on lifecycle, not orchestration.
+     */
+    create(opts: CreateSessionOptions): Session;
+    /**
+     * Run a message through the agent loop and yield SSE events.
+     *
+     * This is the main execution path. The caller iterates over the
+     * generator and sends events to the client (HTTP SSE, WebSocket, etc.).
+     */
+    runMessage(sessionId: string, userMessage: string, opts?: {
+        signal?: AbortSignal;
+        loopConfig?: Partial<AgentLoopConfig>;
+        onUsage?: (usage: TurnUsage) => void;
+        onAutomationResult?: (result: AutomationResult) => void;
+        waitForConfirmation?: (callId: string) => Promise<boolean>;
+        buildToolContext?: (callId: string) => ToolContext;
+        summarizeToolResult?: AgentContext['summarizeToolResult'];
+    }): AsyncGenerator<SSEEvent>;
+    /**
+     * Persist a session to the backing store.
+     */
+    persist(session: Session): Promise<void>;
+    /**
+     * Resume a session from the backing store.
+     *
+     * Loads persisted messages and token usage, then creates a new live
+     * session with a fresh provider/registry/prompt. The system prompt is
+     * recompiled (not stale) — the caller provides the current prompt.
+     *
+     * Concurrent resume calls for the same session ID are deduplicated —
+     * only one store fetch runs, and all callers share the result.
+     */
+    resume(sessionId: string, opts: CreateSessionOptions): Promise<Session | null>;
+    private doResume;
+    /** Get a session by ID. Returns undefined if not found. */
+    get(sessionId: string): Session | undefined;
+    /** Check if a session exists in memory. */
+    has(sessionId: string): boolean;
+    /** List persisted sessions from the backing store, newest first. */
+    listPersisted(opts?: {
+        limit?: number;
+    }): Promise<PersistedSession[]>;
+    /** Number of active in-memory sessions. */
+    get size(): number;
+    /**
+     * Destroy a session. Removes from memory and optionally from store.
+     */
+    destroy(sessionId: string, opts?: {
+        deleteFromStore?: boolean;
+    }): Promise<void>;
+    /**
+     * Clean up expired sessions (idle > TTL).
+     * Returns the number of sessions destroyed.
+     */
+    cleanup(): Promise<number>;
+    private getOrThrow;
+}