@ax-llm/ax 12.0.9 → 12.0.13

package/index.d.ts

@@ -209,7 +209,13 @@ type AxFunction = {
     parameters?: AxFunctionJSONSchema;
     func: AxFunctionHandler;
 };
+type AxFunctionResult = Extract<AxChatRequest['chatPrompt'][number], {
+    role: 'function';
+}> & {
+    index: number;
+};
 type AxChatResponseResult = {
+    index: number;
     content?: string;
     thought?: string;
     name?: string;
@@ -415,8 +421,21 @@ declare enum AxAIAnthropicVertexModel {
     Claude3Haiku = "claude-3-haiku",
     Claude3Opus = "claude-3-opus"
 }
+type AxAIAnthropicThinkingConfig = {
+    type: 'enabled';
+    budget_tokens: number;
+};
+type AxAIAnthropicThinkingTokenBudgetLevels = {
+    minimal?: number;
+    low?: number;
+    medium?: number;
+    high?: number;
+    highest?: number;
+};
 type AxAIAnthropicConfig = AxModelConfig & {
     model: AxAIAnthropicModel | AxAIAnthropicVertexModel;
+    thinking?: AxAIAnthropicThinkingConfig;
+    thinkingTokenBudgetLevels?: AxAIAnthropicThinkingTokenBudgetLevels;
 };
 type AxAIAnthropicChatRequestCacheParam = {
     cache_control?: {
@@ -464,6 +483,14 @@ type AxAIAnthropicChatRequest = {
             id: string;
             name: string;
             input: object;
+        } | {
+            type: 'thinking';
+            thinking: string;
+            signature?: string;
+        } | {
+            type: 'redacted_thinking';
+            thinking: string;
+            signature?: string;
         })[];
     })[];
     tools?: ({
@@ -487,6 +514,7 @@ type AxAIAnthropicChatRequest = {
     temperature?: number;
     top_p?: number;
     top_k?: number;
+    thinking?: AxAIAnthropicThinkingConfig;
     metadata?: {
         user_id: string;
     };
@@ -503,6 +531,14 @@ type AxAIAnthropicChatResponse = {
         name: string;
         type: 'tool_use';
         input?: string;
+    } | {
+        type: 'thinking';
+        thinking: string;
+        signature?: string;
+    } | {
+        type: 'redacted_thinking';
+        thinking: string;
+        signature?: string;
     })[];
     model: string;
     stop_reason: 'end_turn' | 'max_tokens' | 'stop_sequence' | 'tool_use';
@@ -546,6 +582,9 @@ interface AxAIAnthropicContentBlockStartEvent {
         id: string;
         name: string;
         input: object;
+    } | {
+        type: 'thinking';
+        thinking: string;
     };
 }
 interface AxAIAnthropicContentBlockDeltaEvent {
@@ -557,6 +596,12 @@ interface AxAIAnthropicContentBlockDeltaEvent {
     } | {
         type: 'input_json_delta';
         partial_json: string;
+    } | {
+        type: 'thinking_delta';
+        thinking: string;
+    } | {
+        type: 'signature_delta';
+        signature: string;
     };
 }
 interface AxAIAnthropicContentBlockStopEvent {
@@ -2086,20 +2131,26 @@ declare class AxAIGrok extends AxAIOpenAIBase<AxAIGrokModel, AxAIGrokEmbedModels
     constructor({ apiKey, config, options, models, modelInfo, }: Readonly<Omit<AxAIGrokArgs, 'name'>>);
 }
 
+type AxMemoryData = {
+    tags?: string[];
+    role: AxChatRequest['chatPrompt'][number]['role'];
+    chat: {
+        index: number;
+        value: Omit<AxChatRequest['chatPrompt'][number], 'role'>;
+    }[];
+}[];
 interface AxAIMemory {
-    add(result: Readonly<AxChatRequest['chatPrompt']> | Readonly<AxChatRequest['chatPrompt'][number]>, sessionId?: string): void;
-    addResult(result: Readonly<AxChatResponseResult>, sessionId?: string): void;
-    updateResult(result: Readonly<AxChatResponseResult> & {
+    addRequest(result: AxChatRequest['chatPrompt'], sessionId?: string): void;
+    addResponse(results: Readonly<AxChatResponseResult[]>, sessionId?: string): void;
+    updateResult(results: Readonly<AxChatResponseResult> & {
         delta?: string;
     }, sessionId?: string): void;
-    history(sessionId?: string): AxChatRequest['chatPrompt'];
+    addFunctionResults(results: Readonly<AxFunctionResult[]>, sessionId?: string): void;
+    history(index: number, sessionId?: string): AxChatRequest['chatPrompt'];
     reset(sessionId?: string): void;
-    getLast(sessionId?: string): {
-        chat: AxChatRequest['chatPrompt'][number];
-        tags?: string[];
-    } | undefined;
+    getLast(sessionId?: string): AxMemoryData[number] | undefined;
     addTag(name: string, sessionId?: string): void;
-    rewindToTag(name: string, sessionId?: string): AxChatRequest['chatPrompt'];
+    rewindToTag(name: string, sessionId?: string): AxMemoryData;
 }
 
 interface AxField {
@@ -2157,6 +2208,33 @@ declare class AxSignature {
     };
 }
 
+type AxFieldValue = string | string[] | number | boolean | object | null | undefined | {
+    mimeType: string;
+    data: string;
+} | {
+    mimeType: string;
+    data: string;
+}[] | {
+    format?: 'wav';
+    data: string;
+} | {
+    format?: 'wav';
+    data: string;
+}[];
+type AxGenIn = {
+    [key: string]: AxFieldValue;
+};
+type AxGenOut = {
+    [key: string]: AxFieldValue;
+};
+type AxMessage<IN extends AxGenIn> = {
+    role: 'user';
+    values: IN;
+} | {
+    role: 'assistant';
+    values: IN;
+};
+
 interface AxAssertion {
     fn(values: Record<string, unknown>): Promise<boolean | undefined> | boolean | undefined;
     message?: string;
@@ -2179,63 +2257,23 @@ declare class AxAssertionError extends Error {
 }
 
 declare class AxMemory implements AxAIMemory {
-    private limit;
     private options?;
     private memories;
     private defaultMemory;
-    constructor(limit?: number, options?: {
+    constructor(options?: {
         debug?: boolean;
         debugHideSystemPrompt?: boolean;
     } | undefined);
     private getMemory;
-    add(value: AxChatRequest['chatPrompt'][number] | AxChatRequest['chatPrompt'], sessionId?: string): void;
-    addResult(result: Readonly<AxChatResponseResult>, sessionId?: string): void;
-    updateResult(result: Readonly<AxChatResponseResult>, sessionId?: string): void;
+    addRequest(value: AxChatRequest['chatPrompt'], sessionId?: string): void;
+    addResponse(results: Readonly<AxChatResponseResult[]>, sessionId?: string): void;
+    addFunctionResults(results: Readonly<AxFunctionResult[]>, sessionId?: string): void;
+    updateResult(result: Readonly<AxChatResponseResult & {
+        delta?: string;
+    }>, sessionId?: string): void;
     addTag(name: string, sessionId?: string): void;
-    rewindToTag(name: string, sessionId?: string): ({
-        role: "system";
-        content: string;
-        cache?: boolean;
-    } | {
-        role: "user";
-        name?: string;
-        content: string | ({
-            type: "text";
-            text: string;
-            cache?: boolean;
-        } | {
-            type: "image";
-            mimeType: string;
-            image: string;
-            details?: "high" | "low" | "auto";
-            cache?: boolean;
-        } | {
-            type: "audio";
-            data: string;
-            format?: "wav";
-            cache?: boolean;
-        })[];
-    } | {
-        role: "assistant";
-        content?: string;
-        name?: string;
-        functionCalls?: {
-            id: string;
-            type: "function";
-            function: {
-                name: string;
-                params?: string | object;
-            };
-        }[];
-        cache?: boolean;
-    } | {
-        role: "function";
-        result: string;
-        isError?: boolean;
-        functionId: string;
-        cache?: boolean;
-    })[];
-    history(sessionId?: string): ({
+    rewindToTag(name: string, sessionId?: string): AxMemoryData;
+    history(index: number, sessionId?: string): ({
         role: "system";
         content: string;
         cache?: boolean;
@@ -2279,8 +2317,12 @@ declare class AxMemory implements AxAIMemory {
         cache?: boolean;
     })[];
     getLast(sessionId?: string): {
-        chat: AxChatRequest["chatPrompt"][number];
         tags?: string[];
+        role: AxChatRequest["chatPrompt"][number]["role"];
+        chat: {
+            index: number;
+            value: Omit<AxChatRequest["chatPrompt"][number], "role">;
+        }[];
     } | undefined;
     reset(sessionId?: string): void;
 }
@@ -2312,33 +2354,6 @@ type AxInputFunctionType = (AxFunction | {
     toFunction: () => AxFunction | AxFunction[];
 })[];
 
-type AxFieldValue = string | string[] | number | boolean | object | null | undefined | {
-    mimeType: string;
-    data: string;
-} | {
-    mimeType: string;
-    data: string;
-}[] | {
-    format?: 'wav';
-    data: string;
-} | {
-    format?: 'wav';
-    data: string;
-}[];
-type AxGenIn = {
-    [key: string]: AxFieldValue;
-};
-type AxGenOut = {
-    [key: string]: AxFieldValue;
-};
-type AxMessage<IN extends AxGenIn> = {
-    role: 'user';
-    values: IN;
-} | {
-    role: 'assistant';
-    values: IN;
-};
-
 type Writeable<T> = {
     -readonly [P in keyof T]: T[P];
 };
@@ -2364,7 +2379,9 @@ declare class AxPromptTemplate {
         skipSystemPrompt?: boolean;
         examples?: Record<string, AxFieldValue>[];
         demos?: Record<string, AxFieldValue>[];
-    }>) => AxChatRequest["chatPrompt"];
+    }>) => Extract<AxChatRequest["chatPrompt"][number], {
+        role: "user" | "system" | "assistant";
+    }>[];
     renderExtraFields: (extraFields: readonly AxIField[]) => ({
         type: "text";
         text: string;
@@ -2397,6 +2414,25 @@ type AxProgramDemos<IN extends AxGenIn, OUT extends AxGenOut> = {
     programId: string;
 };
 type AxProgramExamples<IN extends AxGenIn, OUT extends AxGenOut> = AxProgramDemos<IN, OUT> | AxProgramDemos<IN, OUT>['traces'];
+type AxResultPickerFunctionFieldResults<OUT extends AxGenOut> = {
+    type: 'fields';
+    results: readonly {
+        index: number;
+        sample: Partial<OUT>;
+    }[];
+};
+type AxResultPickerFunctionFunctionResults = {
+    type: 'function';
+    results: readonly {
+        index: number;
+        functionName: string;
+        functionId: string;
+        args: string | object;
+        result: string;
+        isError?: boolean;
+    }[];
+};
+type AxResultPickerFunction<OUT extends AxGenOut> = (data: AxResultPickerFunctionFieldResults<OUT> | AxResultPickerFunctionFunctionResults) => number | Promise<number>;
 type AxProgramForwardOptions = {
     maxRetries?: number;
     maxSteps?: number;
@@ -2409,6 +2445,8 @@ type AxProgramForwardOptions = {
     tracer?: Tracer;
     rateLimiter?: AxRateLimiterFunction;
     stream?: boolean;
+    sampleCount?: number;
+    resultPicker?: AxResultPickerFunction<AxGenOut>;
     functions?: AxInputFunctionType;
     functionCall?: AxChatRequest['functionCall'];
     stopFunction?: string;
@@ -2431,9 +2469,10 @@ type AxProgramForwardOptions = {
 type AxProgramStreamingForwardOptions = Omit<AxProgramForwardOptions, 'stream'>;
 type AxGenDeltaOut<OUT extends AxGenOut> = {
     version: number;
+    index: number;
     delta: Partial<OUT>;
 };
-type AxGenStreamingOut<OUT extends AxGenOut> = AsyncGenerator<AxGenDeltaOut<OUT>, void | OUT, unknown>;
+type AxGenStreamingOut<OUT extends AxGenOut> = AsyncGenerator<AxGenDeltaOut<OUT>, void, unknown>;
 type AxSetExamplesOptions = {};
 interface AxTunable<IN extends AxGenIn, OUT extends AxGenOut> {
     setExamples: (examples: Readonly<AxProgramExamples<IN, OUT>>, options?: Readonly<AxSetExamplesOptions>) => void;
@@ -3271,109 +3310,6 @@ declare class AxDBManager {
     }> | undefined) => Promise<AxDBMatch[][]>;
 }
 
-type AxFieldProcessorProcess = (value: AxFieldValue, context?: Readonly<{
-    values?: AxGenOut;
-    sessionId?: string;
-    done?: boolean;
-}>) => unknown | Promise<unknown>;
-type AxStreamingFieldProcessorProcess = (value: string, context?: Readonly<{
-    values?: AxGenOut;
-    sessionId?: string;
-    done?: boolean;
-}>) => unknown | Promise<unknown>;
-interface AxFieldProcessor {
-    field: Readonly<AxField>;
-    /**
-     * Process the field value and return a new value (or undefined if no update is needed).
-     * The returned value may be merged back into memory.
-     * @param value - The current field value.
-     * @param context - Additional context (e.g. memory and session id).
-     */
-    process: AxFieldProcessorProcess | AxStreamingFieldProcessorProcess;
-}
-
-type AxGenerateResult<OUT extends AxGenOut> = OUT & {
-    thought?: string;
-};
-interface AxResponseHandlerArgs<T> {
-    ai: Readonly<AxAIService>;
-    model?: string;
-    res: T;
-    mem: AxAIMemory;
-    sessionId?: string;
-    traceId?: string;
-    functions?: Readonly<AxFunction[]>;
-    strictMode?: boolean;
-    span?: Span;
-}
-interface AxStreamingEvent<T> {
-    event: 'delta' | 'done' | 'error';
-    data: {
-        contentDelta?: string;
-        partialValues?: Partial<T>;
-        error?: string;
-        functions?: AxChatResponseFunctionCall[];
-    };
-}
-declare class AxGen<IN extends AxGenIn, OUT extends AxGenerateResult<AxGenOut> = AxGenerateResult<AxGenOut>> extends AxProgramWithSignature<IN, OUT> {
-    private promptTemplate;
-    private asserts;
-    private streamingAsserts;
-    private options?;
-    private functions?;
-    private functionsExecuted;
-    private fieldProcessors;
-    private streamingFieldProcessors;
-    private values;
-    private excludeContentFromTrace;
-    private thoughtFieldName;
-    constructor(signature: NonNullable<ConstructorParameters<typeof AxSignature>[0]>, options?: Readonly<AxProgramForwardOptions>);
-    addAssert: (fn: AxAssertion["fn"], message?: string) => void;
-    addStreamingAssert: (fieldName: string, fn: AxStreamingAssertion["fn"], message?: string) => void;
-    private addFieldProcessorInternal;
-    addStreamingFieldProcessor: (fieldName: string, fn: AxFieldProcessor["process"]) => void;
-    addFieldProcessor: (fieldName: string, fn: AxFieldProcessor["process"]) => void;
-    private forwardSendRequest;
-    private forwardCore;
-    private processStreamingResponse;
-    private processResponse;
-    private _forward2;
-    private shouldContinueSteps;
-    _forward1(ai: Readonly<AxAIService>, values: IN | AxMessage<IN>[], options: Readonly<AxProgramForwardOptions>): AsyncGenerator<{
-        version: number;
-        delta: Partial<OUT>;
-    }, void, unknown>;
-    forward(ai: Readonly<AxAIService>, values: IN | AxMessage<IN>[], options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
-    streamingForward(ai: Readonly<AxAIService>, values: IN | AxMessage<IN>[], options?: Readonly<AxProgramStreamingForwardOptions>): AsyncGenerator<{
-        version: number;
-        delta: Partial<OUT>;
-    }, void, unknown>;
-    setExamples(examples: Readonly<AxProgramExamples<IN, OUT>>, options?: Readonly<AxSetExamplesOptions>): void;
-    private isDebug;
-    private getLogger;
-}
-type AxGenerateErrorDetails = {
-    model?: string;
-    maxTokens?: number;
-    streaming: boolean;
-    signature: {
-        input: Readonly<AxIField[]>;
-        output: Readonly<AxIField[]>;
-        description?: string;
-    };
-};
-declare class AxGenerateError extends Error {
-    readonly details: AxGenerateErrorDetails;
-    constructor(message: string, details: Readonly<AxGenerateErrorDetails>, options?: ErrorOptions);
-}
-
-declare class AxDefaultQueryRewriter extends AxGen<AxRewriteIn, AxRewriteOut> {
-    constructor(options?: Readonly<AxProgramForwardOptions>);
-}
-declare class AxRewriter extends AxGen<AxRewriteIn, AxRewriteOut> {
-    constructor(options?: Readonly<AxProgramForwardOptions>);
-}
-
 interface AxDockerContainer {
     Id: string;
     Names: string[];
@@ -3475,6 +3411,95 @@ declare class AxDockerSession {
     toFunction(): AxFunction;
 }
 
+type AxFieldProcessorProcess = (value: AxFieldValue, context?: Readonly<{
+    values?: AxGenOut;
+    sessionId?: string;
+    done?: boolean;
+}>) => unknown | Promise<unknown>;
+type AxStreamingFieldProcessorProcess = (value: string, context?: Readonly<{
+    values?: AxGenOut;
+    sessionId?: string;
+    done?: boolean;
+}>) => unknown | Promise<unknown>;
+interface AxFieldProcessor {
+    field: Readonly<AxField>;
+    /**
+     * Process the field value and return a new value (or undefined if no update is needed).
+     * The returned value may be merged back into memory.
+     * @param value - The current field value.
+     * @param context - Additional context (e.g. memory and session id).
+     */
+    process: AxFieldProcessorProcess | AxStreamingFieldProcessorProcess;
+}
+
+type AxGenerateResult<OUT extends AxGenOut> = OUT & {
+    thought?: string;
+};
+interface AxResponseHandlerArgs<T> {
+    ai: Readonly<AxAIService>;
+    model?: string;
+    res: T;
+    mem: AxAIMemory;
+    sessionId?: string;
+    traceId?: string;
+    functions: Readonly<AxFunction[]>;
+    strictMode?: boolean;
+    span?: Span;
+}
+interface AxStreamingEvent<T> {
+    event: 'delta' | 'done' | 'error';
+    data: {
+        contentDelta?: string;
+        partialValues?: Partial<T>;
+        error?: string;
+        functions?: AxChatResponseFunctionCall[];
+    };
+}
+declare class AxGen<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxProgramWithSignature<IN, OUT> {
+    private promptTemplate;
+    private asserts;
+    private streamingAsserts;
+    private options?;
+    private functions?;
+    private fieldProcessors;
+    private streamingFieldProcessors;
+    private excludeContentFromTrace;
+    private thoughtFieldName;
+    constructor(signature: NonNullable<ConstructorParameters<typeof AxSignature>[0]>, options?: Readonly<AxProgramForwardOptions>);
+    private createStates;
+    addAssert: (fn: AxAssertion["fn"], message?: string) => void;
+    addStreamingAssert: (fieldName: string, fn: AxStreamingAssertion["fn"], message?: string) => void;
+    private addFieldProcessorInternal;
+    addStreamingFieldProcessor: (fieldName: string, fn: AxFieldProcessor["process"]) => void;
+    addFieldProcessor: (fieldName: string, fn: AxFieldProcessor["process"]) => void;
+    private forwardSendRequest;
+    private forwardCore;
+    private _forward2;
+    _forward1(ai: Readonly<AxAIService>, values: IN | AxMessage<IN>[], options: Readonly<AxProgramForwardOptions>): AxGenStreamingOut<OUT>;
+    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>;
+    setExamples(examples: Readonly<AxProgramExamples<IN, OUT>>, options?: Readonly<AxSetExamplesOptions>): void;
+    private isDebug;
+    private getLogger;
+}
+type AxGenerateErrorDetails = {
+    model?: string;
+    maxTokens?: number;
+    streaming: boolean;
+    signature: {
+        input: Readonly<AxIField[]>;
+        output: Readonly<AxIField[]>;
+        description?: string;
+    };
+};
+type ErrorOptions = {
+    cause?: Error;
+};
+declare class AxGenerateError extends Error {
+    readonly details: AxGenerateErrorDetails;
+    constructor(message: string, details: Readonly<AxGenerateErrorDetails>, options?: ErrorOptions);
+}
+
 type AxDataRow = {
     row: Record<string, AxFieldValue>;
 };
@@ -3934,6 +3959,20 @@ declare const axCreateOptimizerLogger: (output?: (message: string) => void) => A
  */
 declare const axDefaultOptimizerLogger: AxLoggerFunction;
 
+type AxChatRequestMessage = AxChatRequest['chatPrompt'][number];
+/**
+ * Validates a chat request message item to ensure it meets the required criteria
+ * @param item - The chat request message to validate
+ * @throws {Error} When validation fails with a descriptive error message
+ */
+declare function axValidateChatRequestMessage(item: AxChatRequestMessage): void;
+/**
+ * Validates a chat response result to ensure it meets the required criteria
+ * @param results - The chat response results to validate (single result or array)
+ * @throws {Error} When validation fails with a descriptive error message
+ */
+declare function axValidateChatResponseResult(results: Readonly<AxChatResponseResult[]> | Readonly<AxChatResponseResult>): void;
+
 declare class AxAIOpenAIResponsesImpl<TModel, TEmbedModel, // Kept for interface compatibility, but not used by this impl.
 TResponsesReq extends AxAIOpenAIResponsesRequest<TModel>> implements AxAIServiceImpl<TModel, TEmbedModel, Readonly<AxAIOpenAIResponsesRequest<TModel>>, // ChatReq (now ResponsesReq)
 Readonly<AxAIOpenAIEmbedRequest<TEmbedModel>>, // EmbedReq
@@ -4273,4 +4312,8 @@ 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, 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, axCreateDefaultLogger, axCreateDefaultTextLogger, axCreateOptimizerLogger, axDefaultOptimizerLogger, axGlobals, axModelInfoAnthropic, axModelInfoCohere, axModelInfoDeepSeek, axModelInfoGoogleGemini, axModelInfoGrok, axModelInfoGroq, axModelInfoHuggingFace, axModelInfoMistral, axModelInfoOpenAI, axModelInfoReka, axModelInfoTogether, axSpanAttributes, axSpanEvents, f, s };
+interface AxSamplePickerOptions<OUT extends AxGenOut> {
+    resultPicker?: AxResultPickerFunction<OUT>;
+}
+
+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, type AxAIAnthropicThinkingConfig, type AxAIAnthropicThinkingTokenBudgetLevels, 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, 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, type AxFunctionResult, 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 AxMemoryData, 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 AxResultPickerFunction, type AxResultPickerFunctionFieldResults, type AxResultPickerFunctionFunctionResults, type AxRewriteIn, type AxRewriteOut, type AxSamplePickerOptions, 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, axCreateDefaultLogger, axCreateDefaultTextLogger, axCreateOptimizerLogger, axDefaultOptimizerLogger, axGlobals, axModelInfoAnthropic, axModelInfoCohere, axModelInfoDeepSeek, axModelInfoGoogleGemini, axModelInfoGrok, axModelInfoGroq, axModelInfoHuggingFace, axModelInfoMistral, axModelInfoOpenAI, axModelInfoReka, axModelInfoTogether, axSpanAttributes, axSpanEvents, axValidateChatRequestMessage, axValidateChatResponseResult, f, s };