@mastra/core 0.15.3-alpha.3 → 0.15.3-alpha.5

package/dist/{chunk-WOTBMZCN.js → chunk-LOYT3WUA.js}

@@ -9,7 +9,6 @@ var NoOpAISpan = class _NoOpAISpan {
   type;
   attributes;
   parent;
-  trace;
   traceId;
   startTime;
   endTime;
@@ -26,7 +25,6 @@ var NoOpAISpan = class _NoOpAISpan {
     this.attributes = options.attributes || {};
     this.metadata = options.metadata;
     this.parent = options.parent;
-    this.trace = options.parent ? options.parent.trace : this;
     this.traceId = "no-op-trace";
     this.startTime = /* @__PURE__ */ new Date();
     this.aiTracing = aiTracing;
@@ -900,6 +898,232 @@ var DefaultExporter = class {
   }
 };
 
+// src/ai-tracing/registry.ts
+var AITracingRegistry = class {
+  instances = /* @__PURE__ */ new Map();
+  defaultInstance;
+  selector;
+  /**
+   * Register a tracing instance
+   */
+  register(name, instance, isDefault = false) {
+    if (this.instances.has(name)) {
+      throw new Error(`AI Tracing instance '${name}' already registered`);
+    }
+    this.instances.set(name, instance);
+    if (isDefault || !this.defaultInstance) {
+      this.defaultInstance = instance;
+    }
+  }
+  /**
+   * Get a tracing instance by name
+   */
+  get(name) {
+    return this.instances.get(name);
+  }
+  /**
+   * Get the default tracing instance
+   */
+  getDefault() {
+    return this.defaultInstance;
+  }
+  /**
+   * Set the tracing selector function
+   */
+  setSelector(selector) {
+    this.selector = selector;
+  }
+  /**
+   * Get the selected tracing instance based on context
+   */
+  getSelected(context) {
+    if (this.selector) {
+      const selected = this.selector(context, this.instances);
+      if (selected && this.instances.has(selected)) {
+        return this.instances.get(selected);
+      }
+    }
+    return this.defaultInstance;
+  }
+  /**
+   * Unregister a tracing instance
+   */
+  unregister(name) {
+    return this.instances.delete(name);
+  }
+  /**
+   * Shutdown all instances and clear the registry
+   */
+  async shutdown() {
+    const shutdownPromises = Array.from(this.instances.values()).map((instance) => instance.shutdown());
+    await Promise.allSettled(shutdownPromises);
+    this.instances.clear();
+  }
+  /**
+   * Clear all instances without shutdown
+   */
+  clear() {
+    this.instances.clear();
+    this.defaultInstance = void 0;
+    this.selector = void 0;
+  }
+  /**
+   * Get all registered instances
+   */
+  getAll() {
+    return new Map(this.instances);
+  }
+};
+var aiTracingRegistry = new AITracingRegistry();
+function registerAITracing(name, instance, isDefault = false) {
+  aiTracingRegistry.register(name, instance, isDefault);
+}
+function getAITracing(name) {
+  return aiTracingRegistry.get(name);
+}
+function getDefaultAITracing() {
+  return aiTracingRegistry.getDefault();
+}
+function setAITracingSelector(selector) {
+  aiTracingRegistry.setSelector(selector);
+}
+function getSelectedAITracing(context) {
+  return aiTracingRegistry.getSelected(context);
+}
+function unregisterAITracing(name) {
+  return aiTracingRegistry.unregister(name);
+}
+async function shutdownAITracingRegistry() {
+  await aiTracingRegistry.shutdown();
+}
+function clearAITracingRegistry() {
+  aiTracingRegistry.clear();
+}
+function getAllAITracing() {
+  return aiTracingRegistry.getAll();
+}
+function hasAITracing(name) {
+  const tracing = getAITracing(name);
+  if (!tracing) return false;
+  const config = tracing.getConfig();
+  const sampling = config.sampling;
+  return sampling.type !== "never" /* NEVER */;
+}
+function isAITracingInstance(obj) {
+  return obj instanceof MastraAITracing;
+}
+function setupAITracing(config) {
+  const entries = Object.entries(config.instances);
+  entries.forEach(([name, tracingDef], index) => {
+    const instance = isAITracingInstance(tracingDef) ? tracingDef : new DefaultAITracing({ ...tracingDef, instanceName: name });
+    const isDefault = index === 0;
+    registerAITracing(name, instance, isDefault);
+  });
+  if (config.selector) {
+    setAITracingSelector(config.selector);
+  }
+}
+
+// src/ai-tracing/utils.ts
+function shallowCleanObject(obj) {
+  const cleaned = {};
+  for (const [key, value] of Object.entries(obj)) {
+    try {
+      JSON.stringify(value);
+      cleaned[key] = value;
+    } catch (error) {
+      cleaned[key] = `[${error instanceof Error ? error.message : String(error)}]`;
+    }
+  }
+  return cleaned;
+}
+function shallowCleanArray(arr) {
+  return arr.map((item) => {
+    if (item && typeof item === "object" && !Array.isArray(item)) {
+      return shallowCleanObject(item);
+    }
+    try {
+      JSON.stringify(item);
+      return item;
+    } catch (error) {
+      return `[${error instanceof Error ? error.message : String(error)}]`;
+    }
+  });
+}
+function shallowClean(value) {
+  if (value === null || value === void 0) {
+    return value;
+  }
+  if (Array.isArray(value)) {
+    return shallowCleanArray(value);
+  }
+  if (typeof value === "object") {
+    return shallowCleanObject(value);
+  }
+  try {
+    JSON.stringify(value);
+    return value;
+  } catch (error) {
+    return `[${error instanceof Error ? error.message : String(error)}]`;
+  }
+}
+function omitKeys(obj, keysToOmit) {
+  return Object.fromEntries(Object.entries(obj).filter(([key]) => !keysToOmit.includes(key)));
+}
+function selectFields(obj, fields) {
+  if (!obj || typeof obj !== "object") {
+    return obj;
+  }
+  const result = {};
+  for (const field of fields) {
+    const value = getNestedValue(obj, field);
+    if (value !== void 0) {
+      setNestedValue(result, field, value);
+    }
+  }
+  return result;
+}
+function getNestedValue(obj, path) {
+  return path.split(".").reduce((current, key) => {
+    return current && typeof current === "object" ? current[key] : void 0;
+  }, obj);
+}
+function setNestedValue(obj, path, value) {
+  const keys = path.split(".");
+  const lastKey = keys.pop();
+  if (!lastKey) {
+    return;
+  }
+  const target = keys.reduce((current, key) => {
+    if (!current[key] || typeof current[key] !== "object") {
+      current[key] = {};
+    }
+    return current[key];
+  }, obj);
+  target[lastKey] = value;
+}
+function getOrCreateSpan(options) {
+  const { type, attributes, tracingContext, runtimeContext, ...rest } = options;
+  if (tracingContext?.currentSpan) {
+    return tracingContext.currentSpan.createChildSpan({
+      type,
+      attributes,
+      ...rest
+    });
+  }
+  const aiTracing = getSelectedAITracing({
+    runtimeContext
+  });
+  return aiTracing?.startSpan({
+    type,
+    attributes,
+    startOptions: {
+      runtimeContext
+    },
+    ...rest
+  });
+}
+
 // src/ai-tracing/default.ts
 function generateSpanId() {
   const bytes = new Uint8Array(8);
@@ -929,7 +1153,6 @@ var DefaultAISpan = class {
   type;
   attributes;
   parent;
-  trace;
   traceId;
   startTime;
   endTime;
@@ -944,13 +1167,12 @@ var DefaultAISpan = class {
     this.id = generateSpanId();
     this.name = options.name;
     this.type = options.type;
-    this.attributes = options.attributes || {};
-    this.metadata = options.metadata;
+    this.attributes = shallowClean(options.attributes) || {};
+    this.metadata = shallowClean(options.metadata);
     this.parent = options.parent;
-    this.trace = options.parent ? options.parent.trace : this;
     this.startTime = /* @__PURE__ */ new Date();
     this.aiTracing = aiTracing;
-    this.input = options.input;
+    this.input = shallowClean(options.input);
     this.isEvent = options.isEvent;
     this.logger = new ConsoleLogger({
       name: "default-ai-span",
@@ -960,10 +1182,10 @@ var DefaultAISpan = class {
     if (!options.parent) {
       this.traceId = generateTraceId();
     } else {
-      this.traceId = options.parent.trace.traceId;
+      this.traceId = options.parent.traceId;
     }
     if (this.isEvent) {
-      this.output = options.output;
+      this.output = shallowClean(options.output);
     }
   }
   end(options) {
@@ -973,13 +1195,13 @@ var DefaultAISpan = class {
     }
     this.endTime = /* @__PURE__ */ new Date();
     if (options?.output !== void 0) {
-      this.output = options.output;
+      this.output = shallowClean(options.output);
     }
     if (options?.attributes) {
-      this.attributes = { ...this.attributes, ...options.attributes };
+      this.attributes = { ...this.attributes, ...shallowClean(options.attributes) };
     }
     if (options?.metadata) {
-      this.metadata = { ...this.metadata, ...options.metadata };
+      this.metadata = { ...this.metadata, ...shallowClean(options.metadata) };
     }
   }
   error(options) {
@@ -998,10 +1220,10 @@ var DefaultAISpan = class {
       message: error.message
     };
     if (attributes) {
-      this.attributes = { ...this.attributes, ...attributes };
+      this.attributes = { ...this.attributes, ...shallowClean(attributes) };
     }
     if (metadata) {
-      this.metadata = { ...this.metadata, ...metadata };
+      this.metadata = { ...this.metadata, ...shallowClean(metadata) };
     }
     if (endSpan) {
       this.end();
@@ -1029,16 +1251,16 @@ var DefaultAISpan = class {
       return;
     }
     if (options?.input !== void 0) {
-      this.input = options.input;
+      this.input = shallowClean(options.input);
     }
     if (options?.output !== void 0) {
-      this.output = options.output;
+      this.output = shallowClean(options.output);
     }
     if (options?.attributes) {
-      this.attributes = { ...this.attributes, ...options.attributes };
+      this.attributes = { ...this.attributes, ...shallowClean(options.attributes) };
     }
     if (options?.metadata) {
-      this.metadata = { ...this.metadata, ...options.metadata };
+      this.metadata = { ...this.metadata, ...shallowClean(options.metadata) };
     }
   }
   get isRootSpan() {
@@ -1139,155 +1361,6 @@ var DefaultAITracing = class extends MastraAITracing {
   }
 };
 
-// src/ai-tracing/registry.ts
-var AITracingRegistry = class {
-  instances = /* @__PURE__ */ new Map();
-  defaultInstance;
-  selector;
-  /**
-   * Register a tracing instance
-   */
-  register(name, instance, isDefault = false) {
-    if (this.instances.has(name)) {
-      throw new Error(`AI Tracing instance '${name}' already registered`);
-    }
-    this.instances.set(name, instance);
-    if (isDefault || !this.defaultInstance) {
-      this.defaultInstance = instance;
-    }
-  }
-  /**
-   * Get a tracing instance by name
-   */
-  get(name) {
-    return this.instances.get(name);
-  }
-  /**
-   * Get the default tracing instance
-   */
-  getDefault() {
-    return this.defaultInstance;
-  }
-  /**
-   * Set the tracing selector function
-   */
-  setSelector(selector) {
-    this.selector = selector;
-  }
-  /**
-   * Get the selected tracing instance based on context
-   */
-  getSelected(context) {
-    if (this.selector) {
-      const selected = this.selector(context, this.instances);
-      if (selected && this.instances.has(selected)) {
-        return this.instances.get(selected);
-      }
-    }
-    return this.defaultInstance;
-  }
-  /**
-   * Unregister a tracing instance
-   */
-  unregister(name) {
-    return this.instances.delete(name);
-  }
-  /**
-   * Shutdown all instances and clear the registry
-   */
-  async shutdown() {
-    const shutdownPromises = Array.from(this.instances.values()).map((instance) => instance.shutdown());
-    await Promise.allSettled(shutdownPromises);
-    this.instances.clear();
-  }
-  /**
-   * Clear all instances without shutdown
-   */
-  clear() {
-    this.instances.clear();
-    this.defaultInstance = void 0;
-    this.selector = void 0;
-  }
-  /**
-   * Get all registered instances
-   */
-  getAll() {
-    return new Map(this.instances);
-  }
-};
-var aiTracingRegistry = new AITracingRegistry();
-function registerAITracing(name, instance, isDefault = false) {
-  aiTracingRegistry.register(name, instance, isDefault);
-}
-function getAITracing(name) {
-  return aiTracingRegistry.get(name);
-}
-function getDefaultAITracing() {
-  return aiTracingRegistry.getDefault();
-}
-function setAITracingSelector(selector) {
-  aiTracingRegistry.setSelector(selector);
-}
-function getSelectedAITracing(context) {
-  return aiTracingRegistry.getSelected(context);
-}
-function unregisterAITracing(name) {
-  return aiTracingRegistry.unregister(name);
-}
-async function shutdownAITracingRegistry() {
-  await aiTracingRegistry.shutdown();
-}
-function clearAITracingRegistry() {
-  aiTracingRegistry.clear();
-}
-function getAllAITracing() {
-  return aiTracingRegistry.getAll();
-}
-function hasAITracing(name) {
-  const tracing = getAITracing(name);
-  if (!tracing) return false;
-  const config = tracing.getConfig();
-  const sampling = config.sampling;
-  return sampling.type !== "never" /* NEVER */;
-}
-function isAITracingInstance(obj) {
-  return obj instanceof MastraAITracing;
-}
-function setupAITracing(config) {
-  const entries = Object.entries(config.instances);
-  entries.forEach(([name, tracingDef], index) => {
-    const instance = isAITracingInstance(tracingDef) ? tracingDef : new DefaultAITracing({ ...tracingDef, instanceName: name });
-    const isDefault = index === 0;
-    registerAITracing(name, instance, isDefault);
-  });
-  if (config.selector) {
-    setAITracingSelector(config.selector);
-  }
-}
-
-// src/ai-tracing/utils.ts
-function sanitizeMetadata(metadata) {
-  if (!metadata) return {};
-  const sanitized = {};
-  for (const [key, value] of Object.entries(metadata)) {
-    if (isSerializable(value)) {
-      sanitized[key] = value;
-    }
-  }
-  return sanitized;
-}
-function isSerializable(value) {
-  try {
-    JSON.stringify(value);
-    return true;
-  } catch {
-    return false;
-  }
-}
-function omitKeys(obj, keysToOmit) {
-  return Object.fromEntries(Object.entries(obj).filter(([key]) => !keysToOmit.includes(key)));
-}
-
 // src/ai-tracing/context.ts
 var AGENT_GETTERS = ["getAgent", "getAgentById"];
 var AGENT_METHODS_TO_WRAP = ["generate", "stream", "generateVNext", "streamVNext"];
@@ -1316,10 +1389,12 @@ function wrapMastra(mastra, tracingContext) {
               return wrapWorkflow(workflow, tracingContext);
             };
           }
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         } catch (error) {
           console.warn("AI Tracing: Failed to wrap method, falling back to original", error);
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         }
       }
     });
@@ -1344,10 +1419,12 @@ function wrapAgent(agent, tracingContext) {
               });
             };
           }
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         } catch (error) {
           console.warn("AI Tracing: Failed to wrap agent method, falling back to original", error);
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         }
       }
     });
