@fallom/trace 0.2.26 → 0.2.28

package/dist/core-5BF6KLNO.mjs

@@ -0,0 +1,21 @@
+import {
+  DEFAULT_JUDGE_MODEL,
+  _apiKey,
+  _baseUrl,
+  _initialized,
+  compareModels,
+  evaluate,
+  init,
+  uploadResultsPublic
+} from "./chunk-MSI4HGK6.mjs";
+import "./chunk-7P6ASYW6.mjs";
+export {
+  DEFAULT_JUDGE_MODEL,
+  _apiKey,
+  _baseUrl,
+  _initialized,
+  compareModels,
+  evaluate,
+  init,
+  uploadResultsPublic
+};

package/dist/core-SL7FAAJN.mjs

@@ -0,0 +1,21 @@
+import {
+  DEFAULT_JUDGE_MODEL,
+  _apiKey,
+  _baseUrl,
+  _initialized,
+  compareModels,
+  evaluate,
+  init,
+  uploadResultsPublic
+} from "./chunk-TNNLTWRG.mjs";
+import "./chunk-7P6ASYW6.mjs";
+export {
+  DEFAULT_JUDGE_MODEL,
+  _apiKey,
+  _baseUrl,
+  _initialized,
+  compareModels,
+  evaluate,
+  init,
+  uploadResultsPublic
+};

package/dist/index.d.mts

