import { ExecutionNode } from "./checkpoint-types.js";
import { InputRequest, WorkflowExecutionContext } from "./workflow-context.js";
import { WorkflowMessageListener } from "./workflow-state.js";
type WorkflowContext = Record<symbol, unknown>;
export declare const CURRENT_NODE_SYMBOL: unique symbol;
export declare class ExecutionContext {
    context: WorkflowContext;
    private parent?;
    constructor(context: WorkflowContext, parent?: ExecutionContext | undefined, { messageListener, onRequestInput, onRestoreCheckpoint, checkpoint, executionScope, }?: {
        messageListener?: WorkflowMessageListener;
        onRequestInput?: (inputRequest: InputRequest) => Promise<void>;
        onRestoreCheckpoint?: (node: ExecutionNode, feedback: unknown) => Promise<void>;
        checkpoint?: ExecutionNode;
        executionScope?: Record<string, unknown>;
    });
    init(): Promise<void>;
    withContext(newContext: Partial<WorkflowContext>): ExecutionContext;
    get<K extends keyof WorkflowContext>(key: K): WorkflowContext[K] | undefined;
    getWorkflowContext(): WorkflowExecutionContext;
    getCurrentNode(): ExecutionNode | undefined;
    withCurrentNode<T>(node: ExecutionNode, fn: () => T): T;
}
export type RunInContext = <T>(fn: () => T) => T;
export declare function withContext<T>(context: ExecutionContext, fn: () => T): T;
export declare function getCurrentContext(): ExecutionContext;
export declare function getContextSnapshot(): RunInContext;
export declare function getCurrentNodeCheckpointManager(): {
    node: undefined;
    completeNode: () => void;
    updateNode: () => void;
    addMetadata: () => void;
} | {
    node: ExecutionNode;
    completeNode: (output: unknown, opts?: {
        wrapInPromise?: boolean;
    }) => void;
    updateNode: (updates: Partial<ExecutionNode>) => void;
    addMetadata: (metadata: Record<string, unknown>) => void;
};
export {};
//# sourceMappingURL=context.d.ts.map