@ax-llm/ax 11.0.13 → 11.0.15

package/index.d.cts

@@ -1374,6 +1374,7 @@ type AxProgramForwardOptions = {
     functions?: AxFunction[];
     functionCall?: AxChatRequest['functionCall'];
     stopFunction?: string;
+    earlyFail?: boolean;
 };
 type AxProgramStreamingForwardOptions = Omit<AxProgramForwardOptions, 'stream'>;
 type AxGenDeltaOut<OUT> = {
@@ -1576,6 +1577,7 @@ interface AxResponseHandlerArgs<T> {
     sessionId?: string;
     traceId?: string;
     functions?: Readonly<AxFunction[]>;
+    earlyFail?: boolean;
 }
 interface AxStreamingEvent<T> {
     event: 'delta' | 'done' | 'error';

package/index.d.ts

@@ -1374,6 +1374,7 @@ type AxProgramForwardOptions = {
     functions?: AxFunction[];
     functionCall?: AxChatRequest['functionCall'];
     stopFunction?: string;
+    earlyFail?: boolean;
 };
 type AxProgramStreamingForwardOptions = Omit<AxProgramForwardOptions, 'stream'>;
 type AxGenDeltaOut<OUT> = {
@@ -1576,6 +1577,7 @@ interface AxResponseHandlerArgs<T> {
     sessionId?: string;
     traceId?: string;
     functions?: Readonly<AxFunction[]>;
+    earlyFail?: boolean;
 }
 interface AxStreamingEvent<T> {
     event: 'delta' | 'done' | 'error';

package/index.js

@@ -4358,6 +4358,8 @@ var validateValue = (field, value) => {
         return val instanceof Date || typeof val === "string";
       case "datetime":
         return val instanceof Date || typeof val === "string";
+      case "json":
+        return typeof val === "object" || typeof val === "string";
       default:
         return false;
     }
@@ -4724,12 +4726,12 @@ var AxProgram = class {
 
 // dsp/prompt.ts
 var functionCallInstructions = `
-### Function Call Instructions
+## Function Call Instructions
 - Complete the task, using the functions defined earlier in this prompt. 
 - Call functions step-by-step, using the output of one function as input to the next.
 - Use the function results to generate the output fields.`;
 var formattingRules = `
-### Output Formatting Rules
+## Output Formatting Rules
 - Output must strictly follow the defined plain-text \`key: value\` field format.
 - Each output key, value must strictly adhere to the specified output field formatting rules.
 - No preamble, postscript, or supplementary information.
@@ -4755,14 +4757,14 @@ var AxPromptTemplate = class {
       (fn) => `- \`${fn.name}\`: ${capitalizeFirstLetter(fn.description)}.`
     ).join("\n");
     if (funcList && funcList.length > 0) {
-      task.push(`### Available Functions
+      task.push(`## Available Functions
 ${funcList}`);
     }
     const inputFields = this.renderFields(this.sig.getInputFields());
-    task.push(`### Input Fields
+    task.push(`## Input Fields
 ${inputFields}`);
     const outputFields = this.renderFields(this.sig.getOutputFields());
-    task.push(`### Output Fields
+    task.push(`## Output Fields
 ${outputFields}`);
     if (funcList && funcList.length > 0) {
       task.push(functionCallInstructions.trim());
@@ -5118,8 +5120,8 @@ var ValidationError = class extends Error {
   getFixingInstructions = () => {
     return this.fields.map((field) => ({
       name: "outputError",
-      title: "Errors In Output Fields",
-      description: `Please fix and return the field \`${field.title}\` of type \`${toFieldType(field.type)}\`, ${this.message}.`
+      title: "Output Correction Required",
+      description: `The section labeled '${field.title}' either was not generated by the LLM or does not match the expected format of '${toFieldType(field.type)}'. ${this.message} Please revise your response to ensure it conforms to the specified format.`
     }));
   };
 };
@@ -5716,7 +5718,7 @@ var AxGen = class extends AxProgramWithSignature {
     mem,
     options
   }) {
-    const { sessionId, traceId, model, functions } = options ?? {};
+    const { sessionId, traceId, model, functions, earlyFail } = options ?? {};
     const usageInfo = {
       ai: ai.getName(),
       model: ai.getModelInfo().name
@@ -5735,7 +5737,8 @@ var AxGen = class extends AxProgramWithSignature {
         mem,
         traceId,
         sessionId,
-        functions
+        functions,
+        earlyFail
       });
     } else {
       yield await this.processResponse({
@@ -5758,9 +5761,10 @@ var AxGen = class extends AxProgramWithSignature {
     mem,
     sessionId,
     traceId,
-    functions
+    functions,
+    earlyFail
   }) {
-    const streamingValidation = ai.getFeatures().functionCot !== true;
+    const streamingValidation = earlyFail ?? ai.getFeatures().functionCot !== true;
     const functionCalls = [];
     const values = {};
     const xstate = {