@ax-llm/ax 11.0.11 → 11.0.13

package/index.d.cts

@@ -1680,6 +1680,7 @@ declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut = AxGenOut> imple
      * @throws Error if description is too short
      */
     setDescription(description: string): void;
+    setDefinition(definition: string): void;
 }
 
 interface AxApacheTikaArgs {

package/index.d.ts

@@ -1680,6 +1680,7 @@ declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut = AxGenOut> imple
      * @throws Error if description is too short
      */
     setDescription(description: string): void;
+    setDefinition(definition: string): void;
 }
 
 interface AxApacheTikaArgs {

package/index.js

@@ -4743,16 +4743,10 @@ var AxPromptTemplate = class {
     this.sig = sig;
     this.fieldTemplates = fieldTemplates;
     const task = [];
-    const desc = this.sig.getDescription();
-    if (desc) {
-      const capitalized = capitalizeFirstLetter(desc.trim());
-      task.push(capitalized.endsWith(".") ? capitalized : capitalized + ".");
-    }
     const inArgs = this.renderDescFields(this.sig.getInputFields());
     const outArgs = this.renderDescFields(this.sig.getOutputFields());
     task.push(
-      `## Processing Instructions
-You will be provided with the following fields: ${inArgs}. Your task is to generate new fields: ${outArgs}.`
+      `You will be provided with the following fields: ${inArgs}. Your task is to generate new fields: ${outArgs}.`
     );
     const funcs = functions?.map(
       (f) => "toFunction" in f ? f.toFunction() : f
@@ -4774,6 +4768,13 @@ ${outputFields}`);
       task.push(functionCallInstructions.trim());
     }
     task.push(formattingRules.trim());
+    const desc = this.sig.getDescription();
+    if (desc) {
+      const capitalized = capitalizeFirstLetter(desc.trim());
+      const text = capitalized.endsWith(".") ? capitalized : capitalized + ".";
+      task.push(`---
+${text}`);
+    }
     this.task = {
       type: "text",
       text: task.join("\n\n")
@@ -5018,7 +5019,7 @@ ${outputFields}`);
       const type = field.type?.name ? toFieldType(field.type) : "string";
       const required = field.isOptional ? "optional" : "required";
       const description = field.description ? `: ${capitalizeFirstLetter(field.description)}` : "";
-      return `- \`${name}\` (${type}, ${required})${description}.`.trim();
+      return `- \`${name}:\` (${type}, ${required}) ${description}.`.trim();
     });
     return rows.join("\n");
   };
@@ -6053,6 +6054,12 @@ function processChildAgentFunction(childFunction, parentValues, parentInputKeys,
   }
   return processedFunction;
 }
+var descriptionError = new Error(
+  "Agent description must be at least 20 characters (explain in detail what the agent does)"
+);
+var definitionError = new Error(
+  "Agent definition is the prompt you give to the LLM for the agent. It must be detailed and at least 100 characters"
+);
 var AxAgent = class {
   ai;
   program;
@@ -6080,22 +6087,18 @@ var AxAgent = class {
     this.excludeFieldsFromPassthrough = excludeFieldsFromPassthrough ?? [];
     if (!name || name.length < 5) {
       throw new Error(
-        `Agent name must be at least 10 characters (more descriptive): ${name}`
+        `Agent name must be at least 10 characters (more descriptive)`
       );
     }
     if (!description || description.length < 20) {
-      throw new Error(
-        `Agent description must be at least 20 characters (explain in detail what the agent does): ${description}`
-      );
+      throw descriptionError;
     }
     if (definition && definition.length < 100) {
-      throw new Error(
-        `Agent definition is the prompt you give to the LLM for the agent. It must be detailed and at least 100 characters`
-      );
+      throw definitionError;
     }
     this.program = new AxGen(signature, {
       ...options,
-      description: definition
+      description: definition ?? description
     });
     for (const agent of agents ?? []) {
       this.program.register(agent);
@@ -6217,13 +6220,17 @@ var AxAgent = class {
    */
   setDescription(description) {
     if (!description || description.length < 20) {
-      throw new Error(
-        "Agent description must be at least 20 characters (explain in detail what the agent does)"
-      );
+      throw descriptionError;
     }
     this.program.getSignature().setDescription(description);
     this.func.description = description;
   }
+  setDefinition(definition) {
+    if (!definition || definition.length < 100) {
+      throw definitionError;
+    }
+    this.program.getSignature().setDescription(definition);
+  }
 };
 function toCamelCase(inputString) {
   const words = inputString.split(/[^a-zA-Z0-9]/);