@@ -130,6 +130,92 @@ declare function init$4(options?: {
  */
 declare function shutdown(): Promise<void>;
 
+/**
+ * FallomSpan - Manual span for custom operations.
+ *
+ * Use for non-LLM operations like RAG retrieval, preprocessing, tool execution, etc.
+ *
+ * @example
+ * ```typescript
+ * const session = fallom.session({ configKey: "my-agent", sessionId });
+ *
+ * // Create a manual span
+ * const span = session.span("rag.retrieve");
+ * span.set({ "rag.query": userQuery, "rag.topK": 5 });
+ *
+ * const docs = await retrieveDocuments(userQuery);
+ * span.set({ "rag.documents.count": docs.length });
+ *
+ * span.end(); // Sends the span
+ * ```
+ */
+
+interface SpanOptions {
+    /** Parent span ID for nested spans */
+    parentSpanId?: string;
+    /** Trace ID to continue an existing trace */
+    traceId?: string;
+    /** Span kind (defaults to "custom") */
+    kind?: "custom" | "tool" | "retrieval" | "preprocessing" | "postprocessing";
+}
+declare class FallomSpan {
+    private name;
+    private ctx;
+    private attrs;
+    private startTime;
+    private ended;
+    private _status;
+    private _errorMessage?;
+    readonly spanId: string;
+    readonly traceId: string;
+    readonly parentSpanId?: string;
+    readonly kind: string;
+    constructor(name: string, ctx: SessionContext, options?: SpanOptions);
+    /**
+     * Set attributes on the span.
+     * Can be called multiple times - attributes are merged.
+     */
+    set(attributes: Record<string, unknown>): this;
+    /**
+     * Mark the span as errored.
+     */
+    setError(error: Error | string): this;
+    /**
+     * Get span context for creating child spans.
+     */
+    context(): {
+        traceId: string;
+        spanId: string;
+    };
+    /**
+     * End the span and send it.
+     * Must be called for the span to be recorded.
+     */
+    end(): void;
+}
+/**
+ * Wrap a function to automatically create a span around it.
+ * Similar to Braintrust's wrapTraced().
+ *
+ * @example
+ * ```typescript
+ * const fetchDocuments = wrapTraced(
+ *   session,
+ *   "rag.fetch",
+ *   async (query: string) => {
+ *     const docs = await vectorDb.search(query);
+ *     return docs;
+ *   }
+ * );
+ *
+ * // Function input/output automatically captured
+ * const docs = await fetchDocuments("user query");
+ * ```
+ */
+declare function wrapTraced<T extends (...args: any[]) => Promise<any>>(session: {
+    span: (name: string, options?: SpanOptions) => FallomSpan;
+}, name: string, fn: T, options?: SpanOptions): T;
+
 /**
  * FallomSession - Session-scoped tracing for concurrent-safe operations.
  */
@@ -162,6 +248,24 @@ declare class FallomSession {
     constructor(options: SessionOptions);
     /** Get the session context. */
     getContext(): SessionContext;
+    /**
+     * Create a manual span for custom operations.
+     *
+     * Use for non-LLM operations like RAG retrieval, preprocessing, tool execution, etc.
+     * The span uses the session's context (configKey, sessionId, etc.).
+     *
+     * @example
+     * ```typescript
+     * const span = session.span("rag.retrieve");
+     * span.set({ "rag.query": userQuery, "rag.topK": 5 });
+     *
+     * const docs = await retrieveDocuments(userQuery);
+     * span.set({ "rag.documents.count": docs.length });
+     *
+     * span.end(); // Must call to send the span
+     * ```
+     */
+    span(name: string, options?: SpanOptions): FallomSpan;
     /**
      * Get model assignment for this session (A/B testing).
      */
@@ -233,15 +337,19 @@ declare function session(options: SessionOptions): FallomSession;
 
 type trace_FallomSession = FallomSession;
 declare const trace_FallomSession: typeof FallomSession;
+type trace_FallomSpan = FallomSpan;
+declare const trace_FallomSpan: typeof FallomSpan;
 type trace_SessionContext = SessionContext;
 type trace_SessionOptions = SessionOptions;
+type trace_SpanOptions = SpanOptions;
 type trace_TraceContext = TraceContext;
 type trace_TraceData = TraceData;
 type trace_WrapAISDKOptions = WrapAISDKOptions;
 declare const trace_session: typeof session;
 declare const trace_shutdown: typeof shutdown;
+declare const trace_wrapTraced: typeof wrapTraced;
 declare namespace trace {
-  export { trace_FallomSession as FallomSession, type trace_SessionContext as SessionContext, type trace_SessionOptions as SessionOptions, type trace_TraceContext as TraceContext, type trace_TraceData as TraceData, type trace_WrapAISDKOptions as WrapAISDKOptions, init$4 as init, trace_session as session, trace_shutdown as shutdown };
+  export { trace_FallomSession as FallomSession, trace_FallomSpan as FallomSpan, type trace_SessionContext as SessionContext, type trace_SessionOptions as SessionOptions, type trace_SpanOptions as SpanOptions, type trace_TraceContext as TraceContext, type trace_TraceData as TraceData, type trace_WrapAISDKOptions as WrapAISDKOptions, init$4 as init, trace_session as session, trace_shutdown as shutdown, trace_wrapTraced as wrapTraced };
 }
 
 /**
@@ -531,6 +639,12 @@ interface EvaluateOptions {
     /** List of metrics to run (built-in or custom). Default: all built-in metrics */
     metrics?: MetricInput[];
     judgeModel?: string;
+    /**
+     * Context to provide the LLM judge about the product/domain being evaluated.
+     * This helps the judge make better evaluations by understanding what features
+     * or capabilities are valid (e.g., won't mark valid features as hallucinations).
+     */
+    judgeContext?: string;
     name?: string;
     description?: string;
     verbose?: boolean;
@@ -566,7 +680,7 @@ declare const METRIC_PROMPTS: Record<MetricName, {
 /**
  * Build the G-Eval prompt for the LLM judge.
  */
-declare function buildGEvalPrompt(criteria: string, steps: string[], systemMessage: string | undefined, inputText: string, outputText: string): string;
+declare function buildGEvalPrompt(criteria: string, steps: string[], systemMessage: string | undefined, inputText: string, outputText: string, judgeContext?: string): string;
 /**
  * Result of running G-Eval on a single metric.
  */
@@ -600,6 +714,8 @@ interface RunGEvalOptions {
     traceSessionId?: string;
     /** Optional customer ID for tracing (e.g., organization ID) */
     traceCustomerId?: string;
+    /** Optional context to provide the judge about the product/domain being evaluated */
+    judgeContext?: string;
 }
 /**
  * Run G-Eval for a single metric using OpenRouter.
@@ -1114,4 +1230,4 @@ declare const _default: {
     session: typeof session;
 };
 
-export { type CompareModelsOptions, type DatasetItem, type EvalResult, type EvaluateOptions, FallomExporter, type FallomExporterOptions, FallomSession, type GEvalScore, type InitOptions, type MetricName, type PromptResult, type SessionContext, type SessionOptions, buildGEvalPrompt, calculateAggregateScores, clearMastraPrompt, _default as default, detectRegression, evals, init, models, prompts, runGEval, session, setMastraPrompt, setMastraPromptAB, trace };
+export { type CompareModelsOptions, type DatasetItem, type EvalResult, type EvaluateOptions, FallomExporter, type FallomExporterOptions, FallomSession, FallomSpan, type GEvalScore, type InitOptions, type MetricName, type PromptResult, type SessionContext, type SessionOptions, type SpanOptions, buildGEvalPrompt, calculateAggregateScores, clearMastraPrompt, _default as default, detectRegression, evals, init, models, prompts, runGEval, session, setMastraPrompt, setMastraPromptAB, trace, wrapTraced };

package/dist/index.d.ts

@@ -130,6 +130,92 @@ declare function init$4(options?: {
  */
 declare function shutdown(): Promise<void>;
 
+/**
+ * FallomSpan - Manual span for custom operations.
+ *
+ * Use for non-LLM operations like RAG retrieval, preprocessing, tool execution, etc.
+ *
+ * @example
+ * ```typescript
+ * const session = fallom.session({ configKey: "my-agent", sessionId });
+ *
+ * // Create a manual span
+ * const span = session.span("rag.retrieve");
+ * span.set({ "rag.query": userQuery, "rag.topK": 5 });
+ *
+ * const docs = await retrieveDocuments(userQuery);
+ * span.set({ "rag.documents.count": docs.length });
+ *
+ * span.end(); // Sends the span
+ * ```
+ */
+
+interface SpanOptions {
+    /** Parent span ID for nested spans */
+    parentSpanId?: string;
+    /** Trace ID to continue an existing trace */
+    traceId?: string;
+    /** Span kind (defaults to "custom") */
+    kind?: "custom" | "tool" | "retrieval" | "preprocessing" | "postprocessing";
+}
+declare class FallomSpan {
+    private name;
+    private ctx;
+    private attrs;
+    private startTime;
+    private ended;
+    private _status;
+    private _errorMessage?;
+    readonly spanId: string;
+    readonly traceId: string;
+    readonly parentSpanId?: string;
+    readonly kind: string;
+    constructor(name: string, ctx: SessionContext, options?: SpanOptions);
+    /**
+     * Set attributes on the span.
+     * Can be called multiple times - attributes are merged.
+     */
+    set(attributes: Record<string, unknown>): this;
+    /**
+     * Mark the span as errored.
+     */
+    setError(error: Error | string): this;
+    /**
+     * Get span context for creating child spans.
+     */
+    context(): {
+        traceId: string;
+        spanId: string;
+    };
+    /**
+     * End the span and send it.
+     * Must be called for the span to be recorded.
+     */
+    end(): void;
+}
+/**
+ * Wrap a function to automatically create a span around it.
+ * Similar to Braintrust's wrapTraced().
+ *
+ * @example
+ * ```typescript
+ * const fetchDocuments = wrapTraced(
+ *   session,
+ *   "rag.fetch",
+ *   async (query: string) => {
+ *     const docs = await vectorDb.search(query);
+ *     return docs;
+ *   }
+ * );
+ *
+ * // Function input/output automatically captured
+ * const docs = await fetchDocuments("user query");
+ * ```
+ */
+declare function wrapTraced<T extends (...args: any[]) => Promise<any>>(session: {
+    span: (name: string, options?: SpanOptions) => FallomSpan;
+}, name: string, fn: T, options?: SpanOptions): T;
+
 /**
  * FallomSession - Session-scoped tracing for concurrent-safe operations.
  */
@@ -162,6 +248,24 @@ declare class FallomSession {
     constructor(options: SessionOptions);
     /** Get the session context. */
     getContext(): SessionContext;
+    /**
+     * Create a manual span for custom operations.
+     *
+     * Use for non-LLM operations like RAG retrieval, preprocessing, tool execution, etc.
+     * The span uses the session's context (configKey, sessionId, etc.).
+     *
+     * @example
+     * ```typescript
+     * const span = session.span("rag.retrieve");
+     * span.set({ "rag.query": userQuery, "rag.topK": 5 });
+     *
+     * const docs = await retrieveDocuments(userQuery);
+     * span.set({ "rag.documents.count": docs.length });
+     *
+     * span.end(); // Must call to send the span
+     * ```
+     */
+    span(name: string, options?: SpanOptions): FallomSpan;
     /**
      * Get model assignment for this session (A/B testing).
      */
@@ -233,15 +337,19 @@ declare function session(options: SessionOptions): FallomSession;
 
 type trace_FallomSession = FallomSession;
 declare const trace_FallomSession: typeof FallomSession;
+type trace_FallomSpan = FallomSpan;
+declare const trace_FallomSpan: typeof FallomSpan;
 type trace_SessionContext = SessionContext;
 type trace_SessionOptions = SessionOptions;
+type trace_SpanOptions = SpanOptions;
 type trace_TraceContext = TraceContext;
 type trace_TraceData = TraceData;
 type trace_WrapAISDKOptions = WrapAISDKOptions;
 declare const trace_session: typeof session;
 declare const trace_shutdown: typeof shutdown;
+declare const trace_wrapTraced: typeof wrapTraced;
 declare namespace trace {
-  export { trace_FallomSession as FallomSession, type trace_SessionContext as SessionContext, type trace_SessionOptions as SessionOptions, type trace_TraceContext as TraceContext, type trace_TraceData as TraceData, type trace_WrapAISDKOptions as WrapAISDKOptions, init$4 as init, trace_session as session, trace_shutdown as shutdown };
+  export { trace_FallomSession as FallomSession, trace_FallomSpan as FallomSpan, type trace_SessionContext as SessionContext, type trace_SessionOptions as SessionOptions, type trace_SpanOptions as SpanOptions, type trace_TraceContext as TraceContext, type trace_TraceData as TraceData, type trace_WrapAISDKOptions as WrapAISDKOptions, init$4 as init, trace_session as session, trace_shutdown as shutdown, trace_wrapTraced as wrapTraced };
 }
 
 /**
@@ -531,6 +639,12 @@ interface EvaluateOptions {
     /** List of metrics to run (built-in or custom). Default: all built-in metrics */
     metrics?: MetricInput[];
     judgeModel?: string;
+    /**
+     * Context to provide the LLM judge about the product/domain being evaluated.
+     * This helps the judge make better evaluations by understanding what features
+     * or capabilities are valid (e.g., won't mark valid features as hallucinations).
+     */
+    judgeContext?: string;
     name?: string;
     description?: string;
     verbose?: boolean;
@@ -566,7 +680,7 @@ declare const METRIC_PROMPTS: Record<MetricName, {
 /**
  * Build the G-Eval prompt for the LLM judge.
  */
-declare function buildGEvalPrompt(criteria: string, steps: string[], systemMessage: string | undefined, inputText: string, outputText: string): string;
+declare function buildGEvalPrompt(criteria: string, steps: string[], systemMessage: string | undefined, inputText: string, outputText: string, judgeContext?: string): string;
 /**
  * Result of running G-Eval on a single metric.
  */
@@ -600,6 +714,8 @@ interface RunGEvalOptions {
     traceSessionId?: string;
     /** Optional customer ID for tracing (e.g., organization ID) */
     traceCustomerId?: string;
+    /** Optional context to provide the judge about the product/domain being evaluated */
+    judgeContext?: string;
 }
 /**
  * Run G-Eval for a single metric using OpenRouter.
@@ -1114,4 +1230,4 @@ declare const _default: {
     session: typeof session;
 };
 
-export { type CompareModelsOptions, type DatasetItem, type EvalResult, type EvaluateOptions, FallomExporter, type FallomExporterOptions, FallomSession, type GEvalScore, type InitOptions, type MetricName, type PromptResult, type SessionContext, type SessionOptions, buildGEvalPrompt, calculateAggregateScores, clearMastraPrompt, _default as default, detectRegression, evals, init, models, prompts, runGEval, session, setMastraPrompt, setMastraPromptAB, trace };
+export { type CompareModelsOptions, type DatasetItem, type EvalResult, type EvaluateOptions, FallomExporter, type FallomExporterOptions, FallomSession, FallomSpan, type GEvalScore, type InitOptions, type MetricName, type PromptResult, type SessionContext, type SessionOptions, type SpanOptions, buildGEvalPrompt, calculateAggregateScores, clearMastraPrompt, _default as default, detectRegression, evals, init, models, prompts, runGEval, session, setMastraPrompt, setMastraPromptAB, trace, wrapTraced };

package/dist/index.js

@@ -346,10 +346,15 @@ var init_types = __esm({
 });
 
 // src/evals/prompts.ts
-function buildGEvalPrompt(criteria, steps, systemMessage, inputText, outputText) {
+function buildGEvalPrompt(criteria, steps, systemMessage, inputText, outputText, judgeContext) {
   const stepsText = steps.map((s, i) => `${i + 1}. ${s}`).join("\n");
   return `You are an expert evaluator assessing LLM outputs using the G-Eval methodology.
+${judgeContext ? `
+## Important Context
+The following context provides background information about the product/domain being evaluated. Use this to inform your evaluation - for example, if the context mentions that certain features or capabilities exist, do not mark responses as hallucinations when they reference those features.
 
+${judgeContext}
+` : ""}
 ## Evaluation Criteria
 ${criteria}
 
@@ -388,7 +393,8 @@ async function runGEval(options) {
     openrouterKey,
     fallomApiKey,
     traceSessionId,
-    traceCustomerId
+    traceCustomerId,
+    judgeContext
   } = options;
   const apiKey4 = openrouterKey || process.env.OPENROUTER_API_KEY;
   if (!apiKey4) {
@@ -406,7 +412,8 @@ async function runGEval(options) {
     config.steps,
     systemMessage,
     inputText,
-    outputText
+    outputText,
+    judgeContext
   );
   const startTime = Date.now();
   const response = await fetch(
@@ -982,14 +989,15 @@ function init4(options = {}) {
   }
   _initialized = true;
 }
-async function runGEval2(metric, inputText, outputText, systemMessage, judgeModel) {
+async function runGEval2(metric, inputText, outputText, systemMessage, judgeModel, judgeContext) {
   const metricArg = isCustomMetric(metric) ? { name: metric.name, criteria: metric.criteria, steps: metric.steps } : metric;
   return runGEval({
     metric: metricArg,
     inputText,
     outputText,
     systemMessage,
-    judgeModel
+    judgeModel,
+    judgeContext
   });
 }
 async function resolveDataset(datasetInput) {
@@ -1040,6 +1048,7 @@ async function evaluate(options) {
     dataset: datasetInput,
     metrics = [...AVAILABLE_METRICS],
     judgeModel = DEFAULT_JUDGE_MODEL,
+    judgeContext,
     name,
     description,
     verbose = true,
@@ -1102,7 +1111,8 @@ async function evaluate(options) {
           item.input,
           item.output,
           item.systemMessage,
-          judgeModel
+          judgeModel,
+          judgeContext
         );
         const key = isCustomMetric(metric) ? metricName : metricName.replace(/_([a-z])/g, (_, c) => c.toUpperCase());
         result[key] = score;
@@ -1133,6 +1143,7 @@ async function compareModels(options) {
     models,
     metrics = [...AVAILABLE_METRICS],
     judgeModel = DEFAULT_JUDGE_MODEL,
+    judgeContext,
     includeProduction = true,
     modelKwargs = {},
     name,
@@ -1150,6 +1161,7 @@ async function compareModels(options) {
       dataset,
       metrics,
       judgeModel,
+      judgeContext,
       verbose,
       _skipUpload: true
     });
@@ -1206,7 +1218,8 @@ async function compareModels(options) {
               item.input,
               output,
               item.systemMessage,
-              judgeModel
+              judgeModel,
+              judgeContext
             );
             const key = isCustomMetric(metric) ? metricName : metricName.replace(/_([a-z])/g, (_, c) => c.toUpperCase());
             result[key] = score;
@@ -1378,6 +1391,7 @@ var index_exports = {};
 __export(index_exports, {
   FallomExporter: () => FallomExporter,
   FallomSession: () => FallomSession,
+  FallomSpan: () => FallomSpan,
   buildGEvalPrompt: () => buildGEvalPrompt,
   calculateAggregateScores: () => calculateAggregateScores,
   clearMastraPrompt: () => clearMastraPrompt,
@@ -1391,7 +1405,8 @@ __export(index_exports, {
   session: () => session,
   setMastraPrompt: () => setMastraPrompt,
   setMastraPromptAB: () => setMastraPromptAB,
-  trace: () => trace_exports
+  trace: () => trace_exports,
+  wrapTraced: () => wrapTraced
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -1399,9 +1414,11 @@ module.exports = __toCommonJS(index_exports);
 var trace_exports = {};
 __export(trace_exports, {
   FallomSession: () => FallomSession,
+  FallomSpan: () => FallomSpan,
   init: () => init,
   session: () => session,
-  shutdown: () => shutdown
+  shutdown: () => shutdown,
+  wrapTraced: () => wrapTraced
 });
 
 // src/trace/core.ts
@@ -2187,6 +2204,109 @@ function generateHexId(length) {
   return Array.from(bytes).map((b) => b.toString(16).padStart(2, "0")).join("");
 }
 
+// src/trace/span.ts
+var FallomSpan = class {
+  constructor(name, ctx, options = {}) {
+    this.name = name;
+    this.ctx = ctx;
+    this.attrs = {};
+    this.ended = false;
+    this._status = "OK";
+    this.spanId = generateHexId(16);
+    this.traceId = options.traceId || generateHexId(32);
+    this.parentSpanId = options.parentSpanId;
+    this.kind = options.kind || "custom";
+    this.startTime = Date.now();
+  }
+  /**
+   * Set attributes on the span.
+   * Can be called multiple times - attributes are merged.
+   */
+  set(attributes) {
+    if (this.ended) {
+      console.warn("[Fallom] Cannot set attributes on ended span");
+      return this;
+    }
+    Object.assign(this.attrs, attributes);
+    return this;
+  }
+  /**
+   * Mark the span as errored.
+   */
+  setError(error) {
+    this._status = "ERROR";
+    this._errorMessage = error instanceof Error ? error.message : error;
+    return this;
+  }
+  /**
+   * Get span context for creating child spans.
+   */
+  context() {
+    return {
+      traceId: this.traceId,
+      spanId: this.spanId
+    };
+  }
+  /**
+   * End the span and send it.
+   * Must be called for the span to be recorded.
+   */
+  end() {
+    if (this.ended) {
+      console.warn("[Fallom] Span already ended");
+      return;
+    }
+    this.ended = true;
+    if (!isInitialized()) {
+      return;
+    }
+    const endTime = Date.now();
+    sendTrace({
+      config_key: this.ctx.configKey,
+      session_id: this.ctx.sessionId,
+      customer_id: this.ctx.customerId,
+      metadata: this.ctx.metadata,
+      tags: this.ctx.tags,
+      trace_id: this.traceId,
+      span_id: this.spanId,
+      parent_span_id: this.parentSpanId,
+      name: this.name,
+      kind: this.kind,
+      start_time: new Date(this.startTime).toISOString(),
+      end_time: new Date(endTime).toISOString(),
+      duration_ms: endTime - this.startTime,
+      status: this._status,
+      error_message: this._errorMessage,
+      attributes: {
+        "fallom.sdk_version": "2",
+        "fallom.span_type": "manual",
+        ...this.attrs
+      }
+    }).catch(() => {
+    });
+  }
+};
+function wrapTraced(session2, name, fn, options = {}) {
+  return (async (...args) => {
+    const span = session2.span(name, options);
+    if (args.length === 1) {
+      span.set({ input: args[0] });
+    } else if (args.length > 1) {
+      span.set({ input: args });
+    }
+    try {
+      const result = await fn(...args);
+      span.set({ output: result });
+      span.end();
+      return result;
+    } catch (error) {
+      span.setError(error instanceof Error ? error : String(error));
+      span.end();
+      throw error;
+    }
+  });
+}
+
 // src/prompts.ts
 var prompts_exports = {};
 __export(prompts_exports, {
@@ -4056,6 +4176,26 @@ var FallomSession = class {
   getContext() {
     return { ...this.ctx };
   }
+  /**
+   * Create a manual span for custom operations.
+   *
+   * Use for non-LLM operations like RAG retrieval, preprocessing, tool execution, etc.
+   * The span uses the session's context (configKey, sessionId, etc.).
+   *
+   * @example
+   * ```typescript
+   * const span = session.span("rag.retrieve");
+   * span.set({ "rag.query": userQuery, "rag.topK": 5 });
+   *
+   * const docs = await retrieveDocuments(userQuery);
+   * span.set({ "rag.documents.count": docs.length });
+   *
+   * span.end(); // Must call to send the span
+   * ```
+   */
+  span(name, options) {
+    return new FallomSpan(name, this.ctx, options);
+  }
   /**
    * Get model assignment for this session (A/B testing).
    */
@@ -4294,7 +4434,7 @@ async function init5(options = {}) {
 }
 
 // src/mastra.ts
-var import_core13 = require("@opentelemetry/core");
+var import_core14 = require("@opentelemetry/core");
 var promptContext2 = {};
 function setMastraPrompt(promptKey, version) {
   promptContext2 = {
@@ -4344,7 +4484,7 @@ var FallomExporter = class {
    */
   export(spans, resultCallback) {
     if (spans.length === 0) {
-      resultCallback({ code: import_core13.ExportResultCode.SUCCESS });
+      resultCallback({ code: import_core14.ExportResultCode.SUCCESS });
       return;
     }
     this.log(`Exporting ${spans.length} spans...`);
@@ -4361,11 +4501,11 @@ var FallomExporter = class {
     }
     const exportPromise = this.sendSpans(spans).then(() => {
       this.log("Export successful");
-      resultCallback({ code: import_core13.ExportResultCode.SUCCESS });
+      resultCallback({ code: import_core14.ExportResultCode.SUCCESS });
     }).catch((error) => {
       console.error("[FallomExporter] Export failed:", error);
       resultCallback({
-        code: import_core13.ExportResultCode.FAILED,
+        code: import_core14.ExportResultCode.FAILED,
         error: error instanceof Error ? error : new Error(String(error))
       });
     });
@@ -4545,6 +4685,7 @@ var index_default = {
 0 && (module.exports = {
   FallomExporter,
   FallomSession,
+  FallomSpan,
   buildGEvalPrompt,
   calculateAggregateScores,
   clearMastraPrompt,
@@ -4557,5 +4698,6 @@ var index_default = {
   session,
   setMastraPrompt,
   setMastraPromptAB,
-  trace
+  trace,
+  wrapTraced
 });