@ax-llm/ax 11.0.41 → 11.0.44

package/index.d.cts

@@ -65,6 +65,7 @@ interface AxAIFeatures {
     functions: boolean;
     streaming: boolean;
     functionCot?: boolean;
+    thinkingTokenBudget?: boolean;
 }
 interface AxBaseAIArgs<TModel, TEmbedModel> {
     name: string;
@@ -87,6 +88,8 @@ declare class AxBaseAI<TModel, TEmbedModel, TChatRequest, TEmbedRequest, TChatRe
     private rt?;
     private fetch?;
     private tracer?;
+    private timeout?;
+    private excludeContentFromTrace?;
     private models?;
     private modelInfo;
     private modelUsage?;
@@ -159,6 +162,7 @@ type AxTokenUsage = {
     promptTokens: number;
     completionTokens: number;
     totalTokens: number;
+    thoughtsTokens?: number;
 };
 type AxModelConfig = {
     maxTokens?: number;
@@ -195,6 +199,7 @@ type AxFunction = {
 };
 type AxChatResponseResult = {
     content?: string;
+    thought?: string;
     name?: string;
     id?: string;
     functionCalls?: {
@@ -324,17 +329,21 @@ type AxRateLimiterFunction = <T = unknown>(reqFunc: () => Promise<T | ReadableSt
 }>) => Promise<T | ReadableStream$1<T>>;
 type AxAIPromptConfig = {
     stream?: boolean;
+    thinkingTokenBudget?: number;
 };
 type AxAIServiceOptions = {
     debug?: boolean;
     rateLimiter?: AxRateLimiterFunction;
     fetch?: typeof fetch;
     tracer?: Tracer;
+    timeout?: number;
+    excludeContentFromTrace?: boolean;
 };
 type AxAIServiceActionOptions<TModel = unknown, TEmbedModel = unknown> = {
     ai?: Readonly<AxAIService<TModel, TEmbedModel>>;
     sessionId?: string;
     traceId?: string;
+    timeout?: number;
     rateLimiter?: AxRateLimiterFunction;
     debug?: boolean;
     debugHideSystemPrompt?: boolean;
@@ -923,7 +932,7 @@ declare class AxAIDeepSeek extends AxAIOpenAIBase<AxAIDeepSeekModel, undefined>
 }
 
 declare enum AxAIGoogleGeminiModel {
-    Gemini25Pro = "gemini-2.5-pro-preview-03-25",
+    Gemini25Pro = "gemini-2.5-pro-preview-05-06",
     Gemini25Flash = "gemini-2.5-flash-preview-04-17",
     Gemini20Flash = "gemini-2.0-flash",
     Gemini20FlashLite = "gemini-2.0-flash-lite-preview-02-05",
@@ -967,6 +976,7 @@ type AxAIGoogleGeminiContent = {
     role: 'user';
     parts: ({
         text: string;
+        thought?: string;
     } | {
         inlineData: {
             mimeType: string;
@@ -1012,6 +1022,7 @@ type AxAIGoogleGeminiTool = {
     function_declarations?: AxAIGoogleGeminiToolFunctionDeclaration[];
     code_execution?: object;
     google_search_retrieval?: AxAIGoogleGeminiToolGoogleSearchRetrieval;
+    url_context?: object;
 };
 type AxAIGoogleGeminiToolConfig = {
     function_calling_config: {
@@ -1023,9 +1034,15 @@ type AxAIGoogleGeminiGenerationConfig = {
     temperature?: number;
     topP?: number;
     topK?: number;
+    frequencyPenalty?: number;
     candidateCount?: number;
     maxOutputTokens?: number;
     stopSequences?: readonly string[];
+    responseMimeType?: string;
+    thinkingConfig?: {
+        thinkingBudget?: number;
+        includeThoughts?: boolean;
+    };
 };
 type AxAIGoogleGeminiSafetySettings = {
     category: AxAIGoogleGeminiSafetyCategory;
@@ -1062,11 +1079,13 @@ type AxAIGoogleGeminiChatResponse = {
         promptTokenCount: number;
         candidatesTokenCount: number;
         totalTokenCount: number;
+        thoughtsTokenCount: number;
     };
 };
 type AxAIGoogleGeminiChatResponseDelta = AxAIGoogleGeminiChatResponse;
 type AxAIGoogleGeminiThinkingConfig = {
-    thinkingBudget: number;
+    thinkingTokenBudget?: number;
+    includeThoughts?: boolean;
 };
 /**
  * AxAIGoogleGeminiConfig: Configuration options for Google Gemini API
@@ -1078,7 +1097,8 @@ type AxAIGoogleGeminiConfig = AxModelConfig & {
     embedType?: AxAIGoogleGeminiEmbedTypes;
     dimensions?: number;
     autoTruncate?: boolean;
-    thinkingConfig?: AxAIGoogleGeminiThinkingConfig;
+    thinking?: AxAIGoogleGeminiThinkingConfig;
+    urlContext?: string;
 };
 /**
  * AxAIGoogleGeminiEmbedRequest: Structure for making an embedding request to the Google Gemini API.
@@ -1137,6 +1157,7 @@ interface AxAIGoogleGeminiOptionsTools {
         dynamicThreshold?: number;
     };
     googleSearch?: boolean;
+    urlContext?: boolean;
 }
 interface AxAIGoogleGeminiArgs {
     name: 'google-gemini';
@@ -1564,6 +1585,12 @@ declare class AxSignature {
     private updateHash;
     hash: () => string;
     toString: () => string;
+    toJSON: () => {
+        id: string;
+        description: string | undefined;
+        inputFields: AxIField[];
+        outputFields: AxIField[];
+    };
 }
 
 type AxFieldValue = string | string[] | number | boolean | object | null | undefined | {
@@ -1610,6 +1637,8 @@ type AxProgramForwardOptions = {
     fastFail?: boolean;
     debug?: boolean;
     debugHideSystemPrompt?: boolean;
+    thinkingTokenBudget?: number;
+    traceLabel?: string;
 };
 type AxProgramStreamingForwardOptions = Omit<AxProgramForwardOptions, 'stream'>;
 type AxGenDeltaOut<OUT> = {
@@ -1774,9 +1803,11 @@ interface AxGenOptions {
     asserts?: AxAssertion[];
     streamingAsserts?: AxStreamingAssertion[];
     fastFail?: boolean;
+    excludeContentFromTrace?: boolean;
+    traceLabel?: string;
 }
 type AxGenerateResult<OUT extends AxGenOut> = OUT & {
-    functions?: AxChatResponseFunctionCall[];
+    thought?: string;
 };
 interface AxResponseHandlerArgs<T> {
     ai: Readonly<AxAIService>;
@@ -1787,6 +1818,7 @@ interface AxResponseHandlerArgs<T> {
     traceId?: string;
     functions?: Readonly<AxFunction[]>;
     fastFail?: boolean;
+    span?: Span;
 }
 interface AxStreamingEvent<T> {
     event: 'delta' | 'done' | 'error';
@@ -1806,6 +1838,8 @@ declare class AxGen<IN extends AxGenIn = AxGenIn, OUT extends AxGenerateResult<A
     private functionsExecuted;
     private fieldProcessors;
     private streamingFieldProcessors;
+    private values;
+    private excludeContentFromTrace;
     constructor(signature: Readonly<AxSignature | string>, options?: Readonly<AxGenOptions>);
     addAssert: (fn: AxAssertion["fn"], message?: string) => void;
     addStreamingAssert: (fieldName: string, fn: AxStreamingAssertion["fn"], message?: string) => void;
@@ -2379,6 +2413,7 @@ declare class AxJSInterpreter {
 
 declare const axSpanAttributes: {
     LLM_SYSTEM: string;
+    LLM_OPERATION_NAME: string;
     LLM_REQUEST_MODEL: string;
     LLM_REQUEST_MAX_TOKENS: string;
     LLM_REQUEST_TEMPERATURE: string;
@@ -2388,8 +2423,10 @@ declare const axSpanAttributes: {
     LLM_REQUEST_STOP_SEQUENCES: string;
     LLM_REQUEST_LLM_IS_STREAMING: string;
     LLM_REQUEST_TOP_P: string;
-    LLM_USAGE_PROMPT_TOKENS: string;
-    LLM_USAGE_COMPLETION_TOKENS: string;
+    LLM_USAGE_INPUT_TOKENS: string;
+    LLM_USAGE_OUTPUT_TOKENS: string;
+    LLM_USAGE_TOTAL_TOKENS: string;
+    LLM_USAGE_THOUGHTS_TOKENS: string;
     DB_SYSTEM: string;
     DB_TABLE: string;
     DB_NAMESPACE: string;
@@ -2409,7 +2446,12 @@ declare const axSpanAttributes: {
     DB_QUERY_RESULT_DOCUMENT: string;
 };
 declare const axSpanEvents: {
-    LLM_PROMPT: string;
+    GEN_AI_USER_MESSAGE: string;
+    GEN_AI_SYSTEM_MESSAGE: string;
+    GEN_AI_ASSISTANT_MESSAGE: string;
+    GEN_AI_TOOL_MESSAGE: string;
+    GEN_AI_CHOICE: string;
+    GEN_AI_USAGE: string;
 };
 declare enum AxLLMRequestTypeValues {
     COMPLETION = "completion",

package/index.d.ts

@@ -65,6 +65,7 @@ interface AxAIFeatures {
     functions: boolean;
     streaming: boolean;
     functionCot?: boolean;
+    thinkingTokenBudget?: boolean;
 }
 interface AxBaseAIArgs<TModel, TEmbedModel> {
     name: string;
@@ -87,6 +88,8 @@ declare class AxBaseAI<TModel, TEmbedModel, TChatRequest, TEmbedRequest, TChatRe
     private rt?;
     private fetch?;
     private tracer?;
+    private timeout?;
+    private excludeContentFromTrace?;
     private models?;
     private modelInfo;
     private modelUsage?;
@@ -159,6 +162,7 @@ type AxTokenUsage = {
     promptTokens: number;
     completionTokens: number;
     totalTokens: number;
+    thoughtsTokens?: number;
 };
 type AxModelConfig = {
     maxTokens?: number;
@@ -195,6 +199,7 @@ type AxFunction = {
 };
 type AxChatResponseResult = {
     content?: string;
+    thought?: string;
     name?: string;
     id?: string;
     functionCalls?: {
@@ -324,17 +329,21 @@ type AxRateLimiterFunction = <T = unknown>(reqFunc: () => Promise<T | ReadableSt
 }>) => Promise<T | ReadableStream$1<T>>;
 type AxAIPromptConfig = {
     stream?: boolean;
+    thinkingTokenBudget?: number;
 };
 type AxAIServiceOptions = {
     debug?: boolean;
     rateLimiter?: AxRateLimiterFunction;
     fetch?: typeof fetch;
     tracer?: Tracer;
+    timeout?: number;
+    excludeContentFromTrace?: boolean;
 };
 type AxAIServiceActionOptions<TModel = unknown, TEmbedModel = unknown> = {
     ai?: Readonly<AxAIService<TModel, TEmbedModel>>;
     sessionId?: string;
     traceId?: string;
+    timeout?: number;
     rateLimiter?: AxRateLimiterFunction;
     debug?: boolean;
     debugHideSystemPrompt?: boolean;
@@ -923,7 +932,7 @@ declare class AxAIDeepSeek extends AxAIOpenAIBase<AxAIDeepSeekModel, undefined>
 }
 
 declare enum AxAIGoogleGeminiModel {
-    Gemini25Pro = "gemini-2.5-pro-preview-03-25",
+    Gemini25Pro = "gemini-2.5-pro-preview-05-06",
     Gemini25Flash = "gemini-2.5-flash-preview-04-17",
     Gemini20Flash = "gemini-2.0-flash",
     Gemini20FlashLite = "gemini-2.0-flash-lite-preview-02-05",
@@ -967,6 +976,7 @@ type AxAIGoogleGeminiContent = {
     role: 'user';
     parts: ({
         text: string;
+        thought?: string;
     } | {
         inlineData: {
             mimeType: string;
@@ -1012,6 +1022,7 @@ type AxAIGoogleGeminiTool = {
     function_declarations?: AxAIGoogleGeminiToolFunctionDeclaration[];
     code_execution?: object;
     google_search_retrieval?: AxAIGoogleGeminiToolGoogleSearchRetrieval;
+    url_context?: object;
 };
 type AxAIGoogleGeminiToolConfig = {
     function_calling_config: {
@@ -1023,9 +1034,15 @@ type AxAIGoogleGeminiGenerationConfig = {
     temperature?: number;
     topP?: number;
     topK?: number;
+    frequencyPenalty?: number;
     candidateCount?: number;
     maxOutputTokens?: number;
     stopSequences?: readonly string[];
+    responseMimeType?: string;
+    thinkingConfig?: {
+        thinkingBudget?: number;
+        includeThoughts?: boolean;
+    };
 };
 type AxAIGoogleGeminiSafetySettings = {
     category: AxAIGoogleGeminiSafetyCategory;
@@ -1062,11 +1079,13 @@ type AxAIGoogleGeminiChatResponse = {
         promptTokenCount: number;
         candidatesTokenCount: number;
         totalTokenCount: number;
+        thoughtsTokenCount: number;
     };
 };
 type AxAIGoogleGeminiChatResponseDelta = AxAIGoogleGeminiChatResponse;
 type AxAIGoogleGeminiThinkingConfig = {
-    thinkingBudget: number;
+    thinkingTokenBudget?: number;
+    includeThoughts?: boolean;
 };
 /**
  * AxAIGoogleGeminiConfig: Configuration options for Google Gemini API
@@ -1078,7 +1097,8 @@ type AxAIGoogleGeminiConfig = AxModelConfig & {
     embedType?: AxAIGoogleGeminiEmbedTypes;
     dimensions?: number;
     autoTruncate?: boolean;
-    thinkingConfig?: AxAIGoogleGeminiThinkingConfig;
+    thinking?: AxAIGoogleGeminiThinkingConfig;
+    urlContext?: string;
 };
 /**
  * AxAIGoogleGeminiEmbedRequest: Structure for making an embedding request to the Google Gemini API.
@@ -1137,6 +1157,7 @@ interface AxAIGoogleGeminiOptionsTools {
         dynamicThreshold?: number;
     };
     googleSearch?: boolean;
+    urlContext?: boolean;
 }
 interface AxAIGoogleGeminiArgs {
     name: 'google-gemini';
@@ -1564,6 +1585,12 @@ declare class AxSignature {
     private updateHash;
     hash: () => string;
     toString: () => string;
+    toJSON: () => {
+        id: string;
+        description: string | undefined;
+        inputFields: AxIField[];
+        outputFields: AxIField[];
+    };
 }
 
 type AxFieldValue = string | string[] | number | boolean | object | null | undefined | {
@@ -1610,6 +1637,8 @@ type AxProgramForwardOptions = {
     fastFail?: boolean;
     debug?: boolean;
     debugHideSystemPrompt?: boolean;
+    thinkingTokenBudget?: number;
+    traceLabel?: string;
 };
 type AxProgramStreamingForwardOptions = Omit<AxProgramForwardOptions, 'stream'>;
 type AxGenDeltaOut<OUT> = {
@@ -1774,9 +1803,11 @@ interface AxGenOptions {
     asserts?: AxAssertion[];
     streamingAsserts?: AxStreamingAssertion[];
     fastFail?: boolean;
+    excludeContentFromTrace?: boolean;
+    traceLabel?: string;
 }
 type AxGenerateResult<OUT extends AxGenOut> = OUT & {
-    functions?: AxChatResponseFunctionCall[];
+    thought?: string;
 };
 interface AxResponseHandlerArgs<T> {
     ai: Readonly<AxAIService>;
@@ -1787,6 +1818,7 @@ interface AxResponseHandlerArgs<T> {
     traceId?: string;
     functions?: Readonly<AxFunction[]>;
     fastFail?: boolean;
+    span?: Span;
 }
 interface AxStreamingEvent<T> {
     event: 'delta' | 'done' | 'error';
@@ -1806,6 +1838,8 @@ declare class AxGen<IN extends AxGenIn = AxGenIn, OUT extends AxGenerateResult<A
     private functionsExecuted;
     private fieldProcessors;
     private streamingFieldProcessors;
+    private values;
+    private excludeContentFromTrace;
     constructor(signature: Readonly<AxSignature | string>, options?: Readonly<AxGenOptions>);
     addAssert: (fn: AxAssertion["fn"], message?: string) => void;
     addStreamingAssert: (fieldName: string, fn: AxStreamingAssertion["fn"], message?: string) => void;
@@ -2379,6 +2413,7 @@ declare class AxJSInterpreter {
 
 declare const axSpanAttributes: {
     LLM_SYSTEM: string;
+    LLM_OPERATION_NAME: string;
     LLM_REQUEST_MODEL: string;
     LLM_REQUEST_MAX_TOKENS: string;
     LLM_REQUEST_TEMPERATURE: string;
@@ -2388,8 +2423,10 @@ declare const axSpanAttributes: {
     LLM_REQUEST_STOP_SEQUENCES: string;
     LLM_REQUEST_LLM_IS_STREAMING: string;
     LLM_REQUEST_TOP_P: string;
-    LLM_USAGE_PROMPT_TOKENS: string;
-    LLM_USAGE_COMPLETION_TOKENS: string;
+    LLM_USAGE_INPUT_TOKENS: string;
+    LLM_USAGE_OUTPUT_TOKENS: string;
+    LLM_USAGE_TOTAL_TOKENS: string;
+    LLM_USAGE_THOUGHTS_TOKENS: string;
     DB_SYSTEM: string;
     DB_TABLE: string;
     DB_NAMESPACE: string;
@@ -2409,7 +2446,12 @@ declare const axSpanAttributes: {
     DB_QUERY_RESULT_DOCUMENT: string;
 };
 declare const axSpanEvents: {
-    LLM_PROMPT: string;
+    GEN_AI_USER_MESSAGE: string;
+    GEN_AI_SYSTEM_MESSAGE: string;
+    GEN_AI_ASSISTANT_MESSAGE: string;
+    GEN_AI_TOOL_MESSAGE: string;
+    GEN_AI_CHOICE: string;
+    GEN_AI_USAGE: string;
 };
 declare enum AxLLMRequestTypeValues {
     COMPLETION = "completion",