@mastra/observability 0.0.0-fix-thread-list-20251105222841 → 0.0.0-fix-ai-sdk-dependency-20251124104209

package/dist/index.cjs

@@ -1,15 +1,14 @@
 'use strict';
 
 var base = require('@mastra/core/base');
-var logger = require('@mastra/core/logger');
 var error = require('@mastra/core/error');
+var logger = require('@mastra/core/logger');
+var zod = require('zod');
 var observability = require('@mastra/core/observability');
 var utils = require('@mastra/core/utils');
 var web = require('stream/web');
 
 // src/default.ts
-
-// src/config.ts
 var SamplingStrategyType = /* @__PURE__ */ ((SamplingStrategyType2) => {
   SamplingStrategyType2["ALWAYS"] = "always";
   SamplingStrategyType2["NEVER"] = "never";
@@ -17,6 +16,66 @@ var SamplingStrategyType = /* @__PURE__ */ ((SamplingStrategyType2) => {
   SamplingStrategyType2["CUSTOM"] = "custom";
   return SamplingStrategyType2;
 })(SamplingStrategyType || {});
+var samplingStrategySchema = zod.z.discriminatedUnion("type", [
+  zod.z.object({
+    type: zod.z.literal("always" /* ALWAYS */)
+  }),
+  zod.z.object({
+    type: zod.z.literal("never" /* NEVER */)
+  }),
+  zod.z.object({
+    type: zod.z.literal("ratio" /* RATIO */),
+    probability: zod.z.number().min(0, "Probability must be between 0 and 1").max(1, "Probability must be between 0 and 1")
+  }),
+  zod.z.object({
+    type: zod.z.literal("custom" /* CUSTOM */),
+    sampler: zod.z.function().args(zod.z.any().optional()).returns(zod.z.boolean())
+  })
+]);
+var observabilityInstanceConfigSchema = zod.z.object({
+  name: zod.z.string().min(1, "Name is required"),
+  serviceName: zod.z.string().min(1, "Service name is required"),
+  sampling: samplingStrategySchema.optional(),
+  exporters: zod.z.array(zod.z.any()).optional(),
+  spanOutputProcessors: zod.z.array(zod.z.any()).optional(),
+  includeInternalSpans: zod.z.boolean().optional(),
+  requestContextKeys: zod.z.array(zod.z.string()).optional()
+});
+var observabilityConfigValueSchema = zod.z.object({
+  serviceName: zod.z.string().min(1, "Service name is required"),
+  sampling: samplingStrategySchema.optional(),
+  exporters: zod.z.array(zod.z.any()).optional(),
+  spanOutputProcessors: zod.z.array(zod.z.any()).optional(),
+  includeInternalSpans: zod.z.boolean().optional(),
+  requestContextKeys: zod.z.array(zod.z.string()).optional()
+});
+var observabilityRegistryConfigSchema = zod.z.object({
+  default: zod.z.object({
+    enabled: zod.z.boolean().optional()
+  }).optional().nullable(),
+  configs: zod.z.union([zod.z.record(zod.z.string(), zod.z.any()), zod.z.array(zod.z.any()), zod.z.null()]).optional(),
+  configSelector: zod.z.function().optional()
+}).passthrough().refine(
+  (data) => {
+    const isDefaultEnabled = data.default?.enabled === true;
+    const hasConfigs = data.configs && typeof data.configs === "object" && !Array.isArray(data.configs) ? Object.keys(data.configs).length > 0 : false;
+    return !(isDefaultEnabled && hasConfigs);
+  },
+  {
+    message: 'Cannot specify both "default" (when enabled) and "configs". Use either default observability or custom configs, but not both.'
+  }
+).refine(
+  (data) => {
+    const configCount = data.configs && typeof data.configs === "object" && !Array.isArray(data.configs) ? Object.keys(data.configs).length : 0;
+    if (configCount > 1 && !data.configSelector) {
+      return false;
+    }
+    return true;
+  },
+  {
+    message: 'A "configSelector" function is required when multiple configs are specified to determine which config to use.'
+  }
+);
 var BaseExporter = class {
   /** Mastra logger instance */
   logger;
@@ -94,10 +153,10 @@ var CloudExporter = class extends BaseExporter {
     const accessToken = config.accessToken ?? process.env.MASTRA_CLOUD_ACCESS_TOKEN;
     if (!accessToken) {
       this.setDisabled(
-        "MASTRA_CLOUD_ACCESS_TOKEN environment variable not set. \u{1F680} Sign up for Mastra Cloud at https://cloud.mastra.ai to see your traces online and obtain your access token."
+        "MASTRA_CLOUD_ACCESS_TOKEN environment variable not set.\n\u{1F680} Sign up at https://cloud.mastra.ai to see your AI traces online and obtain your access token."
       );
     }
-    const endpoint = config.endpoint ?? process.env.MASTRA_CLOUD_AI_TRACES_ENDPOINT ?? "https://api.mastra.ai/ai/spans/publish";
+    const endpoint = config.endpoint ?? process.env.MASTRA_CLOUD_TRACES_ENDPOINT ?? "https://api.mastra.ai/ai/spans/publish";
     this.config = {
       logger: this.logger,
       logLevel: config.logLevel ?? logger.LogLevel.INFO,
@@ -1589,7 +1648,7 @@ var BaseObservabilityInstance = class extends base.MastraBase {
   // Utility Methods
   // ============================================================================
   /**
-   * Check if an AI trace should be sampled
+   * Check if a trace should be sampled
    */
   shouldSample(options) {
     const { sampling } = this.config;
@@ -1996,10 +2055,38 @@ var Observability = class extends base.MastraBase {
     if (config === void 0) {
       config = {};
     }
-    if (config.default?.enabled && config.configs?.["default"]) {
-      throw new Error(
-        "Cannot use 'default' as a custom config name when default tracing is enabled. Please rename your custom config to avoid conflicts."
-      );
+    const validationResult = observabilityRegistryConfigSchema.safeParse(config);
+    if (!validationResult.success) {
+      const errorMessages = validationResult.error.errors.map((err) => `${err.path.join(".") || "config"}: ${err.message}`).join("; ");
+      throw new error.MastraError({
+        id: "OBSERVABILITY_INVALID_CONFIG",
+        text: `Invalid observability configuration: ${errorMessages}`,
+        domain: error.ErrorDomain.MASTRA_OBSERVABILITY,
+        category: error.ErrorCategory.USER,
+        details: {
+          validationErrors: errorMessages
+        }
+      });
+    }
+    if (config.configs) {
+      for (const [name, configValue] of Object.entries(config.configs)) {
+        if (!isInstance(configValue)) {
+          const configValidation = observabilityConfigValueSchema.safeParse(configValue);
+          if (!configValidation.success) {
+            const errorMessages = configValidation.error.errors.map((err) => `${err.path.join(".")}: ${err.message}`).join("; ");
+            throw new error.MastraError({
+              id: "OBSERVABILITY_INVALID_INSTANCE_CONFIG",
+              text: `Invalid configuration for observability instance '${name}': ${errorMessages}`,
+              domain: error.ErrorDomain.MASTRA_OBSERVABILITY,
+              category: error.ErrorCategory.USER,
+              details: {
+                instanceName: name,
+                validationErrors: errorMessages
+              }
+            });
+          }
+        }
+      }
     }
     if (config.default?.enabled) {
       const defaultInstance = new DefaultObservabilityInstance({
@@ -2098,5 +2185,9 @@ exports.Observability = Observability;
 exports.SamplingStrategyType = SamplingStrategyType;
 exports.SensitiveDataFilter = SensitiveDataFilter;
 exports.deepClean = deepClean;
+exports.observabilityConfigValueSchema = observabilityConfigValueSchema;
+exports.observabilityInstanceConfigSchema = observabilityInstanceConfigSchema;
+exports.observabilityRegistryConfigSchema = observabilityRegistryConfigSchema;
+exports.samplingStrategySchema = samplingStrategySchema;
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map