@ax-llm/ax 12.0.7 → 12.0.8

package/index.d.ts

@@ -2328,7 +2328,9 @@ type AxFieldValue = string | string[] | number | boolean | object | null | undef
 type AxGenIn = {
     [key: string]: AxFieldValue;
 };
-type AxGenOut = Record<string, AxFieldValue>;
+type AxGenOut = {
+    [key: string]: AxFieldValue;
+};
 type AxMessage<IN extends AxGenIn> = {
     role: 'user';
     values: IN;
@@ -2386,15 +2388,15 @@ declare class AxPromptTemplate {
     private defaultRenderInField;
 }
 
-type AxProgramTrace = {
-    trace: Record<string, AxFieldValue>;
+type AxProgramTrace<IN extends AxGenIn, OUT extends AxGenOut> = {
+    trace: OUT & IN;
     programId: string;
 };
-type AxProgramDemos = {
-    traces: Record<string, AxFieldValue>[];
+type AxProgramDemos<IN extends AxGenIn, OUT extends AxGenOut> = {
+    traces: (OUT & IN)[];
     programId: string;
 };
-type AxProgramExamples = AxProgramDemos | AxProgramDemos['traces'];
+type AxProgramExamples<IN extends AxGenIn, OUT extends AxGenOut> = AxProgramDemos<IN, OUT> | AxProgramDemos<IN, OUT>['traces'];
 type AxProgramForwardOptions = {
     maxRetries?: number;
     maxSteps?: number;
@@ -2432,12 +2434,12 @@ type AxGenDeltaOut<OUT extends AxGenOut> = {
 };
 type AxGenStreamingOut<OUT extends AxGenOut> = AsyncGenerator<AxGenDeltaOut<OUT>, void | OUT, unknown>;
 type AxSetExamplesOptions = {};
-interface AxTunable {
-    setExamples: (examples: Readonly<AxProgramExamples>, options?: Readonly<AxSetExamplesOptions>) => void;
+interface AxTunable<IN extends AxGenIn, OUT extends AxGenOut> {
+    setExamples: (examples: Readonly<AxProgramExamples<IN, OUT>>, options?: Readonly<AxSetExamplesOptions>) => void;
     setId: (id: string) => void;
     setParentId: (parentId: string) => void;
-    getTraces: () => AxProgramTrace[];
-    setDemos: (demos: readonly AxProgramDemos[]) => void;
+    getTraces: () => AxProgramTrace<IN, OUT>[];
+    setDemos: (demos: readonly AxProgramDemos<IN, OUT>[]) => void;
 }
 interface AxUsable {
     getUsage: () => AxProgramUsage[];
@@ -2450,53 +2452,53 @@ type AxProgramUsage = AxChatResponse['modelUsage'] & {
 interface AxProgramWithSignatureOptions {
     description?: string;
 }
-declare class AxProgramWithSignature<IN extends AxGenIn, OUT extends AxGenOut> implements AxTunable, AxUsable {
+declare class AxProgramWithSignature<IN extends AxGenIn, OUT extends AxGenOut> implements AxTunable<IN, OUT>, AxUsable {
     protected signature: AxSignature;
     protected sigHash: string;
-    protected examples?: Record<string, AxFieldValue>[];
+    protected examples?: OUT[];
     protected examplesOptions?: AxSetExamplesOptions;
-    protected demos?: Record<string, AxFieldValue>[];
-    protected trace?: Record<string, AxFieldValue>;
+    protected demos?: OUT[];
+    protected trace?: OUT;
     protected usage: AxProgramUsage[];
     private key;
     private children;
     constructor(signature: NonNullable<ConstructorParameters<typeof AxSignature>[0]>, options?: Readonly<AxProgramWithSignatureOptions>);
     getSignature(): AxSignature;
-    register(prog: Readonly<AxTunable & AxUsable>): void;
+    register(prog: Readonly<AxTunable<IN, OUT> & AxUsable>): void;
     forward(_ai: Readonly<AxAIService>, _values: IN | AxMessage<IN>[], _options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
     streamingForward(_ai: Readonly<AxAIService>, _values: IN | AxMessage<IN>[], _options?: Readonly<AxProgramStreamingForwardOptions>): AxGenStreamingOut<OUT>;
     setId(id: string): void;
     setParentId(parentId: string): void;
-    setExamples(examples: Readonly<AxProgramExamples>, options?: Readonly<AxSetExamplesOptions>): void;
+    setExamples(examples: Readonly<AxProgramExamples<IN, OUT>>, options?: Readonly<AxSetExamplesOptions>): void;
     private _setExamples;
-    getTraces(): AxProgramTrace[];
+    getTraces(): AxProgramTrace<IN, OUT>[];
     getUsage(): AxProgramUsage[];
     resetUsage(): void;
-    setDemos(demos: readonly AxProgramDemos[]): void;
+    setDemos(demos: readonly AxProgramDemos<IN, OUT>[]): void;
 }
-declare class AxProgram<IN extends AxGenIn, OUT extends AxGenOut> implements AxTunable, AxUsable {
-    protected trace?: Record<string, AxFieldValue>;
+declare class AxProgram<IN extends AxGenIn, OUT extends AxGenOut> implements AxTunable<IN, OUT>, AxUsable {
+    protected trace?: OUT;
     protected usage: AxProgramUsage[];
     private key;
     private children;
     constructor();
-    register(prog: Readonly<AxTunable & AxUsable>): void;
+    register(prog: Readonly<AxTunable<IN, OUT> & AxUsable>): void;
     forward(_ai: Readonly<AxAIService>, _values: IN | AxMessage<IN>[], _options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
     streamingForward(_ai: Readonly<AxAIService>, _values: IN | AxMessage<IN>[], _options?: Readonly<AxProgramStreamingForwardOptions>): AxGenStreamingOut<OUT>;
     setId(id: string): void;
     setParentId(parentId: string): void;
-    setExamples(examples: Readonly<AxProgramExamples>, options?: Readonly<AxSetExamplesOptions>): void;
-    getTraces(): AxProgramTrace[];
+    setExamples(examples: Readonly<AxProgramExamples<IN, OUT>>, options?: Readonly<AxSetExamplesOptions>): void;
+    getTraces(): AxProgramTrace<IN, OUT>[];
     getUsage(): AxProgramUsage[];
     resetUsage(): void;
-    setDemos(demos: readonly AxProgramDemos[]): void;
+    setDemos(demos: readonly AxProgramDemos<IN, OUT>[]): void;
 }
 
 /**
  * Interface for agents that can be used as child agents.
  * Provides methods to get the agent's function definition and features.
  */
-interface AxAgentic extends AxTunable, AxUsable {
+interface AxAgentic<IN extends AxGenIn, OUT extends AxGenOut> extends AxTunable<IN, OUT>, AxUsable {
     getFunction(): AxFunction;
     getFeatures(): AxAgentFeatures;
 }
@@ -2516,7 +2518,7 @@ interface AxAgentFeatures {
  * An AI agent that can process inputs using an AI service and coordinate with child agents.
  * Supports features like smart model routing and automatic input field passing to child agents.
  */
-declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut = AxGenOut> implements AxAgentic {
+declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAgentic<IN, OUT> {
     private ai?;
     private program;
     private functions?;
@@ -2532,14 +2534,14 @@ declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut = AxGenOut> imple
         description: string;
         definition?: string;
         signature: NonNullable<ConstructorParameters<typeof AxSignature>[0]>;
-        agents?: AxAgentic[];
+        agents?: AxAgentic<IN, OUT>[];
         functions?: AxInputFunctionType;
     }>, options?: Readonly<AxAgentOptions>);
-    setExamples(examples: Readonly<AxProgramExamples>, options?: Readonly<AxSetExamplesOptions>): void;
+    setExamples(examples: Readonly<AxProgramExamples<IN, OUT>>, options?: Readonly<AxSetExamplesOptions>): void;
     setId(id: string): void;
     setParentId(parentId: string): void;
-    getTraces(): AxProgramTrace[];
-    setDemos(demos: readonly AxProgramDemos[]): void;
+    getTraces(): AxProgramTrace<IN, OUT>[];
+    setDemos(demos: readonly AxProgramDemos<IN, OUT>[]): void;
     getUsage(): (AxModelUsage & {
         ai: string;
         model: string;
@@ -2634,6 +2636,434 @@ declare class AxBalancer implements AxAIService<unknown, unknown> {
     getLogger(): AxLoggerFunction;
 }
 
+type AxExample = Record<string, AxFieldValue>;
+type AxMetricFn = <T extends AxGenOut = AxGenOut>(arg0: Readonly<{
+    prediction: T;
+    example: AxExample;
+}>) => number | Promise<number>;
+type AxMetricFnArgs = Parameters<AxMetricFn>[0];
+type AxMultiMetricFn = <T extends AxGenOut = AxGenOut>(arg0: Readonly<{
+    prediction: T;
+    example: AxExample;
+}>) => Record<string, number>;
+interface AxOptimizationProgress {
+    round: number;
+    totalRounds: number;
+    currentScore: number;
+    bestScore: number;
+    tokensUsed: number;
+    timeElapsed: number;
+    successfulExamples: number;
+    totalExamples: number;
+    currentConfiguration?: Record<string, unknown>;
+    convergenceInfo?: {
+        improvement: number;
+        stagnationRounds: number;
+        isConverging: boolean;
+    };
+}
+interface AxCostTracker {
+    trackTokens(count: number, model: string): void;
+    getCurrentCost(): number;
+    getTokenUsage(): Record<string, number>;
+    getTotalTokens(): number;
+    isLimitReached(): boolean;
+    reset(): void;
+}
+interface AxOptimizationCheckpoint {
+    version: string;
+    timestamp: number;
+    optimizerType: string;
+    optimizerConfig: Record<string, unknown>;
+    currentRound: number;
+    totalRounds: number;
+    bestScore: number;
+    bestConfiguration?: Record<string, unknown>;
+    scoreHistory: number[];
+    configurationHistory: Record<string, unknown>[];
+    stats: AxOptimizationStats;
+    optimizerState: Record<string, unknown>;
+    examples: readonly AxExample[];
+    validationSet?: readonly AxExample[];
+}
+type AxCheckpointSaveFn = (checkpoint: Readonly<AxOptimizationCheckpoint>) => Promise<string>;
+type AxCheckpointLoadFn = (checkpointId: string) => Promise<AxOptimizationCheckpoint | null>;
+interface AxCostTrackerOptions {
+    costPerModel?: Record<string, number>;
+    maxCost?: number;
+    maxTokens?: number;
+}
+type AxOptimizerArgs = {
+    studentAI: AxAIService;
+    teacherAI?: AxAIService;
+    examples: readonly AxExample[];
+    validationSet?: readonly AxExample[];
+    minSuccessRate?: number;
+    targetScore?: number;
+    onProgress?: (progress: Readonly<AxOptimizationProgress>) => void;
+    onEarlyStop?: (reason: string, stats: Readonly<AxOptimizationStats>) => void;
+    costTracker?: AxCostTracker;
+    checkpointSave?: AxCheckpointSaveFn;
+    checkpointLoad?: AxCheckpointLoadFn;
+    checkpointInterval?: number;
+    resumeFromCheckpoint?: string;
+    seed?: number;
+};
+interface AxOptimizationStats {
+    totalCalls: number;
+    successfulDemos: number;
+    estimatedTokenUsage: number;
+    earlyStopped: boolean;
+    earlyStopping?: {
+        bestScoreRound: number;
+        patienceExhausted: boolean;
+        reason: string;
+    };
+    resourceUsage: {
+        totalTokens: number;
+        totalTime: number;
+        avgLatencyPerEval: number;
+        peakMemoryUsage?: number;
+        costByModel: Record<string, number>;
+    };
+    convergenceInfo: {
+        converged: boolean;
+        finalImprovement: number;
+        stagnationRounds: number;
+        convergenceThreshold: number;
+    };
+    evaluationBreakdown?: {
+        trainingScore: number;
+        validationScore: number;
+        crossValidationScores?: number[];
+        standardDeviation?: number;
+    };
+}
+interface AxOptimizerResult<OUT extends AxGenOut> {
+    demos?: AxProgramDemos<AxGenIn, OUT>[];
+    stats: AxOptimizationStats;
+    bestScore: number;
+    finalConfiguration?: Record<string, unknown>;
+    scoreHistory?: number[];
+    configurationHistory?: Record<string, unknown>[];
+}
+interface AxParetoResult<OUT extends AxGenOut = AxGenOut> extends AxOptimizerResult<OUT> {
+    paretoFront: ReadonlyArray<{
+        demos: readonly AxProgramDemos<AxGenIn, OUT>[];
+        scores: Readonly<Record<string, number>>;
+        configuration: Readonly<Record<string, unknown>>;
+        dominatedSolutions: number;
+    }>;
+    hypervolume?: number;
+    paretoFrontSize: number;
+    convergenceMetrics?: Record<string, number>;
+}
+interface AxCompileOptions {
+    maxIterations?: number;
+    earlyStoppingPatience?: number;
+    verbose?: boolean;
+    overrideValidationSet?: readonly AxExample[];
+    overrideTargetScore?: number;
+    overrideCostTracker?: AxCostTracker;
+    overrideTeacherAI?: AxAIService;
+    overrideOnProgress?: (progress: Readonly<AxOptimizationProgress>) => void;
+    overrideOnEarlyStop?: (reason: string, stats: Readonly<AxOptimizationStats>) => void;
+    overrideCheckpointSave?: AxCheckpointSaveFn;
+    overrideCheckpointLoad?: AxCheckpointLoadFn;
+    overrideCheckpointInterval?: number;
+    saveCheckpointOnComplete?: boolean;
+}
+interface AxOptimizer<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> {
+    /**
+     * Optimize a program using the provided metric function
+     * @param program The program to optimize (moved from constructor)
+     * @param metricFn Evaluation metric function to assess program performance
+     * @param options Optional configuration options that can override constructor settings
+     * @returns Optimization result containing demos, stats, and configuration
+     */
+    compile(program: Readonly<AxProgram<IN, OUT>>, metricFn: AxMetricFn, options?: AxCompileOptions): Promise<AxOptimizerResult<OUT>>;
+    /**
+     * Optimize a program with real-time streaming updates
+     * @param program The program to optimize
+     * @param metricFn Evaluation metric function
+     * @param options Optional configuration options
+     * @returns Async iterator yielding optimization progress
+     */
+    compileStream?(program: Readonly<AxProgram<IN, OUT>>, metricFn: AxMetricFn, options?: AxCompileOptions): AsyncIterableIterator<AxOptimizationProgress>;
+    /**
+     * Multi-objective optimization using Pareto frontier
+     * @param program The program to optimize
+     * @param metricFn Multi-objective metric function
+     * @param options Optional configuration options
+     * @returns Pareto optimization result
+     */
+    compilePareto?(program: Readonly<AxProgram<IN, OUT>>, metricFn: AxMultiMetricFn, options?: AxCompileOptions): Promise<AxParetoResult<OUT>>;
+    /**
+     * Get current optimization statistics
+     * @returns Current optimization statistics
+     */
+    getStats(): AxOptimizationStats;
+    /**
+     * Cancel ongoing optimization gracefully
+     * @returns Promise that resolves when cancellation is complete
+     */
+    cancel?(): Promise<void>;
+    /**
+     * Reset optimizer state for reuse with different programs
+     */
+    reset?(): void;
+    /**
+     * Get optimizer-specific configuration
+     * @returns Current optimizer configuration
+     */
+    getConfiguration?(): Record<string, unknown>;
+    /**
+     * Update optimizer configuration
+     * @param config New configuration to merge with existing
+     */
+    updateConfiguration?(config: Readonly<Record<string, unknown>>): void;
+    /**
+     * Validate that the optimizer can handle the given program
+     * @param program Program to validate
+     * @returns Validation result with any issues found
+     */
+    validateProgram?(program: Readonly<AxProgram<IN, OUT>>): {
+        isValid: boolean;
+        issues: string[];
+        suggestions: string[];
+    };
+}
+interface AxBootstrapOptimizerOptions {
+    maxRounds?: number;
+    maxExamples?: number;
+    maxDemos?: number;
+    batchSize?: number;
+    earlyStoppingPatience?: number;
+    teacherAI?: AxAIService;
+    costMonitoring?: boolean;
+    maxTokensPerGeneration?: number;
+    verboseMode?: boolean;
+    debugMode?: boolean;
+    adaptiveBatching?: boolean;
+    dynamicTemperature?: boolean;
+    qualityThreshold?: number;
+    diversityWeight?: number;
+}
+interface AxMiPROOptimizerOptions {
+    numCandidates?: number;
+    initTemperature?: number;
+    maxBootstrappedDemos?: number;
+    maxLabeledDemos?: number;
+    numTrials?: number;
+    minibatch?: boolean;
+    minibatchSize?: number;
+    minibatchFullEvalSteps?: number;
+    programAwareProposer?: boolean;
+    dataAwareProposer?: boolean;
+    viewDataBatchSize?: number;
+    tipAwareProposer?: boolean;
+    fewshotAwareProposer?: boolean;
+    verbose?: boolean;
+    earlyStoppingTrials?: number;
+    minImprovementThreshold?: number;
+    bayesianOptimization?: boolean;
+    acquisitionFunction?: 'expected_improvement' | 'upper_confidence_bound' | 'probability_improvement';
+    explorationWeight?: number;
+}
+interface AxBootstrapCompileOptions extends AxCompileOptions {
+    valset?: readonly AxExample[];
+    maxDemos?: number;
+    teacherProgram?: Readonly<AxProgram<AxGenIn, AxGenOut>>;
+}
+interface AxMiPROCompileOptions extends AxCompileOptions {
+    valset?: readonly AxExample[];
+    teacher?: Readonly<AxProgram<AxGenIn, AxGenOut>>;
+    auto?: 'light' | 'medium' | 'heavy';
+    instructionCandidates?: string[];
+    customProposer?: (context: Readonly<{
+        programSummary: string;
+        dataSummary: string;
+        previousInstructions: string[];
+    }>) => Promise<string[]>;
+}
+declare class AxDefaultCostTracker implements AxCostTracker {
+    private tokenUsage;
+    private totalTokens;
+    private readonly costPerModel;
+    private readonly maxCost?;
+    private readonly maxTokens?;
+    constructor(options?: AxCostTrackerOptions);
+    trackTokens(count: number, model: string): void;
+    getCurrentCost(): number;
+    getTokenUsage(): Record<string, number>;
+    getTotalTokens(): number;
+    isLimitReached(): boolean;
+    reset(): void;
+}
+/**
+ * Abstract base class for optimizers that provides common functionality
+ * and standardized handling of AxOptimizerArgs
+ */
+declare abstract class AxBaseOptimizer<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> implements AxOptimizer<IN, OUT> {
+    protected readonly studentAI: AxAIService;
+    protected readonly teacherAI?: AxAIService;
+    protected readonly examples: readonly AxExample[];
+    protected readonly validationSet?: readonly AxExample[];
+    protected readonly targetScore?: number;
+    protected readonly minSuccessRate?: number;
+    protected readonly onProgress?: (progress: Readonly<AxOptimizationProgress>) => void;
+    protected readonly onEarlyStop?: (reason: string, stats: Readonly<AxOptimizationStats>) => void;
+    protected readonly costTracker?: AxCostTracker;
+    protected readonly seed?: number;
+    protected readonly checkpointSave?: AxCheckpointSaveFn;
+    protected readonly checkpointLoad?: AxCheckpointLoadFn;
+    protected readonly checkpointInterval?: number;
+    protected readonly resumeFromCheckpoint?: string;
+    private currentRound;
+    private scoreHistory;
+    private configurationHistory;
+    protected stats: AxOptimizationStats;
+    constructor(args: Readonly<AxOptimizerArgs>);
+    /**
+     * Initialize the optimization statistics structure
+     */
+    protected initializeStats(): AxOptimizationStats;
+    /**
+     * Set up reproducible random seed if provided
+     */
+    protected setupRandomSeed(): void;
+    /**
+     * Check if optimization should stop early due to cost limits
+     */
+    protected checkCostLimits(): boolean;
+    /**
+     * Check if target score has been reached
+     */
+    protected checkTargetScore(currentScore: number): boolean;
+    /**
+     * Update resource usage statistics
+     */
+    protected updateResourceUsage(startTime: number, tokensUsed?: number): void;
+    /**
+     * Trigger early stopping with appropriate callbacks
+     */
+    protected triggerEarlyStopping(reason: string, bestScoreRound: number): void;
+    /**
+     * Get the validation set, with fallback to a split of examples
+     */
+    protected getValidationSet(options?: AxCompileOptions): readonly AxExample[];
+    /**
+     * Get the AI service to use for a specific task, preferring teacher when available
+     * @param preferTeacher Whether to prefer teacher AI over student AI
+     * @param options Optional compile options that may override teacher AI
+     * @returns The appropriate AI service to use
+     */
+    protected getAIService(preferTeacher?: boolean, options?: AxCompileOptions): AxAIService;
+    /**
+     * Check if teacher AI is available (including overrides)
+     * @param options Optional compile options that may override teacher AI
+     * @returns True if teacher AI is configured or overridden
+     */
+    protected hasTeacherAI(options?: AxCompileOptions): boolean;
+    /**
+     * Get teacher AI if available, otherwise return student AI
+     * @param options Optional compile options that may override teacher AI
+     * @returns Teacher AI if available, otherwise student AI
+     */
+    protected getTeacherOrStudentAI(options?: AxCompileOptions): AxAIService;
+    /**
+     * Execute a task with teacher AI if available, otherwise use student AI
+     * @param task Function that takes an AI service and returns a promise
+     * @param preferTeacher Whether to prefer teacher AI (default: true)
+     * @param options Optional compile options that may override teacher AI
+     * @returns Result of the task execution
+     */
+    protected executeWithTeacher<T>(task: (ai: AxAIService) => Promise<T>, preferTeacher?: boolean, options?: AxCompileOptions): Promise<T>;
+    /**
+     * Abstract method that must be implemented by concrete optimizers
+     */
+    abstract compile(program: Readonly<AxProgram<IN, OUT>>, metricFn: AxMetricFn, options?: AxCompileOptions): Promise<AxOptimizerResult<OUT>>;
+    /**
+     * Get current optimization statistics
+     */
+    getStats(): AxOptimizationStats;
+    /**
+     * Reset optimizer state for reuse with different programs
+     */
+    reset(): void;
+    /**
+     * Basic program validation that can be extended by concrete optimizers
+     */
+    validateProgram(program: Readonly<AxProgram<IN, OUT>>): {
+        isValid: boolean;
+        issues: string[];
+        suggestions: string[];
+    };
+    /**
+     * Multi-objective optimization using Pareto frontier
+     * Default implementation that leverages the single-objective compile method
+     * @param program The program to optimize
+     * @param metricFn Multi-objective metric function that returns multiple scores
+     * @param options Optional configuration options
+     * @returns Pareto optimization result with frontier of non-dominated solutions
+     */
+    compilePareto(program: Readonly<AxProgram<IN, OUT>>, metricFn: AxMultiMetricFn, options?: AxCompileOptions): Promise<AxParetoResult<OUT>>;
+    /**
+     * Generate solutions using different weighted combinations of objectives
+     */
+    private generateWeightedSolutions;
+    /**
+     * Generate solutions using constraint-based optimization
+     */
+    private generateConstraintSolutions;
+    /**
+     * Generate different weight combinations for objectives
+     */
+    private generateWeightCombinations;
+    /**
+     * Evaluate a single-objective result with multi-objective metrics
+     */
+    private evaluateWithMultiObjective;
+    /**
+     * Find the Pareto frontier from a set of solutions
+     */
+    private findParetoFrontier;
+    /**
+     * Check if solution A dominates solution B
+     * A dominates B if A is better or equal in all objectives and strictly better in at least one
+     */
+    private dominates;
+    /**
+     * Calculate hypervolume of the Pareto frontier
+     * Simplified implementation using reference point at origin
+     */
+    private calculateHypervolume;
+    /**
+     * Save current optimization state to checkpoint
+     */
+    protected saveCheckpoint(optimizerType: string, optimizerConfig: Record<string, unknown>, bestScore: number, bestConfiguration?: Record<string, unknown>, optimizerState?: Record<string, unknown>, options?: AxCompileOptions): Promise<string | undefined>;
+    /**
+     * Load optimization state from checkpoint
+     */
+    protected loadCheckpoint(checkpointId: string, options?: AxCompileOptions): Promise<AxOptimizationCheckpoint | null>;
+    /**
+     * Restore optimizer state from checkpoint
+     */
+    protected restoreFromCheckpoint(checkpoint: Readonly<AxOptimizationCheckpoint>): void;
+    /**
+     * Check if checkpoint should be saved
+     */
+    protected shouldSaveCheckpoint(round: number, options?: AxCompileOptions): boolean;
+    /**
+     * Update optimization progress and handle checkpointing
+     */
+    protected updateOptimizationProgress(round: number, score: number, configuration: Record<string, unknown>, optimizerType: string, optimizerConfig: Record<string, unknown>, bestScore: number, bestConfiguration?: Record<string, unknown>, optimizerState?: Record<string, unknown>, options?: AxCompileOptions): Promise<void>;
+    /**
+     * Save final checkpoint on completion
+     */
+    protected saveFinalCheckpoint(optimizerType: string, optimizerConfig: Record<string, unknown>, bestScore: number, bestConfiguration?: Record<string, unknown>, optimizerState?: Record<string, unknown>, options?: AxCompileOptions): Promise<void>;
+}
+
 type AxDBUpsertRequest = {
     id: string;
     text?: string;
@@ -2901,7 +3331,7 @@ declare class AxGen<IN extends AxGenIn, OUT extends AxGenerateResult<AxGenOut> =
         version: number;
         delta: Partial<OUT>;
     }, void, unknown>;
-    setExamples(examples: Readonly<AxProgramExamples>, options?: Readonly<AxSetExamplesOptions>): void;
+    setExamples(examples: Readonly<AxProgramExamples<IN, OUT>>, options?: Readonly<AxSetExamplesOptions>): void;
     private isDebug;
     private getLogger;
 }
@@ -3251,111 +3681,10 @@ declare class AxMCPStreambleHTTPTransport implements AxMCPTransport {
     close(): void;
 }
 
-type AxExample = Record<string, AxFieldValue>;
-type AxMetricFn = <T extends AxGenOut = AxGenOut>(arg0: Readonly<{
-    prediction: T;
-    example: AxExample;
-}>) => number;
-type AxMetricFnArgs = Parameters<AxMetricFn>[0];
-type AxOptimizerArgs<IN extends AxGenIn, OUT extends AxGenOut> = {
-    ai: AxAIService;
-    program: Readonly<AxProgram<IN, OUT>>;
-    examples: Readonly<AxExample[]>;
-    options?: Record<string, unknown>;
-};
-interface AxOptimizationStats {
-    totalCalls: number;
-    successfulDemos: number;
-    estimatedTokenUsage: number;
-    earlyStopped: boolean;
-    earlyStopping?: {
-        bestScoreRound: number;
-        patienceExhausted: boolean;
-    };
-}
-interface AxOptimizerResult<IN extends AxGenIn, OUT extends AxGenOut> {
-    program?: Readonly<AxProgram<IN, OUT>>;
-    demos?: AxProgramDemos[];
-    stats?: AxOptimizationStats;
-}
-interface AxOptimizer<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> {
-    /**
-     * Main optimization method that optimizes the program using the provided metric function
-     * @param metricFn Evaluation metric function to assess program performance
-     * @param options Optional configuration options specific to the optimizer
-     * @returns Optimization result containing the optimized program, demos, and/or stats
-     */
-    compile(metricFn: AxMetricFn, options?: Record<string, unknown>): Promise<AxOptimizerResult<IN, OUT>>;
-    /**
-     * Get optimization statistics if available
-     * @returns Optimization statistics or undefined if not supported
-     */
-    getStats?(): AxOptimizationStats | undefined;
-}
-interface AxBootstrapOptimizerOptions {
-    maxRounds?: number;
-    maxExamples?: number;
-    maxDemos?: number;
-    batchSize?: number;
-    earlyStoppingPatience?: number;
-    teacherAI?: AxAIService;
-    costMonitoring?: boolean;
-    maxTokensPerGeneration?: number;
-    verboseMode?: boolean;
-    debugMode?: boolean;
-}
-interface AxMiPROOptimizerOptions {
-    numCandidates?: number;
-    initTemperature?: number;
-    maxBootstrappedDemos?: number;
-    maxLabeledDemos?: number;
-    numTrials?: number;
-    minibatch?: boolean;
-    minibatchSize?: number;
-    minibatchFullEvalSteps?: number;
-    programAwareProposer?: boolean;
-    dataAwareProposer?: boolean;
-    viewDataBatchSize?: number;
-    tipAwareProposer?: boolean;
-    fewshotAwareProposer?: boolean;
-    seed?: number;
-    verbose?: boolean;
-    earlyStoppingTrials?: number;
-    minImprovementThreshold?: number;
-}
-interface AxBootstrapCompileOptions {
-    valset?: readonly AxExample[];
-    maxDemos?: number;
-}
-interface AxMiPROCompileOptions {
-    valset?: readonly AxExample[];
-    teacher?: Readonly<AxProgram<AxGenIn, AxGenOut>>;
-    auto?: 'light' | 'medium' | 'heavy';
-}
-
-interface AxMiPROOptions {
-    numCandidates?: number;
-    initTemperature?: number;
-    maxBootstrappedDemos?: number;
-    maxLabeledDemos?: number;
-    numTrials?: number;
-    minibatch?: boolean;
-    minibatchSize?: number;
-    minibatchFullEvalSteps?: number;
-    programAwareProposer?: boolean;
-    dataAwareProposer?: boolean;
-    viewDataBatchSize?: number;
-    tipAwareProposer?: boolean;
-    fewshotAwareProposer?: boolean;
-    seed?: number;
-    verbose?: boolean;
-    earlyStoppingTrials?: number;
-    minImprovementThreshold?: number;
+interface AxMiPROResult<IN extends AxGenIn, OUT extends AxGenOut> extends AxOptimizerResult<OUT> {
+    optimizedGen?: AxGen<IN, OUT>;
 }
-declare class AxMiPRO<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> implements AxOptimizer<IN, OUT> {
-    private ai;
-    private program;
-    private examples;
+declare class AxMiPRO<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxBaseOptimizer<IN, OUT> {
     private maxBootstrappedDemos;
     private maxLabeledDemos;
     private numCandidates;
@@ -3369,14 +3698,15 @@ declare class AxMiPRO<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGen
     private viewDataBatchSize;
     private tipAwareProposer;
     private fewshotAwareProposer;
-    private seed?;
     private verbose;
-    private bootstrapper;
     private earlyStoppingTrials;
     private minImprovementThreshold;
-    constructor({ ai, program, examples, options, }: Readonly<AxOptimizerArgs<IN, OUT>> & {
-        options?: AxMiPROOptions;
-    });
+    private bayesianOptimization;
+    private acquisitionFunction;
+    private explorationWeight;
+    constructor(args: Readonly<AxOptimizerArgs & {
+        options?: AxMiPROOptimizerOptions;
+    }>);
     /**
      * Configures the optimizer for light, medium, or heavy optimization
      * @param level The optimization level: "light", "medium", or "heavy"
@@ -3387,21 +3717,11 @@ declare class AxMiPRO<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGen
      */
     private generateTips;
     /**
-     * Generates instruction candidates for each predictor in the program
+     * Generates instruction candidates using the teacher model if available
+     * @param options Optional compile options that may override teacher AI
      * @returns Array of generated instruction candidates
      */
     private proposeInstructionCandidates;
-    /**
-     * Generates a summary of the program structure for instruction proposal
-     */
-    private generateProgramSummary;
-    /**
-     * Generates a summary of the dataset for instruction proposal
-     */
-    private generateDataSummary;
-    /**
-     * Generates a specific instruction candidate
-     */
     private generateInstruction;
     /**
      * Bootstraps few-shot examples for the program
@@ -3412,43 +3732,43 @@ declare class AxMiPRO<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGen
      */
     private selectLabeledExamples;
     /**
-     * Runs Bayesian optimization to find the best combination of few-shot examples and instructions
-     */
-    private runBayesianOptimization;
-    /**
-     * Evaluates a configuration on the validation set
+     * Runs optimization to find the best combination of few-shot examples and instructions
      */
+    private runOptimization;
     private evaluateConfig;
+    private applyConfigToProgram;
     /**
-     * Run full evaluation on the entire validation set
+     * The main compile method to run MIPROv2 optimization
      */
-    private fullEvaluation;
+    compile(program: Readonly<AxProgram<IN, OUT>>, metricFn: AxMetricFn, options?: AxCompileOptions): Promise<AxMiPROResult<IN, OUT>>;
     /**
-     * Implements a Bayesian-inspired selection of the next configuration to try
-     * This is a simplified version using Upper Confidence Bound (UCB) strategy
+     * Applies a configuration to an AxGen instance
      */
-    private selectNextConfiguration;
+    private applyConfigToAxGen;
     /**
-     * Applies a configuration to a program instance
+     * Get optimizer-specific configuration
+     * @returns Current optimizer configuration
      */
-    private applyConfigToProgram;
+    getConfiguration(): Record<string, unknown>;
     /**
-     * Sets instruction to a program
-     * Note: Workaround since setInstruction may not be available directly
+     * Update optimizer configuration
+     * @param config New configuration to merge with existing
      */
-    private setInstructionToProgram;
+    updateConfiguration(config: Readonly<Record<string, unknown>>): void;
     /**
-     * The main compile method to run MIPROv2 optimization
-     * @param metricFn Evaluation metric function
-     * @param options Optional configuration options
-     * @returns The optimization result
+     * Reset optimizer state for reuse with different programs
      */
-    compile(metricFn: AxMetricFn, options?: Record<string, unknown>): Promise<AxOptimizerResult<IN, OUT>>;
+    reset(): void;
     /**
-     * Get optimization statistics from the internal bootstrapper
-     * @returns Optimization statistics or undefined if not available
+     * Validate that the optimizer can handle the given program
+     * @param program Program to validate
+     * @returns Validation result with any issues found
      */
-    getStats(): AxOptimizationStats | undefined;
+    validateProgram(program: Readonly<AxProgram<IN, OUT>>): {
+        isValid: boolean;
+        issues: string[];
+        suggestions: string[];
+    };
 }
 
 type AxMockAIServiceConfig = {
@@ -3608,11 +3928,7 @@ Readonly<AxAIOpenAIEmbedResponse>> {
     createEmbedReq(req: Readonly<AxInternalEmbedRequest<TEmbedModel>>): [AxAPI, AxAIOpenAIEmbedRequest<TEmbedModel>];
 }
 
-declare class AxBootstrapFewShot<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> implements AxOptimizer<IN, OUT> {
-    private ai;
-    private teacherAI?;
-    private program;
-    private examples;
+declare class AxBootstrapFewShot<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxBaseOptimizer<IN, OUT> {
     private maxRounds;
     private maxDemos;
     private maxExamples;
@@ -3623,11 +3939,11 @@ declare class AxBootstrapFewShot<IN extends AxGenIn = AxGenIn, OUT extends AxGen
     private verboseMode;
     private debugMode;
     private traces;
-    private stats;
-    constructor({ ai, program, examples, options, }: Readonly<AxOptimizerArgs<IN, OUT>>);
+    constructor(args: Readonly<AxOptimizerArgs & {
+        options?: AxBootstrapOptimizerOptions;
+    }>);
     private compileRound;
-    compile(metricFn: AxMetricFn, options?: Record<string, unknown>): Promise<AxOptimizerResult<IN, OUT>>;
-    getStats(): AxOptimizationStats;
+    compile(program: Readonly<AxProgram<IN, OUT>>, metricFn: AxMetricFn, options?: AxBootstrapCompileOptions): Promise<AxOptimizerResult<OUT>>;
 }
 
 declare class AxChainOfThought<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxGen<IN, OUT> {
@@ -3701,11 +4017,12 @@ declare const AxEvalUtil: {
     novelF1ScoreOptimized: typeof novelF1ScoreOptimized;
 };
 
-declare class AxInstanceRegistry<T> {
+type AxInstanceRegistryItem<T extends AxTunable<IN, OUT>, IN extends AxGenIn, OUT extends AxGenOut> = T & AxUsable;
+declare class AxInstanceRegistry<T extends AxTunable<IN, OUT>, IN extends AxGenIn, OUT extends AxGenOut> {
     private reg;
     constructor();
-    register(instance: T): void;
-    [Symbol.iterator](): Generator<T, void, unknown>;
+    register(instance: AxInstanceRegistryItem<T, IN, OUT>): void;
+    [Symbol.iterator](): Generator<AxInstanceRegistryItem<T, IN, OUT> | undefined, void, unknown>;
 }
 
 /**
@@ -3928,4 +4245,4 @@ declare const axModelInfoReka: AxModelInfo[];
 
 declare const axModelInfoTogether: AxModelInfo[];
 
-export { AxAI, AxAIAnthropic, type AxAIAnthropicArgs, type AxAIAnthropicChatError, type AxAIAnthropicChatRequest, type AxAIAnthropicChatRequestCacheParam, type AxAIAnthropicChatResponse, type AxAIAnthropicChatResponseDelta, type AxAIAnthropicConfig, type AxAIAnthropicContentBlockDeltaEvent, type AxAIAnthropicContentBlockStartEvent, type AxAIAnthropicContentBlockStopEvent, type AxAIAnthropicErrorEvent, type AxAIAnthropicMessageDeltaEvent, type AxAIAnthropicMessageStartEvent, type AxAIAnthropicMessageStopEvent, AxAIAnthropicModel, type AxAIAnthropicPingEvent, AxAIAnthropicVertexModel, type AxAIArgs, AxAIAzureOpenAI, type AxAIAzureOpenAIArgs, type AxAIAzureOpenAIConfig, AxAICohere, type AxAICohereArgs, type AxAICohereChatRequest, type AxAICohereChatRequestToolResults, type AxAICohereChatResponse, type AxAICohereChatResponseDelta, type AxAICohereChatResponseToolCalls, type AxAICohereConfig, AxAICohereEmbedModel, type AxAICohereEmbedRequest, type AxAICohereEmbedResponse, AxAICohereModel, AxAIDeepSeek, type AxAIDeepSeekArgs, AxAIDeepSeekModel, type AxAIEmbedModels, type AxAIFeatures, AxAIGoogleGemini, type AxAIGoogleGeminiArgs, type AxAIGoogleGeminiBatchEmbedRequest, type AxAIGoogleGeminiBatchEmbedResponse, type AxAIGoogleGeminiChatRequest, type AxAIGoogleGeminiChatResponse, type AxAIGoogleGeminiChatResponseDelta, type AxAIGoogleGeminiConfig, type AxAIGoogleGeminiContent, type AxAIGoogleGeminiContentPart, AxAIGoogleGeminiEmbedModel, AxAIGoogleGeminiEmbedTypes, type AxAIGoogleGeminiGenerationConfig, AxAIGoogleGeminiModel, type AxAIGoogleGeminiOptionsTools, AxAIGoogleGeminiSafetyCategory, type AxAIGoogleGeminiSafetySettings, AxAIGoogleGeminiSafetyThreshold, type AxAIGoogleGeminiThinkingConfig, type AxAIGoogleGeminiThinkingTokenBudgetLevels, type AxAIGoogleGeminiTool, type AxAIGoogleGeminiToolConfig, type AxAIGoogleGeminiToolFunctionDeclaration, type AxAIGoogleGeminiToolGoogleSearchRetrieval, type AxAIGoogleVertexBatchEmbedRequest, type AxAIGoogleVertexBatchEmbedResponse, AxAIGrok, type AxAIGrokArgs, type AxAIGrokChatRequest, AxAIGrokEmbedModels, AxAIGrokModel, type AxAIGrokOptionsTools, type AxAIGrokSearchSource, AxAIGroq, type AxAIGroqArgs, AxAIGroqModel, AxAIHuggingFace, type AxAIHuggingFaceArgs, type AxAIHuggingFaceConfig, AxAIHuggingFaceModel, type AxAIHuggingFaceRequest, type AxAIHuggingFaceResponse, type AxAIInputModelList, type AxAIMemory, AxAIMistral, type AxAIMistralArgs, type AxAIMistralChatRequest, AxAIMistralEmbedModels, AxAIMistralModel, type AxAIModelList, type AxAIModelListBase, type AxAIModels, AxAIOllama, type AxAIOllamaAIConfig, type AxAIOllamaArgs, AxAIOpenAI, type AxAIOpenAIArgs, AxAIOpenAIBase, type AxAIOpenAIBaseArgs, type AxAIOpenAIChatRequest, type AxAIOpenAIChatResponse, type AxAIOpenAIChatResponseDelta, type AxAIOpenAIConfig, AxAIOpenAIEmbedModel, type AxAIOpenAIEmbedRequest, type AxAIOpenAIEmbedResponse, type AxAIOpenAILogprob, AxAIOpenAIModel, type AxAIOpenAIResponseDelta, AxAIOpenAIResponses, type AxAIOpenAIResponsesArgs, AxAIOpenAIResponsesBase, type AxAIOpenAIResponsesCodeInterpreterToolCall, type AxAIOpenAIResponsesComputerToolCall, type AxAIOpenAIResponsesConfig, type AxAIOpenAIResponsesContentPartAddedEvent, type AxAIOpenAIResponsesContentPartDoneEvent, type AxAIOpenAIResponsesDefineFunctionTool, type AxAIOpenAIResponsesErrorEvent, type AxAIOpenAIResponsesFileSearchCallCompletedEvent, type AxAIOpenAIResponsesFileSearchCallInProgressEvent, type AxAIOpenAIResponsesFileSearchCallSearchingEvent, type AxAIOpenAIResponsesFileSearchToolCall, type AxAIOpenAIResponsesFunctionCallArgumentsDeltaEvent, type AxAIOpenAIResponsesFunctionCallArgumentsDoneEvent, type AxAIOpenAIResponsesFunctionCallItem, type AxAIOpenAIResponsesImageGenerationCallCompletedEvent, type AxAIOpenAIResponsesImageGenerationCallGeneratingEvent, type AxAIOpenAIResponsesImageGenerationCallInProgressEvent, type AxAIOpenAIResponsesImageGenerationCallPartialImageEvent, type AxAIOpenAIResponsesImageGenerationToolCall, AxAIOpenAIResponsesImpl, type AxAIOpenAIResponsesInputAudioContentPart, type AxAIOpenAIResponsesInputContentPart, type AxAIOpenAIResponsesInputFunctionCallItem, type AxAIOpenAIResponsesInputFunctionCallOutputItem, type AxAIOpenAIResponsesInputImageUrlContentPart, type AxAIOpenAIResponsesInputItem, type AxAIOpenAIResponsesInputMessageItem, type AxAIOpenAIResponsesInputTextContentPart, type AxAIOpenAIResponsesLocalShellToolCall, type AxAIOpenAIResponsesMCPCallArgumentsDeltaEvent, type AxAIOpenAIResponsesMCPCallArgumentsDoneEvent, type AxAIOpenAIResponsesMCPCallCompletedEvent, type AxAIOpenAIResponsesMCPCallFailedEvent, type AxAIOpenAIResponsesMCPCallInProgressEvent, type AxAIOpenAIResponsesMCPListToolsCompletedEvent, type AxAIOpenAIResponsesMCPListToolsFailedEvent, type AxAIOpenAIResponsesMCPListToolsInProgressEvent, type AxAIOpenAIResponsesMCPToolCall, AxAIOpenAIResponsesModel, type AxAIOpenAIResponsesOutputItem, type AxAIOpenAIResponsesOutputItemAddedEvent, type AxAIOpenAIResponsesOutputItemDoneEvent, type AxAIOpenAIResponsesOutputMessageItem, type AxAIOpenAIResponsesOutputRefusalContentPart, type AxAIOpenAIResponsesOutputTextAnnotationAddedEvent, type AxAIOpenAIResponsesOutputTextContentPart, type AxAIOpenAIResponsesOutputTextDeltaEvent, type AxAIOpenAIResponsesOutputTextDoneEvent, type AxAIOpenAIResponsesReasoningDeltaEvent, type AxAIOpenAIResponsesReasoningDoneEvent, type AxAIOpenAIResponsesReasoningItem, type AxAIOpenAIResponsesReasoningSummaryDeltaEvent, type AxAIOpenAIResponsesReasoningSummaryDoneEvent, type AxAIOpenAIResponsesReasoningSummaryPart, type AxAIOpenAIResponsesReasoningSummaryPartAddedEvent, type AxAIOpenAIResponsesReasoningSummaryPartDoneEvent, type AxAIOpenAIResponsesReasoningSummaryTextDeltaEvent, type AxAIOpenAIResponsesReasoningSummaryTextDoneEvent, type AxAIOpenAIResponsesRefusalDeltaEvent, type AxAIOpenAIResponsesRefusalDoneEvent, type AxAIOpenAIResponsesRequest, type AxAIOpenAIResponsesResponse, type AxAIOpenAIResponsesResponseCompletedEvent, type AxAIOpenAIResponsesResponseCreatedEvent, type AxAIOpenAIResponsesResponseDelta, type AxAIOpenAIResponsesResponseFailedEvent, type AxAIOpenAIResponsesResponseInProgressEvent, type AxAIOpenAIResponsesResponseIncompleteEvent, type AxAIOpenAIResponsesResponseQueuedEvent, type AxAIOpenAIResponsesStreamEvent, type AxAIOpenAIResponsesStreamEventBase, type AxAIOpenAIResponsesToolCall, type AxAIOpenAIResponsesToolCallBase, type AxAIOpenAIResponsesToolChoice, type AxAIOpenAIResponsesToolDefinition, type AxAIOpenAIResponsesWebSearchCallCompletedEvent, type AxAIOpenAIResponsesWebSearchCallInProgressEvent, type AxAIOpenAIResponsesWebSearchCallSearchingEvent, type AxAIOpenAIResponsesWebSearchToolCall, type AxAIOpenAIUsage, type AxAIPromptConfig, AxAIReka, type AxAIRekaArgs, type AxAIRekaChatRequest, type AxAIRekaChatResponse, type AxAIRekaChatResponseDelta, type AxAIRekaConfig, AxAIRekaModel, type AxAIRekaUsage, type AxAIService, AxAIServiceAbortedError, type AxAIServiceActionOptions, AxAIServiceAuthenticationError, AxAIServiceError, type AxAIServiceImpl, type AxAIServiceMetrics, AxAIServiceNetworkError, type AxAIServiceOptions, AxAIServiceResponseError, AxAIServiceStatusError, AxAIServiceStreamTerminatedError, AxAIServiceTimeoutError, AxAITogether, type AxAITogetherArgs, type AxAPI, type AxAPIConfig, AxAgent, type AxAgentFeatures, type AxAgentOptions, type AxAgentic, AxApacheTika, type AxApacheTikaArgs, type AxApacheTikaConvertOptions, type AxAssertion, AxAssertionError, AxBalancer, type AxBalancerOptions, AxBaseAI, type AxBaseAIArgs, type AxBootstrapCompileOptions, AxBootstrapFewShot, type AxBootstrapOptimizerOptions, AxChainOfThought, type AxChatRequest, type AxChatResponse, type AxChatResponseFunctionCall, type AxChatResponseResult, AxDB, type AxDBArgs, AxDBBase, type AxDBBaseArgs, type AxDBBaseOpOptions, AxDBCloudflare, type AxDBCloudflareArgs, type AxDBCloudflareOpOptions, type AxDBLoaderOptions, AxDBManager, type AxDBManagerArgs, type AxDBMatch, AxDBMemory, type AxDBMemoryArgs, type AxDBMemoryOpOptions, AxDBPinecone, type AxDBPineconeArgs, type AxDBPineconeOpOptions, type AxDBQueryRequest, type AxDBQueryResponse, type AxDBQueryService, type AxDBService, type AxDBState, type AxDBUpsertRequest, type AxDBUpsertResponse, AxDBWeaviate, type AxDBWeaviateArgs, type AxDBWeaviateOpOptions, type AxDataRow, AxDefaultQueryRewriter, AxDefaultResultReranker, type AxDockerContainer, AxDockerSession, type AxEmbedRequest, type AxEmbedResponse, AxEmbeddingAdapter, AxEvalUtil, type AxEvaluateArgs, type AxExample, type AxField, type AxFieldDescriptor, type AxFieldProcessor, type AxFieldProcessorProcess, type AxFieldTemplateFn, type AxFieldType, type AxFieldValue, type AxFunction, AxFunctionError, type AxFunctionHandler, type AxFunctionJSONSchema, AxFunctionProcessor, AxGen, type AxGenDeltaOut, type AxGenIn, type AxGenOut, type AxGenStreamingOut, AxGenerateError, type AxGenerateErrorDetails, type AxGenerateResult, AxHFDataLoader, type AxIField, type AxInputFunctionType, AxInstanceRegistry, type AxInternalChatRequest, type AxInternalEmbedRequest, AxJSInterpreter, AxJSInterpreterPermission, AxLLMRequestTypeValues, type AxLoggerFunction, type AxLoggerTag, AxMCPClient, AxMCPHTTPSSETransport, AxMCPStdioTransport, type AxMCPStreamableHTTPTransportOptions, AxMCPStreambleHTTPTransport, type AxMCPTransport, AxMemory, type AxMessage, type AxMetricFn, type AxMetricFnArgs, AxMiPRO, type AxMiPROCompileOptions, type AxMiPROOptimizerOptions, type AxMiPROOptions, AxMockAIService, type AxMockAIServiceConfig, type AxModelConfig, type AxModelInfo, type AxModelInfoWithProvider, type AxModelUsage, AxMultiServiceRouter, type AxOptimizationStats, type AxOptimizer, type AxOptimizerArgs, type AxOptimizerResult, AxProgram, type AxProgramDemos, type AxProgramExamples, type AxProgramForwardOptions, type AxProgramStreamingForwardOptions, type AxProgramTrace, type AxProgramUsage, AxProgramWithSignature, type AxProgramWithSignatureOptions, AxPromptTemplate, type AxPromptTemplateOptions, AxRAG, type AxRateLimiterFunction, AxRateLimiterTokenUsage, type AxRateLimiterTokenUsageOptions, type AxRerankerIn, type AxRerankerOut, type AxResponseHandlerArgs, type AxRewriteIn, type AxRewriteOut, AxRewriter, type AxSetExamplesOptions, AxSignature, type AxSignatureConfig, type AxSignatureTemplateValue, AxSimpleClassifier, AxSimpleClassifierClass, type AxSimpleClassifierForwardOptions, AxSpanKindValues, type AxStreamingAssertion, type AxStreamingEvent, type AxStreamingFieldProcessorProcess, AxStringUtil, AxTestPrompt, type AxTokenUsage, type AxTunable, type AxUsable, ax, axAIAnthropicDefaultConfig, axAIAnthropicVertexDefaultConfig, axAIAzureOpenAIBestConfig, axAIAzureOpenAICreativeConfig, axAIAzureOpenAIDefaultConfig, axAIAzureOpenAIFastConfig, axAICohereCreativeConfig, axAICohereDefaultConfig, axAIDeepSeekCodeConfig, axAIDeepSeekDefaultConfig, axAIGoogleGeminiDefaultConfig, axAIGoogleGeminiDefaultCreativeConfig, axAIGrokBestConfig, axAIGrokDefaultConfig, axAIHuggingFaceCreativeConfig, axAIHuggingFaceDefaultConfig, axAIMistralBestConfig, axAIMistralDefaultConfig, axAIOllamaDefaultConfig, axAIOllamaDefaultCreativeConfig, axAIOpenAIBestConfig, axAIOpenAICreativeConfig, axAIOpenAIDefaultConfig, axAIOpenAIFastConfig, axAIOpenAIResponsesBestConfig, axAIOpenAIResponsesCreativeConfig, axAIOpenAIResponsesDefaultConfig, axAIRekaBestConfig, axAIRekaCreativeConfig, axAIRekaDefaultConfig, axAIRekaFastConfig, axAITogetherDefaultConfig, axBaseAIDefaultConfig, axBaseAIDefaultCreativeConfig, axGlobals, axModelInfoAnthropic, axModelInfoCohere, axModelInfoDeepSeek, axModelInfoGoogleGemini, axModelInfoGrok, axModelInfoGroq, axModelInfoHuggingFace, axModelInfoMistral, axModelInfoOpenAI, axModelInfoReka, axModelInfoTogether, axSpanAttributes, axSpanEvents, f, s };
+export { AxAI, AxAIAnthropic, type AxAIAnthropicArgs, type AxAIAnthropicChatError, type AxAIAnthropicChatRequest, type AxAIAnthropicChatRequestCacheParam, type AxAIAnthropicChatResponse, type AxAIAnthropicChatResponseDelta, type AxAIAnthropicConfig, type AxAIAnthropicContentBlockDeltaEvent, type AxAIAnthropicContentBlockStartEvent, type AxAIAnthropicContentBlockStopEvent, type AxAIAnthropicErrorEvent, type AxAIAnthropicMessageDeltaEvent, type AxAIAnthropicMessageStartEvent, type AxAIAnthropicMessageStopEvent, AxAIAnthropicModel, type AxAIAnthropicPingEvent, AxAIAnthropicVertexModel, type AxAIArgs, AxAIAzureOpenAI, type AxAIAzureOpenAIArgs, type AxAIAzureOpenAIConfig, AxAICohere, type AxAICohereArgs, type AxAICohereChatRequest, type AxAICohereChatRequestToolResults, type AxAICohereChatResponse, type AxAICohereChatResponseDelta, type AxAICohereChatResponseToolCalls, type AxAICohereConfig, AxAICohereEmbedModel, type AxAICohereEmbedRequest, type AxAICohereEmbedResponse, AxAICohereModel, AxAIDeepSeek, type AxAIDeepSeekArgs, AxAIDeepSeekModel, type AxAIEmbedModels, type AxAIFeatures, AxAIGoogleGemini, type AxAIGoogleGeminiArgs, type AxAIGoogleGeminiBatchEmbedRequest, type AxAIGoogleGeminiBatchEmbedResponse, type AxAIGoogleGeminiChatRequest, type AxAIGoogleGeminiChatResponse, type AxAIGoogleGeminiChatResponseDelta, type AxAIGoogleGeminiConfig, type AxAIGoogleGeminiContent, type AxAIGoogleGeminiContentPart, AxAIGoogleGeminiEmbedModel, AxAIGoogleGeminiEmbedTypes, type AxAIGoogleGeminiGenerationConfig, AxAIGoogleGeminiModel, type AxAIGoogleGeminiOptionsTools, AxAIGoogleGeminiSafetyCategory, type AxAIGoogleGeminiSafetySettings, AxAIGoogleGeminiSafetyThreshold, type AxAIGoogleGeminiThinkingConfig, type AxAIGoogleGeminiThinkingTokenBudgetLevels, type AxAIGoogleGeminiTool, type AxAIGoogleGeminiToolConfig, type AxAIGoogleGeminiToolFunctionDeclaration, type AxAIGoogleGeminiToolGoogleSearchRetrieval, type AxAIGoogleVertexBatchEmbedRequest, type AxAIGoogleVertexBatchEmbedResponse, AxAIGrok, type AxAIGrokArgs, type AxAIGrokChatRequest, AxAIGrokEmbedModels, AxAIGrokModel, type AxAIGrokOptionsTools, type AxAIGrokSearchSource, AxAIGroq, type AxAIGroqArgs, AxAIGroqModel, AxAIHuggingFace, type AxAIHuggingFaceArgs, type AxAIHuggingFaceConfig, AxAIHuggingFaceModel, type AxAIHuggingFaceRequest, type AxAIHuggingFaceResponse, type AxAIInputModelList, type AxAIMemory, AxAIMistral, type AxAIMistralArgs, type AxAIMistralChatRequest, AxAIMistralEmbedModels, AxAIMistralModel, type AxAIModelList, type AxAIModelListBase, type AxAIModels, AxAIOllama, type AxAIOllamaAIConfig, type AxAIOllamaArgs, AxAIOpenAI, type AxAIOpenAIArgs, AxAIOpenAIBase, type AxAIOpenAIBaseArgs, type AxAIOpenAIChatRequest, type AxAIOpenAIChatResponse, type AxAIOpenAIChatResponseDelta, type AxAIOpenAIConfig, AxAIOpenAIEmbedModel, type AxAIOpenAIEmbedRequest, type AxAIOpenAIEmbedResponse, type AxAIOpenAILogprob, AxAIOpenAIModel, type AxAIOpenAIResponseDelta, AxAIOpenAIResponses, type AxAIOpenAIResponsesArgs, AxAIOpenAIResponsesBase, type AxAIOpenAIResponsesCodeInterpreterToolCall, type AxAIOpenAIResponsesComputerToolCall, type AxAIOpenAIResponsesConfig, type AxAIOpenAIResponsesContentPartAddedEvent, type AxAIOpenAIResponsesContentPartDoneEvent, type AxAIOpenAIResponsesDefineFunctionTool, type AxAIOpenAIResponsesErrorEvent, type AxAIOpenAIResponsesFileSearchCallCompletedEvent, type AxAIOpenAIResponsesFileSearchCallInProgressEvent, type AxAIOpenAIResponsesFileSearchCallSearchingEvent, type AxAIOpenAIResponsesFileSearchToolCall, type AxAIOpenAIResponsesFunctionCallArgumentsDeltaEvent, type AxAIOpenAIResponsesFunctionCallArgumentsDoneEvent, type AxAIOpenAIResponsesFunctionCallItem, type AxAIOpenAIResponsesImageGenerationCallCompletedEvent, type AxAIOpenAIResponsesImageGenerationCallGeneratingEvent, type AxAIOpenAIResponsesImageGenerationCallInProgressEvent, type AxAIOpenAIResponsesImageGenerationCallPartialImageEvent, type AxAIOpenAIResponsesImageGenerationToolCall, AxAIOpenAIResponsesImpl, type AxAIOpenAIResponsesInputAudioContentPart, type AxAIOpenAIResponsesInputContentPart, type AxAIOpenAIResponsesInputFunctionCallItem, type AxAIOpenAIResponsesInputFunctionCallOutputItem, type AxAIOpenAIResponsesInputImageUrlContentPart, type AxAIOpenAIResponsesInputItem, type AxAIOpenAIResponsesInputMessageItem, type AxAIOpenAIResponsesInputTextContentPart, type AxAIOpenAIResponsesLocalShellToolCall, type AxAIOpenAIResponsesMCPCallArgumentsDeltaEvent, type AxAIOpenAIResponsesMCPCallArgumentsDoneEvent, type AxAIOpenAIResponsesMCPCallCompletedEvent, type AxAIOpenAIResponsesMCPCallFailedEvent, type AxAIOpenAIResponsesMCPCallInProgressEvent, type AxAIOpenAIResponsesMCPListToolsCompletedEvent, type AxAIOpenAIResponsesMCPListToolsFailedEvent, type AxAIOpenAIResponsesMCPListToolsInProgressEvent, type AxAIOpenAIResponsesMCPToolCall, AxAIOpenAIResponsesModel, type AxAIOpenAIResponsesOutputItem, type AxAIOpenAIResponsesOutputItemAddedEvent, type AxAIOpenAIResponsesOutputItemDoneEvent, type AxAIOpenAIResponsesOutputMessageItem, type AxAIOpenAIResponsesOutputRefusalContentPart, type AxAIOpenAIResponsesOutputTextAnnotationAddedEvent, type AxAIOpenAIResponsesOutputTextContentPart, type AxAIOpenAIResponsesOutputTextDeltaEvent, type AxAIOpenAIResponsesOutputTextDoneEvent, type AxAIOpenAIResponsesReasoningDeltaEvent, type AxAIOpenAIResponsesReasoningDoneEvent, type AxAIOpenAIResponsesReasoningItem, type AxAIOpenAIResponsesReasoningSummaryDeltaEvent, type AxAIOpenAIResponsesReasoningSummaryDoneEvent, type AxAIOpenAIResponsesReasoningSummaryPart, type AxAIOpenAIResponsesReasoningSummaryPartAddedEvent, type AxAIOpenAIResponsesReasoningSummaryPartDoneEvent, type AxAIOpenAIResponsesReasoningSummaryTextDeltaEvent, type AxAIOpenAIResponsesReasoningSummaryTextDoneEvent, type AxAIOpenAIResponsesRefusalDeltaEvent, type AxAIOpenAIResponsesRefusalDoneEvent, type AxAIOpenAIResponsesRequest, type AxAIOpenAIResponsesResponse, type AxAIOpenAIResponsesResponseCompletedEvent, type AxAIOpenAIResponsesResponseCreatedEvent, type AxAIOpenAIResponsesResponseDelta, type AxAIOpenAIResponsesResponseFailedEvent, type AxAIOpenAIResponsesResponseInProgressEvent, type AxAIOpenAIResponsesResponseIncompleteEvent, type AxAIOpenAIResponsesResponseQueuedEvent, type AxAIOpenAIResponsesStreamEvent, type AxAIOpenAIResponsesStreamEventBase, type AxAIOpenAIResponsesToolCall, type AxAIOpenAIResponsesToolCallBase, type AxAIOpenAIResponsesToolChoice, type AxAIOpenAIResponsesToolDefinition, type AxAIOpenAIResponsesWebSearchCallCompletedEvent, type AxAIOpenAIResponsesWebSearchCallInProgressEvent, type AxAIOpenAIResponsesWebSearchCallSearchingEvent, type AxAIOpenAIResponsesWebSearchToolCall, type AxAIOpenAIUsage, type AxAIPromptConfig, AxAIReka, type AxAIRekaArgs, type AxAIRekaChatRequest, type AxAIRekaChatResponse, type AxAIRekaChatResponseDelta, type AxAIRekaConfig, AxAIRekaModel, type AxAIRekaUsage, type AxAIService, AxAIServiceAbortedError, type AxAIServiceActionOptions, AxAIServiceAuthenticationError, AxAIServiceError, type AxAIServiceImpl, type AxAIServiceMetrics, AxAIServiceNetworkError, type AxAIServiceOptions, AxAIServiceResponseError, AxAIServiceStatusError, AxAIServiceStreamTerminatedError, AxAIServiceTimeoutError, AxAITogether, type AxAITogetherArgs, type AxAPI, type AxAPIConfig, AxAgent, type AxAgentFeatures, type AxAgentOptions, type AxAgentic, AxApacheTika, type AxApacheTikaArgs, type AxApacheTikaConvertOptions, type AxAssertion, AxAssertionError, AxBalancer, type AxBalancerOptions, AxBaseAI, type AxBaseAIArgs, AxBaseOptimizer, type AxBootstrapCompileOptions, AxBootstrapFewShot, type AxBootstrapOptimizerOptions, AxChainOfThought, type AxChatRequest, type AxChatResponse, type AxChatResponseFunctionCall, type AxChatResponseResult, type AxCheckpointLoadFn, type AxCheckpointSaveFn, type AxCompileOptions, type AxCostTracker, type AxCostTrackerOptions, AxDB, type AxDBArgs, AxDBBase, type AxDBBaseArgs, type AxDBBaseOpOptions, AxDBCloudflare, type AxDBCloudflareArgs, type AxDBCloudflareOpOptions, type AxDBLoaderOptions, AxDBManager, type AxDBManagerArgs, type AxDBMatch, AxDBMemory, type AxDBMemoryArgs, type AxDBMemoryOpOptions, AxDBPinecone, type AxDBPineconeArgs, type AxDBPineconeOpOptions, type AxDBQueryRequest, type AxDBQueryResponse, type AxDBQueryService, type AxDBService, type AxDBState, type AxDBUpsertRequest, type AxDBUpsertResponse, AxDBWeaviate, type AxDBWeaviateArgs, type AxDBWeaviateOpOptions, type AxDataRow, AxDefaultCostTracker, AxDefaultQueryRewriter, AxDefaultResultReranker, type AxDockerContainer, AxDockerSession, type AxEmbedRequest, type AxEmbedResponse, AxEmbeddingAdapter, AxEvalUtil, type AxEvaluateArgs, type AxExample, type AxField, type AxFieldDescriptor, type AxFieldProcessor, type AxFieldProcessorProcess, type AxFieldTemplateFn, type AxFieldType, type AxFieldValue, type AxFunction, AxFunctionError, type AxFunctionHandler, type AxFunctionJSONSchema, AxFunctionProcessor, AxGen, type AxGenDeltaOut, type AxGenIn, type AxGenOut, type AxGenStreamingOut, AxGenerateError, type AxGenerateErrorDetails, type AxGenerateResult, AxHFDataLoader, type AxIField, type AxInputFunctionType, AxInstanceRegistry, type AxInternalChatRequest, type AxInternalEmbedRequest, AxJSInterpreter, AxJSInterpreterPermission, AxLLMRequestTypeValues, type AxLoggerFunction, type AxLoggerTag, AxMCPClient, AxMCPHTTPSSETransport, AxMCPStdioTransport, type AxMCPStreamableHTTPTransportOptions, AxMCPStreambleHTTPTransport, type AxMCPTransport, AxMemory, type AxMessage, type AxMetricFn, type AxMetricFnArgs, AxMiPRO, type AxMiPROCompileOptions, type AxMiPROOptimizerOptions, type AxMiPROResult, AxMockAIService, type AxMockAIServiceConfig, type AxModelConfig, type AxModelInfo, type AxModelInfoWithProvider, type AxModelUsage, type AxMultiMetricFn, AxMultiServiceRouter, type AxOptimizationCheckpoint, type AxOptimizationProgress, type AxOptimizationStats, type AxOptimizer, type AxOptimizerArgs, type AxOptimizerResult, type AxParetoResult, AxProgram, type AxProgramDemos, type AxProgramExamples, type AxProgramForwardOptions, type AxProgramStreamingForwardOptions, type AxProgramTrace, type AxProgramUsage, AxProgramWithSignature, type AxProgramWithSignatureOptions, AxPromptTemplate, type AxPromptTemplateOptions, AxRAG, type AxRateLimiterFunction, AxRateLimiterTokenUsage, type AxRateLimiterTokenUsageOptions, type AxRerankerIn, type AxRerankerOut, type AxResponseHandlerArgs, type AxRewriteIn, type AxRewriteOut, AxRewriter, type AxSetExamplesOptions, AxSignature, type AxSignatureConfig, type AxSignatureTemplateValue, AxSimpleClassifier, AxSimpleClassifierClass, type AxSimpleClassifierForwardOptions, AxSpanKindValues, type AxStreamingAssertion, type AxStreamingEvent, type AxStreamingFieldProcessorProcess, AxStringUtil, AxTestPrompt, type AxTokenUsage, type AxTunable, type AxUsable, ax, axAIAnthropicDefaultConfig, axAIAnthropicVertexDefaultConfig, axAIAzureOpenAIBestConfig, axAIAzureOpenAICreativeConfig, axAIAzureOpenAIDefaultConfig, axAIAzureOpenAIFastConfig, axAICohereCreativeConfig, axAICohereDefaultConfig, axAIDeepSeekCodeConfig, axAIDeepSeekDefaultConfig, axAIGoogleGeminiDefaultConfig, axAIGoogleGeminiDefaultCreativeConfig, axAIGrokBestConfig, axAIGrokDefaultConfig, axAIHuggingFaceCreativeConfig, axAIHuggingFaceDefaultConfig, axAIMistralBestConfig, axAIMistralDefaultConfig, axAIOllamaDefaultConfig, axAIOllamaDefaultCreativeConfig, axAIOpenAIBestConfig, axAIOpenAICreativeConfig, axAIOpenAIDefaultConfig, axAIOpenAIFastConfig, axAIOpenAIResponsesBestConfig, axAIOpenAIResponsesCreativeConfig, axAIOpenAIResponsesDefaultConfig, axAIRekaBestConfig, axAIRekaCreativeConfig, axAIRekaDefaultConfig, axAIRekaFastConfig, axAITogetherDefaultConfig, axBaseAIDefaultConfig, axBaseAIDefaultCreativeConfig, axGlobals, axModelInfoAnthropic, axModelInfoCohere, axModelInfoDeepSeek, axModelInfoGoogleGemini, axModelInfoGrok, axModelInfoGroq, axModelInfoHuggingFace, axModelInfoMistral, axModelInfoOpenAI, axModelInfoReka, axModelInfoTogether, axSpanAttributes, axSpanEvents, f, s };