@ax-llm/ax 16.0.5 → 16.0.7

package/index.d.cts

@@ -680,6 +680,16 @@ type AxAIServiceOptions = {
      * Currently supported by: Google Gemini/Vertex AI
      */
     contextCache?: AxContextCacheOptions;
+    /**
+     * When true, examples/demos are embedded in system prompt (legacy).
+     * When false (default), they are rendered as alternating user/assistant message pairs.
+     */
+    examplesInSystem?: boolean;
+    /**
+     * Custom labels to include in OpenTelemetry metrics.
+     * These labels are merged with axGlobals.customLabels (service-level overrides global).
+     */
+    customLabels?: Record<string, string>;
 };
 interface AxAIService<TModel = unknown, TEmbedModel = unknown, TModelKey = string> {
     getId(): string;
@@ -2055,6 +2065,7 @@ declare class AxGen<IN = any, OUT extends AxGenOut = any> extends AxProgram<IN,
     getInstruction(): string | undefined;
     private getSignatureName;
     private getMetricsInstruments;
+    private getMergedCustomLabels;
     updateMeter(meter?: Meter): void;
     private createStates;
     addAssert: (fn: AxAssertion<OUT>["fn"], message?: string) => void;
@@ -2662,6 +2673,7 @@ declare class AxBaseAI<TModel, TEmbedModel, TChatRequest, TEmbedRequest, TChatRe
     private logger;
     private corsProxy?;
     private retry?;
+    private customLabels?;
     private modelInfo;
     private modelUsage?;
     private embedModelUsage?;
@@ -2685,6 +2697,7 @@ declare class AxBaseAI<TModel, TEmbedModel, TChatRequest, TEmbedRequest, TChatRe
     setOptions(options: Readonly<AxAIServiceOptions>): void;
     getOptions(): Readonly<AxAIServiceOptions>;
     getLogger(): AxLoggerFunction;
+    private getMergedCustomLabels;
     getModelList(): ({
         readonly key: TModelKey;
         readonly description: string;
@@ -2863,15 +2876,10 @@ type AxAIAnthropicChatRequest = {
         })[];
     } | {
         role: 'assistant';
-        content: string | ({
+        content: string | (({
             type: 'text';
             text: string;
-        } | {
-            type: 'tool_use';
-            id: string;
-            name: string;
-            input: object;
-        } | {
+        } & AxAIAnthropicChatRequestCacheParam) | {
             type: 'thinking';
             thinking: string;
             signature?: string;
@@ -2879,7 +2887,12 @@ type AxAIAnthropicChatRequest = {
             type: 'redacted_thinking';
             data: string;
             signature?: string;
-        })[];
+        } | ({
+            type: 'tool_use';
+            id: string;
+            name: string;
+            input: object;
+        } & AxAIAnthropicChatRequestCacheParam))[];
     })[];
     tools?: AxAIAnthropicRequestTool[];
     tool_choice?: {
@@ -6138,6 +6151,7 @@ declare const axGlobals: {
     optimizerLogger: AxOptimizerLoggerFunction | undefined;
     debug: boolean | undefined;
     abortSignal: AbortSignal | undefined;
+    customLabels: Record<string, string> | undefined;
     cachingFunction: ((key: string, value?: AxGenOut) => AxGenOut | undefined | Promise<AxGenOut | undefined>) | undefined;
     functionResultFormatter: AxFunctionResultFormatter;
 };

package/index.d.ts

@@ -680,6 +680,16 @@ type AxAIServiceOptions = {
      * Currently supported by: Google Gemini/Vertex AI
      */
     contextCache?: AxContextCacheOptions;
+    /**
+     * When true, examples/demos are embedded in system prompt (legacy).
+     * When false (default), they are rendered as alternating user/assistant message pairs.
+     */
+    examplesInSystem?: boolean;
+    /**
+     * Custom labels to include in OpenTelemetry metrics.
+     * These labels are merged with axGlobals.customLabels (service-level overrides global).
+     */
+    customLabels?: Record<string, string>;
 };
 interface AxAIService<TModel = unknown, TEmbedModel = unknown, TModelKey = string> {
     getId(): string;
@@ -2055,6 +2065,7 @@ declare class AxGen<IN = any, OUT extends AxGenOut = any> extends AxProgram<IN,
     getInstruction(): string | undefined;
     private getSignatureName;
     private getMetricsInstruments;
+    private getMergedCustomLabels;
     updateMeter(meter?: Meter): void;
     private createStates;
     addAssert: (fn: AxAssertion<OUT>["fn"], message?: string) => void;
@@ -2662,6 +2673,7 @@ declare class AxBaseAI<TModel, TEmbedModel, TChatRequest, TEmbedRequest, TChatRe
     private logger;
     private corsProxy?;
     private retry?;
+    private customLabels?;
     private modelInfo;
     private modelUsage?;
     private embedModelUsage?;
@@ -2685,6 +2697,7 @@ declare class AxBaseAI<TModel, TEmbedModel, TChatRequest, TEmbedRequest, TChatRe
     setOptions(options: Readonly<AxAIServiceOptions>): void;
     getOptions(): Readonly<AxAIServiceOptions>;
     getLogger(): AxLoggerFunction;
+    private getMergedCustomLabels;
     getModelList(): ({
         readonly key: TModelKey;
         readonly description: string;
@@ -2863,15 +2876,10 @@ type AxAIAnthropicChatRequest = {
         })[];
     } | {
         role: 'assistant';
-        content: string | ({
+        content: string | (({
             type: 'text';
             text: string;
-        } | {
-            type: 'tool_use';
-            id: string;
-            name: string;
-            input: object;
-        } | {
+        } & AxAIAnthropicChatRequestCacheParam) | {
             type: 'thinking';
             thinking: string;
             signature?: string;
@@ -2879,7 +2887,12 @@ type AxAIAnthropicChatRequest = {
             type: 'redacted_thinking';
             data: string;
             signature?: string;
-        })[];
+        } | ({
+            type: 'tool_use';
+            id: string;
+            name: string;
+            input: object;
+        } & AxAIAnthropicChatRequestCacheParam))[];
     })[];
     tools?: AxAIAnthropicRequestTool[];
     tool_choice?: {
@@ -6138,6 +6151,7 @@ declare const axGlobals: {
     optimizerLogger: AxOptimizerLoggerFunction | undefined;
     debug: boolean | undefined;
     abortSignal: AbortSignal | undefined;
+    customLabels: Record<string, string> | undefined;
     cachingFunction: ((key: string, value?: AxGenOut) => AxGenOut | undefined | Promise<AxGenOut | undefined>) | undefined;
     functionResultFormatter: AxFunctionResultFormatter;
 };