@ax-llm/ax 11.0.21 → 11.0.22

package/index.d.ts

@@ -60,10 +60,14 @@ declare class AxAIServiceAuthenticationError extends AxAIServiceError {
     constructor(url: string, requestBody?: unknown, context?: Record<string, unknown>);
 }
 
-type AxAIModelList<T = unknown> = {
+type AxAIInputModelList<TModel> = (AxAIModelList[number] & {
+    model: TModel;
+    isInternal?: boolean;
+})[];
+type AxAIModelList = {
     key: string;
-    model: T;
     description: string;
+    model: string;
 }[];
 type AxModelInfo = {
     name: string;
@@ -256,10 +260,12 @@ type AxAIServiceActionOptions<TModel = unknown, TEmbedModel = unknown> = {
 interface AxAIService<TModel = unknown, TEmbedModel = unknown> {
     getId(): string;
     getName(): string;
-    getModelInfo(): Readonly<AxModelInfoWithProvider>;
-    getEmbedModelInfo(): Readonly<AxModelInfoWithProvider> | undefined;
     getFeatures(model?: TModel): AxAIFeatures;
-    getModelList(): AxAIModelList<TModel> | undefined;
+    getModelList(): AxAIModelList | undefined;
+    getDefaultModels(): Readonly<{
+        model: string;
+        embedModel?: string;
+    }>;
     getMetrics(): AxAIServiceMetrics;
     chat(req: Readonly<AxChatRequest<TModel>>, options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions<TModel, TEmbedModel>>): Promise<AxChatResponse | ReadableStream<AxChatResponse>>;
     embed(req: Readonly<AxEmbedRequest<TEmbedModel>>, options?: Readonly<AxAIServiceActionOptions<TModel, TEmbedModel>>): Promise<AxEmbedResponse>;
@@ -291,7 +297,7 @@ interface AxBaseAIArgs<TModel, TEmbedModel> {
     }>;
     options?: Readonly<AxAIServiceOptions>;
     supportFor: AxAIFeatures | ((model: TModel) => AxAIFeatures);
-    models?: AxAIModelList<TModel>;
+    models?: AxAIInputModelList<TModel>;
 }
 declare class AxBaseAI<TModel, TEmbedModel, TChatRequest, TEmbedRequest, TChatResponse, TChatResponseDelta, TEmbedResponse> implements AxAIService<TModel, TEmbedModel> {
     private readonly aiImpl;
@@ -317,9 +323,11 @@ declare class AxBaseAI<TModel, TEmbedModel, TChatRequest, TEmbedRequest, TChatRe
     setHeaders(headers: () => Promise<Record<string, string>>): void;
     setOptions(options: Readonly<AxAIServiceOptions>): void;
     getOptions(): Readonly<AxAIServiceOptions>;
-    getModelInfo(): Readonly<AxModelInfoWithProvider>;
-    getEmbedModelInfo(): AxModelInfoWithProvider | undefined;
-    getModelList(): AxAIModelList<TModel> | undefined;
+    getModelList(): AxAIModelList | undefined;
+    getDefaultModels(): Readonly<{
+        model: string;
+        embedModel?: string;
+    }>;
     getName(): string;
     getFeatures(model?: TModel): AxAIFeatures;
     private calculatePercentile;
@@ -532,7 +540,7 @@ interface AxAIAnthropicArgs {
     region?: string;
     config?: Readonly<Partial<AxAIAnthropicConfig>>;
     options?: Readonly<AxAIServiceOptions>;
-    models?: AxAIModelList<AxAIAnthropicModel | AxAIAnthropicVertexModel>;
+    models?: AxAIInputModelList<AxAIAnthropicModel | AxAIAnthropicVertexModel>;
 }
 declare class AxAIAnthropic extends AxBaseAI<AxAIAnthropicModel | AxAIAnthropicVertexModel, unknown, AxAIAnthropicChatRequest, unknown, AxAIAnthropicChatResponse, AxAIAnthropicChatResponseDelta, unknown> {
     constructor({ apiKey, projectId, region, config, options, models, }: Readonly<Omit<AxAIAnthropicArgs, 'name'>>);
@@ -656,7 +664,7 @@ type AxAIOpenAIChatRequest<TModel> = {
     response_format?: {
         type: string;
     };
-    max_tokens: number;
+    max_completion_tokens: number;
     temperature?: number;
     top_p?: number;
     n?: number;
@@ -732,7 +740,7 @@ interface AxAIOpenAIBaseArgs<TModel, TEmbedModel> {
         streamingUsage?: boolean;
     }>;
     modelInfo: Readonly<AxModelInfo[]>;
-    models?: AxAIModelList<TModel>;
+    models?: AxAIInputModelList<TModel>;
 }
 declare class AxAIOpenAIBase<TModel, TEmbedModel> extends AxBaseAI<TModel, TEmbedModel, AxAIOpenAIChatRequest<TModel>, AxAIOpenAIEmbedRequest<TEmbedModel>, AxAIOpenAIChatResponse, AxAIOpenAIChatResponseDelta, AxAIOpenAIEmbedResponse> {
     constructor({ apiKey, config, options, apiURL, modelInfo, models, }: Readonly<Omit<AxAIOpenAIBaseArgs<TModel, TEmbedModel>, 'name'>>);
@@ -855,7 +863,7 @@ interface AxAICohereArgs {
     apiKey: string;
     config?: Readonly<Partial<AxAICohereConfig>>;
     options?: Readonly<AxAIServiceOptions>;
-    models?: AxAIModelList<AxAICohereModel>;
+    models?: AxAIInputModelList<AxAICohereModel>;
 }
 declare class AxAICohere extends AxBaseAI<AxAICohereModel, AxAICohereEmbedModel, AxAICohereChatRequest, AxAICohereEmbedRequest, AxAICohereChatResponse, AxAICohereChatResponseDelta, AxAICohereEmbedResponse> {
     constructor({ apiKey, config, options, models, }: Readonly<Omit<AxAICohereArgs, 'name'>>);
@@ -1060,6 +1068,7 @@ interface AxAIGoogleGeminiOptionsTools {
         mode?: 'MODE_DYNAMIC';
         dynamicThreshold?: number;
     };
+    googleSearch?: boolean;
 }
 interface AxAIGoogleGeminiArgs {
     name: 'google-gemini';
@@ -1069,7 +1078,7 @@ interface AxAIGoogleGeminiArgs {
     endpointId?: string;
     config?: Readonly<Partial<AxAIGoogleGeminiConfig>>;
     options?: Readonly<AxAIServiceOptions & AxAIGoogleGeminiOptionsTools>;
-    models?: AxAIModelList<AxAIGoogleGeminiModel>;
+    models?: AxAIInputModelList<AxAIGoogleGeminiModel>;
 }
 /**
  * AxAIGoogleGemini: AI Service
@@ -1135,7 +1144,7 @@ interface AxAIHuggingFaceArgs {
     apiKey: string;
     config?: Readonly<Partial<AxAIHuggingFaceConfig>>;
     options?: Readonly<AxAIServiceOptions>;
-    models?: AxAIModelList<AxAIHuggingFaceModel>;
+    models?: AxAIInputModelList<AxAIHuggingFaceModel>;
 }
 declare class AxAIHuggingFace extends AxBaseAI<AxAIHuggingFaceModel, unknown, AxAIHuggingFaceRequest, unknown, AxAIHuggingFaceResponse, unknown, unknown> {
     constructor({ apiKey, config, options, models, }: Readonly<Omit<AxAIHuggingFaceArgs, 'name'>>);
@@ -1253,7 +1262,7 @@ interface AxAIRekaArgs {
         streamingUsage?: boolean;
     }>;
     modelInfo?: Readonly<AxModelInfo[]>;
-    models?: AxAIModelList<AxAIRekaModel>;
+    models?: AxAIInputModelList<AxAIRekaModel>;
 }
 declare class AxAIReka extends AxBaseAI<AxAIRekaModel, undefined, AxAIRekaChatRequest, unknown, AxAIRekaChatResponse, AxAIRekaChatResponseDelta, unknown> {
     constructor({ apiKey, config, options, apiURL, modelInfo, models, }: Readonly<Omit<AxAIRekaArgs, 'name'>>);
@@ -1272,13 +1281,15 @@ declare class AxAI implements AxAIService {
     constructor(options: Readonly<AxAIArgs>);
     getName(): string;
     getId(): string;
-    getModelInfo(): Readonly<AxModelInfoWithProvider>;
-    getEmbedModelInfo(): Readonly<AxModelInfoWithProvider> | undefined;
     getFeatures(model?: string): {
         functions: boolean;
         streaming: boolean;
     };
     getModelList(): AxAIModelList | undefined;
+    getDefaultModels(): Readonly<{
+        model: string;
+        embedModel?: string;
+    }>;
     getMetrics(): AxAIServiceMetrics;
     chat(req: Readonly<AxChatRequest>, options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions>): Promise<AxChatResponse | ReadableStream<AxChatResponse>>;
     embed(req: Readonly<AxEmbedRequest>, options?: Readonly<AxAIServiceActionOptions & AxAIServiceActionOptions>): Promise<AxEmbedResponse>;
@@ -1470,6 +1481,16 @@ declare class AxAssertionError extends Error {
     }[];
 }
 
+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>;
     /**
@@ -1478,11 +1499,7 @@ interface AxFieldProcessor {
      * @param value - The current field value.
      * @param context - Additional context (e.g. memory and session id).
      */
-    process: (value: AxFieldValue, context?: Readonly<{
-        values?: AxGenOut;
-        sessionId?: string;
-        done?: boolean;
-    }>) => any | Promise<any>;
+    process: AxFieldProcessorProcess | AxStreamingFieldProcessorProcess;
 }
 
 declare class AxMemory implements AxAIMemory {
@@ -1710,7 +1727,9 @@ declare class AxGen<IN extends AxGenIn = AxGenIn, OUT extends AxGenerateResult<A
     constructor(signature: Readonly<AxSignature | string>, options?: Readonly<AxGenOptions>);
     addAssert: (fn: AxAssertion["fn"], message?: string) => void;
     addStreamingAssert: (fieldName: string, fn: AxStreamingAssertion["fn"], message?: string) => void;
-    addFieldProcessor: (fieldName: string, fn: AxFieldProcessor["process"], streaming?: boolean) => void;
+    private addFieldProcessorInternal;
+    addStreamingFieldProcessor: (fieldName: string, fn: AxFieldProcessor["process"]) => void;
+    addFieldProcessor: (fieldName: string, fn: AxFieldProcessor["process"]) => void;
     private forwardSendRequest;
     private forwardCore;
     private processStreamingResponse;
@@ -1832,7 +1851,7 @@ type AxBalancerOptions = {
 /**
  * Balancer that rotates through services.
  */
-declare class AxBalancer implements AxAIService {
+declare class AxBalancer implements AxAIService<unknown, unknown> {
     private services;
     private currentServiceIndex;
     private currentService;
@@ -1849,14 +1868,16 @@ declare class AxBalancer implements AxAIService {
     /**
      * Service comparator that sorts services by cost.
      */
-    static costComparator: (a: AxAIService, b: AxAIService) => number;
+    static metricComparator: (a: AxAIService, b: AxAIService) => number;
     getModelList(): AxAIModelList | undefined;
+    getDefaultModels(): Readonly<{
+        model: string;
+        embedModel?: string;
+    }>;
     private getNextService;
     private reset;
     getName(): string;
     getId(): string;
-    getModelInfo(): Readonly<AxModelInfoWithProvider>;
-    getEmbedModelInfo(): Readonly<AxModelInfoWithProvider> | undefined;
     getFeatures(model?: string): AxAIFeatures;
     getMetrics(): AxAIServiceMetrics;
     private canRetryService;
@@ -2269,13 +2290,15 @@ declare class AxMockAIService implements AxAIService {
     constructor(config?: AxMockAIServiceConfig);
     getName(): string;
     getId(): string;
-    getModelInfo(): Readonly<AxModelInfoWithProvider>;
-    getEmbedModelInfo(): Readonly<AxModelInfoWithProvider> | undefined;
     getFeatures(_model?: string): {
         functions: boolean;
         streaming: boolean;
     };
     getModelList(): AxAIModelList | undefined;
+    getDefaultModels(): Readonly<{
+        model: string;
+        embedModel?: string;
+    }>;
     getMetrics(): AxAIServiceMetrics;
     chat(req: Readonly<AxChatRequest<unknown>>, options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions<unknown, unknown>>): Promise<AxChatResponse | ReadableStream<AxChatResponse>>;
     embed(req: Readonly<AxEmbedRequest>, _options?: Readonly<AxAIServiceActionOptions>): Promise<AxEmbedResponse>;
@@ -2299,25 +2322,25 @@ declare class AxRateLimiterTokenUsage {
     acquire(tokens: number): Promise<void>;
 }
 
-interface AxRouterForwardOptions {
+interface AxSimpleClassifierForwardOptions {
     cutoff?: number;
 }
-declare class AxRoute {
+declare class AxSimpleClassifierClass {
     private readonly name;
     private readonly context;
     constructor(name: string, context: readonly string[]);
     getName(): string;
     getContext(): readonly string[];
 }
-declare class AxRouter {
+declare class AxSimpleClassifier {
     private readonly ai;
     private db;
     private debug?;
     constructor(ai: AxAIService);
     getState(): AxDBState | undefined;
     setState(state: AxDBState): void;
-    setRoutes: (routes: readonly AxRoute[]) => Promise<void>;
-    forward(text: string, options?: Readonly<AxRouterForwardOptions>): Promise<string>;
+    setClasses: (classes: readonly AxSimpleClassifierClass[]) => Promise<void>;
+    forward(text: string, options?: Readonly<AxSimpleClassifierForwardOptions>): Promise<string>;
     setOptions(options: Readonly<{
         debug?: boolean;
     }>): void;
@@ -2375,6 +2398,70 @@ declare class AxInstanceRegistry<T> {
     [Symbol.iterator](): Generator<T, void, unknown>;
 }
 
+type AxAIServiceListItem = {
+    key: string;
+    service: AxAIService;
+    description: string;
+    isInternal?: boolean;
+};
+declare class AxMultiServiceRouter implements AxAIService<string, string> {
+    private services;
+    /**
+     * Constructs a new multi-service router.
+     * It validates that each service provides a unique set of model keys,
+     * then builds a lookup (map) for routing the chat/embed requests.
+     */
+    constructor(services: (AxAIServiceListItem | AxAIService)[]);
+    /**
+     * Delegates the chat call to the service matching the provided model key.
+     */
+    chat(req: Readonly<AxChatRequest<string>>, options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions<string, string>>): Promise<AxChatResponse | ReadableStream<AxChatResponse>>;
+    /**
+     * Delegates the embed call to the service matching the provided embed model key.
+     */
+    embed(req: Readonly<AxEmbedRequest<string>>, options?: Readonly<AxAIServiceActionOptions<string, string>>): Promise<AxEmbedResponse>;
+    /**
+     * Returns a composite ID built from the IDs of the underlying services.
+     */
+    getId(): string;
+    /**
+     * Returns the name of this router.
+     */
+    getName(): string;
+    /**
+     * Aggregates all available models across the underlying services.
+     */
+    getModelList(): AxAIModelList;
+    getDefaultModels(): Readonly<{
+        model: string;
+        embedModel?: string;
+    }>;
+    /**
+     * If a model key is provided, delegate to the corresponding service's features.
+     * Otherwise, returns a default feature set.
+     */
+    getFeatures(model?: string): {
+        functions: boolean;
+        streaming: boolean;
+        functionCot?: boolean;
+    };
+    /**
+     * Returns aggregated metrics from the underlying service.
+     * Uses the metrics from the last service that was used,
+     * or falls back to the first service if none has been used.
+     */
+    getMetrics(): AxAIServiceMetrics;
+    /**
+     * Sets options on all underlying services.
+     */
+    setOptions(options: Readonly<AxAIServiceOptions>): void;
+    /**
+     * Returns the options from the last used service,
+     * or falls back to the first service if none has been used.
+     */
+    getOptions(): Readonly<AxAIServiceOptions>;
+}
+
 declare class AxRAG extends AxChainOfThought<{
     context: string[];
     question: string;
@@ -2395,4 +2482,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, 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, AxAIOpenAIBase, type AxAIOpenAIBaseArgs, 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 AxAgentFeatures, 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 AxFieldProcessor, 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 AxMockAIServiceConfig, 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 AxAIInputModelList, type AxAIMemory, AxAIMistral, type AxAIMistralArgs, AxAIMistralEmbedModels, AxAIMistralModel, type AxAIModelList, 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, 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 AxAgentFeatures, 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 AxFieldProcessor, type AxFieldProcessorProcess, 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 AxMockAIServiceConfig, type AxModelConfig, type AxModelInfo, type AxModelInfoWithProvider, AxMultiServiceRouter, 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, AxSignature, AxSimpleClassifier, AxSimpleClassifierClass, type AxSimpleClassifierForwardOptions, AxSpanKindValues, type AxStreamingAssertion, type AxStreamingEvent, type AxStreamingFieldProcessorProcess, AxTestPrompt, type AxTokenUsage, type AxTunable, type AxUsable };