@ax-llm/ax 10.0.28 → 10.0.30

package/index.js

@@ -702,11 +702,11 @@ var logChatRequest = (req) => {
         return `${colorLog.blueBright("System:")}
 ${colorLog.whiteBright(msg.content)}`;
       case "function":
-        return `${colorLog.blueBright("Function Result:")}
+        return `${colorLog.blueBright("\nFunction Result:")}
 ${colorLog.whiteBright(msg.result)}`;
       case "user": {
         if (typeof msg.content === "string") {
-          return `${colorLog.blueBright("User:")}
+          return `${colorLog.blueBright("\nUser:")}
 ${colorLog.whiteBright(msg.content)}`;
         }
         const items2 = msg.content.map((v) => {
@@ -719,7 +719,7 @@ ${colorLog.whiteBright(msg.content)}`;
               throw new Error("Invalid content type");
           }
         });
-        return `${colorLog.blueBright("User:")}
+        return `${colorLog.blueBright("\nUser:")}
 ${items2.join("\n")}`;
       }
       case "assistant": {
@@ -728,10 +728,10 @@ ${items2.join("\n")}`;
             const args = typeof fn.params !== "string" ? JSON.stringify(fn.params, null, 2) : fn.params;
             return `${fn.name}(${args})`;
           });
-          return `${colorLog.blueBright("Functions:")}
+          return `${colorLog.blueBright("\nFunctions:")}
 ${colorLog.whiteBright(fns.join("\n"))}`;
         }
-        return `${colorLog.blueBright("Assistant:")}
+        return `${colorLog.blueBright("\nAssistant:")}
 ${colorLog.whiteBright(msg.content ?? "<empty>")}`;
       }
       default:
@@ -4007,6 +4007,15 @@ var AxProgram = class {
 };
 
 // dsp/prompt.ts
+var formattingRules = `
+When providing responses:
+1. Only output the exact requested content - no additional text, commentary, explanations, or clarifications
+2. Each key's value must strictly adhere to the formatting rules specified in the reference documentation 
+3. Follow all formatting conventions precisely as defined
+4. Do not add any extra content beyond what is explicitly requested
+5. Match the exact structure and format specifications for each field
+6. No preamble, postscript, or supplementary information
+7. Pure output only - conforming exactly to the documented requirements`;
 var AxPromptTemplate = class {
   sig;
   fieldTemplates;
@@ -4017,43 +4026,39 @@ var AxPromptTemplate = class {
     const inArgs = this.renderDescFields(this.sig.getInputFields());
     const outArgs = this.renderDescFields(this.sig.getOutputFields());
     const task = [
-      `#Task
-Given the fields ${inArgs}, produce the fields ${outArgs}.`
+      `You will be provided with the following fields: ${inArgs}. Your task is to generate two new fields: ${outArgs}.`
     ];
-    const fnNames = functions?.map((f) => {
-      if ("toFunction" in f) {
-        return f.toFunction().name;
-      }
-      return f.name;
-    });
-    const funcList = fnNames?.map((fname) => `'${fname}'`).join(", ");
+    const funcs = functions?.map(
+      (f) => "toFunction" in f ? f.toFunction() : f
+    );
+    const funcList = funcs?.map((fn) => `\`${fn.name}\`: ${capitalizeFirstLetter(fn.description)}`).join("\n");
     if (funcList && funcList.length > 0) {
+      task.push(`## Available Functions
+${funcList}`);
       task.push(
-        `Use the following functions ${funcList} to complete the task. The functions must be used to resolve the output field values.`
+        `Complete the task, using the functions defined earlier in this prompt, as needed. The output field values may be generated by applying these functions if appropriate for the task.`
       );
     }
-    const desc = this.sig.getDescription();
-    if (desc) {
-      task.push(desc.endsWith(".") ? desc : desc + ".");
-    }
-    task.push(
-      "Ensure the output strictly follows a plain text format, `key: value` separated by a new line."
-    );
     const inputFields = this.renderFields(this.sig.getInputFields());
+    task.push(`## Input Fields
+${inputFields}`);
     const outputFields = this.renderFields(this.sig.getOutputFields());
+    task.push(`## Output Fields
+${outputFields}`);
     task.push(
-      [
-        "\n",
-        "## Input Fields",
-        inputFields,
-        "\n",
-        "## Output Fields",
-        outputFields
-      ].join("\n")
+      "Output must be in plain text, with each `key: value` pair on a new line. The format of each `value` should strictly adhere to the formatting instructions for its corresponding `key`, as defined earlier in this prompt."
     );
