@ax-llm/ax 11.0.13 → 11.0.14

package/index.cjs

@@ -4826,12 +4826,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.
@@ -4857,14 +4857,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());
@@ -5220,8 +5220,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.`
     }));
   };
 };
@@ -5818,7 +5818,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
@@ -5837,7 +5837,8 @@ var AxGen = class extends AxProgramWithSignature {
         mem,
         traceId,
         sessionId,
-        functions
+        functions,
+        earlyFail
       });
     } else {
       yield await this.processResponse({
@@ -5860,9 +5861,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 = {