@ax-llm/ax 12.0.8 → 12.0.9

package/index.js

@@ -320,6 +320,17 @@ var AxAIServiceAuthenticationError = class extends AxAIServiceError {
     this.name = this.constructor.name;
   }
 };
+async function safeReadResponseBody(response) {
+  try {
+    if (response.headers.get("content-type")?.includes("application/json")) {
+      return await response.json();
+    }
+    const clonedResponse = response.clone();
+    return await clonedResponse.text();
+  } catch (e) {
+    return `[ReadableStream - read failed: ${e.message}]`;
+  }
+}
 function calculateRetryDelay(attempt, config) {
   const delay = Math.min(
     config.maxDelayMs,
@@ -413,9 +424,15 @@ var apiCall = async (api, json) => {
       });
       clearTimeout(timeoutId);
       if (res.status === 401 || res.status === 403) {
-        throw new AxAIServiceAuthenticationError(apiUrl.href, json, res.body, {
-          metrics
-        });
+        const responseBody = await safeReadResponseBody(res);
+        throw new AxAIServiceAuthenticationError(
+          apiUrl.href,
+          json,
+          responseBody,
+          {
+            metrics
+          }
+        );
       }
       if (res.status >= 400 && shouldRetry(new Error(), res.status, attempt, retryConfig)) {
         const delay = calculateRetryDelay(attempt, retryConfig);
@@ -433,12 +450,13 @@ var apiCall = async (api, json) => {
         continue;
       }
       if (res.status >= 400) {
+        const responseBody = await safeReadResponseBody(res);
         throw new AxAIServiceStatusError(
           res.status,
           res.statusText,
           apiUrl.href,
           json,
-          res.body,
+          responseBody,
           { metrics }
         );
       }
@@ -536,7 +554,7 @@ var apiCall = async (api, json) => {
                     error,
                     apiUrl.href,
                     json,
-                    res.body,
+                    "[ReadableStream - consumed during streaming]",
                     {
                       streamMetrics
                     }
@@ -667,12 +685,12 @@ var ColorLog = class {
   }
 };
 
-// ai/debug.ts
+// dsp/loggers.ts
 var colorLog = new ColorLog();
 var defaultOutput = (message) => {
   process.stdout.write(message);
 };
-var createDefaultLogger = (output = defaultOutput) => {
+var axCreateDefaultLogger = (output = defaultOutput) => {
   return (message, options) => {
     const tags = options?.tags ?? [];
     let formattedMessage = message;
@@ -681,11 +699,18 @@ var createDefaultLogger = (output = defaultOutput) => {
     } else if (tags.includes("success") || tags.includes("responseContent")) {
       formattedMessage = colorLog.greenBright(formattedMessage);
     } else if (tags.includes("functionName")) {
-      formattedMessage = colorLog.whiteBright(formattedMessage);
-    } else if (tags.includes("functionArg") || tags.includes("systemContent") || tags.includes("assistantContent")) {
+      if (tags.includes("firstFunction")) {
+        formattedMessage = `
+${colorLog.whiteBright(formattedMessage)}`;
+      } else {
+        formattedMessage = `${colorLog.whiteBright(formattedMessage)}`;
+      }
+    } else if (tags.includes("systemContent") || tags.includes("assistantContent")) {
       formattedMessage = colorLog.blueBright(formattedMessage);
     } else if (tags.includes("warning") || tags.includes("discovery")) {
       formattedMessage = colorLog.yellow(formattedMessage);
+    } else if (tags.includes("functionArg")) {
+      formattedMessage = "";
     }
     if (tags.includes("responseStart") || tags.includes("systemStart") || tags.includes("userStart")) {
       formattedMessage = `
@@ -700,6 +725,31 @@ ${formattedMessage}
     } else if (tags.includes("error")) {
       formattedMessage = `
 ${formattedMessage}
+`;
+    } else if (tags.includes("functionEnd")) {
+      formattedMessage = `
+`;
+    }
+    output(formattedMessage);
+  };
+};
+var axCreateDefaultTextLogger = (output = defaultOutput) => {
+  return (message, options) => {
+    const tags = options?.tags ?? [];
+    let formattedMessage = message;
+    if (tags.includes("responseStart") || tags.includes("systemStart") || tags.includes("userStart")) {
+      formattedMessage = `
+${formattedMessage}`;
+    } else if (tags.includes("responseEnd") || tags.includes("systemEnd") || tags.includes("userEnd")) {
+      formattedMessage = `${formattedMessage}
+`;
+    } else if (tags.includes("assistantStart")) {
+      formattedMessage = `
+${formattedMessage}
+`;
+    } else if (tags.includes("error")) {
+      formattedMessage = `
+${formattedMessage}
 `;
     } else if (tags.includes("functionEnd")) {
       formattedMessage = `${formattedMessage}
@@ -708,7 +758,137 @@ ${formattedMessage}
     output(formattedMessage);
   };
 };
-var defaultLogger = createDefaultLogger();
+var axCreateOptimizerLogger = (output = (msg) => process.stdout.write(msg)) => {
+  const baseLogger = axCreateDefaultLogger(output);
+  let isFirstPhase = true;
+  return (message, options) => {
+    const tags = options?.tags ?? [];
+    let formattedMessage = message;
+    if (tags.includes("optimizer")) {
+      if (tags.includes("start")) {
+        const trialsMatch = message.match(/with (\d+) trials?/) || message.match(/(\d+) trials?/);
+        const optimizerMatch = message.match(
+          /(MIPROv2|BootstrapFewshot|[A-Z][a-zA-Z]+)/
+        );
+        const optimizerName = optimizerMatch ? optimizerMatch[1] : "Optimizer";
+        if (trialsMatch && trialsMatch[1]) {
+          formattedMessage = `
+\u250C\u2500 ${optimizerName} optimization (${trialsMatch[1]} trials)
+`;
+        } else {
+          formattedMessage = `
+\u250C\u2500 ${optimizerName} optimization
+`;
+        }
+        isFirstPhase = true;
+      } else if (tags.includes("config")) {
+        if (message.includes("examples") && message.includes("training")) {
+          const match = message.match(
+            /(\d+) examples for training and (\d+) for validation/
+          ) || message.match(/(\d+) training.*?(\d+) validation/);
+          if (match && match[1] && match[2]) {
+            formattedMessage = `\u2502  Dataset: ${match[1]} training, ${match[2]} validation
+`;
+          } else {
+            const simpleMatch = message.match(/(\d+) examples/);
+            if (simpleMatch && simpleMatch[1]) {
+              formattedMessage = `\u2502  Dataset: ${simpleMatch[1]} examples
+`;
+            }
+          }
+        } else if (message.includes("teacher")) {
+          formattedMessage = `\u2502  Using teacher model
+`;
+        } else {
+          formattedMessage = `\u2502  ${message}
+`;
+        }
+      } else if (tags.includes("phase")) {
+        if (isFirstPhase) {
+          formattedMessage = `\u251C\u2500 ${message}
+`;
+          isFirstPhase = false;
+        } else {
+          formattedMessage = `\u251C\u2500 ${message}
+`;
+        }
+      } else if (tags.includes("result")) {
+        if (message.includes("Generated") || message.includes("Selected")) {
+          const match = message.match(/(\d+)/);
+          if (match && match[1]) {
+            formattedMessage = `\u2502  \u2713 ${message}
+`;
+          } else {
+            formattedMessage = `\u2502  \u2713 ${message}
+`;
+          }
+        } else if (message.includes("configuration")) {
+          formattedMessage = `\u2502  Applied best configuration
+`;
+        } else {
+          formattedMessage = `\u2502  ${message}
+`;
+        }
+      } else if (tags.includes("progress")) {
+        formattedMessage = `\u2502  ${message}
+`;
+      } else if (tags.includes("complete")) {
+        const scoreMatch = message.match(/(score|performance):\s*([\d.]+)/);
+        if (scoreMatch && scoreMatch[2]) {
+          const score = parseFloat(scoreMatch[2]);
+          const percentage = score <= 1 ? (score * 100).toFixed(1) + "%" : score.toFixed(3);
+          formattedMessage = `\u251C\u2500 Complete! Best: ${percentage}
+`;
+        } else if (message.includes("Bootstrap")) {
+          formattedMessage = `\u251C\u2500 ${message}
+`;
+        } else {
+          formattedMessage = `\u251C\u2500 Optimization complete
+`;
+        }
+      } else if (tags.includes("checkpoint")) {
+        if (message.includes("Resuming")) {
+          formattedMessage = `\u2502  ${message}
+`;
+        } else {
+          const match = message.match(/checkpoint:\s*(.+)/) || message.match(/Saved\s+(.+)/);
+          if (match && match[1]) {
+            formattedMessage = `\u2514\u2500 Saved: ${match[1]}
+`;
+          } else {
+            formattedMessage = `\u2514\u2500 Checkpoint saved
+`;
+          }
+        }
+      }
+    } else if (tags.includes("discovery")) {
+      if (message.includes("Found") && message.includes("examples")) {
+        const match = message.match(/Found (\d+)/);
+        if (match && match[1]) {
+          formattedMessage = `\u2502  Found ${match[1]} examples
+`;
+        }
+      }
+    }
+    if (tags.includes("error")) {
+      formattedMessage = `
+\u2717 ${message}
+`;
+    } else if (tags.includes("warning")) {
+      formattedMessage = `
+\u26A0 ${message}
+`;
+    } else if (tags.includes("success") && !tags.includes("optimizer")) {
+      formattedMessage = `\u2713 ${message}
+`;
+    }
+    baseLogger(formattedMessage, options);
+  };
+};
+var axDefaultOptimizerLogger = axCreateOptimizerLogger();
+
+// ai/debug.ts
+var defaultLogger = axCreateDefaultLogger();
 var formatChatMessage = (msg, hideContent, hideSystemPrompt) => {
   switch (msg.role) {
     case "system":
@@ -785,9 +965,14 @@ var logResponseResult = (r, logger = defaultLogger) => {
   if (r.functionCalls && r.functionCalls.length > 0) {
     for (const [i, f2] of r.functionCalls.entries()) {
       if (f2.function.name) {
-        logger(`[${i + 1}] ${f2.function.name}`, {
-          tags: ["functionName"]
-        });
+        const tags = ["functionName"];
+        if (i === 0) {
+          tags.push("firstFunction");
+        }
+        if (r.functionCalls.length > 1) {
+          tags.push("multipleFunctions");
+        }
+        logger(`[${i + 1}] ${f2.function.name}`, { tags });
       }
       if (f2.function.params) {
         const params = typeof f2.function.params === "string" ? f2.function.params : JSON.stringify(f2.function.params, null, 2);
@@ -1458,6 +1643,16 @@ function validateAxMessageArray(values) {
 function validateChatPrompt(chatPrompt) {
   for (let i = 0; i < chatPrompt.length; i++) {
     const message = chatPrompt[i];
+    if (message && "functionCalls" in message && Array.isArray(message.functionCalls) && message.functionCalls.length === 0) {
+      throw new Error(
+        `Chat prompt validation failed: Message at index ${i} has empty functionCalls`
+      );
+    }
+    if (message && "content" in message && Array.isArray(message.content) && message.content.length === 0) {
+      throw new Error(
+        `Chat prompt validation failed: Message at index ${i} has empty content`
+      );
+    }
     if (message && "content" in message && typeof message.content === "string" && message.content.trim() === "") {
       throw new Error(
         `Chat prompt validation failed: Message at index ${i} has empty content`
@@ -5785,9 +5980,11 @@ var updateProgressBar = (current, total, success, elapsedTime, msg, progressBarW
   const emptyBarLength = progressBarWidth - filledBarLength;
   const filledBar = colorLog3.blueBright("\u2588".repeat(filledBarLength));
   const emptyBar = " ".repeat(emptyBarLength);
-  const itemsPerSecond = elapsedTime > 0 ? (current / elapsedTime).toFixed(2) : "0.00";
+  const successRate = total > 0 ? (success / total * 100).toFixed(1) : "0.0";
+  const friendlyMsg = msg.includes("Running MIPROv2 optimization") ? "Testing prompt variations" : msg.includes("Tuning Prompt") ? "Generating training examples" : msg;
   process.stdout.write(
-    `\r${msg}: ${current} / ${total} (${colorLog3.yellow(percentage)}%): 100%|${filledBar}${emptyBar}| Success: ${success}/${total} [${colorLog3.red(elapsedTime.toFixed(2))}, ${itemsPerSecond}it/s]`
+    `\u2502  ${friendlyMsg}: ${current}/${total} (${colorLog3.yellow(percentage)}%) |${filledBar}${emptyBar}| Success rate: ${colorLog3.greenBright(successRate)}%
+`
   );
 };
 var validateValue = (field, value) => {
@@ -5994,19 +6191,15 @@ function matchesContent(content, prefix, startIndex = 0, prefixCache = globalPre
   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++) {
+  let longestPartialMatch = -1;
+  for (let i = prefixes.length - 1; i >= 0; i--) {
     const partialPrefix = prefixes[i];
-    if (partialPrefix === "\n" || partialPrefix === ":") {
-      continue;
-    }
-    if (partialPrefix && contentEnd.endsWith(partialPrefix)) {
-      return -2;
+    if (content.endsWith(partialPrefix)) {
+      longestPartialMatch = i;
+      break;
     }
   }
-  return -1;
+  return longestPartialMatch >= 0 ? -2 : -1;
 }
 var formatTime = (ms) => {
   const seconds = Math.floor(ms / 1e3);
@@ -6029,11 +6222,10 @@ var updateDetailedProgress = (roundIndex, current, total, elapsedTime, example,
   process.stdout.write("\r\x1B[K");
   const percentage = (current / total * 100).toFixed(1);
   const formattedTime = formatTime(elapsedTime);
-  const itemsPerSecond = elapsedTime > 0 ? (current / elapsedTime * 1e3).toFixed(2) : "0.00";
   const eta = calculateETA(current, total, elapsedTime);
-  let output = `Round ${roundIndex + 1}/${configInfo.maxRounds}: ${current}/${total} (${percentage}%) [${formattedTime}, ${itemsPerSecond} it/s, ETA: ${eta}]`;
+  let output = `Training round ${roundIndex + 1}/${configInfo.maxRounds}: ${current}/${total} (${percentage}%) [${formattedTime}, ETA: ${eta}]`;
   const successRate = stats.totalCalls > 0 ? stats.successfulDemos / stats.totalCalls * 100 : 0;
-  output += ` | Success: ${stats.successfulDemos}/${stats.totalCalls} (${successRate.toFixed(1)}%)`;
+  output += ` | Success rate: ${successRate.toFixed(1)}% (${stats.successfulDemos}/${stats.totalCalls})`;
   if (configInfo.verboseMode || configInfo.debugMode) {
     if (configInfo.costMonitoring) {
       output += `
@@ -6159,7 +6351,7 @@ ${outputFields}`);
       content: systemContent
     };
     if (Array.isArray(values)) {
-      let userMessages = [];
+      let messages = [];
       const history = values;
       for (const [index, message] of history.entries()) {
         let content;
@@ -6179,7 +6371,7 @@ ${outputFields}`);
           );
         }
         if (message.role === "user") {
-          userMessages.push({ role: "user", content });
+          messages.push({ role: "user", content });
           continue;
         }
         if (message.role !== "assistant") {
@@ -6190,9 +6382,9 @@ ${outputFields}`);
             "Assistant message cannot contain non-text content like images, files,etc"
           );
         }
-        userMessages.push({ role: "assistant", content });
+        messages.push({ role: "assistant", content });
       }
-      return [systemPrompt, ...userMessages];
+      return [systemPrompt, ...messages];
     }
     const userContent = this.renderSingleValueUserContent(
       values,
@@ -6645,9 +6837,9 @@ var formatDateWithTimezone = (date) => {
 };
 
 // dsp/extract.ts
-var extractValues = (sig, values, content) => {
+var extractValues = (sig, values, content, strictMode = false) => {
   const xstate = { extractedFields: [], streamedIndex: {}, s: -1 };
-  streamingExtractValues(sig, values, xstate, content);
+  streamingExtractValues(sig, values, xstate, content, strictMode);
   streamingExtractFinalValue(sig, values, xstate, content);
   for (const field of sig.getOutputFields()) {
     if (field.isInternal) {
@@ -6655,10 +6847,9 @@ var extractValues = (sig, values, content) => {
     }
   }
 };
-var checkMissingRequiredFields = (xstate, values, currentIndex) => {
+var checkMissingRequiredFields = (xstate, values, outputFields) => {
   const missingFields = [];
-  for (let i = 0; i < currentIndex; i++) {
-    const field = xstate.extractedFields[i];
+  for (const field of outputFields) {
     if (field && !field.isOptional && values[field.name] === void 0) {
       missingFields.push(field);
     }
@@ -6670,23 +6861,34 @@ var checkMissingRequiredFields = (xstate, values, currentIndex) => {
     });
   }
 };
-var streamingExtractValues = (sig, values, xstate, content, streamingValidation = false) => {
+var streamingExtractValues = (sig, values, xstate, content, strictMode = false) => {
   const fields = sig.getOutputFields();
+  let expectedField;
   for (const [index, field] of fields.entries()) {
+    if (index === xstate.currFieldIndex) {
+      continue;
+    }
     if (field.name in values) {
       continue;
     }
     const isFirst = xstate.extractedFields.length === 0;
     const prefix = (isFirst ? "" : "\n") + field.title + ":";
     let e = matchesContent(content, prefix, xstate.s);
+    let prefixLen = prefix.length;
     switch (e) {
       case -1:
-        if (streamingValidation && values.length == 0 && !field.isOptional) {
+        if (!strictMode && fields.length === 1 && xstate.currField === void 0) {
+          prefixLen = 0;
+          e = 0;
+          break;
+        }
+        if (xstate.currField === void 0 && !field.isOptional) {
           throw new ValidationError({
-            message: "Required field not found",
+            message: "Expected (Required) field not found",
             fields: [field]
           });
         }
+        expectedField = field.isOptional ? void 0 : field;
         continue;
       // Field is not found, continue to the next field
       case -2:
@@ -6699,7 +6901,12 @@ var streamingExtractValues = (sig, values, xstate, content, streamingValidation
         xstate.inBlock = true;
         return true;
     }
-    let prefixLen = prefix.length;
+    if (expectedField && expectedField.name !== field.name) {
+      throw new ValidationError({
+        message: "Expected (Required) field not found",
+        fields: [expectedField]
+      });
+    }
     if (xstate.currField) {
       const val = content.substring(xstate.s, e).trim();
       const parsedValue = validateAndParseFieldValue(xstate.currField, val);
@@ -6712,7 +6919,6 @@ var streamingExtractValues = (sig, values, xstate, content, streamingValidation
         xstate.prevFields = [{ field: xstate.currField, s: xstate.s, e }];
       }
     }
-    checkMissingRequiredFields(xstate, values, index);
     xstate.s = e + prefixLen;
     xstate.currField = field;
     xstate.currFieldIndex = index;
@@ -6732,8 +6938,7 @@ var streamingExtractFinalValue = (sig, values, xstate, content) => {
       values[xstate.currField.name] = parsedValue;
     }
   }
-  const sigFields = sig.getOutputFields();
-  checkMissingRequiredFields(xstate, values, sigFields.length);
+  checkMissingRequiredFields(xstate, values, sig.getOutputFields());
 };
 var convertValueToType = (field, val, required = false) => {
   switch (field.type?.name) {
@@ -8936,7 +9141,7 @@ var AxGen = class extends AxProgramWithSignature {
     traceContext
   }) {
     const { sessionId, traceId, functions: _functions } = options ?? {};
-    const fastFail = options?.fastFail ?? this.options?.fastFail;
+    const strictMode = options?.strictMode ?? false;
     const model = options.model;
     const functions = _functions?.map((f2) => "toFunction" in f2 ? f2.toFunction() : f2)?.flat();
     const res = await this.forwardSendRequest({
@@ -8955,7 +9160,7 @@ var AxGen = class extends AxProgramWithSignature {
         traceId,
         sessionId,
         functions,
-        fastFail,
+        strictMode,
         span
       });
       this.getLogger(ai, options)?.("", { tags: ["responseEnd"] });
@@ -8968,7 +9173,8 @@ var AxGen = class extends AxProgramWithSignature {
         traceId,
         sessionId,
         functions,
-        span
+        span,
+        strictMode
       });
     }
   }
@@ -8980,10 +9186,9 @@ var AxGen = class extends AxProgramWithSignature {
     sessionId,
     traceId,
     functions,
-    fastFail,
+    strictMode,
     span
   }) {
-    const streamingValidation = fastFail ?? ai.getFeatures(model).functionCot !== true;
     const functionCalls = [];
     this.values = {};
     const xstate = {
@@ -9034,7 +9239,7 @@ var AxGen = class extends AxProgramWithSignature {
           this.values,
           xstate,
           content,
-          streamingValidation
+          strictMode
         );
         if (skip) {
           continue;
@@ -9135,7 +9340,8 @@ Content: ${content}`
     sessionId,
     traceId,
     functions,
-    span
+    span,
+    strictMode
   }) {
     this.values = {};
     let results = res.results ?? [];
@@ -9169,7 +9375,7 @@ Content: ${content}`
         if (result.thought && result.thought.length > 0) {
           this.values[this.thoughtFieldName] = result.thought;
         }
-        extractValues(this.signature, this.values, result.content);
+        extractValues(this.signature, this.values, result.content, strictMode);
         await assertAssertions(this.asserts, this.values);
         if (this.fieldProcessors.length) {
           await processFieldProcessors(
@@ -9341,8 +9547,7 @@ Content: ${result.content}`
       ...options?.thinkingTokenBudget ? { thinking_token_budget: options.thinkingTokenBudget } : {},
       ...options?.showThoughts ? { show_thoughts: options.showThoughts } : {},
       ...options?.maxSteps ? { max_steps: options.maxSteps } : {},
-      ...options?.maxRetries ? { max_retries: options.maxRetries } : {},
-      ...options?.fastFail ? { fast_fail: options.fastFail } : {}
+      ...options?.maxRetries ? { max_retries: options.maxRetries } : {}
     };
     const traceLabel = options.traceLabel ?? this.options?.traceLabel;
     const spanName = traceLabel ? `${traceLabel} (AxGen)` : "AxGen";
@@ -10107,6 +10312,9 @@ var AxBaseOptimizer = class {
   checkpointLoad;
   checkpointInterval;
   resumeFromCheckpoint;
+  // Logging fields
+  logger;
+  verbose;
   // Checkpoint state
   currentRound = 0;
   scoreHistory = [];
@@ -10130,6 +10338,8 @@ var AxBaseOptimizer = class {
     this.checkpointLoad = args.checkpointLoad;
     this.checkpointInterval = args.checkpointInterval ?? 10;
     this.resumeFromCheckpoint = args.resumeFromCheckpoint;
+    this.logger = args.logger;
+    this.verbose = args.verbose;
     const costTracker = new AxDefaultCostTracker({
       maxTokens: 1e6
     });
@@ -10308,8 +10518,14 @@ var AxBaseOptimizer = class {
   async compilePareto(program, metricFn, options) {
     const startTime = Date.now();
     if (options?.verbose) {
-      console.log("Starting Pareto optimization using base implementation");
-      console.log("This will run multiple single-objective optimizations");
+      this.getLogger(options)?.(
+        "Starting Pareto optimization using base implementation",
+        { tags: ["discovery"] }
+      );
+      this.getLogger(options)?.(
+        "This will run multiple single-objective optimizations",
+        { tags: ["discovery"] }
+      );
     }
     const solutions = await this.generateWeightedSolutions(
       program,
@@ -10323,13 +10539,22 @@ var AxBaseOptimizer = class {
     );
     const allSolutions = [...solutions, ...constraintSolutions];
     if (options?.verbose) {
-      console.log(`Generated ${allSolutions.length} candidate solutions`);
+      this.getLogger(options)?.(
+        `Generated ${allSolutions.length} candidate solutions`,
+        { tags: ["discovery"] }
+      );
     }
     const paretoFront = this.findParetoFrontier(allSolutions);
     const hypervolume = this.calculateHypervolume(paretoFront);
     if (options?.verbose) {
-      console.log(`Found ${paretoFront.length} non-dominated solutions`);
-      console.log(`Hypervolume: ${hypervolume?.toFixed(4) || "N/A"}`);
+      this.getLogger(options)?.(
+        `Found ${paretoFront.length} non-dominated solutions`,
+        { tags: ["discovery"] }
+      );
+      this.getLogger(options)?.(
+        `Hypervolume: ${hypervolume?.toFixed(4) || "N/A"}`,
+        { tags: ["discovery"] }
+      );
     }
     this.updateResourceUsage(startTime);
     this.stats.convergenceInfo.converged = true;
@@ -10367,13 +10592,19 @@ var AxBaseOptimizer = class {
     });
     const objectives = Object.keys(sampleScores);
     if (options?.verbose) {
-      console.log(`Detected objectives: ${objectives.join(", ")}`);
+      this.getLogger(options)?.(
+        `Detected objectives: ${objectives.join(", ")}`,
+        { tags: ["discovery"] }
+      );
     }
     const weightCombinations = this.generateWeightCombinations(objectives);
     for (let i = 0; i < weightCombinations.length; i++) {
       const weights = weightCombinations[i];
       if (options?.verbose) {
-        console.log(`Optimizing with weights: ${JSON.stringify(weights)}`);
+        this.getLogger(options)?.(
+          `Optimizing with weights: ${JSON.stringify(weights)}`,
+          { tags: ["discovery"] }
+        );
       }
       const weightedMetric = async ({ prediction, example }) => {
         const scores = await metricFn({ prediction, example });
@@ -10405,9 +10636,9 @@ var AxBaseOptimizer = class {
         });
       } catch (error) {
         if (options?.verbose) {
-          console.warn(
-            `Failed optimization with weights ${JSON.stringify(weights)}:`,
-            error
+          this.getLogger(options)?.(
+            `Failed optimization with weights ${JSON.stringify(weights)}: ${error}`,
+            { tags: ["warning"] }
           );
         }
         continue;
@@ -10432,8 +10663,9 @@ var AxBaseOptimizer = class {
     const objectives = Object.keys(sampleScores);
     for (const primaryObjective of objectives) {
       if (options?.verbose) {
-        console.log(
-          `Optimizing ${primaryObjective} with constraints on other objectives`
+        this.getLogger(options)?.(
+          `Optimizing ${primaryObjective} with constraints on other objectives`,
+          { tags: ["discovery"] }
         );
       }
       const constraintMetric = async ({ prediction, example }) => {
@@ -10470,9 +10702,9 @@ var AxBaseOptimizer = class {
         });
       } catch (error) {
         if (options?.verbose) {
-          console.warn(
-            `Failed constraint optimization for ${primaryObjective}:`,
-            error
+          this.getLogger(options)?.(
+            `Failed constraint optimization for ${primaryObjective}: ${error}`,
+            { tags: ["warning"] }
           );
         }
         continue;
@@ -10707,6 +10939,39 @@ var AxBaseOptimizer = class {
       );
     }
   }
+  /**
+   * Get the logger function with fallback hierarchy:
+   * 1. Explicit logger passed to optimizer
+   * 2. Logger from student AI service
+   * 3. Default optimizer logger
+   * 4. undefined if verbose is false
+   */
+  getLogger(options) {
+    const isVerbose = this.isLoggingEnabled(options);
+    if (!isVerbose) {
+      return void 0;
+    }
+    if (this.logger) {
+      return this.logger;
+    }
+    try {
+      const aiLogger = this.studentAI.getLogger();
+      if (aiLogger) {
+        return aiLogger;
+      }
+    } catch {
+    }
+    return axDefaultOptimizerLogger;
+  }
+  /**
+   * Check if logging is enabled based on verbose settings
+   */
+  isLoggingEnabled(options) {
+    if (options?.verbose !== void 0) {
+      return options.verbose;
+    }
+    return this.verbose ?? true;
+  }
 };
 
 // db/base.ts
@@ -12289,9 +12554,9 @@ var AxBootstrapFewShot = class extends AxBaseOptimizer {
         this.stats.earlyStopped = true;
         this.stats.earlyStopping.reason = `No improvement for ${this.earlyStoppingPatience} rounds`;
         if (this.verboseMode || this.debugMode) {
-          console.log(
-            `
-Early stopping triggered after ${roundIndex + 1} rounds. No improvement for ${this.earlyStoppingPatience} rounds.`
+          this.getLogger()?.(
+            `Early stopping after ${roundIndex + 1} rounds (no improvement for ${this.earlyStoppingPatience} rounds)`,
+            { tags: ["optimizer", "warning"] }
           );
         }
         return;
@@ -12302,6 +12567,16 @@ Early stopping triggered after ${roundIndex + 1} rounds. No improvement for ${th
     const maxRounds = options?.maxIterations ?? this.maxRounds;
     this.traces = [];
     this.reset();
+    if (this.verboseMode || this.debugMode) {
+      this.getLogger()?.(
+        `Starting BootstrapFewshot optimization with ${maxRounds} rounds`,
+        { tags: ["optimizer", "start"] }
+      );
+      this.getLogger()?.(
+        `Using ${this.examples.length} examples, max ${this.maxDemos} demos`,
+        { tags: ["optimizer", "config"] }
+      );
+    }
     for (let i = 0; i < maxRounds; i++) {
       await this.compileRound(program, i, metricFn, options);
       if (this.stats.earlyStopped) {
@@ -12318,6 +12593,12 @@ Early stopping triggered after ${roundIndex + 1} rounds. No improvement for ${th
     if (this.traces.length > 0) {
       bestScore = this.stats.successfulDemos / Math.max(1, this.stats.totalCalls);
     }
+    if (this.verboseMode || this.debugMode) {
+      this.getLogger()?.(
+        `Bootstrap complete. Generated ${demos.length} demos with ${bestScore.toFixed(3)} success rate`,
+        { tags: ["optimizer", "complete"] }
+      );
+    }
     return {
       demos,
       stats: this.stats,
@@ -12383,7 +12664,6 @@ var AxMiPRO = class extends AxBaseOptimizer {
   viewDataBatchSize;
   tipAwareProposer;
   fewshotAwareProposer;
-  verbose;
   earlyStoppingTrials;
   minImprovementThreshold;
   bayesianOptimization;
@@ -12405,7 +12685,6 @@ var AxMiPRO = class extends AxBaseOptimizer {
     this.viewDataBatchSize = options.viewDataBatchSize ?? 10;
     this.tipAwareProposer = options.tipAwareProposer ?? true;
     this.fewshotAwareProposer = options.fewshotAwareProposer ?? true;
-    this.verbose = options.verbose ?? false;
     this.earlyStoppingTrials = options.earlyStoppingTrials ?? 5;
     this.minImprovementThreshold = options.minImprovementThreshold ?? 0.01;
     this.bayesianOptimization = options.bayesianOptimization ?? false;
@@ -12496,8 +12775,10 @@ var AxMiPRO = class extends AxBaseOptimizer {
    * Bootstraps few-shot examples for the program
    */
   async bootstrapFewShotExamples(program, metricFn) {
-    if (this.verbose) {
-      console.log("Bootstrapping few-shot examples...");
+    if (this.isLoggingEnabled()) {
+      this.getLogger()?.("Bootstrapping few-shot examples...", {
+        tags: ["optimizer", "phase"]
+      });
     }
     const bootstrapper = new AxBootstrapFewShot({
       studentAI: this.studentAI,
@@ -12505,7 +12786,7 @@ var AxMiPRO = class extends AxBaseOptimizer {
       options: {
         maxDemos: this.maxBootstrappedDemos,
         maxRounds: 3,
-        verboseMode: this.verbose
+        verboseMode: this.isLoggingEnabled()
       }
     });
     const result = await bootstrapper.compile(program, metricFn, {
@@ -12550,9 +12831,10 @@ var AxMiPRO = class extends AxBaseOptimizer {
         options
       );
       if (checkpoint && checkpoint.optimizerType === "MiPRO") {
-        if (this.verbose || options?.verbose) {
-          console.log(
-            `Resuming from checkpoint at round ${checkpoint.currentRound}`
+        if (this.isLoggingEnabled(options)) {
+          this.getLogger(options)?.(
+            `Resuming from checkpoint at round ${checkpoint.currentRound}`,
+            { tags: ["optimizer", "checkpoint"] }
           );
         }
         this.restoreFromCheckpoint(checkpoint);
@@ -12562,6 +12844,12 @@ var AxMiPRO = class extends AxBaseOptimizer {
         stagnationRounds = checkpoint.stats.convergenceInfo?.stagnationRounds || 0;
       }
     }
+    if (this.isLoggingEnabled(options)) {
+      this.getLogger(options)?.(
+        `Running optimization trials (${this.numTrials} total)`,
+        { tags: ["optimizer", "phase"] }
+      );
+    }
     for (let i = startRound; i < this.numTrials; i++) {
       const config = {
         instruction: instructions[i % instructions.length] || instructions[0] || "",
@@ -12588,6 +12876,12 @@ var AxMiPRO = class extends AxBaseOptimizer {
         bestScore = score;
         bestConfig = config;
         stagnationRounds = 0;
+        if (this.isLoggingEnabled(options)) {
+          this.getLogger(options)?.(
+            `Trial ${i + 1}/${this.numTrials}: New best score ${bestScore.toFixed(3)}`,
+            { tags: ["optimizer", "progress"] }
+          );
+        }
       } else {
         stagnationRounds++;
       }
@@ -12706,38 +13000,53 @@ var AxMiPRO = class extends AxBaseOptimizer {
       this.configureAuto(miproOptions.auto);
     }
     const valset = this.getValidationSet(options) || (miproOptions?.valset ?? this.examples.slice(0, Math.floor(this.examples.length * 0.2)));
-    if (this.verbose || options?.verbose) {
-      console.log(`Starting MIPROv2 optimization with ${this.numTrials} trials`);
-      console.log(
-        `Using ${this.examples.length} examples for training and ${valset.length} for validation`
+    if (this.isLoggingEnabled(options)) {
+      this.getLogger(options)?.(
+        `Starting MIPROv2 optimization with ${this.numTrials} trials`,
+        { tags: ["optimizer", "start"] }
+      );
+      this.getLogger(options)?.(
+        `Using ${this.examples.length} examples for training and ${valset.length} for validation`,
+        { tags: ["optimizer", "config"] }
       );
       if (this.teacherAI) {
-        console.log("Using separate teacher model for instruction generation");
+        this.getLogger(options)?.(
+          "Using separate teacher model for instruction generation",
+          { tags: ["optimizer", "config"] }
+        );
       }
     }
     let bootstrappedDemos = [];
     if (this.maxBootstrappedDemos > 0) {
       bootstrappedDemos = await this.bootstrapFewShotExamples(program, metricFn);
-      if (this.verbose) {
-        console.log(
-          `Generated ${bootstrappedDemos.length} bootstrapped demonstrations`
+      if (this.isLoggingEnabled(options)) {
+        this.getLogger(options)?.(
+          `Generated ${bootstrappedDemos.length} bootstrapped demonstrations`,
+          { tags: ["optimizer", "result"] }
         );
       }
     }
     let labeledExamples = [];
     if (this.maxLabeledDemos > 0) {
       labeledExamples = this.selectLabeledExamples();
-      if (this.verbose) {
-        console.log(
-          `Selected ${labeledExamples.length} labeled examples from training set`
+      if (this.isLoggingEnabled(options)) {
+        this.getLogger(options)?.(
+          `Selected ${labeledExamples.length} labeled examples from training set`,
+          { tags: ["optimizer", "result"] }
         );
       }
     }
     const instructions = await this.proposeInstructionCandidates(options);
-    if (this.verbose) {
-      console.log(`Generated ${instructions.length} instruction candidates`);
+    if (this.isLoggingEnabled(options)) {
+      this.getLogger(options)?.(
+        `Generated ${instructions.length} instruction candidates`,
+        { tags: ["optimizer", "result"] }
+      );
       if (this.hasTeacherAI(options)) {
-        console.log("Using teacher AI for instruction generation");
+        this.getLogger(options)?.(
+          "Using teacher AI for instruction generation",
+          { tags: ["optimizer", "config"] }
+        );
       }
     }
     const { bestConfig, bestScore } = await this.runOptimization(
@@ -12749,9 +13058,15 @@ var AxMiPRO = class extends AxBaseOptimizer {
       metricFn,
       options
     );
-    if (this.verbose || options?.verbose) {
-      console.log(`Optimization complete. Best score: ${bestScore}`);
-      console.log(`Best configuration: ${JSON.stringify(bestConfig)}`);
+    if (this.isLoggingEnabled(options)) {
+      this.getLogger(options)?.(
+        `Optimization complete. Best score: ${bestScore}`,
+        { tags: ["optimizer", "complete"] }
+      );
+      this.getLogger(options)?.(
+        `Best configuration: ${JSON.stringify(bestConfig)}`,
+        { tags: ["optimizer", "result"] }
+      );
     }
     if (this.checkTargetScore(bestScore)) {
       this.triggerEarlyStopping(
@@ -12878,9 +13193,6 @@ var AxMiPRO = class extends AxBaseOptimizer {
     if (config.minImprovementThreshold !== void 0) {
       this.minImprovementThreshold = config.minImprovementThreshold;
     }
-    if (config.verbose !== void 0) {
-      this.verbose = config.verbose;
-    }
   }
   /**
    * Reset optimizer state for reuse with different programs
@@ -15354,6 +15666,10 @@ export {
   axAITogetherDefaultConfig,
   axBaseAIDefaultConfig,
   axBaseAIDefaultCreativeConfig,
+  axCreateDefaultLogger,
+  axCreateDefaultTextLogger,
+  axCreateOptimizerLogger,
+  axDefaultOptimizerLogger,
   axGlobals,
   axModelInfoAnthropic,
   axModelInfoCohere,