@ax-llm/ax 11.0.39 → 11.0.42

package/index.cjs

@@ -90,6 +90,7 @@ __export(index_exports, {
   AxFunctionError: () => AxFunctionError,
   AxFunctionProcessor: () => AxFunctionProcessor,
   AxGen: () => AxGen,
+  AxGenerateError: () => AxGenerateError,
   AxHFDataLoader: () => AxHFDataLoader,
   AxInstanceRegistry: () => AxInstanceRegistry,
   AxJSInterpreter: () => AxJSInterpreter,
@@ -900,6 +901,7 @@ var AxBaseAI = class {
   rt;
   fetch;
   tracer;
+  timeout;
   models;
   modelInfo;
   modelUsage;
@@ -958,6 +960,7 @@ var AxBaseAI = class {
     this.debug = options.debug ?? false;
     this.rt = options.rateLimiter;
     this.fetch = options.fetch;
+    this.timeout = options.timeout;
     this.tracer = options.tracer;
   }
   getOptions() {
@@ -1141,6 +1144,7 @@ var AxBaseAI = class {
           url: this.apiURL,
           headers: await this.buildHeaders(apiConfig.headers),
           stream: modelConfig.stream,
+          timeout: this.timeout,
           debug,
           fetch: this.fetch,
           span
@@ -1270,6 +1274,7 @@ var AxBaseAI = class {
           headers: await this.buildHeaders(apiConfig.headers),
           debug,
           fetch: this.fetch,
+          timeout: this.timeout,
           span
         },
         reqValue
@@ -5948,7 +5953,7 @@ function* streamValues(sig, content, values, xstate) {
       const v = value.slice(s);
       if (v && v.length > 0) {
         yield { [key]: v };
-        xstate.streamedIndex[key] = s + 1;
+        xstate.streamedIndex[key] = s + v.length;
       }
       continue;
     }
@@ -6794,21 +6799,31 @@ Content: ${result.content}`
     });
   }
 };
+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 = [
-    `name=${ai.getName()}`,
-    `model=${model}`,
-    `maxTokens=${modelConfig?.maxTokens ?? "N/A"}`,
-    `streaming=${modelConfig?.stream ?? false}`
-  ].join(", ");
-  return new Error(
-    `Generate Failed:${signature.toString()}
-Details: ${details}`,
-    { cause: originalError }
-  );
+  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
@@ -11560,6 +11575,7 @@ var AxRAG = class extends AxChainOfThought {
   AxFunctionError,
   AxFunctionProcessor,
   AxGen,
+  AxGenerateError,
   AxHFDataLoader,
   AxInstanceRegistry,
   AxJSInterpreter,