@ax-llm/ax 10.0.39 → 10.0.41

package/index.cjs

@@ -33,6 +33,7 @@ __export(index_exports, {
   AxAI: () => AxAI,
   AxAIAnthropic: () => AxAIAnthropic,
   AxAIAnthropicModel: () => AxAIAnthropicModel,
+  AxAIAnthropicVertexModel: () => AxAIAnthropicVertexModel,
   AxAIAzureOpenAI: () => AxAIAzureOpenAI,
   AxAICohere: () => AxAICohere,
   AxAICohereEmbedModel: () => AxAICohereEmbedModel,
@@ -84,6 +85,7 @@ __export(index_exports, {
   AxJSInterpreterPermission: () => AxJSInterpreterPermission,
   AxLLMRequestTypeValues: () => AxLLMRequestTypeValues,
   AxMemory: () => AxMemory,
+  AxMockAIService: () => AxMockAIService,
   AxProgram: () => AxProgram,
   AxProgramWithSignature: () => AxProgramWithSignature,
   AxPromptTemplate: () => AxPromptTemplate,
@@ -618,9 +620,11 @@ var AxBaseAI = class {
           }
         },
         async (span) => {
-          const res = await this._chat2(model, modelConfig, req, options, span);
-          span.end();
-          return res;
+          try {
+            return await this._chat2(model, modelConfig, req, options, span);
+          } finally {
+            span.end();
+          }
         }
       );
     }
@@ -743,9 +747,11 @@ var AxBaseAI = class {
           }
         },
         async (span) => {
-          const res = await this._embed2(embedModel, req, options, span);
-          span.end();
-          return res;
+          try {
+            return await this._embed2(embedModel, req, options, span);
+          } finally {
+            span.end();
+          }
         }
       );
     }
@@ -813,7 +819,7 @@ ${colorLog.whiteBright(msg.content)}`;
         const items2 = msg.content.map((v) => {
           switch (v.type) {
             case "text":
-              return `(Text) ${colorLog.whiteBright(v.text)}`;
+              return `${colorLog.whiteBright(v.text)}`;
             case "image":
               return `(Image, ${v.mimeType}) ${colorLog.whiteBright(v.image.substring(0, 10))}`;
             default:
@@ -878,6 +884,61 @@ var setResponseAttr = (res, span) => {
   }
 };
 
+// ai/google-vertex/auth.ts
+var import_google_auth_library = require("google-auth-library");
+var GoogleVertexAuth = class {
+  auth;
+  client;
+  currentToken;
+  tokenExpiry;
+  constructor(config = {}) {
+    this.auth = new import_google_auth_library.GoogleAuth({
+      scopes: ["https://www.googleapis.com/auth/cloud-platform"],
+      ...config
+    });
+  }
+  async getAuthenticatedClient() {
+    if (!this.client) {
+      this.client = await this.auth.getClient();
+    }
+    return this.client;
+  }
+  async getAccessToken() {
+    if (this.currentToken && this.tokenExpiry && Date.now() < this.tokenExpiry) {
+      return this.currentToken;
+    }
+    const client = await this.getAuthenticatedClient();
+    const tokenResponse = await client.getAccessToken();
+    this.currentToken = tokenResponse.token ?? void 0;
+    const expiry = this.getExpiry(tokenResponse);
+    const fiveMinutes = 5 * 60 * 1e3;
+    this.tokenExpiry = expiry - fiveMinutes;
+    return this.currentToken;
+  }
+  /**
+   * Get the expiry date from the token response.
+   */
+  getExpiry(tokenResponse) {
+    const oneHour = 3600 * 1e3;
+    let expiry = Date.now() + oneHour;
+    let responseExpiry = tokenResponse.res?.data?.expiry_date;
+    if (responseExpiry) {
+      if (typeof responseExpiry === "number") {
+        expiry = responseExpiry;
+      } else if (responseExpiry instanceof Date) {
+        expiry = responseExpiry.getTime();
+      } else if (typeof responseExpiry === "string") {
+        expiry = new Date(responseExpiry).getTime();
+      } else {
+        console.warn("Unknown expiry type", responseExpiry);
+      }
+    } else {
+      console.warn("No expiry date found in response", tokenResponse.res?.data);
+    }
+    return expiry;
+  }
+};
+
 // ai/anthropic/types.ts
 var AxAIAnthropicModel = /* @__PURE__ */ ((AxAIAnthropicModel2) => {
   AxAIAnthropicModel2["Claude35Sonnet"] = "claude-3-5-sonnet-latest";
@@ -889,6 +950,14 @@ var AxAIAnthropicModel = /* @__PURE__ */ ((AxAIAnthropicModel2) => {
   AxAIAnthropicModel2["ClaudeInstant12"] = "claude-instant-1.2";
   return AxAIAnthropicModel2;
 })(AxAIAnthropicModel || {});
+var AxAIAnthropicVertexModel = /* @__PURE__ */ ((AxAIAnthropicVertexModel2) => {
+  AxAIAnthropicVertexModel2["Claude35Haiku"] = "claude-3-5-haiku";
+  AxAIAnthropicVertexModel2["Claude35Sonnet"] = "claude-3-5-sonnet";
+  AxAIAnthropicVertexModel2["Claude35SonnetV2"] = "claude-3-5-sonnet-v2";
+  AxAIAnthropicVertexModel2["Claude3Haiku"] = "claude-3-haiku";
+  AxAIAnthropicVertexModel2["Claude3Opus"] = "claude-3-opus";
+  return AxAIAnthropicVertexModel2;
+})(AxAIAnthropicVertexModel || {});
 
 // ai/anthropic/info.ts
 var axModelInfoAnthropic = [
@@ -945,8 +1014,9 @@ var axAIAnthropicDefaultConfig = () => structuredClone({
   ...axBaseAIDefaultConfig()
 });
 var AxAIAnthropicImpl = class {
-  constructor(config) {
+  constructor(config, isVertex) {
     this.config = config;
+    this.isVertex = isVertex;
   }
   getModelConfig() {
     const { config } = this;
@@ -965,9 +1035,41 @@ var AxAIAnthropicImpl = class {
   }
   createChatReq = (req) => {
     const model = req.model;
-    const apiConfig = {
-      name: "/messages"
-    };
+    const stream = req.modelConfig?.stream ?? this.config.stream;
+    let apiConfig;
+    if (this.isVertex) {
+      apiConfig = {
+        name: stream ? `/models/${model}:streamRawPredict?alt=sse` : `/models/${model}:rawPredict`
+      };
+    } else {
+      apiConfig = {
+        name: "/messages"
+      };
+    }
+    let toolsChoice;
+    if (req.functionCall && req.functions && req.functions.length > 0) {
+      if (typeof req.functionCall === "string") {
+        switch (req.functionCall) {
+          case "auto":
+            toolsChoice = { tool_choice: { type: "auto" } };
+            break;
+          case "required":
+            toolsChoice = { tool_choice: { type: "any" } };
+            break;
+          case "none":
+            throw new Error("functionCall none not supported");
+        }
+      } else if ("function" in req.functionCall) {
+        toolsChoice = {
+          tool_choice: {
+            type: "tool",
+            name: req.functionCall.function.name
+          }
+        };
+      } else {
+        throw new Error("Invalid function call type, must be string or object");
+      }
+    }
     const system = req.chatPrompt.filter((msg) => msg.role === "system").map((msg) => ({
       type: "text",
       text: msg.content,
@@ -982,15 +1084,15 @@ var AxAIAnthropicImpl = class {
         input_schema: v.parameters
       })
     );
-    const stream = req.modelConfig?.stream ?? this.config.stream;
     const reqValue = {
-      model,
+      ...this.isVertex ? { anthropic_version: "vertex-2023-10-16" } : { model },
       max_tokens: req.modelConfig?.maxTokens ?? this.config.maxTokens,
       stop_sequences: req.modelConfig?.stopSequences ?? this.config.stopSequences,
       temperature: req.modelConfig?.temperature ?? this.config.temperature,
       top_p: req.modelConfig?.topP ?? this.config.topP,
       top_k: req.modelConfig?.topK ?? this.config.topK,
-      ...tools && tools.length > 0 ? { tools, tool_choice: { type: "auto" } } : {},
+      ...toolsChoice,
+      ...tools && tools.length > 0 ? { tools } : {},
       ...stream ? { stream: true } : {},
       ...system ? { system } : {},
       messages
@@ -1138,26 +1240,45 @@ var AxAIAnthropicImpl = class {
 var AxAIAnthropic = class extends AxBaseAI {
   constructor({
     apiKey,
+    projectId,
+    region,
     config,
     options,
     modelMap
   }) {
-    if (!apiKey || apiKey === "") {
-      throw new Error("Anthropic API key not set");
+    const isVertex = projectId !== void 0 && region !== void 0;
+    let apiURL;
+    let headers;
+    if (isVertex) {
+      apiURL = `https://${region}-aiplatform.googleapis.com/v1/projects/${projectId}/locations/${region}/publishers/anthropic/`;
+      if (apiKey) {
+        headers = async () => ({ Authorization: `Bearer ${apiKey}` });
+      } else {
+        const vertexAuth = new GoogleVertexAuth();
+        headers = async () => ({
+          Authorization: `Bearer ${await vertexAuth.getAccessToken()}`
+        });
+      }
+    } else {
+      if (!apiKey) {
+        throw new Error("Anthropic API key not set");
+      }
+      apiURL = "https://api.anthropic.com/v1";
+      headers = async () => ({
+        "anthropic-version": "2023-06-01",
+        "anthropic-beta": "prompt-caching-2024-07-31",
+        "x-api-key": apiKey
+      });
     }
     const _config = {
       ...axAIAnthropicDefaultConfig(),
       ...config
     };
