@ax-llm/ax 12.0.6 → 12.0.7

package/index.cjs

@@ -2623,48 +2623,59 @@ function createMessages2(req) {
       case "system":
         return { role: "system", content: msg.content };
       case "user":
-        if (Array.isArray(msg.content)) {
+        const content = Array.isArray(msg.content) ? msg.content.map((c) => {
+          switch (c.type) {
+            case "text":
+              return { type: "text", text: c.text };
+            case "image": {
+              const url = `data:${c.mimeType};base64,` + c.image;
+              return {
+                type: "image_url",
+                image_url: { url, details: c.details ?? "auto" }
+              };
+            }
+            case "audio": {
+              const data = c.data;
+              return {
+                type: "input_audio",
+                input_audio: { data, format: c.format ?? "wav" }
+              };
+            }
+            default:
+              throw new Error("Invalid content type");
+          }
+        }) : msg.content;
+        return {
+          role: "user",
+          ...msg.name ? { name: msg.name } : {},
+          content
+        };
+      case "assistant":
+        const toolCalls = msg.functionCalls?.map((v) => ({
+          id: v.id,
+          type: "function",
+          function: {
+            name: v.function.name,
+            arguments: typeof v.function.params === "object" ? JSON.stringify(v.function.params) : v.function.params
+          }
+        }));
+        if (toolCalls && toolCalls.length > 0) {
           return {
-            role: "user",
+            role: "assistant",
+            ...msg.content ? { content: msg.content } : {},
             name: msg.name,
-            content: msg.content.map((c) => {
-              switch (c.type) {
-                case "text":
-                  return { type: "text", text: c.text };
-                case "image": {
-                  const url = `data:${c.mimeType};base64,` + c.image;
-                  return {
-                    type: "image_url",
-                    image_url: { url, details: c.details ?? "auto" }
-                  };
-                }
-                case "audio": {
-                  const data = c.data;
-                  return {
-                    type: "input_audio",
-                    input_audio: { data, format: c.format ?? "wav" }
-                  };
-                }
-                default:
-                  throw new Error("Invalid content type");
-              }
-            })
+            tool_calls: toolCalls
           };
         }
-        return { role: "user", content: msg.content, name: msg.name };
-      case "assistant":
+        if (!msg.content) {
+          throw new Error(
+            "Assistant content is required when no tool calls are provided"
+          );
+        }
         return {
           role: "assistant",
           content: msg.content,
-          name: msg.name,
-          tool_calls: msg.functionCalls?.map((v) => ({
-            id: v.id,
-            type: "function",
-            function: {
-              name: v.function.name,
-              arguments: typeof v.function.params === "object" ? JSON.stringify(v.function.params) : v.function.params
-            }
-          }))
+          ...msg.name ? { name: msg.name } : {}
         };
       case "function":
         return {
@@ -8215,11 +8226,41 @@ var AxSignature = class _AxSignature {
       if (signature.validatedAtHash === this.sigHash) {
         this.validatedAtHash = this.sigHash;
       }
+    } else if (typeof signature === "object" && signature !== null) {
+      if (!("inputs" in signature) || !("outputs" in signature)) {
+        throw new AxSignatureValidationError(
+          "Invalid signature object: missing inputs or outputs",
+          void 0,
+          'Signature object must have "inputs" and "outputs" arrays. Example: { inputs: [...], outputs: [...] }'
+        );
+      }
+      if (!Array.isArray(signature.inputs) || !Array.isArray(signature.outputs)) {
+        throw new AxSignatureValidationError(
+          "Invalid signature object: inputs and outputs must be arrays",
+          void 0,
+          'Both "inputs" and "outputs" must be arrays of AxField objects'
+        );
+      }
+      try {
+        this.description = signature.description;
+        this.inputFields = signature.inputs.map((v) => this.parseField(v));
+        this.outputFields = signature.outputs.map((v) => this.parseField(v));
+        [this.sigHash, this.sigString] = this.updateHash();
+      } catch (error) {
+        if (error instanceof AxSignatureValidationError) {
+          throw error;
+        }
+        throw new AxSignatureValidationError(
+          `Failed to create signature from object: ${error instanceof Error ? error.message : "Unknown error"}`,
+          void 0,
+          "Check that all fields in inputs and outputs arrays are valid AxField objects"
+        );
+      }
     } else {
       throw new AxSignatureValidationError(
         "Invalid signature argument type",
         void 0,
-        "Signature must be a string or another AxSignature instance"
+        "Signature must be a string, another AxSignature instance, or an object with inputs and outputs arrays"
       );
     }
   }
@@ -8262,7 +8303,7 @@ var AxSignature = class _AxSignature {
     }
     this.description = desc;
     this.invalidateValidationCache();
-    this.updateHash();
+    this.updateHashLight();
   };
   addInputField = (field) => {
     try {
@@ -8938,11 +8979,9 @@ var AxGen = class extends AxProgramWithSignature {
   values = {};
   excludeContentFromTrace = false;
   thoughtFieldName;
-  logger;
   constructor(signature, options) {
     super(signature, { description: options?.description });
     this.options = options;
-    this.logger = options?.logger;
     this.thoughtFieldName = options?.thoughtFieldName ?? "thought";
     const promptTemplateOptions = {
       functions: options?.functions,
@@ -9032,6 +9071,7 @@ var AxGen = class extends AxProgramWithSignature {
         rateLimiter,
         stream,
         debug: false,
+        // we do our own debug logging
         thinkingTokenBudget,
         showThoughts,
         traceContext,
@@ -9071,6 +9111,7 @@ var AxGen = class extends AxProgramWithSignature {
         fastFail,
         span
       });
+      this.getLogger(ai, options)?.("", { tags: ["responseEnd"] });
     } else {
       yield await this.processResponse({
         ai,
@@ -9107,7 +9148,6 @@ var AxGen = class extends AxProgramWithSignature {
     mem.addResult(
       {
         content: "",
-        name: "initial",
         functionCalls: []
       },
       sessionId
@@ -9240,10 +9280,6 @@ Content: ${content}`
         xstate
       );
     }
-    if (ai.getOptions().debug) {
-      const logger = ai.getLogger();
-      logger("", { tags: ["responseEnd"] });
-    }
   }
   async processResponse({
     ai,
@@ -9315,9 +9351,11 @@ Content: ${result.content}`
     const stopFunction = (options?.stopFunction ?? this.options?.stopFunction)?.toLowerCase();
     const maxRetries = options.maxRetries ?? this.options?.maxRetries ?? 10;
     const maxSteps = options.maxSteps ?? this.options?.maxSteps ?? 10;
-    const debug = options.debug ?? ai.getOptions().debug;
     const debugHideSystemPrompt = options.debugHideSystemPrompt;
-    const memOptions = { debug, debugHideSystemPrompt };
+    const memOptions = {
+      debug: this.isDebug(ai, options),
+      debugHideSystemPrompt
+    };
     const mem = options.mem ?? this.options?.mem ?? new AxMemory(1e4, memOptions);
     let err;
     if (options?.functions && options.functions.length > 0) {
@@ -9371,10 +9409,7 @@ Content: ${result.content}`
           if (shouldContinue) {
             continue multiStepLoop;
           }
-          if (debug) {
-            const logger = options.logger ?? this.logger ?? ai.getLogger();
-            logger("", { tags: ["responseEnd"] });
-          }
+          this.getLogger(ai, options)?.("", { tags: ["responseEnd"] });
           return;
         } catch (e) {
           let errorFields;
@@ -9516,6 +9551,12 @@ Content: ${result.content}`
   setExamples(examples, options) {
     super.setExamples(examples, options);
   }
+  isDebug(ai, options) {
+    return options?.debug ?? this.options?.debug ?? ai.getOptions().debug ?? false;
+  }
+  getLogger(ai, options) {
+    return options?.logger ?? this.options?.logger ?? ai.getLogger();
+  }
 };
 var AxGenerateError = class extends Error {
   details;