@ax-llm/ax 14.0.1 → 14.0.3

package/index.d.cts

@@ -1237,7 +1237,7 @@ declare class AxAIOpenAIBase<TModel, TEmbedModel, TModelKey, TChatReq extends Ax
     constructor({ apiKey, config, options, apiURL, modelInfo, models, chatReqUpdater, supportFor, }: Readonly<Omit<AxAIOpenAIBaseArgs<TModel, TEmbedModel, TModelKey, TChatReq>, 'name'>>);
 }
 declare class AxAIOpenAI<TModelKey = string> extends AxAIOpenAIBase<AxAIOpenAIModel, AxAIOpenAIEmbedModel, TModelKey> {
-    constructor({ apiKey, config, options, models, modelInfo, }: Readonly<Omit<AxAIOpenAIArgs<'openai', AxAIOpenAIModel, AxAIOpenAIEmbedModel, TModelKey>, 'name'>>);
+    constructor({ apiKey, apiURL, config, options, models, modelInfo, }: Readonly<Omit<AxAIOpenAIArgs<'openai', AxAIOpenAIModel, AxAIOpenAIEmbedModel, TModelKey>, 'name'>>);
 }
 
 declare const axAIAzureOpenAIDefaultConfig: () => AxAIOpenAIConfig<AxAIOpenAIModel, AxAIOpenAIEmbedModel>;