+    task.push(formattingRules);
+    const desc = this.sig.getDescription();
+    if (desc) {
+      task.push(
+        `## TASK DESCRIPTION
+${capitalizeFirstLetter(desc.endsWith(".") ? desc : desc + ".")}`
+      );
+    }
     this.task = {
       type: "text",
-      text: task.join(" ")
+      text: task.join("\n\n")
     };
   }
   render = (values, {
@@ -4061,21 +4066,22 @@ Given the fields ${inArgs}, produce the fields ${outArgs}.`
     demos
   }) => {
     const renderedExamples = examples ? [
-      { type: "text", text: "Examples:\n" },
+      { type: "text", text: "## Examples:\n" },
       ...this.renderExamples(examples)
     ] : [];
     const renderedDemos = demos ? this.renderDemos(demos) : [];
     const completion = this.renderInputFields(values);
     const allTextExamples = renderedExamples.every((v) => v.type === "text");
     const allTextDemos = renderedDemos.every((v) => v.type === "text");
+    const examplesInSystemPrompt = allTextExamples && allTextDemos;
     let systemContent = this.task.text;
-    if (allTextExamples && allTextDemos) {
+    if (examplesInSystemPrompt) {
       const combinedItems = [
-        { type: "text", text: this.task.text },
+        { type: "text", text: systemContent + "\n\n" },
         ...renderedExamples,
         ...renderedDemos
       ];
-      combinedItems.reduce(combineConsecutiveStrings("\n"), []);
+      combinedItems.reduce(combineConsecutiveStrings(""), []);
       if (combinedItems && combinedItems[0]) {
         systemContent = combinedItems[0].text;
       }
@@ -4084,7 +4090,7 @@ Given the fields ${inArgs}, produce the fields ${outArgs}.`
       role: "system",
       content: systemContent
     };
-    const promptList = allTextExamples && allTextDemos ? completion : [...renderedExamples, ...renderedDemos, ...completion];
+    const promptList = examplesInSystemPrompt ? completion : [...renderedExamples, ...renderedDemos, ...completion];
     const prompt = promptList.filter((v) => v !== void 0);
     const userContent = prompt.every((v) => v.type === "text") ? prompt.map((v) => v.text).join("\n") : prompt.reduce(combineConsecutiveStrings("\n"), []);
     const userPrompt = {
@@ -4119,16 +4125,13 @@ Given the fields ${inArgs}, produce the fields ${outArgs}.`
       const renderedItem = [...renderedInputItem, ...renderedOutputItem];
       renderedItem.forEach((v) => {
         if ("text" in v) {
-          v.text = v.text;
+          v.text = v.text + "\n";
         }
         if ("image" in v) {
           v.image = v.image;
         }
         list.push(v);
       });
-      if (renderedItem.length > 0) {
-        list.push({ type: "text", text: "\n" });
-      }
     }
     return list;
   };
@@ -4139,23 +4142,20 @@ Given the fields ${inArgs}, produce the fields ${outArgs}.`
       const renderedItem = fields.map((field) => this.renderInField(field, item, true)).filter((v) => v !== void 0).flat();
       renderedItem.slice(0, -1).forEach((v) => {
         if ("text" in v) {
-          v.text = v.text;
+          v.text = v.text + "\n";
         }
         if ("image" in v) {
           v.image = v.image;
         }
         list.push(v);
       });
-      if (renderedItem.length > 0) {
-        list.push({ type: "text", text: "\n" });
-      }
     }
     return list;
   };
   renderInputFields = (values) => {
     const renderedItems = this.sig.getInputFields().map((field) => this.renderInField(field, values)).filter((v) => v !== void 0).flat();
     renderedItems.filter((v) => v.type === "text").forEach((v) => {
-      v.text = v.text + "\n\n";
+      v.text = v.text + "\n";
     });
     return renderedItems;
   };
@@ -4269,16 +4269,14 @@ Given the fields ${inArgs}, produce the fields ${outArgs}.`
   };
   renderDescFields = (list) => list.map((v) => `\`${v.title}\``).join(", ");
   renderFields = (fields) => {
-    const header = "Field Name | Field Type | Required/Optional | Description";
-    const separator = "|";
     const rows = fields.map((field) => {
       const name = field.title;
       const type = field.type?.name ? toFieldType(field.type) : "string";
       const required = field.isOptional ? "optional" : "required";
-      const description = field.description ?? "";
-      return [name, type, required, description].join(` ${separator} `).trim();
+      const description = field.description ? `: ${capitalizeFirstLetter(field.description)}` : "";
+      return `- \`${name}\` (${type}, ${required})${description}`.trim();
     });
-    return [header, ...rows].join("\n");
+    return rows.join("\n");
   };
 };
 var processValue = (field, value) => {
@@ -4353,6 +4351,12 @@ var isEmptyValue = (field, value) => {
   }
   return false;
 };
+function capitalizeFirstLetter(str) {
+  if (str.length === 0) {
+    return "";
+  }
+  return `${str.charAt(0).toUpperCase()}${str.slice(1)}`;
+}
 
 // dsp/validate.ts
 var AxValidationError = class extends Error {