@ax-llm/ax 11.0.38 → 11.0.41

package/index.cjs

@@ -90,6 +90,7 @@ __export(index_exports, {
   AxFunctionError: () => AxFunctionError,
   AxFunctionProcessor: () => AxFunctionProcessor,
   AxGen: () => AxGen,
+  AxGenerateError: () => AxGenerateError,
   AxHFDataLoader: () => AxHFDataLoader,
   AxInstanceRegistry: () => AxInstanceRegistry,
   AxJSInterpreter: () => AxJSInterpreter,
@@ -6700,7 +6701,7 @@ Content: ${result.content}`
             err = e;
           } else if (e instanceof AxAIServiceStreamTerminatedError) {
           } else {
-            throw enhanceError(e, ai);
+            throw enhanceError(e, ai, this.signature);
           }
           if (errorFields) {
             handleValidationError(
@@ -6713,9 +6714,17 @@ Content: ${result.content}`
           }
         }
       }
-      throw new Error(`Unable to fix validation error: ${err?.toString()}`);
+      throw enhanceError(
+        new Error(`Unable to fix validation error: ${err?.toString()}`),
+        ai,
+        this.signature
+      );
     }
-    throw enhanceError(new Error(`Max steps reached: ${maxSteps}`), ai);
+    throw enhanceError(
+      new Error(`Max steps reached: ${maxSteps}`),
+      ai,
+      this.signature
+    );
   }
   shouldContinueSteps(lastMemItem, stopFunction) {
     const stopFunctionExecuted = stopFunction && this.functionsExecuted.has(stopFunction);
@@ -6786,19 +6795,31 @@ Content: ${result.content}`
     });
   }
 };
-function enhanceError(e, ai) {
+var AxGenerateError = class extends Error {
+  details;
+  constructor(message, details, options) {
+    super(message, options);
+    this.name = "AxGenerateError";
+    this.details = details;
+  }
+};
+function enhanceError(e, ai, signature) {
   const originalError = e instanceof Error ? e : new Error(String(e));
   const model = ai.getLastUsedChatModel();
   const modelConfig = ai.getLastUsedModelConfig();
-  const details = [
-    `model=${model}`,
-    `maxTokens=${modelConfig?.maxTokens ?? "N/A"}`,
-    `streaming=${modelConfig?.stream ?? false}`
-  ].join(", ");
-  const enhancedError = new Error();
-  enhancedError.stack = `Generate Failed: ${details}
-${originalError.stack}`;
-  return enhancedError;
+  const details = {
+    model,
+    maxTokens: modelConfig?.maxTokens,
+    streaming: modelConfig?.stream ?? false,
+    signature: {
+      input: signature.getInputFields(),
+      output: signature.getOutputFields(),
+      description: signature.getDescription()
+    }
+  };
+  return new AxGenerateError("Generate failed", details, {
+    cause: originalError
+  });
 }
 
 // prompts/agent.ts
@@ -11550,6 +11571,7 @@ var AxRAG = class extends AxChainOfThought {
   AxFunctionError,
   AxFunctionProcessor,
   AxGen,
+  AxGenerateError,
   AxHFDataLoader,
   AxInstanceRegistry,
   AxJSInterpreter,