@@ -1372,10 +1449,12 @@ function wrapWorkflow(workflow, tracingContext) {
               });
             };
           }
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         } catch (error) {
           console.warn("AI Tracing: Failed to wrap workflow method, falling back to original", error);
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         }
       }
     });
@@ -1385,6 +1464,6 @@ function wrapWorkflow(workflow, tracingContext) {
   }
 }
 
-export { AISpanType, AITracingEventType, ConsoleExporter, DefaultAITracing, DefaultExporter, MastraAITracing, SamplingStrategyType, SensitiveDataFilter, aiTracingDefaultConfig, clearAITracingRegistry, getAITracing, getAllAITracing, getDefaultAITracing, getSelectedAITracing, hasAITracing, isSerializable, omitKeys, registerAITracing, sanitizeMetadata, setAITracingSelector, setupAITracing, shutdownAITracingRegistry, unregisterAITracing, wrapAgent, wrapMastra, wrapWorkflow };
-//# sourceMappingURL=chunk-WOTBMZCN.js.map
-//# sourceMappingURL=chunk-WOTBMZCN.js.map
+export { AISpanType, AITracingEventType, ConsoleExporter, DefaultAITracing, DefaultExporter, MastraAITracing, SamplingStrategyType, SensitiveDataFilter, aiTracingDefaultConfig, clearAITracingRegistry, getAITracing, getAllAITracing, getDefaultAITracing, getOrCreateSpan, getSelectedAITracing, hasAITracing, omitKeys, registerAITracing, selectFields, setAITracingSelector, setupAITracing, shallowClean, shallowCleanArray, shallowCleanObject, shutdownAITracingRegistry, unregisterAITracing, wrapAgent, wrapMastra, wrapWorkflow };
+//# sourceMappingURL=chunk-LOYT3WUA.js.map
+//# sourceMappingURL=chunk-LOYT3WUA.js.map