@@ -4113,6 +4113,7 @@ type AxProgramForwardOptions<MODEL> = AxAIServiceOptions & {
     functionResultFormatter?: (result: unknown) => string;
     fastFail?: boolean;
     showThoughts?: boolean;
+    functionCallMode?: 'auto' | 'native' | 'prompt';
     traceLabel?: string;
     description?: string;
     thoughtFieldName?: string;
@@ -4369,6 +4370,26 @@ declare class AxSignature<_TInput extends Record<string, any> = Record<string, a
     validate: () => boolean;
     hash: () => string;
     toString: () => string;
+    /**
+     * Inject tool schemas as optional output fields for signature tool calling
+     * Uses dot notation for nested parameters (jq-like syntax)
+     */
+    injectToolFields(tools: readonly AxFunction[]): AxSignature<_TInput, _TOutput>;
+    /**
+     * Generate signature fields for tool parameters using dot notation
+     */
+    private generateToolParameterFields;
+    /**
+     * Infer signature field type from JSON Schema parameter
+     */
+    private inferParameterType;
+    /**
+     * Format parameter title for display
+     */
+    private formatParameterTitle;
+    private sanitizeFieldName;
+    private formatTitle;
+    private inferToolFieldType;
     toJSON: () => {
         id: string;
         description: string | undefined;
@@ -4480,6 +4501,7 @@ declare class AxGen<IN = any, OUT extends AxGenOut = any> extends AxProgram<IN,
     private streamingFieldProcessors;
     private excludeContentFromTrace;
     private thoughtFieldName;
+    private signatureToolCallingManager?;
     constructor(signature: NonNullable<ConstructorParameters<typeof AxSignature>[0]> | AxSignature<any, any>, options?: Readonly<AxProgramForwardOptions<any>>);
     private getSignatureName;
     private getMetricsInstruments;
@@ -5566,6 +5588,7 @@ declare const AxStringUtil: {
 
 declare function s<const T extends string>(signature: T): AxSignature<ParseSignature<T>['inputs'], ParseSignature<T>['outputs']>;
 declare function ax<const T extends string>(signature: T, options?: Readonly<AxProgramForwardOptions<any>>): AxGen<ParseSignature<T>['inputs'], ParseSignature<T>['outputs']>;
+declare function ax<TInput extends Record<string, any>, TOutput extends Record<string, any>>(signature: AxSignature<TInput, TOutput>, options?: Readonly<AxProgramForwardOptions<any>>): AxGen<TInput, TOutput>;
 
 /**
  * Analyzes mapping functions to extract state dependencies.
@@ -7163,6 +7186,7 @@ declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAge
     private disableSmartModelRouting?;
     private excludeFieldsFromPassthrough;
     private debug?;
+    private options?;
     private name;
     private func;
     constructor({ ai, name, description, definition, signature, agents, functions, }: Readonly<{
@@ -7239,21 +7263,30 @@ interface AxAgentConfig<IN extends AxGenIn, OUT extends AxGenOut> extends AxAgen
     functions?: AxInputFunctionType;
 }
 /**
- * Creates a strongly-typed AI agent from a string signature.
+ * Creates a strongly-typed AI agent from a signature.
  * This is the recommended way to create agents, providing better type inference and cleaner syntax.
+ * Supports both string signatures and AxSignature objects.
  *
- * @param signature - The input/output signature as a string (e.g., "userInput:string -> responseText:string")
+ * @param signature - The input/output signature as a string or AxSignature object
  * @param config - Configuration options for the agent
  * @returns A typed agent instance
  *
  * @example
  * ```typescript
+ * // Using string signature
  * const myAgent = agent('userInput:string -> responseText:string', {
  *   name: 'myAgent',
  *   description: 'An agent that processes user input and returns a response',
  *   definition: 'You are a helpful assistant that responds to user queries...'
  * });
  *
+ * // Using AxSignature object
+ * const sig = s('userInput:string -> responseText:string');
+ * const myAgent2 = agent(sig, {
+ *   name: 'myAgent2',
+ *   description: 'Same agent but using AxSignature object'
+ * });
+ *
  * // With child agents
  * const parentAgent = agent('taskDescription:string -> completedTask:string', {
  *   name: 'parentAgent',
@@ -7267,6 +7300,7 @@ interface AxAgentConfig<IN extends AxGenIn, OUT extends AxGenOut> extends AxAgen
  * ```
  */
 declare function agent<const T extends string>(signature: T, config: AxAgentConfig<ParseSignature<T>['inputs'], ParseSignature<T>['outputs']>): AxAgent<ParseSignature<T>['inputs'], ParseSignature<T>['outputs']>;
+declare function agent<TInput extends Record<string, any>, TOutput extends Record<string, any>>(signature: AxSignature<TInput, TOutput>, config: AxAgentConfig<TInput, TOutput>): AxAgent<TInput, TOutput>;
 
 /**
  * Advanced Multi-hop RAG with iterative query refinement, context accumulation,

package/index.d.ts

@@ -1237,7 +1237,7 @@ declare class AxAIOpenAIBase<TModel, TEmbedModel, TModelKey, TChatReq extends Ax
     constructor({ apiKey, config, options, apiURL, modelInfo, models, chatReqUpdater, supportFor, }: Readonly<Omit<AxAIOpenAIBaseArgs<TModel, TEmbedModel, TModelKey, TChatReq>, 'name'>>);
 }
 declare class AxAIOpenAI<TModelKey = string> extends AxAIOpenAIBase<AxAIOpenAIModel, AxAIOpenAIEmbedModel, TModelKey> {
-    constructor({ apiKey, config, options, models, modelInfo, }: Readonly<Omit<AxAIOpenAIArgs<'openai', AxAIOpenAIModel, AxAIOpenAIEmbedModel, TModelKey>, 'name'>>);
+    constructor({ apiKey, apiURL, config, options, models, modelInfo, }: Readonly<Omit<AxAIOpenAIArgs<'openai', AxAIOpenAIModel, AxAIOpenAIEmbedModel, TModelKey>, 'name'>>);
 }
 
 declare const axAIAzureOpenAIDefaultConfig: () => AxAIOpenAIConfig<AxAIOpenAIModel, AxAIOpenAIEmbedModel>;
@@ -4113,6 +4113,7 @@ type AxProgramForwardOptions<MODEL> = AxAIServiceOptions & {
     functionResultFormatter?: (result: unknown) => string;
     fastFail?: boolean;
     showThoughts?: boolean;
+    functionCallMode?: 'auto' | 'native' | 'prompt';
     traceLabel?: string;
     description?: string;
     thoughtFieldName?: string;
@@ -4369,6 +4370,26 @@ declare class AxSignature<_TInput extends Record<string, any> = Record<string, a
     validate: () => boolean;
     hash: () => string;
     toString: () => string;
+    /**
+     * Inject tool schemas as optional output fields for signature tool calling
+     * Uses dot notation for nested parameters (jq-like syntax)
+     */
+    injectToolFields(tools: readonly AxFunction[]): AxSignature<_TInput, _TOutput>;
+    /**
+     * Generate signature fields for tool parameters using dot notation
+     */
+    private generateToolParameterFields;
+    /**
+     * Infer signature field type from JSON Schema parameter
+     */
+    private inferParameterType;
+    /**
+     * Format parameter title for display
+     */
+    private formatParameterTitle;
+    private sanitizeFieldName;
+    private formatTitle;
+    private inferToolFieldType;
     toJSON: () => {
         id: string;
         description: string | undefined;
@@ -4480,6 +4501,7 @@ declare class AxGen<IN = any, OUT extends AxGenOut = any> extends AxProgram<IN,
     private streamingFieldProcessors;
     private excludeContentFromTrace;
     private thoughtFieldName;
+    private signatureToolCallingManager?;
     constructor(signature: NonNullable<ConstructorParameters<typeof AxSignature>[0]> | AxSignature<any, any>, options?: Readonly<AxProgramForwardOptions<any>>);
     private getSignatureName;
     private getMetricsInstruments;
@@ -5566,6 +5588,7 @@ declare const AxStringUtil: {
 
 declare function s<const T extends string>(signature: T): AxSignature<ParseSignature<T>['inputs'], ParseSignature<T>['outputs']>;
 declare function ax<const T extends string>(signature: T, options?: Readonly<AxProgramForwardOptions<any>>): AxGen<ParseSignature<T>['inputs'], ParseSignature<T>['outputs']>;
+declare function ax<TInput extends Record<string, any>, TOutput extends Record<string, any>>(signature: AxSignature<TInput, TOutput>, options?: Readonly<AxProgramForwardOptions<any>>): AxGen<TInput, TOutput>;
 
 /**
  * Analyzes mapping functions to extract state dependencies.
@@ -7163,6 +7186,7 @@ declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAge
     private disableSmartModelRouting?;
     private excludeFieldsFromPassthrough;
     private debug?;
+    private options?;
     private name;
     private func;
     constructor({ ai, name, description, definition, signature, agents, functions, }: Readonly<{
@@ -7239,21 +7263,30 @@ interface AxAgentConfig<IN extends AxGenIn, OUT extends AxGenOut> extends AxAgen
     functions?: AxInputFunctionType;
 }
 /**
- * Creates a strongly-typed AI agent from a string signature.
+ * Creates a strongly-typed AI agent from a signature.
  * This is the recommended way to create agents, providing better type inference and cleaner syntax.
+ * Supports both string signatures and AxSignature objects.
  *
- * @param signature - The input/output signature as a string (e.g., "userInput:string -> responseText:string")
+ * @param signature - The input/output signature as a string or AxSignature object
  * @param config - Configuration options for the agent
  * @returns A typed agent instance
  *
  * @example
  * ```typescript
+ * // Using string signature
  * const myAgent = agent('userInput:string -> responseText:string', {
  *   name: 'myAgent',
  *   description: 'An agent that processes user input and returns a response',
  *   definition: 'You are a helpful assistant that responds to user queries...'
  * });
  *
+ * // Using AxSignature object
+ * const sig = s('userInput:string -> responseText:string');
+ * const myAgent2 = agent(sig, {
+ *   name: 'myAgent2',
+ *   description: 'Same agent but using AxSignature object'
+ * });
+ *
  * // With child agents
  * const parentAgent = agent('taskDescription:string -> completedTask:string', {
  *   name: 'parentAgent',
@@ -7267,6 +7300,7 @@ interface AxAgentConfig<IN extends AxGenIn, OUT extends AxGenOut> extends AxAgen
  * ```
  */
 declare function agent<const T extends string>(signature: T, config: AxAgentConfig<ParseSignature<T>['inputs'], ParseSignature<T>['outputs']>): AxAgent<ParseSignature<T>['inputs'], ParseSignature<T>['outputs']>;
+declare function agent<TInput extends Record<string, any>, TOutput extends Record<string, any>>(signature: AxSignature<TInput, TOutput>, config: AxAgentConfig<TInput, TOutput>): AxAgent<TInput, TOutput>;
 
 /**
  * Advanced Multi-hop RAG with iterative query refinement, context accumulation,