@amodalai/runtime 0.1.26 → 0.2.0

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

@@ -1,219 +0,0 @@
-/**
- * @license
- * Copyright 2025 Amodal Labs, Inc.
- * SPDX-License-Identifier: MIT
- */
-import { AmodalConfig, type AmodalConfigParameters, Scheduler, type GeminiClient, PlanModeManager, McpManager } from '@amodalai/core';
-import type { AgentBundle, CustomToolExecutor, CustomShellExecutor, StoreBackend } from '@amodalai/core';
-import type { AuthContext } from '../middleware/auth.js';
-export interface PendingAskUser {
-    resolve: (answers: Record<string, string>) => void;
-    reject: (reason: Error) => void;
-}
-export interface SessionMessage {
-    type: 'user' | 'assistant_text' | 'error';
-    id: string;
-    text: string;
-    timestamp: string;
-    toolCalls?: Array<Record<string, unknown>>;
-    skillActivations?: string[];
-    widgets?: Array<Record<string, unknown>>;
-    contentBlocks?: Array<Record<string, unknown>>;
-}
-export interface ManagedSession {
-    id: string;
-    config: AmodalConfig;
-    geminiClient: GeminiClient;
-    scheduler: Scheduler;
-    createdAt: number;
-    lastAccessedAt: number;
-    /** The org this session belongs to */
-    orgId?: string;
-    /** Session type — controls which skills, tools, KB docs load */
-    sessionType?: string;
-    /** Pending ask_user responses awaiting user input */
-    pendingAskUser: Map<string, PendingAskUser>;
-    /** Accumulated messages for session history persistence */
-    accumulatedMessages: SessionMessage[];
-    /** Model used for this session (pinned at creation, survives hydration) */
-    model?: string;
-    /** Provider used for this session (pinned at creation, survives hydration) */
-    provider?: string;
-    /** Store backend for cleanup on session destroy */
-    storeBackend?: StoreBackend;
-    /** Plan mode manager (local dev) */
-    planModeManager?: PlanModeManager;
-    /** MCP server manager (local dev) */
-    mcpManager?: McpManager;
-    /** Session title */
-    title?: string;
-    /** Custom tool executor (local dev) */
-    toolExecutor?: CustomToolExecutor;
-    /** Custom shell executor (local dev) */
-    shellExecutor?: CustomShellExecutor;
-    /** App ID for this session */
-    appId?: string;
-}
-/** Shape of a stored session record (from platform API or session store). */
-export interface StoredSessionRecord {
-    id: string;
-    app_id: string;
-    messages: SessionMessage[];
-    status: string;
-    model?: string;
-    provider?: string;
-}
-/**
- * Pluggable session store for loading stored session history.
- * Implementations are provided by the hosting layer.
- */
-export interface SessionStore {
-    /** Fetch a stored session record by ID. Returns null if not found.
-     *  Optional context provides auth info for implementations that need it. */
-    getSession(sessionId: string, context?: {
-        appId?: string;
-        token?: string;
-    }): Promise<StoredSessionRecord | null>;
-}
-export interface SessionManagerOptions {
-    /** Base config parameters to clone for each session */
-    baseParams: AmodalConfigParameters;
-    /** Session TTL in milliseconds (default 30 minutes) */
-    ttlMs?: number;
-    /** Cleanup interval in milliseconds (default 5 minutes) */
-    cleanupIntervalMs?: number;
-    /** AgentBundle for local dev mode — sessions initialized from bundle config instead of platform API */
-    bundle?: AgentBundle;
-    /** Custom tool executor (local dev) */
-    toolExecutor?: CustomToolExecutor;
-    /** Custom shell executor (local dev) */
-    shellExecutor?: CustomShellExecutor;
-    /** Shared store backend (local dev) */
-    storeBackend?: StoreBackend;
-    /** Pluggable session store for hydration (if not provided, falls back to platform API) */
-    sessionStore?: SessionStore;
-    /** Async callback that resolves an AgentBundle from a deploy ID (used by hosted runtime) */
-    bundleProvider?: (deployId: string, token?: string) => Promise<AgentBundle | null>;
-}
-/**
- * Manages per-request sessions: creates Config + GeminiClient + Scheduler
- * instances, tracks them by ID, and cleans up expired sessions.
- */
-export declare class SessionManager {
-    private readonly sessions;
-    private baseParams;
-    private readonly ttlMs;
-    private bundle?;
-    private readonly toolExecutor?;
-    private readonly shellExecutor?;
-    private readonly sharedStoreBackend?;
-    private readonly sessionStore?;
-    private readonly bundleProvider?;
-    private cleanupTimer;
-    /** Deduplicates concurrent hydration requests for the same conversation */
-    private readonly pendingHydrations;
-    /** Shared MCP manager for all sessions (lazy-initialized, reused) */
-    private sharedMcpManager?;
-    /** Persistent MCP manager for inspect operations (lazy-initialized) */
-    private inspectMcp?;
-    private inspectMcpInitialized;
-    constructor(options: SessionManagerOptions);
-    /**
-     * Create a new session with optional role override and auth context.
-     * When auth context is provided, the session is configured with
-     * the caller's API key and org/app context.
-     */
-    create(role?: string, auth?: AuthContext, sessionType?: string, pinnedModel?: {
-        provider: string;
-        model: string;
-    }, deployId?: string): Promise<ManagedSession>;
-    /**
-     * Get an existing session by ID, updating its last-accessed timestamp.
-     * Returns undefined if the session doesn't exist.
-     */
-    get(id: string): ManagedSession | undefined;
-    /**
-     * Hydrate a session from stored conversation history in the platform API.
-     *
-     * When a user sends a message to a session that expired from the in-memory
-     * cache, this method fetches the conversation from the DB, creates a fresh
-     * runtime session (new config, tools, KB, secrets), seeds its LLM history
-     * with the stored messages, and registers it under the original conversation ID.
-     *
-     * Returns null if hydration cannot proceed (no platform API, missing auth,
-     * fetch failure, or no stored messages).
-     */
-    hydrate(conversationId: string, role?: string, auth?: AuthContext, sessionType?: string): Promise<ManagedSession | null>;
-    private doHydrate;
-    /**
-     * Destroy a session, flushing its audit log.
-     */
-    destroy(id: string): Promise<void>;
-    /**
-     * Get the bundle (local dev mode).
-     */
-    getBundle(): AgentBundle | undefined;
-    /**
-     * Update the bundle for new sessions (hot reload).
-     * Existing sessions keep their old config.
-     */
-    updateBundle(bundle: AgentBundle): void;
-    /**
-     * Re-register a session under a different ID (e.g., restoring original ID on session restore).
-     */
-    reregister(session: ManagedSession, newId: string): void;
-    /**
-     * Create an admin session for the config chat.
-     * Uses admin agent skills/knowledge but the current bundle's connections/stores.
-     *
-     * Temporarily swaps bundle fields so create() builds the prompt with admin
-     * content, then restores the original bundle. This mirrors the old approach
-     * of building an adminRepo overlay.
-     */
-    createAdminSession(getPort?: () => number | null): Promise<ManagedSession>;
-    /**
-     * Get a persistent MCP manager for inspect/health operations.
-     * Lazy-initialized on first call, reused across requests.
-     */
-    getInspectMcpManager(): Promise<McpManager | undefined>;
-    /**
-     * Initialize MCP servers for a session from bundle connections.
-     */
-    /**
-     * Initialize the shared MCP manager (once, reused across sessions).
-     * Avoids reconnecting MCP servers for every eval/judge/admin session.
-     */
-    private initSharedMcp;
-    /**
-     * Build the list of upstream core tools to enable based on bundle config.
-     * Only tools relevant to the Amodal runtime are included.
-     */
-    private buildCoreToolsList;
-    /**
-     * Build MCP server configs from bundle connections.
-     */
-    private buildMcpConfigs;
-    /**
-     * Remove sessions that have been idle longer than the TTL.
-     */
-    cleanup(): Promise<number>;
-    /**
-     * Shutdown all sessions and stop the cleanup timer.
-     */
-    shutdown(): Promise<void>;
-    /**
-     * Wait for a user's response to an ask_user prompt.
-     * Resolves when the user submits answers via the HTTP endpoint.
-     * Rejects on timeout or if the signal is aborted.
-     */
-    waitForAskUserResponse(session: ManagedSession, askId: string, signal: AbortSignal): Promise<Record<string, string>>;
-    /**
-     * Resolve a pending ask_user with user-provided answers.
-     * Returns true if the ask_id was found and resolved, false otherwise.
-     */
-    resolveAskUser(session: ManagedSession, askId: string, answers: Record<string, string>): boolean;
-    /**
-     * Number of active sessions.
-     */
-    get size(): number;
-}