@ax-llm/ax 10.0.49 → 11.0.0

package/index.d.ts

@@ -61,7 +61,11 @@ declare class AxAIServiceAuthenticationError extends AxAIServiceError {
     constructor(url: string, requestBody?: unknown, context?: Record<string, unknown>);
 }
 
-type AxAIModelMap = Record<string, string>;
+type AxAIModelList<T = string> = {
+    key: string;
+    model: T;
+    description: string;
+}[];
 type AxModelInfo = {
     name: string;
     currency?: string;
@@ -251,11 +255,8 @@ interface AxAIService {
     getName(): string;
     getModelInfo(): Readonly<AxModelInfoWithProvider>;
     getEmbedModelInfo(): Readonly<AxModelInfoWithProvider> | undefined;
-    getFeatures(model?: string): {
-        functions: boolean;
-        streaming: boolean;
-    };
-    getModelMap(): AxAIModelMap | undefined;
+    getFeatures(model?: string): AxAIFeatures;
+    getModelList(): AxAIModelList | undefined;
     getMetrics(): AxAIServiceMetrics;
     chat(req: Readonly<AxChatRequest>, options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions>): Promise<AxChatResponse | ReadableStream$1<AxChatResponse>>;
     embed(req: Readonly<AxEmbedRequest>, options?: Readonly<AxAIServiceActionOptions & AxAIServiceActionOptions>): Promise<AxEmbedResponse>;
@@ -271,22 +272,23 @@ interface AxAIServiceImpl<TChatRequest, TEmbedRequest, TChatResponse, TChatRespo
     getModelConfig(): AxModelConfig;
 }
 
-interface AxBaseAIFeatures {
+interface AxAIFeatures {
     functions: boolean;
     streaming: boolean;
+    functionCot?: boolean;
 }
 interface AxBaseAIArgs {
     name: string;
     apiURL: string;
     headers: () => Promise<Record<string, string>>;
     modelInfo: Readonly<AxModelInfo[]>;
-    models: Readonly<{
+    defaults: Readonly<{
         model: string;
         embedModel?: string;
     }>;
     options?: Readonly<AxAIServiceOptions>;
-    supportFor: AxBaseAIFeatures | ((model: string) => AxBaseAIFeatures);
-    modelMap?: AxAIModelMap;
+    supportFor: AxAIFeatures | ((model: string) => AxAIFeatures);
+    models?: AxAIModelList;
 }
 declare class AxBaseAI<TChatRequest, TEmbedRequest, TChatResponse, TChatResponseDelta, TEmbedResponse> implements AxAIService {
     private readonly aiImpl;
@@ -294,17 +296,17 @@ declare class AxBaseAI<TChatRequest, TEmbedRequest, TChatResponse, TChatResponse
     private rt?;
     private fetch?;
     private tracer?;
-    private modelMap?;
+    private models?;
     private modelInfo;
     private modelUsage?;
     private embedModelUsage?;
-    private models;
+    private defaults;
     protected apiURL: string;
     protected name: string;
     protected headers: () => Promise<Record<string, string>>;
-    protected supportFor: AxBaseAIFeatures | ((model: string) => AxBaseAIFeatures);
+    protected supportFor: AxAIFeatures | ((model: string) => AxAIFeatures);
     private metrics;
-    constructor(aiImpl: Readonly<AxAIServiceImpl<TChatRequest, TEmbedRequest, TChatResponse, TChatResponseDelta, TEmbedResponse>>, { name, apiURL, headers, modelInfo, models, options, supportFor, modelMap, }: Readonly<AxBaseAIArgs>);
+    constructor(aiImpl: Readonly<AxAIServiceImpl<TChatRequest, TEmbedRequest, TChatResponse, TChatResponseDelta, TEmbedResponse>>, { name, apiURL, headers, modelInfo, defaults, options, supportFor, models, }: Readonly<AxBaseAIArgs>);
     setName(name: string): void;
     setAPIURL(apiURL: string): void;
     setHeaders(headers: () => Promise<Record<string, string>>): void;
@@ -312,9 +314,9 @@ declare class AxBaseAI<TChatRequest, TEmbedRequest, TChatResponse, TChatResponse
     getOptions(): Readonly<AxAIServiceOptions>;
     getModelInfo(): Readonly<AxModelInfoWithProvider>;
     getEmbedModelInfo(): AxModelInfoWithProvider | undefined;
-    getModelMap(): AxAIModelMap | undefined;
+    getModelList(): AxAIModelList | undefined;
     getName(): string;
-    getFeatures(model?: string): AxBaseAIFeatures;
+    getFeatures(model?: string): AxAIFeatures;
     private calculatePercentile;
     private updateLatencyMetrics;
     private updateErrorMetrics;
@@ -524,10 +526,10 @@ interface AxAIAnthropicArgs {
     region?: string;
     config?: Readonly<Partial<AxAIAnthropicConfig>>;
     options?: Readonly<AxAIServiceOptions>;
-    modelMap?: Record<string, AxAIAnthropicModel | string>;
+    models?: AxAIModelList<AxAIAnthropicModel | string>;
 }
 declare class AxAIAnthropic extends AxBaseAI<AxAIAnthropicChatRequest, unknown, AxAIAnthropicChatResponse, AxAIAnthropicChatResponseDelta, unknown> {
-    constructor({ apiKey, projectId, region, config, options, modelMap, }: Readonly<Omit<AxAIAnthropicArgs, 'name'>>);
+    constructor({ apiKey, projectId, region, config, options, models, }: Readonly<Omit<AxAIAnthropicArgs, 'name'>>);
 }
 
 declare enum AxAIOpenAIModel {
@@ -550,9 +552,9 @@ declare enum AxAIOpenAIEmbedModel {
     TextEmbedding3Small = "text-embedding-3-small",
     TextEmbedding3Large = "text-embedding-3-large"
 }
-type AxAIOpenAIConfig = Omit<AxModelConfig, 'topK'> & {
-    model: AxAIOpenAIModel | string;
-    embedModel?: AxAIOpenAIEmbedModel | string;
+type AxAIOpenAIConfig<TModel, TEmbedModel> = Omit<AxModelConfig, 'topK'> & {
+    model: TModel | string;
+    embedModel?: TEmbedModel | string;
     user?: string;
     responseFormat?: 'json_object';
     bestOf?: number;
@@ -712,33 +714,29 @@ type AxAIOpenAIEmbedResponse = {
     usage: AxAIOpenAIUsage;
 };
 
-interface AxAIOpenAIArgs {
-    name: 'openai';
+interface AxAIOpenAIArgs<TName = 'openai', TConfig = AxAIOpenAIConfig<AxAIOpenAIModel | string, AxAIOpenAIEmbedModel | string>, TModels = AxAIOpenAIModel | AxAIOpenAIEmbedModel> {
+    name: TName;
     apiKey: string;
     apiURL?: string;
-    config?: Readonly<Partial<AxAIOpenAIConfig>>;
+    config?: Readonly<Partial<TConfig>>;
     options?: Readonly<AxAIServiceOptions & {
         streamingUsage?: boolean;
     }>;
     modelInfo?: Readonly<AxModelInfo[]>;
-    modelMap?: Record<string, AxAIOpenAIModel | AxAIOpenAIEmbedModel | string>;
+    models?: AxAIModelList<TModels | string>;
 }
-declare class AxAIOpenAI extends AxBaseAI<AxAIOpenAIChatRequest, AxAIOpenAIEmbedRequest, AxAIOpenAIChatResponse, AxAIOpenAIChatResponseDelta, AxAIOpenAIEmbedResponse> {
-    constructor({ apiKey, config, options, apiURL, modelInfo, modelMap, }: Readonly<Omit<AxAIOpenAIArgs, 'name'>>);
+declare class AxAIOpenAI<TArgs extends Omit<AxAIOpenAIArgs, 'name'> = Omit<AxAIOpenAIArgs, 'name'>, TModel = AxAIOpenAIModel, TEmbedModel = AxAIOpenAIEmbedModel> extends AxBaseAI<AxAIOpenAIChatRequest, AxAIOpenAIEmbedRequest, AxAIOpenAIChatResponse, AxAIOpenAIChatResponseDelta, AxAIOpenAIEmbedResponse> {
+    constructor({ apiKey, config, options, apiURL, modelInfo, models, }: Readonly<TArgs>);
 }
 
-interface AxAIAzureOpenAIArgs {
-    name: 'azure-openai';
-    apiKey: string;
+type AxAIAzureOpenAIConfig = AxAIOpenAIConfig<AxAIOpenAIModel, AxAIOpenAIEmbedModel>;
+type AxAIAzureOpenAIArgs = AxAIOpenAIArgs<'azure-openai', AxAIAzureOpenAIConfig, AxAIOpenAIModel | AxAIOpenAIEmbedModel> & {
     resourceName: string;
     deploymentName: string;
     version?: string;
-    config?: Readonly<Partial<AxAIOpenAIConfig>>;
-    options?: Readonly<AxAIServiceOptions>;
-    modelMap?: Record<string, AxAIOpenAIModel | AxAIOpenAIEmbedModel>;
-}
+};
 declare class AxAIAzureOpenAI extends AxAIOpenAI {
-    constructor({ apiKey, resourceName, deploymentName, version, config, options, modelMap, }: Readonly<Omit<AxAIAzureOpenAIArgs, 'name'>>);
+    constructor({ apiKey, resourceName, deploymentName, version, config, options, models, }: Readonly<Omit<AxAIAzureOpenAIArgs, 'name'>>);
 }
 
 /**
@@ -845,10 +843,10 @@ interface AxAICohereArgs {
     apiKey: string;
     config?: Readonly<Partial<AxAICohereConfig>>;
     options?: Readonly<AxAIServiceOptions>;
-    modelMap?: Record<string, AxAICohereModel | AxAICohereEmbedModel | string>;
+    models?: AxAIModelList<AxAICohereModel | AxAICohereEmbedModel | string>;
 }
 declare class AxAICohere extends AxBaseAI<AxAICohereChatRequest, AxAICohereEmbedRequest, AxAICohereChatResponse, AxAICohereChatResponseDelta, AxAICohereEmbedResponse> {
-    constructor({ apiKey, config, options, modelMap, }: Readonly<Omit<AxAICohereArgs, 'name'>>);
+    constructor({ apiKey, config, options, models, }: Readonly<Omit<AxAICohereArgs, 'name'>>);
 }
 
 /**
@@ -859,19 +857,17 @@ declare enum AxAIDeepSeekModel {
     DeepSeekCoder = "deepseek-coder"
 }
 
-type DeepSeekConfig = AxAIOpenAIConfig;
-interface AxAIDeepSeekArgs {
-    name: 'deepseek';
-    apiKey: string;
-    config?: Readonly<Partial<DeepSeekConfig>>;
-    options?: Readonly<AxAIServiceOptions>;
-    modelMap?: Record<string, AxAIDeepSeekModel | string>;
-}
-declare class AxAIDeepSeek extends AxAIOpenAI {
-    constructor({ apiKey, config, options, modelMap, }: Readonly<Omit<AxAIDeepSeekArgs, 'name'>>);
+type DeepSeekConfig = AxAIOpenAIConfig<AxAIDeepSeekModel, undefined>;
+type AxAIDeepSeekArgs = AxAIOpenAIArgs<'deepseek', DeepSeekConfig, AxAIDeepSeekModel>;
+declare class AxAIDeepSeek extends AxAIOpenAI<Omit<AxAIDeepSeekArgs, 'name'>, AxAIDeepSeekModel> {
+    constructor({ apiKey, config, options, models, }: Readonly<Omit<AxAIDeepSeekArgs, 'name'>>);
 }
 
 declare enum AxAIGoogleGeminiModel {
+    Gemini20Pro = "gemini-2.0-pro-exp-02-05",
+    Gemini20Flash = "gemini-2.0-flash",
+    Gemini20FlashLite = "gemini-2.0-flash-lite-preview-02-05",
+    Gemini20FlashThinking = "gemini-2.0-flash-thinking-exp-01-21",
     Gemini1Pro = "gemini-1.0-pro",
     Gemini15Flash = "gemini-1.5-flash",
     Gemini15Flash8B = "gemini-1.5-flash-8b",
@@ -1060,13 +1056,13 @@ interface AxAIGoogleGeminiArgs {
     region?: string;
     config?: Readonly<Partial<AxAIGoogleGeminiConfig>>;
     options?: Readonly<AxAIServiceOptions & AxAIGoogleGeminiOptionsTools>;
-    modelMap?: Record<string, AxAIGoogleGeminiModel | AxAIGoogleGeminiEmbedModel | string>;
+    models?: AxAIModelList<AxAIGoogleGeminiModel | AxAIGoogleGeminiEmbedModel | string>;
 }
 /**
  * AxAIGoogleGemini: AI Service
  */
 declare class AxAIGoogleGemini extends AxBaseAI<AxAIGoogleGeminiChatRequest, AxAIGoogleGeminiBatchEmbedRequest | AxAIGoogleVertexBatchEmbedRequest, AxAIGoogleGeminiChatResponse, AxAIGoogleGeminiChatResponseDelta, AxAIGoogleGeminiBatchEmbedResponse | AxAIGoogleVertexBatchEmbedResponse> {
-    constructor({ apiKey, projectId, region, config, options, modelMap, }: Readonly<Omit<AxAIGoogleGeminiArgs, 'name'>>);
+    constructor({ apiKey, projectId, region, config, options, models, }: Readonly<Omit<AxAIGoogleGeminiArgs, 'name'>>);
 }
 
 declare enum AxAIGroqModel {
@@ -1076,18 +1072,14 @@ declare enum AxAIGroqModel {
     Gemma2_9B = "gemma2-9b-it"
 }
 
-type AxAIGroqAIConfig = AxAIOpenAIConfig;
-interface AxAIGroqArgs {
-    name: 'groq';
-    apiKey: string;
-    config?: Readonly<Partial<AxAIGroqAIConfig>>;
+type AxAIGroqAIConfig = AxAIOpenAIConfig<AxAIGroqModel, undefined>;
+type AxAIGroqArgs = AxAIOpenAIArgs<'groq', AxAIGroqAIConfig, AxAIGroqModel> & {
     options?: Readonly<AxAIServiceOptions> & {
         tokensPerMinute?: number;
     };
-    modelMap?: Record<string, AxAIGroqModel>;
-}
-declare class AxAIGroq extends AxAIOpenAI {
-    constructor({ apiKey, config, options, modelMap, }: Readonly<Omit<AxAIGroqArgs, 'groq'>>);
+};
+declare class AxAIGroq extends AxAIOpenAI<Omit<AxAIGroqArgs, 'name'>, AxAIGroqModel> {
+    constructor({ apiKey, config, options, models, }: Readonly<Omit<AxAIGroqArgs, 'name'>>);
     setOptions: (options: Readonly<AxAIServiceOptions>) => void;
     private newRateLimiter;
 }
@@ -1131,10 +1123,10 @@ interface AxAIHuggingFaceArgs {
     apiKey: string;
     config?: Readonly<Partial<AxAIHuggingFaceConfig>>;
     options?: Readonly<AxAIServiceOptions>;
-    modelMap?: Record<string, AxAIHuggingFaceModel>;
+    models?: AxAIModelList<AxAIHuggingFaceModel>;
 }
 declare class AxAIHuggingFace extends AxBaseAI<AxAIHuggingFaceRequest, unknown, AxAIHuggingFaceResponse, unknown, unknown> {
-    constructor({ apiKey, config, options, modelMap, }: Readonly<Omit<AxAIHuggingFaceArgs, 'name'>>);
+    constructor({ apiKey, config, options, models, }: Readonly<Omit<AxAIHuggingFaceArgs, 'name'>>);
 }
 
 declare enum AxAIMistralModel {
@@ -1151,34 +1143,27 @@ declare enum AxAIMistralEmbedModels {
     MistralEmbed = "mistral-embed"
 }
 
-type MistralConfig = AxAIOpenAIConfig;
-interface AxAIMistralArgs {
-    name: 'mistral';
-    apiKey: string;
-    config?: Readonly<Partial<MistralConfig>>;
-    options?: Readonly<AxAIServiceOptions>;
-    modelMap?: Record<string, AxAIMistralModel | AxAIMistralEmbedModels | string>;
-}
-declare class AxAIMistral extends AxAIOpenAI {
-    constructor({ apiKey, config, options, modelMap, }: Readonly<Omit<AxAIMistralArgs, 'name'>>);
+type AxAIMistralConfig = AxAIOpenAIConfig<AxAIMistralModel, AxAIMistralEmbedModels>;
+type AxAIMistralArgs = AxAIOpenAIArgs<'mistral', AxAIMistralConfig, AxAIMistralModel | AxAIMistralEmbedModels> & {
+    options?: Readonly<AxAIServiceOptions> & {
+        tokensPerMinute?: number;
+    };
+};
+declare class AxAIMistral extends AxAIOpenAI<Omit<AxAIMistralArgs, 'name'>, AxAIMistralModel | AxAIMistralEmbedModels> {
+    constructor({ apiKey, config, options, models, }: Readonly<Omit<AxAIMistralArgs, 'name'>>);
 }
 
-type AxAIOllamaAIConfig = AxAIOpenAIConfig;
-type AxAIOllamaArgs = {
-    name: 'ollama';
+type AxAIOllamaAIConfig = AxAIOpenAIConfig<string, undefined>;
+type AxAIOllamaArgs = AxAIOpenAIArgs<'ollama', AxAIOllamaAIConfig, string> & {
     model?: string;
     embedModel?: string;
     url?: string;
-    apiKey?: string;
-    config?: Readonly<Partial<AxAIOllamaAIConfig>>;
-    options?: Readonly<AxAIServiceOptions>;
-    modelMap?: Record<string, string>;
 };
 /**
  * OllamaAI: AI Service
  */
-declare class AxAIOllama extends AxAIOpenAI {
-    constructor({ apiKey, url, config, options, modelMap, }: Readonly<Omit<AxAIOllamaArgs, 'name'>>);
+declare class AxAIOllama extends AxAIOpenAI<Omit<AxAIOllamaArgs, 'name'>, string> {
+    constructor({ apiKey, url, config, options, models, }: Readonly<Omit<AxAIOllamaArgs, 'name'>>);
 }
 
 declare enum AxAIRekaModel {
@@ -1257,22 +1242,16 @@ interface AxAIRekaArgs {
         streamingUsage?: boolean;
     }>;
     modelInfo?: Readonly<AxModelInfo[]>;
-    modelMap?: Record<string, AxAIRekaModel | string>;
+    models?: AxAIModelList<AxAIRekaModel | string>;
 }
 declare class AxAIReka extends AxBaseAI<AxAIRekaChatRequest, unknown, AxAIRekaChatResponse, AxAIRekaChatResponseDelta, unknown> {
-    constructor({ apiKey, config, options, apiURL, modelInfo, modelMap, }: Readonly<Omit<AxAIRekaArgs, 'name'>>);
+    constructor({ apiKey, config, options, apiURL, modelInfo, models, }: Readonly<Omit<AxAIRekaArgs, 'name'>>);
 }
 
-type TogetherAIConfig = AxAIOpenAIConfig;
-interface AxAITogetherArgs {
-    name: 'together';
-    apiKey: string;
-    config?: Readonly<Partial<TogetherAIConfig>>;
-    options?: Readonly<AxAIServiceOptions>;
-    modelMap?: Record<string, string>;
-}
-declare class AxAITogether extends AxAIOpenAI {
-    constructor({ apiKey, config, options, modelMap, }: Readonly<Omit<AxAITogetherArgs, 'name'>>);
+type TogetherAIConfig = AxAIOpenAIConfig<string, undefined>;
+type AxAITogetherArgs = AxAIOpenAIArgs<'together', TogetherAIConfig, string>;
+declare class AxAITogether extends AxAIOpenAI<Omit<AxAITogetherArgs, 'name'>, string> {
+    constructor({ apiKey, config, options, models, }: Readonly<Omit<AxAITogetherArgs, 'name'>>);
 }
 
 type AxAIArgs = AxAIOpenAIArgs | AxAIAzureOpenAIArgs | AxAITogetherArgs | AxAIAnthropicArgs | AxAIGroqArgs | AxAIGoogleGeminiArgs | AxAICohereArgs | AxAIHuggingFaceArgs | AxAIMistralArgs | AxAIDeepSeekArgs | AxAIOllamaArgs | AxAIRekaArgs;
@@ -1288,7 +1267,7 @@ declare class AxAI implements AxAIService {
         functions: boolean;
         streaming: boolean;
     };
-    getModelMap(): AxAIModelMap | undefined;
+    getModelList(): AxAIModelList | undefined;
     getMetrics(): AxAIServiceMetrics;
     chat(req: Readonly<AxChatRequest>, options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions>): Promise<AxChatResponse | ReadableStream$1<AxChatResponse>>;
     embed(req: Readonly<AxEmbedRequest>, options?: Readonly<AxAIServiceActionOptions & AxAIServiceActionOptions>): Promise<AxEmbedResponse>;
@@ -1630,10 +1609,11 @@ interface AxAgentic extends AxTunable, AxUsable {
     getFunction(): AxFunction;
 }
 type AxAgentOptions = Omit<AxGenOptions, 'functions'>;
-declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAgentic {
+declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut = AxGenOut> implements AxAgentic {
     private ai?;
     private signature;
     private program;
+    private functions?;
     private agents?;
     private name;
     private description;
@@ -1659,8 +1639,8 @@ declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAge
     resetUsage(): void;
     getFunction(): AxFunction;
     private init;
-    forward(ai: Readonly<AxAIService>, values: IN, options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
-    streamingForward(ai: Readonly<AxAIService>, values: IN, options?: Readonly<AxProgramStreamingForwardOptions>): AxGenStreamingOut<OUT>;
+    forward(parentAi: Readonly<AxAIService>, values: IN, options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
+    streamingForward(parentAi: Readonly<AxAIService>, values: IN, options?: Readonly<AxProgramStreamingForwardOptions>): AxGenStreamingOut<OUT>;
 }
 
 interface AxApacheTikaArgs {
@@ -1705,16 +1685,13 @@ declare class AxBalancer implements AxAIService {
      * Service comparator that sorts services by cost.
      */
     static costComparator: (a: AxAIService, b: AxAIService) => number;
-    getModelMap(): AxAIModelMap | undefined;
+    getModelList(): AxAIModelList | undefined;
     private getNextService;
     private reset;
     getName(): string;
     getModelInfo(): Readonly<AxModelInfoWithProvider>;
     getEmbedModelInfo(): Readonly<AxModelInfoWithProvider> | undefined;
-    getFeatures(model?: string): {
-        functions: boolean;
-        streaming: boolean;
-    };
+    getFeatures(model?: string): AxAIFeatures;
     getMetrics(): AxAIServiceMetrics;
     chat(req: Readonly<AxChatRequest>, options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions> | undefined): Promise<AxChatResponse | ReadableStream$1<AxChatResponse>>;
     embed(req: Readonly<AxEmbedRequest>, options?: Readonly<AxAIServiceActionOptions> | undefined): Promise<AxEmbedResponse>;
@@ -2203,7 +2180,7 @@ declare class AxMockAIService implements AxAIService {
             functions?: boolean;
             streaming?: boolean;
         };
-        modelMap?: AxAIModelMap;
+        models?: AxAIModelList;
         chatResponse?: AxChatResponse | ((req: Readonly<AxChatRequest>) => AxChatResponse | Promise<AxChatResponse>);
         embedResponse?: AxEmbedResponse | ((req: Readonly<AxEmbedRequest>) => AxEmbedResponse | Promise<AxEmbedResponse>);
         shouldError?: boolean;
@@ -2217,7 +2194,7 @@ declare class AxMockAIService implements AxAIService {
         functions: boolean;
         streaming: boolean;
     };
-    getModelMap(): AxAIModelMap | undefined;
+    getModelList(): AxAIModelList | undefined;
     getMetrics(): AxAIServiceMetrics;
     chat(req: Readonly<AxChatRequest>, _options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions>): Promise<AxChatResponse | ReadableStream<AxChatResponse>>;
     embed(req: Readonly<AxEmbedRequest>, _options?: Readonly<AxAIServiceActionOptions>): Promise<AxEmbedResponse>;
@@ -2246,4 +2223,4 @@ declare class AxRAG extends AxChainOfThought<{
     }>;
 }
 
-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, 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, AxAIGoogleGemini, type AxAIGoogleGeminiArgs, type AxAIGoogleGeminiBatchEmbedRequest, type AxAIGoogleGeminiBatchEmbedResponse, type AxAIGoogleGeminiChatRequest, type AxAIGoogleGeminiChatResponse, type AxAIGoogleGeminiChatResponseDelta, type AxAIGoogleGeminiConfig, type AxAIGoogleGeminiContent, AxAIGoogleGeminiEmbedModel, type AxAIGoogleGeminiGenerationConfig, AxAIGoogleGeminiModel, type AxAIGoogleGeminiOptionsTools, AxAIGoogleGeminiSafetyCategory, type AxAIGoogleGeminiSafetySettings, AxAIGoogleGeminiSafetyThreshold, type AxAIGoogleGeminiTool, type AxAIGoogleGeminiToolConfig, type AxAIGoogleGeminiToolFunctionDeclaration, type AxAIGoogleGeminiToolGoogleSearchRetrieval, type AxAIGoogleVertexBatchEmbedRequest, type AxAIGoogleVertexBatchEmbedResponse, AxAIGroq, type AxAIGroqArgs, AxAIGroqModel, AxAIHuggingFace, type AxAIHuggingFaceArgs, type AxAIHuggingFaceConfig, AxAIHuggingFaceModel, type AxAIHuggingFaceRequest, type AxAIHuggingFaceResponse, type AxAIMemory, AxAIMistral, type AxAIMistralArgs, AxAIMistralEmbedModels, AxAIMistralModel, type AxAIModelMap, type AxAIModels, AxAIOllama, type AxAIOllamaAIConfig, type AxAIOllamaArgs, AxAIOpenAI, type AxAIOpenAIArgs, type AxAIOpenAIChatRequest, type AxAIOpenAIChatResponse, type AxAIOpenAIChatResponseDelta, type AxAIOpenAIConfig, AxAIOpenAIEmbedModel, type AxAIOpenAIEmbedRequest, type AxAIOpenAIEmbedResponse, type AxAIOpenAILogprob, AxAIOpenAIModel, type AxAIOpenAIResponseDelta, type AxAIOpenAIUsage, type AxAIPromptConfig, AxAIReka, type AxAIRekaArgs, type AxAIRekaChatRequest, type AxAIRekaChatResponse, type AxAIRekaChatResponseDelta, type AxAIRekaConfig, AxAIRekaModel, type AxAIRekaUsage, type AxAIService, type AxAIServiceActionOptions, AxAIServiceAuthenticationError, AxAIServiceError, type AxAIServiceImpl, type AxAIServiceMetrics, AxAIServiceNetworkError, type AxAIServiceOptions, AxAIServiceResponseError, AxAIServiceStatusError, AxAIServiceStreamTerminatedError, AxAIServiceTimeoutError, AxAITogether, type AxAITogetherArgs, type AxAPI, type AxAPIConfig, AxAgent, type AxAgentOptions, type AxAgentic, AxApacheTika, type AxApacheTikaArgs, type AxApacheTikaConvertOptions, type AxAssertion, AxAssertionError, AxBalancer, type AxBalancerOptions, AxBaseAI, type AxBaseAIArgs, type AxBaseAIFeatures, AxBootstrapFewShot, 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, type AxEvaluateArgs, type AxExample, type AxField, type AxFieldTemplateFn, type AxFieldValue, type AxFunction, AxFunctionError, type AxFunctionHandler, type AxFunctionJSONSchema, AxFunctionProcessor, AxGen, type AxGenDeltaOut, type AxGenIn, type AxGenOptions, type AxGenOut, type AxGenStreamingOut, type AxGenerateResult, AxHFDataLoader, type AxIField, type AxInputFunctionType, AxInstanceRegistry, type AxInternalChatRequest, type AxInternalEmbedRequest, AxJSInterpreter, AxJSInterpreterPermission, AxLLMRequestTypeValues, AxMemory, type AxMetricFn, type AxMetricFnArgs, AxMockAIService, type AxModelConfig, type AxModelInfo, type AxModelInfoWithProvider, type AxOptimizerArgs, AxProgram, type AxProgramDemos, type AxProgramExamples, type AxProgramForwardOptions, type AxProgramStreamingForwardOptions, type AxProgramTrace, type AxProgramUsage, AxProgramWithSignature, type AxProgramWithSignatureOptions, AxPromptTemplate, AxRAG, type AxRateLimiterFunction, AxRateLimiterTokenUsage, type AxRateLimiterTokenUsageOptions, type AxRerankerIn, type AxRerankerOut, type AxResponseHandlerArgs, type AxRewriteIn, type AxRewriteOut, AxRoute, AxRouter, type AxRouterForwardOptions, AxSignature, AxSpanKindValues, type AxStreamingAssertion, type AxStreamingEvent, AxTestPrompt, type AxTokenUsage, type AxTunable, type AxUsable };
+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, AxAIGoogleGeminiEmbedModel, type AxAIGoogleGeminiGenerationConfig, AxAIGoogleGeminiModel, type AxAIGoogleGeminiOptionsTools, AxAIGoogleGeminiSafetyCategory, type AxAIGoogleGeminiSafetySettings, AxAIGoogleGeminiSafetyThreshold, type AxAIGoogleGeminiTool, type AxAIGoogleGeminiToolConfig, type AxAIGoogleGeminiToolFunctionDeclaration, type AxAIGoogleGeminiToolGoogleSearchRetrieval, type AxAIGoogleVertexBatchEmbedRequest, type AxAIGoogleVertexBatchEmbedResponse, AxAIGroq, type AxAIGroqArgs, AxAIGroqModel, AxAIHuggingFace, type AxAIHuggingFaceArgs, type AxAIHuggingFaceConfig, AxAIHuggingFaceModel, type AxAIHuggingFaceRequest, type AxAIHuggingFaceResponse, type AxAIMemory, AxAIMistral, type AxAIMistralArgs, AxAIMistralEmbedModels, AxAIMistralModel, type AxAIModelList, type AxAIModels, AxAIOllama, type AxAIOllamaAIConfig, type AxAIOllamaArgs, AxAIOpenAI, type AxAIOpenAIArgs, type AxAIOpenAIChatRequest, type AxAIOpenAIChatResponse, type AxAIOpenAIChatResponseDelta, type AxAIOpenAIConfig, AxAIOpenAIEmbedModel, type AxAIOpenAIEmbedRequest, type AxAIOpenAIEmbedResponse, type AxAIOpenAILogprob, AxAIOpenAIModel, type AxAIOpenAIResponseDelta, type AxAIOpenAIUsage, type AxAIPromptConfig, AxAIReka, type AxAIRekaArgs, type AxAIRekaChatRequest, type AxAIRekaChatResponse, type AxAIRekaChatResponseDelta, type AxAIRekaConfig, AxAIRekaModel, type AxAIRekaUsage, type AxAIService, type AxAIServiceActionOptions, AxAIServiceAuthenticationError, AxAIServiceError, type AxAIServiceImpl, type AxAIServiceMetrics, AxAIServiceNetworkError, type AxAIServiceOptions, AxAIServiceResponseError, AxAIServiceStatusError, AxAIServiceStreamTerminatedError, AxAIServiceTimeoutError, AxAITogether, type AxAITogetherArgs, type AxAPI, type AxAPIConfig, AxAgent, type AxAgentOptions, type AxAgentic, AxApacheTika, type AxApacheTikaArgs, type AxApacheTikaConvertOptions, type AxAssertion, AxAssertionError, AxBalancer, type AxBalancerOptions, AxBaseAI, type AxBaseAIArgs, AxBootstrapFewShot, 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, type AxEvaluateArgs, type AxExample, type AxField, type AxFieldTemplateFn, type AxFieldValue, type AxFunction, AxFunctionError, type AxFunctionHandler, type AxFunctionJSONSchema, AxFunctionProcessor, AxGen, type AxGenDeltaOut, type AxGenIn, type AxGenOptions, type AxGenOut, type AxGenStreamingOut, type AxGenerateResult, AxHFDataLoader, type AxIField, type AxInputFunctionType, AxInstanceRegistry, type AxInternalChatRequest, type AxInternalEmbedRequest, AxJSInterpreter, AxJSInterpreterPermission, AxLLMRequestTypeValues, AxMemory, type AxMetricFn, type AxMetricFnArgs, AxMockAIService, type AxModelConfig, type AxModelInfo, type AxModelInfoWithProvider, type AxOptimizerArgs, AxProgram, type AxProgramDemos, type AxProgramExamples, type AxProgramForwardOptions, type AxProgramStreamingForwardOptions, type AxProgramTrace, type AxProgramUsage, AxProgramWithSignature, type AxProgramWithSignatureOptions, AxPromptTemplate, AxRAG, type AxRateLimiterFunction, AxRateLimiterTokenUsage, type AxRateLimiterTokenUsageOptions, type AxRerankerIn, type AxRerankerOut, type AxResponseHandlerArgs, type AxRewriteIn, type AxRewriteOut, AxRoute, AxRouter, type AxRouterForwardOptions, AxSignature, AxSpanKindValues, type AxStreamingAssertion, type AxStreamingEvent, AxTestPrompt, type AxTokenUsage, type AxTunable, type AxUsable };