@mastra/observability 0.0.0-main-test-05-11-2025-2-20251106022801 → 0.0.0-main-test-05-11-2025-2-20251106025330

package/dist/index.js

@@ -1,13 +1,12 @@
 import { MastraBase } from '@mastra/core/base';
-import { ConsoleLogger, LogLevel, RegisteredLogger } from '@mastra/core/logger';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
+import { ConsoleLogger, LogLevel, RegisteredLogger } from '@mastra/core/logger';
+import { z } from 'zod';
 import { TracingEventType, SpanType, InternalSpans } from '@mastra/core/observability';
 import { fetchWithRetry, getNestedValue, setNestedValue } from '@mastra/core/utils';
 import { TransformStream } from 'stream/web';
 
 // src/default.ts
-
-// src/config.ts
 var SamplingStrategyType = /* @__PURE__ */ ((SamplingStrategyType2) => {
   SamplingStrategyType2["ALWAYS"] = "always";
   SamplingStrategyType2["NEVER"] = "never";
@@ -15,6 +14,66 @@ var SamplingStrategyType = /* @__PURE__ */ ((SamplingStrategyType2) => {
   SamplingStrategyType2["CUSTOM"] = "custom";
   return SamplingStrategyType2;
 })(SamplingStrategyType || {});
+var samplingStrategySchema = z.discriminatedUnion("type", [
+  z.object({
+    type: z.literal("always" /* ALWAYS */)
+  }),
+  z.object({
+    type: z.literal("never" /* NEVER */)
+  }),
+  z.object({
+    type: z.literal("ratio" /* RATIO */),
+    probability: z.number().min(0, "Probability must be between 0 and 1").max(1, "Probability must be between 0 and 1")
+  }),
+  z.object({
+    type: z.literal("custom" /* CUSTOM */),
+    sampler: z.function().args(z.any().optional()).returns(z.boolean())
+  })
+]);
+var observabilityInstanceConfigSchema = z.object({
+  name: z.string().min(1, "Name is required"),
+  serviceName: z.string().min(1, "Service name is required"),
+  sampling: samplingStrategySchema.optional(),
+  exporters: z.array(z.any()).optional(),
+  spanOutputProcessors: z.array(z.any()).optional(),
+  includeInternalSpans: z.boolean().optional(),
+  requestContextKeys: z.array(z.string()).optional()
+});
+var observabilityConfigValueSchema = z.object({
+  serviceName: z.string().min(1, "Service name is required"),
+  sampling: samplingStrategySchema.optional(),
+  exporters: z.array(z.any()).optional(),
+  spanOutputProcessors: z.array(z.any()).optional(),
+  includeInternalSpans: z.boolean().optional(),
+  requestContextKeys: z.array(z.string()).optional()
+});
+var observabilityRegistryConfigSchema = z.object({
+  default: z.object({
+    enabled: z.boolean().optional()
+  }).optional().nullable(),
+  configs: z.union([z.record(z.string(), z.any()), z.array(z.any()), z.null()]).optional(),
+  configSelector: 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;
@@ -1994,10 +2053,38 @@ var Observability = class extends 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 MastraError({
+        id: "OBSERVABILITY_INVALID_CONFIG",
+        text: `Invalid observability configuration: ${errorMessages}`,
+        domain: ErrorDomain.MASTRA_OBSERVABILITY,
+        category: 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 MastraError({
+              id: "OBSERVABILITY_INVALID_INSTANCE_CONFIG",
+              text: `Invalid configuration for observability instance '${name}': ${errorMessages}`,
+              domain: ErrorDomain.MASTRA_OBSERVABILITY,
+              category: ErrorCategory.USER,
+              details: {
+                instanceName: name,
+                validationErrors: errorMessages
+              }
+            });
+          }
+        }
+      }
     }
     if (config.default?.enabled) {
       const defaultInstance = new DefaultObservabilityInstance({
@@ -2082,6 +2169,6 @@ var Observability = class extends MastraBase {
   }
 };
 
-export { BaseExporter, BaseObservabilityInstance, BaseSpan, CloudExporter, ConsoleExporter, DefaultExporter, DefaultObservabilityInstance, DefaultSpan, ModelSpanTracker, NoOpSpan, Observability, SamplingStrategyType, SensitiveDataFilter, deepClean };
+export { BaseExporter, BaseObservabilityInstance, BaseSpan, CloudExporter, ConsoleExporter, DefaultExporter, DefaultObservabilityInstance, DefaultSpan, ModelSpanTracker, NoOpSpan, Observability, SamplingStrategyType, SensitiveDataFilter, deepClean, observabilityConfigValueSchema, observabilityInstanceConfigSchema, observabilityRegistryConfigSchema, samplingStrategySchema };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map