-    const aiImpl = new AxAIAnthropicImpl(_config);
+    const aiImpl = new AxAIAnthropicImpl(_config, isVertex);
     super(aiImpl, {
       name: "Anthropic",
-      apiURL: "https://api.anthropic.com/v1",
-      headers: async () => ({
-        "anthropic-version": "2023-06-01",
-        "anthropic-beta": "prompt-caching-2024-07-31",
-        "x-api-key": apiKey
-      }),
+      apiURL,
+      headers,
       modelInfo: axModelInfoAnthropic,
       models: { model: _config.model },
       options,
@@ -2094,61 +2215,6 @@ var AxAIDeepSeek = class extends AxAIOpenAI {
   }
 };
 
-// ai/google-gemini/auth.ts
-var import_google_auth_library = require("google-auth-library");
-var GoogleVertexAuth = class {
-  auth;
-  client;
-  currentToken;
-  tokenExpiry;
-  constructor(config = {}) {
-    this.auth = new import_google_auth_library.GoogleAuth({
-      scopes: ["https://www.googleapis.com/auth/cloud-platform"],
-      ...config
-    });
-  }
-  async getAuthenticatedClient() {
-    if (!this.client) {
-      this.client = await this.auth.getClient();
-    }
-    return this.client;
-  }
-  async getAccessToken() {
-    if (this.currentToken && this.tokenExpiry && Date.now() < this.tokenExpiry) {
-      return this.currentToken;
-    }
-    const client = await this.getAuthenticatedClient();
-    const tokenResponse = await client.getAccessToken();
-    this.currentToken = tokenResponse.token ?? void 0;
-    const expiry = this.getExpiry(tokenResponse);
-    const fiveMinutes = 5 * 60 * 1e3;
-    this.tokenExpiry = expiry - fiveMinutes;
-    return this.currentToken;
-  }
-  /**
-   * Get the expiry date from the token response.
-   */
-  getExpiry(tokenResponse) {
-    const oneHour = 3600 * 1e3;
-    let expiry = Date.now() + oneHour;
-    let responseExpiry = tokenResponse.res?.data?.expiry_date;
-    if (responseExpiry) {
-      if (typeof responseExpiry === "number") {
-        expiry = responseExpiry;
-      } else if (responseExpiry instanceof Date) {
-        expiry = responseExpiry.getTime();
-      } else if (typeof responseExpiry === "string") {
-        expiry = new Date(responseExpiry).getTime();
-      } else {
-        console.warn("Unknown expiry type", responseExpiry);
-      }
-    } else {
-      console.warn("No expiry date found in response", tokenResponse.res?.data);
-    }
-    return expiry;
-  }
-};
-
 // ai/google-gemini/types.ts
 var AxAIGoogleGeminiModel = /* @__PURE__ */ ((AxAIGoogleGeminiModel2) => {
   AxAIGoogleGeminiModel2["Gemini1Pro"] = "gemini-1.0-pro";
@@ -2238,11 +2304,10 @@ var axAIGoogleGeminiDefaultConfig = () => structuredClone({
   ...axBaseAIDefaultConfig()
 });
 var AxAIGoogleGeminiImpl = class {
-  constructor(config, isVertex, apiKey, keyFile, options) {
+  constructor(config, isVertex, apiKey, options) {
     this.config = config;
     this.isVertex = isVertex;
     this.apiKey = apiKey;
-    this.keyFile = keyFile;
     this.options = options;
   }
   getModelConfig() {
@@ -2519,7 +2584,6 @@ var AxAIGoogleGemini = class extends AxBaseAI {
     apiKey,
     projectId,
     region,
-    keyFile,
     config,
     options,
     modelMap
@@ -2532,9 +2596,7 @@ var AxAIGoogleGemini = class extends AxBaseAI {
       if (apiKey) {
         headers = async () => ({ Authorization: `Bearer ${apiKey}` });
       } else {
-        const vertexAuth = new GoogleVertexAuth({
-          keyFile
-        });
+        const vertexAuth = new GoogleVertexAuth();
         headers = async () => ({
           Authorization: `Bearer ${await vertexAuth.getAccessToken()}`
         });
@@ -2550,13 +2612,7 @@ var AxAIGoogleGemini = class extends AxBaseAI {
       ...axAIGoogleGeminiDefaultConfig(),
       ...config
     };
-    const aiImpl = new AxAIGoogleGeminiImpl(
-      _config,
-      isVertex,
-      apiKey,
-      keyFile,
-      options
-    );
+    const aiImpl = new AxAIGoogleGeminiImpl(_config, isVertex, apiKey, options);
     super(aiImpl, {
       name: "GoogleGeminiAI",
       apiURL,
@@ -3299,9 +3355,6 @@ var AxAI = class {
   }
 };
 
-// prompts/agent.ts
-var import_api23 = require("@opentelemetry/api");
-
 // dsp/generate.ts
 var import_web5 = require("stream/web");
 var import_api22 = require("@opentelemetry/api");
@@ -3490,7 +3543,7 @@ var AxAssertionError = class extends Error {
     extraFields.push({
       name: "error",
       title: "Error In Output",
-      description: this.message
+      description: `You must follow the following instructions, "${this.message}".`
     });
     return extraFields;
   };
@@ -3529,18 +3582,6 @@ var assertStreamingAssertions = (asserts, values, xstate, content, final) => {
     }
   }
 };
-var assertRequiredFields = (sig, values) => {
-  const fields = sig.getOutputFields();
-  const missingFields = fields.filter(
-    (f) => !f.isOptional && !(f.name in values)
-  );
-  if (missingFields.length > 0) {
-    throw new AxAssertionError({
-      message: `Output must include: ${missingFields.map((f) => `\`${f.title}:\``).join(", ")}`,
-      values
-    });
-  }
-};
 
 // dsp/extract.ts
 var import_json5 = __toESM(require("json5"), 1);
@@ -4088,6 +4129,67 @@ var parseMarkdownList = (input) => {
   }
   return list;
 };
+function mergeDeltas(base, delta) {
+  for (const key of Object.keys(delta)) {
+    const baseValue = base[key];
+    const deltaValue = delta[key];
+    if ((baseValue === void 0 || Array.isArray(baseValue)) && Array.isArray(deltaValue)) {
+      base[key] = [...baseValue ?? [], ...deltaValue];
+    } else if ((baseValue === void 0 || typeof baseValue === "string") && typeof deltaValue === "string") {
+      base[key] = (baseValue ?? "") + deltaValue;
+    } else {
+      base[key] = deltaValue;
+    }
+  }
+  return base;
+}
+var LRUCache = class {
+  cache = /* @__PURE__ */ new Map();
+  maxSize;
+  constructor(maxSize) {
+    this.maxSize = maxSize;
+  }
+  get(key) {
+    const value = this.cache.get(key);
+    if (value) {
+      this.cache.delete(key);
+      this.cache.set(key, value);
+    }
+    return value;
+  }
+  set(key, value) {
+    if (this.cache.has(key)) {
+      this.cache.delete(key);
+    } else if (this.cache.size >= this.maxSize) {
+      const firstKey = this.cache.keys().next().value;
+      if (firstKey) {
+        this.cache.delete(firstKey);
+      }
+    }
+    this.cache.set(key, value);
+  }
+};
+var globalPrefixCache = new LRUCache(500);
+function matchesContent(content, prefix, startIndex = 0, prefixCache = globalPrefixCache) {
+  const exactMatchIndex = content.indexOf(prefix, startIndex);
+  if (exactMatchIndex !== -1) {
+    return exactMatchIndex;
+  }
+  const prefixes = prefixCache.get(prefix) ?? Array.from({ length: prefix.length }, (_, i) => prefix.slice(0, i + 1));
+  if (!prefixCache.get(prefix)) {
+    prefixCache.set(prefix, prefixes);
+  }
+  const contentEnd = content.slice(
+    Math.max(startIndex, content.length - prefix.length)
+  );
+  for (let i = 0; i < prefixes.length - 1; i++) {
+    const partialPrefix = prefixes[i];
+    if (contentEnd.endsWith(partialPrefix)) {
+      return -2;
+    }
+  }
+  return -1;
+}
 
 // dsp/program.ts
 var AxProgramWithSignature = class {
@@ -4120,6 +4222,9 @@ var AxProgramWithSignature = class {
   async forward(_ai, _values, _options) {
     throw new Error("forward() not implemented");
   }
+  async *streamingForward(_ai, _values, _options) {
+    throw new Error("streamingForward() not implemented");
+  }
   setId(id) {
     this.key = { id, custom: true };
     for (const child of this.children) {
@@ -4214,6 +4319,9 @@ var AxProgram = class {
   async forward(_ai, _values, _options) {
     throw new Error("forward() not implemented");
   }
+  async *streamingForward(_ai, _values, _options) {
+    throw new Error("streamingForward() not implemented");
+  }
   setId(id) {
     this.key = { id, custom: true };
     for (const child of this.children) {
@@ -4311,9 +4419,10 @@ ${outputFields}`);
     task.push(formattingRules.trim());
     const desc = this.sig.getDescription();
     if (desc) {
+      const capitalized = capitalizeFirstLetter(desc.trim());
       task.push(
         `## TASK DESCRIPTION
-${capitalizeFirstLetter(desc.endsWith(".") ? desc : desc + ".")}`
+${capitalized.endsWith(".") ? capitalized : capitalized + "."}`
       );
     }
     this.task = {
@@ -4361,16 +4470,42 @@ ${capitalizeFirstLetter(desc.endsWith(".") ? desc : desc + ".")}`
   };
   renderExtraFields = (extraFields) => {
     const prompt = [];
-    if (extraFields && extraFields.length > 0) {
-      extraFields.forEach((field) => {
-        const fn = this.fieldTemplates?.[field.name] ?? this.defaultRenderInField;
-        prompt.push(...fn(field, field.description));
-      });
-    }
-    if (prompt.every((v) => v.type === "text")) {
-      return prompt.map((v) => v.text).join("\n\n");
-    }
-    return prompt.reduce(combineConsecutiveStrings("\n"), []);
+    if (!extraFields || extraFields.length === 0) {
+      return prompt;
+    }
+    const groupedFields = extraFields.reduce(
+      (acc, field) => {
+        const title = field.title;
+        if (!acc[title]) {
+          acc[title] = [];
+        }
+        acc[title].push(field);
+        return acc;
+      },
+      {}
+    );
+    const formattedGroupedFields = Object.entries(groupedFields).map(([title, fields]) => {
+      if (fields.length === 1) {
+        const field = fields[0];
+        return {
+          title,
+          name: field.name,
+          description: field.description
+        };
+      } else if (fields.length > 1) {
+        const valuesList = fields.map((field) => `- ${field.description}`).join("\n");
+        return {
+          title,
+          name: fields[0].name,
+          description: valuesList
+        };
+      }
+    }).filter(Boolean);
+    formattedGroupedFields.forEach((field) => {
+      const fn = this.fieldTemplates?.[field.name] ?? this.defaultRenderInField;
+      prompt.push(...fn(field, field.description));
+    });
+    return prompt;
   };
   renderExamples = (data) => {
     const list = [];
@@ -4619,34 +4754,44 @@ function capitalizeFirstLetter(str) {
 }
 
 // dsp/validate.ts
+var colorLog4 = new ColorLog();
 var ValidationError = class extends Error {
-  field;
-  value;
+  fields;
   constructor({
     message,
-    field,
-    value
+    fields
   }) {
     super(message);
-    this.field = field;
-    this.value = value;
+    this.fields = fields;
     this.name = this.constructor.name;
     Error.captureStackTrace(this, this.constructor);
   }
-  getField = () => this.field;
-  getValue = () => this.value;
+  getFields = () => this.fields;
   getFixingInstructions = () => {
-    const f = this.field;
-    const extraFields = [
-      {
-        name: `outputError`,
-        title: `Invalid Output Field`,
-        description: `Invalid format for field \`${f.title}\` of type \`${toFieldType(f.type)}\`, format should match: \`${f.description}\``
-      }
-    ];
-    return extraFields;
+    return this.fields.map((field) => ({
+      name: "outputError",
+      title: "Error In Output",
+      description: `Please fix and return the field \`${field.title}\` of type \`${toFieldType(field.type)}\`, ${this.message}.`
+    }));
   };
 };
+function handleValidationError(mem, errorFields, ai, promptTemplate, sessionId) {
+  mem.add(
+    {
+      role: "user",
+      content: promptTemplate.renderExtraFields(errorFields)
+    },
+    sessionId
+  );
+  mem.addTag("error");
+  if (ai.getOptions().debug) {
+    const errors = errorFields.map((field) => `- ${field.title}: ${field.description}`).join("\n");
+    process.stdout.write(colorLog4.red(`
+Error Correction:
+${errors}
+`));
+  }
+}
 
 // dsp/datetime.ts
 function parseLLMFriendlyDate(field, dateStr) {
@@ -4654,7 +4799,7 @@ function parseLLMFriendlyDate(field, dateStr) {
     return _parseLLMFriendlyDate(dateStr);
   } catch (err) {
     const message = err.message;
-    throw new ValidationError({ field, message, value: dateStr });
+    throw new ValidationError({ fields: [field], message, value: dateStr });
   }
 }
 function _parseLLMFriendlyDate(dateStr) {
@@ -4671,7 +4816,7 @@ function parseLLMFriendlyDateTime(field, dateStr) {
     return _parseLLMFriendlyDateTime(dateStr);
   } catch (err) {
     const message = err.message;
-    throw new ValidationError({ field, message, value: dateStr });
+    throw new ValidationError({ fields: [field], message, value: dateStr });
   }
 }
 function _parseLLMFriendlyDateTime(dateTimeStr) {
@@ -4713,41 +4858,73 @@ var formatDateWithTimezone = (date) => {
 
 // dsp/extract.ts
 var extractValues = (sig, values, content) => {
-  const xstate = { s: -1 };
+  const xstate = { extractedFields: [], s: -1 };
   streamingExtractValues(sig, values, xstate, content);
-  streamingExtractFinalValue(values, xstate, content);
+  streamingExtractFinalValue(sig, values, xstate, content);
 };
-var streamingExtractValues = (sig, values, state, content) => {
+var checkMissingRequiredFields = (xstate, values, currentIndex) => {
+  const missingFields = [];
+  for (let i = 0; i < currentIndex; i++) {
+    const field = xstate.extractedFields[i];
+    if (field && !field.isOptional && values[field.name] === void 0) {
+      missingFields.push(field);
+    }
+  }
+  if (missingFields.length > 0) {
+    throw new ValidationError({
+      message: `Required ${missingFields.length === 1 ? "field" : "fields"} not found`,
+      fields: missingFields
+    });
+  }
+};
+var streamingExtractValues = (sig, values, xstate, content) => {
+  if (content.endsWith("\n")) {
+    return true;
+  }
   const fields = sig.getOutputFields();
-  for (const field of fields) {
+  for (const [index, field] of fields.entries()) {
     if (field.name in values) {
       continue;
     }
     const prefix = field.title + ":";
-    const e = content.indexOf(prefix, state.s + 1);
-    if (e === -1) {
-      continue;
+    let e = matchesContent(content, prefix, xstate.s + 1);
+    switch (e) {
+      case -1:
+        continue;
+      // Field is not found, continue to the next field
+      case -2:
+        return true;
     }
-    if (state.currField) {
-      const val = content.substring(state.s, e).trim().replace(/---+$/, "").trim();
-      values[state.currField.name] = validateAndParseFieldValue(
-        state.currField,
-        val
-      );
+    let prefixLen = prefix.length;
+    if (e - 1 >= 0 && content[e - 1] === "\n") {
+      e -= 1;
+      prefixLen += 1;
+    }
+    if (xstate.currField) {
+      const val = content.substring(xstate.s, e).trim();
+      const parsedValue = validateAndParseFieldValue(xstate.currField, val);
+      if (parsedValue !== void 0) {
+        values[xstate.currField.name] = parsedValue;
+      }
+    }
+    checkMissingRequiredFields(xstate, values, index);
+    xstate.s = e + prefixLen;
+    xstate.currField = field;
+    if (!xstate.extractedFields.includes(field)) {
+      xstate.extractedFields.push(field);
     }
-    state.s = e + prefix.length;
-    state.currField = field;
   }
 };
-var streamingExtractFinalValue = (values, state, content) => {
-  if (!state.currField) {
-    return;
+var streamingExtractFinalValue = (sig, values, xstate, content) => {
+  if (xstate.currField) {
+    const val = content.substring(xstate.s).trim();
+    const parsedValue = validateAndParseFieldValue(xstate.currField, val);
+    if (parsedValue !== void 0) {
+      values[xstate.currField.name] = parsedValue;
+    }
   }
-  const val = content.substring(state.s).trim().replace(/---+$/, "").trim();
-  values[state.currField.name] = validateAndParseFieldValue(
-    state.currField,
-    val
-  );
+  const fields = sig.getOutputFields();
+  checkMissingRequiredFields(xstate, values, fields.length - 1);
 };
 var convertValueToType = (field, val) => {
   switch (field.type?.name) {
@@ -4761,6 +4938,9 @@ var convertValueToType = (field, val) => {
       return v;
     }
     case "boolean": {
+      if (typeof val === "boolean") {
+        return val;
+      }
       const v = val.toLowerCase();
       if (v === "true") {
         return true;
@@ -4775,37 +4955,73 @@ var convertValueToType = (field, val) => {
     case "datetime":
       return parseLLMFriendlyDateTime(field, val);
     case "class":
-      if (field.type.classes && !field.type.classes.includes(val)) {
+      const className = val;
+      if (field.type.classes && !field.type.classes.includes(className)) {
         throw new Error(
           `Invalid class '${val}', expected one of the following: ${field.type.classes.join(", ")}`
         );
       }
-      return val;
+      return className;
     default:
       return val;
   }
 };
-var expectedTypeError = (field, err, value = "") => {
-  const exp = field.type?.isArray ? `array of ${field.type.name}` : field.type?.name;
-  const message = `Error '${err.message}', expected '${exp}' got '${value}'`;
-  return new ValidationError({ message, field, value });
-};
+function* streamValues(sig, values, xstate, content) {
+  if (!xstate.currField) {
+    return;
+  }
+  const fieldName = xstate.currField.name;
+  if (!xstate.streamedIndex) {
+    xstate.streamedIndex = { [fieldName]: 0 };
+  }
+  if (!xstate.currField.type || !xstate.currField.type.isArray && xstate.currField.type.name === "string") {
+    const pos = xstate.streamedIndex[fieldName] ?? 0;
+    const s = xstate.s + pos;
+    const v = content.substring(s);
+    yield { [fieldName]: pos === 0 ? v.trimStart() : v };
+    xstate.streamedIndex[fieldName] = pos + v.length;
+    return;
+  }
+  for (const key of Object.keys(values)) {
+    const value = values[key];
+    if (Array.isArray(value)) {
+      const s = xstate.streamedIndex[key] ?? 0;
+      const v = value.slice(s);
+      if (v) {
+        yield { [key]: v };
+        xstate.streamedIndex[key] = s + 1;
+      }
+      continue;
+    }
+    if (!xstate.streamedIndex[key]) {
+      yield { [key]: value };
+      xstate.streamedIndex[key] = 1;
+    }
+  }
+}
 function validateAndParseFieldValue(field, fieldValue) {
-  const fv = fieldValue?.toLocaleLowerCase();
-  if (!fieldValue || !fv || fv === "" || fv === "null" || fv === "undefined") {
+  if (!fieldValue || fieldValue === "" || fieldValue === "null" || fieldValue === "NULL" || fieldValue === "undefined") {
     if (field.isOptional) {
       return;
     }
-    throw expectedTypeError(field, new Error("Empty value"), fieldValue);
+    throw new ValidationError({
+      message: "Required field is missing",
+      fields: [field],
+      value: fieldValue
+    });
   }
-  let value = fieldValue;
+  let value;
   if (field.type?.name === "json") {
     try {
       const text = extractBlock(fieldValue);
       value = import_json5.default.parse(text);
       return value;
     } catch (e) {
-      throw expectedTypeError(field, e, fieldValue);
+      throw new ValidationError({
+        message: "Invalid JSON: " + e.message,
+        fields: [field],
+        value: fieldValue
+      });
     }
   }
   if (field.type?.isArray) {
@@ -4819,23 +5035,33 @@ function validateAndParseFieldValue(field, fieldValue) {
         throw new Error("Expected an array");
       }
     } catch (e) {
-      throw expectedTypeError(field, e, fieldValue);
+      throw new ValidationError({
+        message: "Invalid Array: " + e.message,
+        fields: [field],
+        value: fieldValue
+      });
     }
   }
-  if (Array.isArray(value)) {
-    for (const [index, item] of value.entries()) {
-      try {
-        value[index] = convertValueToType(field, item);
-      } catch (e) {
-        throw expectedTypeError(field, e, item);
+  try {
+    if (Array.isArray(value)) {
+      for (const [index, item] of value.entries()) {
+        if (item !== void 0) {
+          const v = typeof item === "string" ? item.trim() : item;
+          value[index] = convertValueToType(field, v);
+        }
       }
-    }
-  } else {
-    try {
+    } else {
       value = convertValueToType(field, fieldValue);
-    } catch (e) {
-      throw expectedTypeError(field, e, fieldValue);
     }
+  } catch (e) {
+    throw new ValidationError({
+      message: e.message,
+      fields: [field],
+      value: fieldValue
+    });
+  }
+  if (typeof value === "string" && value === "") {
+    return void 0;
   }
   return value;
 }
@@ -5004,7 +5230,6 @@ function parseFunctionCalls(ai, functionCalls, values, model) {
 }
 
 // dsp/generate.ts
-var colorLog4 = new ColorLog();
 var AxGen = class extends AxProgramWithSignature {
   promptTemplate;
   asserts;
@@ -5069,7 +5294,7 @@ var AxGen = class extends AxProgramWithSignature {
     );
     return res;
   }
-  async forwardCore({
+  async *forwardCore({
     ai,
     mem,
     options
@@ -5085,7 +5310,18 @@ var AxGen = class extends AxProgramWithSignature {
       options
     });
     if (res instanceof import_web5.ReadableStream) {
-      return await this.processSteamingResponse({
+      yield* this.processStreamingResponse({
+        ai,
+        model,
+        res,
+        usageInfo,
+        mem,
+        traceId,
+        sessionId,
+        functions
+      });
+    } else {
+      yield await this.processResponse({
         ai,
         model,
         res,
@@ -5096,18 +5332,8 @@ var AxGen = class extends AxProgramWithSignature {
         functions
       });
     }
-    return await this.processResponse({
-      ai,
-      model,
-      res,
-      usageInfo,
-      mem,
-      traceId,
-      sessionId,
-      functions
-    });
   }
-  async processSteamingResponse({
+  async *processStreamingResponse({
     ai,
     model,
     res,
@@ -5119,36 +5345,50 @@ var AxGen = class extends AxProgramWithSignature {
   }) {
     const functionCalls = [];
     const values = {};
-    const xstate = { s: -1 };
+    const xstate = {
+      extractedFields: [],
+      s: -1
+    };
     let content = "";
     for await (const v of res) {
-      for (const result of v.results ?? []) {
-        if (v.modelUsage) {
-          this.usage.push({ ...usageInfo, ...v.modelUsage });
-        }
-        if (result.content) {
-          content += result.content;
-          mem.updateResult({ name: result.name, content }, sessionId);
-          assertStreamingAssertions(
-            this.streamingAsserts,
-            values,
-            xstate,
-            content,
-            false
-          );
-          streamingExtractValues(this.signature, values, xstate, content);
-          assertAssertions(this.asserts, values);
-        }
-        if (result.functionCalls) {
-          mergeFunctionCalls(functionCalls, result.functionCalls);
-          mem.updateResult(
-            { name: result.name, content, functionCalls },
-            sessionId
-          );
-        }
-        if (result.finishReason === "length") {
-          throw new Error("Max tokens reached before completion");
+      const result = v.results[0];
+      if (!result) {
+        continue;
+      }
+      if (v.modelUsage) {
+        this.usage.push({ ...usageInfo, ...v.modelUsage });
+      }
+      if (result.content) {
+        content += result.content;
+        mem.updateResult({ name: result.name, content }, sessionId);
+        const skip = streamingExtractValues(
+          this.signature,
+          values,
+          xstate,
+          content
+        );
+        if (skip) {
+          continue;
         }
+        assertStreamingAssertions(
+          this.streamingAsserts,
+          values,
+          xstate,
+          content,
+          false
+        );
+        assertAssertions(this.asserts, values);
+        yield* streamValues(this.signature, values, xstate, content);
+      }
+      if (result.functionCalls) {
+        mergeFunctionCalls(functionCalls, result.functionCalls);
+        mem.updateResult(
+          { name: result.name, content, functionCalls },
+          sessionId
+        );
+      }
+      if (result.finishReason === "length") {
+        throw new Error("Max tokens reached before completion");
       }
     }
     const funcs = parseFunctionCalls(ai, functionCalls, values, model);
@@ -5166,7 +5406,6 @@ var AxGen = class extends AxProgramWithSignature {
       );
       this.functionsExecuted = /* @__PURE__ */ new Set([...this.functionsExecuted, ...fx]);
     }
-    streamingExtractFinalValue(values, xstate, content);
     assertStreamingAssertions(
       this.streamingAsserts,
       values,
@@ -5174,8 +5413,9 @@ var AxGen = class extends AxProgramWithSignature {
       content,
       true
     );
+    streamingExtractFinalValue(this.signature, values, xstate, content);
     assertAssertions(this.asserts, values);
-    return { ...values };
+    yield* streamValues(this.signature, values, xstate, content);
   }
   async processResponse({
     ai,
@@ -5187,45 +5427,47 @@ var AxGen = class extends AxProgramWithSignature {
     functions
   }) {
     const values = {};
-    for (const result of res.results ?? []) {
-      if (res.modelUsage) {
-        this.usage.push({ ...usageInfo, ...res.modelUsage });
-      }
-      mem.addResult(result, sessionId);
-      if (result.content) {
-        extractValues(this.signature, values, result.content);
-        assertAssertions(this.asserts, values);
-      }
-      if (result.functionCalls) {
-        const funcs = parseFunctionCalls(ai, result.functionCalls, values);
-        if (funcs) {
-          if (!functions) {
-            throw new Error("Functions are not defined");
-          }
-          const fx = await processFunctions(
-            ai,
-            functions,
-            funcs,
-            mem,
-            sessionId,
-            traceId
-          );
-          this.functionsExecuted = /* @__PURE__ */ new Set([...this.functionsExecuted, ...fx]);
+    const result = res.results[0];
+    if (!result) {
+      throw new Error("No result found");
+    }
+    if (res.modelUsage) {
+      this.usage.push({ ...usageInfo, ...res.modelUsage });
+    }
+    mem.addResult(result, sessionId);
+    if (result.content) {
+      extractValues(this.signature, values, result.content);
+      assertAssertions(this.asserts, values);
+    }
+    if (result.functionCalls) {
+      const funcs = parseFunctionCalls(ai, result.functionCalls, values);
+      if (funcs) {
+        if (!functions) {
+          throw new Error("Functions are not defined");
         }
+        const fx = await processFunctions(
+          ai,
+          functions,
+          funcs,
+          mem,
+          sessionId,
+          traceId
+        );
+        this.functionsExecuted = /* @__PURE__ */ new Set([...this.functionsExecuted, ...fx]);
       }
-      if (result.finishReason === "length") {
-        throw new Error("Max tokens reached before completion");
-      }
+    }
+    if (result.finishReason === "length") {
+      throw new Error("Max tokens reached before completion");
     }
     return { ...values };
   }
-  async _forward(ai, values, options, span) {
+  async *_forward2(ai, values, options, span) {
     const stopFunction = (options?.stopFunction ?? this.options?.stopFunction)?.toLowerCase();
-    const maxRetries = options?.maxRetries ?? this.options?.maxRetries ?? 3;
-    const maxSteps = options?.maxSteps ?? this.options?.maxSteps ?? 10;
-    const mem = options?.mem ?? this.options?.mem ?? new AxMemory();
+    const maxRetries = options.maxRetries ?? this.options?.maxRetries ?? 10;
+    const maxSteps = options.maxSteps ?? this.options?.maxSteps ?? 10;
+    const mem = options.mem ?? this.options?.mem ?? new AxMemory();
     let err;
-    if (options?.functions && options?.functions.length > 0) {
+    if (options?.functions && options.functions.length > 0) {
       const promptTemplate = this.options?.promptTemplate ?? AxPromptTemplate;
       this.promptTemplate = new promptTemplate(
         this.signature,
@@ -5240,25 +5482,21 @@ var AxGen = class extends AxProgramWithSignature {
     multiStepLoop: for (let n = 0; n < maxSteps; n++) {
       for (let errCount = 0; errCount < maxRetries; errCount++) {
         try {
-          const output = await this.forwardCore({
-            options,
-            ai,
-            mem
-          });
-          const lastMemItem = mem.getLast(options?.sessionId);
-          if (lastMemItem) {
-            const stopFunctionExecuted = stopFunction && this.functionsExecuted.has(stopFunction);
-            if (lastMemItem.role === "function") {
-              if (!stopFunction || !stopFunctionExecuted) {
-                continue multiStepLoop;
-              }
-            }
-            if (!stopFunctionExecuted) {
-              assertRequiredFields(this.signature, output);
+          const generator = this.forwardCore({ options, ai, mem });
+          for await (const delta of generator) {
+            if (delta !== void 0) {
+              yield { version: errCount, delta };
             }
           }
-          this.trace = { ...values, ...output };
-          return output;
+          const lastMemItem = mem.getLast(options?.sessionId);
+          const shouldContinue = this.shouldContinueSteps(
+            lastMemItem,
+            stopFunction
+          );
+          if (shouldContinue) {
+            continue multiStepLoop;
+          }
+          return;
         } catch (e) {
           let errorFields;
           span?.recordException(e);
@@ -5273,23 +5511,13 @@ var AxGen = class extends AxProgramWithSignature {
             throw e;
           }
           if (errorFields) {
-            mem.add(
-              {
-                role: "user",
-                content: this.promptTemplate.renderExtraFields(errorFields)
-              },
-              options?.sessionId
+            handleValidationError(
+              mem,
+              errorFields,
+              ai,
+              this.promptTemplate,
+              options.sessionId
             );
-            mem.addTag("error");
-            if (ai.getOptions().debug) {
-              process.stdout.write(
-                colorLog4.red(
-                  `Error Correction:
-${JSON.stringify(errorFields, null, 2)}
-`
-                )
-              );
-            }
           }
         }
       }
@@ -5300,35 +5528,73 @@ ${JSON.stringify(errorFields, null, 2)}
     }
     throw new Error(`Max steps reached: ${maxSteps}`);
   }
-  async forward(ai, values, options) {
+  shouldContinueSteps(lastMemItem, stopFunction) {
+    const stopFunctionExecuted = stopFunction && this.functionsExecuted.has(stopFunction);
+    if (lastMemItem?.role === "function" && stopFunction && stopFunctionExecuted) {
+      return false;
+    }
+    if (lastMemItem?.role === "function") {
+      return true;
+    }
+    return false;
+  }
+  async *_forward1(ai, values, options) {
     const tracer = this.options?.tracer ?? options?.tracer;
     let functions = this.functions;
     if (options?.functions) {
       functions = parseFunctions(options.functions, this.functions);
     }
     if (!tracer) {
-      return await this._forward(ai, values, {
+      yield* this._forward2(ai, values, {
         ...options,
         functions
       });
+      return;
     }
     const funcNames = functions?.map((f) => f.name).join(",");
     const attributes = {
       ["generate.signature"]: this.signature.toString(),
       ["generate.functions"]: funcNames ?? ""
     };
-    return await tracer.startActiveSpan(
-      "Generate",
-      {
-        kind: import_api22.SpanKind.SERVER,
-        attributes
-      },
-      async (span) => {
-        const res = this._forward(ai, values, options, span);
-        span.end();
-        return res;
+    const span = tracer.startSpan("Generate", {
+      kind: import_api22.SpanKind.SERVER,
+      attributes
+    });
+    try {
+      yield* this._forward2(
+        ai,
+        values,
+        {
+          ...options,
+          functions
+        },
+        span
+      );
+    } finally {
+      span.end();
+    }
+  }
+  async forward(ai, values, options) {
+    const generator = this._forward1(ai, values, {
+      ...options
+    });
+    let buffer = {};
+    let currentVersion = 0;
+    for await (const item of generator) {
+      if (item.version !== currentVersion) {
+        buffer = {};
       }
-    );
+      currentVersion = item.version;
+      buffer = mergeDeltas(buffer, item.delta);
+    }
+    this.trace = { ...values, ...buffer };
+    return buffer;
+  }
+  async *streamingForward(ai, values, options) {
+    yield* this._forward1(ai, values, {
+      ...options,
+      stream: true
+    });
   }
 };
 
@@ -5418,7 +5684,7 @@ var AxAgent = class {
       func: wrappedFunc
     };
   }
-  async forward(ai, values, options) {
+  init(ai, options) {
     const _ai = this.ai ?? ai;
     const funcs = [
       ...options?.functions ?? [],
@@ -5429,26 +5695,15 @@ var AxAgent = class {
       const opt2 = { ...options, functions: funcs };
       this.program = new AxGen(this.signature, opt2);
     }
-    if (!options?.tracer) {
-      return await this.program.forward(_ai, values, opt);
-    }
-    const attributes = {
-      ["agent.name"]: this.name,
-      ["agent.description"]: this.description,
-      ["agent.subAgents"]: this.subAgentList ?? "none"
-    };
-    return await options?.tracer.startActiveSpan(
-      "Agent",
-      {
-        kind: import_api23.SpanKind.SERVER,
-        attributes
-      },
-      async (span) => {
-        const res = await this.program.forward(_ai, values, opt);
-        span.end();
-        return res;
-      }
-    );
+    return { _ai, opt };
+  }
+  async forward(ai, values, options) {
+    const { _ai, opt } = this.init(ai, options);
+    return await this.program.forward(_ai, values, opt);
+  }
+  async *streamingForward(ai, values, options) {
+    const { _ai, opt } = this.init(ai, options);
+    return yield* this.program.streamingForward(_ai, values, opt);
   }
 };
 function toCamelCase(inputString) {
@@ -5719,7 +5974,7 @@ var randomSample = (array, n) => {
 };
 
 // db/base.ts
-var import_api24 = require("@opentelemetry/api");
+var import_api23 = require("@opentelemetry/api");
 var AxDBBase = class {
   name;
   fetch;
@@ -5746,7 +6001,7 @@ var AxDBBase = class {
     return await this.tracer?.startActiveSpan(
       "DB Upsert Request",
       {
-        kind: import_api24.SpanKind.SERVER,
+        kind: import_api23.SpanKind.SERVER,
         attributes: {
           [axSpanAttributes.DB_SYSTEM]: this.name,
           [axSpanAttributes.DB_OPERATION_NAME]: "upsert",
@@ -5756,9 +6011,11 @@ var AxDBBase = class {
         }
       },
       async (span) => {
-        const res = await this._upsert(req, update, { span });
-        span.end();
-        return res;
+        try {
+          return await this._upsert(req, update, { span });
+        } finally {
+          span.end();
+        }
       }
     );
   }
@@ -5778,7 +6035,7 @@ var AxDBBase = class {
     return await this.tracer?.startActiveSpan(
       "DB Batch Upsert Request",
       {
-        kind: import_api24.SpanKind.SERVER,
+        kind: import_api23.SpanKind.SERVER,
         attributes: {
           [axSpanAttributes.DB_SYSTEM]: this.name,
           [axSpanAttributes.DB_OPERATION_NAME]: "upsert",
@@ -5788,9 +6045,11 @@ var AxDBBase = class {
         }
       },
       async (span) => {
-        const res = await this._batchUpsert(req, update, { span });
-        span.end();
-        return res;
+        try {
+          return await this._batchUpsert(req, update, { span });
+        } finally {
+          span.end();
+        }
       }
     );
   }
@@ -5804,7 +6063,7 @@ var AxDBBase = class {
     return await this.tracer?.startActiveSpan(
       "DB Query Request",
       {
-        kind: import_api24.SpanKind.SERVER,
+        kind: import_api23.SpanKind.SERVER,
         attributes: {
           [axSpanAttributes.DB_SYSTEM]: this.name,
           [axSpanAttributes.DB_OPERATION_NAME]: "upsert",
@@ -5814,9 +6073,11 @@ var AxDBBase = class {
         }
       },
       async (span) => {
-        const res = await this._query(req, { span });
-        span.end();
-        return res;
+        try {
+          return await this._query(req, { span });
+        } finally {
+          span.end();
+        }
       }
     );
   }
@@ -7083,6 +7344,122 @@ var AxEmbeddingAdapter = class {
   }
 };
 
+// ai/mock/api.ts
+var AxMockAIService = class {
+  constructor(config = {}) {
+    this.config = config;
+  }
+  options = {};
+  metrics = {
+    latency: {
+      chat: { mean: 0, p95: 0, p99: 0, samples: [] },
+      embed: { mean: 0, p95: 0, p99: 0, samples: [] }
+    },
+    errors: {
+      chat: { count: 0, rate: 0, total: 0 },
+      embed: { count: 0, rate: 0, total: 0 }
+    }
+  };
+  getName() {
+    return this.config.name ?? "mock-ai-service";
+  }
+  getModelInfo() {
+    return {
+      name: "mock-model",
+      provider: "mock-provider",
+      promptTokenCostPer1M: 100,
+      completionTokenCostPer1M: 100,
+      ...this.config.modelInfo
+    };
+  }
+  getEmbedModelInfo() {
+    return this.config.embedModelInfo;
+  }
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  getFeatures(_model) {
+    return {
+      functions: this.config.features?.functions ?? false,
+      streaming: this.config.features?.streaming ?? false
+    };
+  }
+  getModelMap() {
+    return this.config.modelMap;
+  }
+  getMetrics() {
+    return this.metrics;
+  }
+  async chat(req, _options) {
+    if (this.config.latencyMs) {
+      await new Promise((resolve) => setTimeout(resolve, this.config.latencyMs));
+    }
+    if (this.config.shouldError) {
+      throw new Error(this.config.errorMessage ?? "Mock chat error");
+    }
+    this.updateMetrics("chat");
+    if (typeof this.config.chatResponse === "function") {
+      return this.config.chatResponse(req);
+    }
+    return this.config.chatResponse ?? {
+      results: [
+        {
+          content: "Mock response",
+          finishReason: "stop"
+        }
+      ],
+      modelUsage: {
+        promptTokens: 10,
+        completionTokens: 5,
+        totalTokens: 15
+      }
+    };
+  }
+  async embed(req, _options) {
+    if (this.config.latencyMs) {
+      await new Promise((resolve) => setTimeout(resolve, this.config.latencyMs));
+    }
+    if (this.config.shouldError) {
+      throw new Error(this.config.errorMessage ?? "Mock embed error");
+    }
+    this.updateMetrics("embed");
+    if (typeof this.config.embedResponse === "function") {
+      return this.config.embedResponse(req);
+    }
+    return this.config.embedResponse ?? {
+      embeddings: [[0.1, 0.2, 0.3]],
+      modelUsage: {
+        promptTokens: 5,
+        completionTokens: 0,
+        totalTokens: 5
+      }
+    };
+  }
+  setOptions(options) {
+    this.options = options;
+  }
+  getOptions() {
+    return this.options;
+  }
+  updateMetrics(type) {
+    const latency = this.config.latencyMs ?? 0;
+    this.metrics.latency[type].samples.push(latency);
+    const samples = this.metrics.latency[type].samples;
+    this.metrics.latency[type].mean = samples.reduce((a, b) => a + b, 0) / samples.length;
+    if (samples.length > 0) {
+      const sortedSamples = [...samples].sort((a, b) => a - b);
+      const p95Index = Math.max(0, Math.floor(sortedSamples.length * 0.95) - 1);
+      this.metrics.latency[type].p95 = sortedSamples[p95Index] ?? latency;
+      const p99Index = Math.max(0, Math.floor(sortedSamples.length * 0.99) - 1);
+      this.metrics.latency[type].p99 = sortedSamples[p99Index] ?? latency;
+    }
+    if (this.config.shouldError) {
+      this.metrics.errors[type].count++;
+      this.metrics.errors[type].total++;
+      const totalRequests = this.metrics.latency[type].samples.length;
+      this.metrics.errors[type].rate = totalRequests > 0 ? this.metrics.errors[type].count / totalRequests : 0;
+    }
+  }
+};
+
 // prompts/rag.ts
 var AxRAG = class extends AxChainOfThought {
   genQuery;
@@ -7121,6 +7498,7 @@ var AxRAG = class extends AxChainOfThought {
   AxAI,
   AxAIAnthropic,
   AxAIAnthropicModel,
+  AxAIAnthropicVertexModel,
   AxAIAzureOpenAI,
   AxAICohere,
   AxAICohereEmbedModel,
@@ -7172,6 +7550,7 @@ var AxRAG = class extends AxChainOfThought {
   AxJSInterpreterPermission,
   AxLLMRequestTypeValues,
   AxMemory,
+  AxMockAIService,
   AxProgram,
   AxProgramWithSignature,
   AxPromptTemplate,