autotel 7.1.0 → 7.3.0

package/README.md

@@ -143,20 +143,20 @@ Replace `NODE_OPTIONS` and 30+ lines of SDK boilerplate with `init()`, wrap func
 
 `autotel` has ~35 subpath exports. For most scenarios you only need one or two. Cloudflare Workers and other edge runtimes should use [`autotel-cloudflare`](../autotel-cloudflare) or [`autotel-edge`](../autotel-edge), not `autotel` (which expects Node).
 
-| Scenario                                           | Package                               | Import                                                                      |
-| -------------------------------------------------- | ------------------------------------- | --------------------------------------------------------------------------- |
-| Node service: init + spans                         | `autotel`                             | `import { init, trace, span, withTracing, span } from 'autotel'`            |
-| Node service: pino-style logger with trace context | `autotel`                             | `import { createLogger } from 'autotel/logger'`                             |
-| Node service: class decorators                     | `autotel`                             | `import { Trace, Span } from 'autotel/decorators'`                          |
-| Node service: Drizzle ORM spans                    | `autotel-drizzle`                     | `import { instrumentDrizzleClient } from 'autotel-drizzle'`                 |
-| Node service: testing assertions                   | `autotel`                             | `import { createTraceCollector } from 'autotel/testing'`                    |
-| Cloudflare Worker: fetch with spans                | `autotel-cloudflare`                  | `import { wrapModule, trace, span, withTracing } from 'autotel-cloudflare'` |
-| Cloudflare Worker: logs only, no `nodejs_compat`   | `autotel-cloudflare`                  | `import { createEdgeLogger } from 'autotel-cloudflare/logger'`              |
-| Cloudflare Worker: queue consumer                  | `autotel-cloudflare`                  | `import { wrapModule, getQueueLogger } from 'autotel-cloudflare'`           |
-| Cloudflare Worker: Durable Object                  | `autotel-cloudflare`                  | `import { wrapDurableObject } from 'autotel-cloudflare'`                    |
-| Hono on Workers                                    | `autotel-cloudflare` + `autotel-hono` | `wrapModule` from `autotel-cloudflare`, middleware from `autotel-hono`      |
-| Vercel Edge / Netlify Edge / Deno Deploy           | `autotel-edge`                        | `import { init, trace, span, withTracing } from 'autotel-edge'`             |
-| Edge runtime: logs only                            | `autotel-edge`                        | `import { createEdgeLogger } from 'autotel-edge/logger'`                    |
+| Scenario                                           | Package                               | Import                                                                         |
+| -------------------------------------------------- | ------------------------------------- | ------------------------------------------------------------------------------ |
+| Node service: init + spans                         | `autotel`                             | `import { init, trace, span, withTracing, span } from 'autotel'`               |
+| Node service: pino-style logger with trace context | `autotel`                             | `import { createLogger } from 'autotel/logger'`                                |
+| Node service: class decorators                     | `autotel`                             | `import { Trace, Span } from 'autotel/decorators'`                             |
+| Node service: Drizzle ORM spans                    | `autotel-drizzle`                     | `import { instrumentDrizzleClient } from 'autotel-drizzle'`                    |
+| Node service: testing assertions                   | `autotel`                             | `import { createTraceCollector, createMemoryExporter } from 'autotel/testing'` |
+| Cloudflare Worker: fetch with spans                | `autotel-cloudflare`                  | `import { wrapModule, trace, span, withTracing } from 'autotel-cloudflare'`    |
+| Cloudflare Worker: logs only, no `nodejs_compat`   | `autotel-cloudflare`                  | `import { createEdgeLogger } from 'autotel-cloudflare/logger'`                 |
+| Cloudflare Worker: queue consumer                  | `autotel-cloudflare`                  | `import { wrapModule, getQueueLogger } from 'autotel-cloudflare'`              |
+| Cloudflare Worker: Durable Object                  | `autotel-cloudflare`                  | `import { wrapDurableObject } from 'autotel-cloudflare'`                       |
+| Hono on Workers                                    | `autotel-cloudflare` + `autotel-hono` | `wrapModule` from `autotel-cloudflare`, middleware from `autotel-hono`         |
+| Vercel Edge / Netlify Edge / Deno Deploy           | `autotel-edge`                        | `import { init, trace, span, withTracing } from 'autotel-edge'`                |
+| Edge runtime: logs only                            | `autotel-edge`                        | `import { createEdgeLogger } from 'autotel-edge/logger'`                       |
 
 ## Quick Start
 
@@ -434,6 +434,8 @@ calling a model, this is why.
 
 Autotel defaults to production-ready adaptive sampling: a 10% baseline, with errors and slow requests kept automatically.
 
+A sampler configured here also governs the functions you wrap. `trace()`, `withTracing()` and `instrument()` resolve their sampler per call as `options.sampler ?? init()'s sampler ?? keep everything`, so `sampling: 'production'` applies to wrapped spans and not only to the SDK. Resolution is per call because a wrapper is usually created at module load, before `init()` has run. A sampler passed at the wrapper still wins.
+
 ### Preset Shorthand
 
 Use the `sampling` field on `init()` when you want the shortest path:
@@ -775,6 +777,31 @@ app.use((req, res, next) => {
 - Baggage values are strings (convert numbers/objects before setting)
 - Never put PII in baggage - it propagates in HTTP headers across services!
 
+### Getting the trace id, and a link to it
+
+`getActiveTraceContext()` reads the trace currently in scope, and
+`resolveTraceUrl()` turns its id into a link to wherever you view traces. Printing
+that link beside a slow request is the shortest path from "this was slow" to
+looking at why:
+
+```typescript
+import { getActiveTraceContext, resolveTraceUrl, span } from 'autotel';
+
+await span('GET /feed', async () => {
+  const traceId = getActiveTraceContext()?.traceId;
+  const url = resolveTraceUrl(
+    'http://localhost:4318/#trace={traceId}',
+    traceId,
+  );
+  console.log(`feed request ${url ?? traceId}`);
+});
+```
+
+`resolveTraceUrl(template, traceId)` substitutes `{traceId}`. Pass `undefined`
+as the template and it falls back to `OTEL_TRACE_URL_TEMPLATE`, so the same code
+points at devtools locally and at your vendor in production without a branch.
+It returns `undefined` when neither is set.
+
 ### Reusable Middleware Helpers
 
 - `withTracing(options)` : create a preconfigured wrapper (service name, default attributes, skip rules).

package/dist/analysis.cjs

@@ -81,6 +81,23 @@ function compareCohorts(options) {
 	results.sort((a, b) => Math.abs(b.difference) - Math.abs(a.difference) || b.difference - a.difference || a.field.localeCompare(b.field) || a.value.localeCompare(b.value));
 	return results.slice(0, limit);
 }
+/**
+* Label a numeric value with the range it falls in.
+*
+* `compareCohorts` skips a field whose values never repeat, which is what a
+* raw duration or byte count does. Bucketing at instrumentation time turns
+* such a field into one that can describe a cohort.
+*/
+function bucket(value, boundaries) {
+	if (!Number.isFinite(value) || boundaries.length === 0) return "unknown";
+	const ordered = [...boundaries].sort((a, b) => a - b);
+	for (const [index, boundary] of ordered.entries()) if (value < boundary) {
+		const lower = ordered[index - 1];
+		return lower === void 0 ? `<${boundary}` : `${lower}-${boundary}`;
+	}
+	return `>=${ordered.at(-1)}`;
+}
 
 //#endregion
+exports.bucket = bucket;
 exports.compareCohorts = compareCohorts;

package/dist/analysis.d.cts

@@ -82,5 +82,13 @@ interface CompareCohortsOptions {
  * @returns Differences sorted by absolute strength, strongest first
  */
 declare function compareCohorts(options: CompareCohortsOptions): CohortDifference[];
+/**
+ * Label a numeric value with the range it falls in.
+ *
+ * `compareCohorts` skips a field whose values never repeat, which is what a
+ * raw duration or byte count does. Bucketing at instrumentation time turns
+ * such a field into one that can describe a cohort.
+ */
+declare function bucket(value: number, boundaries: readonly number[]): string;
 //#endregion
-export { AnalysisEvent, CohortDifference, CompareCohortsOptions, compareCohorts };
+export { AnalysisEvent, CohortDifference, CompareCohortsOptions, bucket, compareCohorts };

package/dist/analysis.d.ts

@@ -82,5 +82,13 @@ interface CompareCohortsOptions {
  * @returns Differences sorted by absolute strength, strongest first
  */
 declare function compareCohorts(options: CompareCohortsOptions): CohortDifference[];
+/**
+ * Label a numeric value with the range it falls in.
+ *
+ * `compareCohorts` skips a field whose values never repeat, which is what a
+ * raw duration or byte count does. Bucketing at instrumentation time turns
+ * such a field into one that can describe a cohort.
+ */
+declare function bucket(value: number, boundaries: readonly number[]): string;
 //#endregion
-export { AnalysisEvent, CohortDifference, CompareCohortsOptions, compareCohorts };
+export { AnalysisEvent, CohortDifference, CompareCohortsOptions, bucket, compareCohorts };

package/dist/analysis.js

@@ -79,6 +79,22 @@ function compareCohorts(options) {
 	results.sort((a, b) => Math.abs(b.difference) - Math.abs(a.difference) || b.difference - a.difference || a.field.localeCompare(b.field) || a.value.localeCompare(b.value));
 	return results.slice(0, limit);
 }
+/**
+* Label a numeric value with the range it falls in.
+*
+* `compareCohorts` skips a field whose values never repeat, which is what a
+* raw duration or byte count does. Bucketing at instrumentation time turns
+* such a field into one that can describe a cohort.
+*/
+function bucket(value, boundaries) {
+	if (!Number.isFinite(value) || boundaries.length === 0) return "unknown";
+	const ordered = [...boundaries].sort((a, b) => a - b);
+	for (const [index, boundary] of ordered.entries()) if (value < boundary) {
+		const lower = ordered[index - 1];
+		return lower === void 0 ? `<${boundary}` : `${lower}-${boundary}`;
+	}
+	return `>=${ordered.at(-1)}`;
+}
 
 //#endregion
-export { compareCohorts };
+export { bucket, compareCohorts };

package/dist/auto.cjs

@@ -1,6 +1,5 @@
 const require_rolldown_runtime = require('./rolldown-runtime-C_NdSu1c.cjs');
-const require_init = require('./init-C5aHRqi2.cjs');
-const require_yaml_config = require('./yaml-config.cjs');
+const require_init = require('./init-DmM7dprX.cjs');
 let node_module = require("node:module");
 node_module = require_rolldown_runtime.__toESM(node_module, 1);
 let import_in_the_middle = require("import-in-the-middle");
@@ -42,7 +41,7 @@ let import_in_the_middle = require("import-in-the-middle");
 */
 const { registerOptions } = (0, import_in_the_middle.createAddHookMessageChannel)();
 node_module.register("import-in-the-middle/hook.mjs", require("url").pathToFileURL(__filename).href, registerOptions);
-const yamlConfig = require_yaml_config.loadYamlConfig();
+const yamlConfig = require_init.loadYamlConfig();
 const autoInstrumentationsEnv = process.env.AUTOTEL_INTEGRATIONS;
 const autoInstrumentations = autoInstrumentationsEnv === "true" ? true : autoInstrumentationsEnv ? autoInstrumentationsEnv.split(",").map((s) => s.trim()) : yamlConfig?.autoInstrumentations ?? ["http", "express"];
 require_init.init({

package/dist/auto.js

@@ -1,5 +1,4 @@
-import { i as init } from "./init-DDyaLzeg.js";
-import { loadYamlConfig } from "./yaml-config.js";
+import { _ as loadYamlConfig, a as init } from "./init-CsKK9vZO.js";
 import * as nodeModule from "node:module";
 import { createAddHookMessageChannel } from "import-in-the-middle";
 

package/dist/correlation-id.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
-const require_init = require('./init-C5aHRqi2.cjs');
+const require_init = require('./init-DmM7dprX.cjs');
 
 exports.CORRELATION_ID_BAGGAGE_KEY = require_init.CORRELATION_ID_BAGGAGE_KEY;
 exports.generateCorrelationId = require_init.generateCorrelationId;

package/dist/correlation-id.js

@@ -1,3 +1,3 @@
-import { C as setCorrelationId, S as runWithCorrelationId, _ as CORRELATION_ID_BAGGAGE_KEY, b as getCorrelationStorage, v as generateCorrelationId, w as setCorrelationIdInBaggage, x as getOrCreateCorrelationId, y as getCorrelationId } from "./init-DDyaLzeg.js";
+import { $ as getCorrelationStorage, Q as getCorrelationId, X as CORRELATION_ID_BAGGAGE_KEY, Z as generateCorrelationId, et as getOrCreateCorrelationId, nt as setCorrelationId, rt as setCorrelationIdInBaggage, tt as runWithCorrelationId } from "./init-CsKK9vZO.js";
 
 export { CORRELATION_ID_BAGGAGE_KEY, generateCorrelationId, getCorrelationId, getCorrelationStorage, getOrCreateCorrelationId, runWithCorrelationId, setCorrelationId, setCorrelationIdInBaggage };

package/dist/decorators.cjs

@@ -1,7 +1,7 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 const require_values = require('./values-DApyq53y.cjs');
 const require_config = require('./config.cjs');
-const require_init = require('./init-C5aHRqi2.cjs');
+const require_init = require('./init-DmM7dprX.cjs');
 let _opentelemetry_api = require("@opentelemetry/api");
 
 //#region src/decorators.ts

package/dist/decorators.d.cts

@@ -1,5 +1,5 @@
 import { n as TraceContext } from "./trace-context-C1Cn1q0y.cjs";
-import { m as TracingOptions } from "./functional-CJn2zY4o.cjs";
+import { m as TracingOptions } from "./functional-CgWcRXGX.cjs";
 //#region src/decorators.d.ts
 /**
  * Options for @Trace method decorator

package/dist/decorators.d.ts

@@ -1,5 +1,5 @@
 import { n as TraceContext } from "./trace-context-f89xPJbK.js";
-import { m as TracingOptions } from "./functional-DXy1cHFF.js";
+import { m as TracingOptions } from "./functional-RFFpQzAV.js";
 //#region src/decorators.d.ts
 /**
  * Options for @Trace method decorator

package/dist/decorators.js

@@ -1,6 +1,6 @@
 import { y as splitNameOrOptions } from "./values-BenyvWKz.js";
 import { getConfig } from "./config.js";
-import { T as createTraceContext } from "./init-DDyaLzeg.js";
+import { U as createTraceContext } from "./init-CsKK9vZO.js";
 import { SpanStatusCode } from "@opentelemetry/api";
 
 //#region src/decorators.ts

package/dist/event.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
-const require_init = require('./init-C5aHRqi2.cjs');
+const require_init = require('./init-DmM7dprX.cjs');
 
 exports.Event = require_init.Event;
 exports.getEvents = require_init.getEvents;

package/dist/event.js

@@ -1,3 +1,3 @@
-import { d as getEvents, f as resetEvents, u as Event } from "./init-DDyaLzeg.js";
+import { d as Event, f as getEvents, p as resetEvents } from "./init-CsKK9vZO.js";
 
 export { Event, getEvents, resetEvents };

package/dist/{functional-Cub1ku-C.js → functional-C4js2vTM.js}

@@ -1,14 +1,43 @@
 import { l as asString, m as isDevelopment } from "./values-BenyvWKz.js";
 import { getForceFlushableProvider } from "./tracer-provider.js";
 import { getConfig } from "./config.js";
-import { AUTOTEL_SAMPLING_RATE, AUTOTEL_SAMPLING_TAIL_EVALUATED, AUTOTEL_SAMPLING_TAIL_KEEP, AlwaysSampler } from "./sampling.js";
-import { D as getActiveContextWithBaggage, O as getContextStorage, T as createTraceContext, h as getEventQueue, k as hasExplicitSpanStatus, m as runInOperationContext, r as getSdk, t as getConfig$1 } from "./init-DDyaLzeg.js";
+import { C as AUTOTEL_SAMPLING_RATE, D as AlwaysSampler, G as getActiveContextWithBaggage, I as debugCaptureRequested, J as getEventQueue, K as getContextStorage, T as AUTOTEL_SAMPLING_TAIL_KEEP, U as createTraceContext, h as runInOperationContext, i as getSdk, n as getConfiguredSampler, q as hasExplicitSpanStatus, t as getConfig$1, w as AUTOTEL_SAMPLING_TAIL_EVALUATED, z as isForceKept } from "./init-CsKK9vZO.js";
 import { evidenceAttribute } from "./evidence.js";
 import { setSpanName } from "./trace-helpers.js";
 import { SpanStatusCode, context, propagation, trace } from "@opentelemetry/api";
 import * as nodeFs from "node:fs";
 import * as nodeUrl from "node:url";
 
+//#region src/is-thenable.ts
+/**
+* Convert anything `await` would wait on into a native Promise, or return
+* `undefined` for a synchronous value.
+*
+* `instanceof Promise` is not enough. ORM query builders (drizzle, knex,
+* Prisma, Sequelize) are lazy thenables that are not native Promises, and a
+* promise created in another realm fails the check too. Treating those as
+* synchronous ends the span before the work runs, which produces a short
+* parent span with its real children orphaned outside it.
+*/
+function promiseFromThenable(value) {
+	if (typeof value !== "object" && typeof value !== "function" || value === null) return;
+	let then;
+	try {
+		then = value.then;
+	} catch (error) {
+		return Promise.reject(error);
+	}
+	if (typeof then !== "function") return void 0;
+	return new Promise((resolve, reject) => {
+		try {
+			then.call(value, resolve, reject);
+		} catch (error) {
+			reject(error);
+		}
+	});
+}
+
+//#endregion
 //#region src/variable-name-inference.ts
 /**
 * Variable Name Inference Utility
@@ -314,9 +343,11 @@ function runWithTraceContextStorage(fn) {
 	if (storage.getStore()) return fn();
 	return storage.run({ value: context.active() }, fn);
 }
+/** Used when neither the wrapper nor `init()` chose a sampler. */
+const FALLBACK_SAMPLER = new AlwaysSampler();
 function wrapWithTracingSync(fnFactory, options, variableName) {
 	const { tracer, meter } = getConfig();
-	const sampler = options.sampler || new AlwaysSampler();
+	const resolveSampler = () => options.sampler ?? getConfiguredSampler() ?? FALLBACK_SAMPLER;
 	const spanName = getSpanName(options, fnFactory, variableName);
 	const callCounter = options.withMetrics ? meter.createCounter(`${spanName}.calls`, {
 		description: `Call count for ${spanName}`,
@@ -332,10 +363,14 @@ function wrapWithTracingSync(fnFactory, options, variableName) {
 			args,
 			metadata: {}
 		};
+		const sampler = resolveSampler();
 		const shouldSample = sampler.shouldSample(samplingContext);
 		const sampleRate = sampler.sampleRate?.(samplingContext);
 		const needsTailSampling = sampler.needsTailSampling?.() ?? false;
-		if (!shouldSample && !needsTailSampling) return fnFactory(createDummyCtx()).call(this, ...args);
+		if (!shouldSample && !needsTailSampling) {
+			const unsampled = fnFactory(createDummyCtx()).call(this, ...args);
+			return promiseFromThenable(unsampled) ?? unsampled;
+		}
 		const startTime = performance.now();
 		const flushRootTelemetry = createRootTelemetryFlusher(options, options.startNewRoot || trace.getActiveSpan() === void 0);
 		const spanOptions = {};
@@ -351,7 +386,7 @@ function wrapWithTracingSync(fnFactory, options, variableName) {
 				...options.attributesFromArgs?.(args)
 			};
 			const handleTailSampling = (success, duration, error) => {
-				if (needsTailSampling && sampler.shouldKeepTrace) {
+				if (needsTailSampling && sampler.shouldKeepTrace && !isForceKept(span) && !debugCaptureRequested()) {
 					shouldKeepSpan = sampler.shouldKeepTrace(samplingContext, {
 						success,
 						duration,
@@ -406,7 +441,8 @@ function wrapWithTracingSync(fnFactory, options, variableName) {
 					status: "started"
 				});
 				const result = context.with(trace.setSpan(getActiveContextWithBaggage(), span), () => fn.call(this, ...args));
-				if (result instanceof Promise) return result.then(async (value) => {
+				const resultPromise = promiseFromThenable(result);
+				if (resultPromise) return resultPromise.then(async (value) => {
 					const completed = onSuccess(value);
 					await flushRootTelemetry();
 					return completed;
@@ -656,7 +692,8 @@ function span(nameOrOptions, fn) {
 			try {
 				if (attributes) for (const [key, value] of Object.entries(attributes)) span.setAttribute(key, value);
 				const result = fn(span);
-				if (result instanceof Promise) return result.then((resolved) => {
+				const resultPromise = promiseFromThenable(result);
+				if (resultPromise) return resultPromise.then((resolved) => {
 					span.setStatus({ code: SpanStatusCode.OK });
 					span.end();
 					return resolved;

package/dist/{functional-CJn2zY4o.d.cts → functional-CgWcRXGX.d.cts}

@@ -1,5 +1,5 @@
 import { n as TraceContext } from "./trace-context-C1Cn1q0y.cjs";
-import { p as Sampler } from "./sampling-CmRdnXUi.cjs";
+import { m as Sampler } from "./sampling-Er217Mge.cjs";
 import { Attributes, Span, SpanKind } from "@opentelemetry/api";
 //#region src/functional-wrapper.d.ts
 type WrappedFunction<TArgs extends unknown[], TReturn> = (...args: TArgs) => TReturn | Promise<TReturn>;
@@ -192,7 +192,9 @@ declare const ctx: TraceContext;
  * })
  * ```
  */
+declare function traceImpl<TArgs extends unknown[], TReturn>(fn: (...args: TArgs) => PromiseLike<TReturn>): (...args: TArgs) => Promise<TReturn>;
 declare function traceImpl<TArgs extends unknown[], TReturn>(fn: (...args: TArgs) => TReturn): (...args: TArgs) => TReturn;
+declare function traceImpl<TArgs extends unknown[], TReturn>(name: string, fn: (...args: TArgs) => PromiseLike<TReturn>): (...args: TArgs) => Promise<TReturn>;
 declare function traceImpl<TArgs extends unknown[], TReturn>(name: string, fn: (...args: TArgs) => TReturn): (...args: TArgs) => TReturn;
 declare function traceImpl<TArgs extends unknown[], TReturn>(options: TracingOptions<TArgs, TReturn>, fn: (...args: TArgs) => TReturn): (...args: TArgs) => TReturn;
 declare function traceImpl(name: string): <TArgs extends unknown[], TReturn>(fn: (...args: TArgs) => TReturn) => (...args: TArgs) => TReturn;
@@ -258,7 +260,7 @@ declare const trace$1: typeof traceImpl & {
  * )(ctx => async (data) => { })
  * ```
  */
-declare function withTracing<TCfgArgs extends unknown[] = unknown[], TCfgReturn = unknown>(options?: TracingOptions<TCfgArgs, TCfgReturn>): <TArgs extends TCfgArgs, TReturn extends TCfgReturn>(fnFactory: (ctx: TraceContext) => (...args: TArgs) => TReturn | Promise<TReturn>) => WrappedFunction<TArgs, TReturn>;
+declare function withTracing<TCfgArgs extends unknown[] = unknown[], TCfgReturn = unknown>(options?: TracingOptions<TCfgArgs, TCfgReturn>): <TArgs extends TCfgArgs, TReturn extends TCfgReturn>(fnFactory: (ctx: TraceContext) => (...args: TArgs) => TReturn | PromiseLike<TReturn>) => WrappedFunction<TArgs, TReturn>;
 /**
  * Approach 3: instrument() - Batch auto-instrumentation
  *
@@ -309,8 +311,13 @@ declare function withTracing<TCfgArgs extends unknown[] = unknown[], TCfgReturn
  * })
  * ```
  */
+type InstrumentedFunction<TFunction extends AnyInstrumentable> = TFunction extends ((...args: infer TArgs) => infer TReturn) ? (...args: TArgs) => TReturn extends PromiseLike<infer TValue> ? Promise<TValue> : TReturn : never;
+type InstrumentedFunctions<T extends Record<string, AnyInstrumentable>> = { [TKey in keyof T]: InstrumentedFunction<T[TKey]>; };
+declare function instrument<TFunction extends AnyInstrumentable>(options: SingleInstrumentOptions<TFunction> & {
+  fn: TFunction & ((...args: Parameters<TFunction>) => PromiseLike<unknown>);
+}): (...args: Parameters<TFunction>) => Promise<Awaited<ReturnType<TFunction>>>;
 declare function instrument<TFunction extends AnyInstrumentable>(options: SingleInstrumentOptions<TFunction>): TFunction;
-declare function instrument<T extends Record<string, AnyInstrumentable>>(options: InstrumentOptions<T>): T;
+declare function instrument<T extends Record<string, AnyInstrumentable>>(options: InstrumentOptions<T>): InstrumentedFunctions<T>;
 /**
  * Options for span() function
  */
@@ -352,10 +359,10 @@ interface SpanOptions {
  * })
  * ```
  */
+declare function span<T = unknown>(name: string, fn: (span: Span) => PromiseLike<T>): Promise<T>;
+declare function span<T = unknown>(options: SpanOptions, fn: (span: Span) => PromiseLike<T>): Promise<T>;
 declare function span<T = unknown>(name: string, fn: (span: Span) => T): T;
-declare function span<T = unknown>(name: string, fn: (span: Span) => Promise<T>): Promise<T>;
 declare function span<T = unknown>(options: SpanOptions, fn: (span: Span) => T): T;
-declare function span<T = unknown>(options: SpanOptions, fn: (span: Span) => Promise<T>): Promise<T>;
 /**
  * Options for withNewContext() function
  */

package/dist/{functional-BhhP4Er_.cjs → functional-DHlD7V0x.cjs}

@@ -2,8 +2,7 @@ const require_rolldown_runtime = require('./rolldown-runtime-C_NdSu1c.cjs');
 const require_values = require('./values-DApyq53y.cjs');
 const require_tracer_provider = require('./tracer-provider.cjs');
 const require_config = require('./config.cjs');
-const require_sampling = require('./sampling.cjs');
-const require_init = require('./init-C5aHRqi2.cjs');
+const require_init = require('./init-DmM7dprX.cjs');
 const require_evidence = require('./evidence.cjs');
 const require_trace_helpers = require('./trace-helpers.cjs');
 let _opentelemetry_api = require("@opentelemetry/api");
@@ -12,6 +11,36 @@ node_fs = require_rolldown_runtime.__toESM(node_fs, 1);
 let node_url = require("node:url");
 node_url = require_rolldown_runtime.__toESM(node_url, 1);
 
+//#region src/is-thenable.ts
+/**
+* Convert anything `await` would wait on into a native Promise, or return
+* `undefined` for a synchronous value.
+*
+* `instanceof Promise` is not enough. ORM query builders (drizzle, knex,
+* Prisma, Sequelize) are lazy thenables that are not native Promises, and a
+* promise created in another realm fails the check too. Treating those as
+* synchronous ends the span before the work runs, which produces a short
+* parent span with its real children orphaned outside it.
+*/
+function promiseFromThenable(value) {
+	if (typeof value !== "object" && typeof value !== "function" || value === null) return;
+	let then;
+	try {
+		then = value.then;
+	} catch (error) {
+		return Promise.reject(error);
+	}
+	if (typeof then !== "function") return void 0;
+	return new Promise((resolve, reject) => {
+		try {
+			then.call(value, resolve, reject);
+		} catch (error) {
+			reject(error);
+		}
+	});
+}
+
+//#endregion
 //#region src/variable-name-inference.ts
 /**
 * Variable Name Inference Utility
@@ -317,9 +346,11 @@ function runWithTraceContextStorage(fn) {
 	if (storage.getStore()) return fn();
 	return storage.run({ value: _opentelemetry_api.context.active() }, fn);
 }
+/** Used when neither the wrapper nor `init()` chose a sampler. */
+const FALLBACK_SAMPLER = new require_init.AlwaysSampler();
 function wrapWithTracingSync(fnFactory, options, variableName) {
 	const { tracer, meter } = require_config.getConfig();
-	const sampler = options.sampler || new require_sampling.AlwaysSampler();
+	const resolveSampler = () => options.sampler ?? require_init.getConfiguredSampler() ?? FALLBACK_SAMPLER;
 	const spanName = getSpanName(options, fnFactory, variableName);
 	const callCounter = options.withMetrics ? meter.createCounter(`${spanName}.calls`, {
 		description: `Call count for ${spanName}`,
@@ -335,10 +366,14 @@ function wrapWithTracingSync(fnFactory, options, variableName) {
 			args,
 			metadata: {}
 		};
+		const sampler = resolveSampler();
 		const shouldSample = sampler.shouldSample(samplingContext);
 		const sampleRate = sampler.sampleRate?.(samplingContext);
 		const needsTailSampling = sampler.needsTailSampling?.() ?? false;
-		if (!shouldSample && !needsTailSampling) return fnFactory(createDummyCtx()).call(this, ...args);
+		if (!shouldSample && !needsTailSampling) {
+			const unsampled = fnFactory(createDummyCtx()).call(this, ...args);
+			return promiseFromThenable(unsampled) ?? unsampled;
+		}
 		const startTime = performance.now();
 		const flushRootTelemetry = createRootTelemetryFlusher(options, options.startNewRoot || _opentelemetry_api.trace.getActiveSpan() === void 0);
 		const spanOptions = {};
@@ -347,21 +382,21 @@ function wrapWithTracingSync(fnFactory, options, variableName) {
 		return tracer.startActiveSpan(spanName, spanOptions, require_init.getActiveContextWithBaggage(), (span) => require_init.runInOperationContext(spanName, () => runWithTraceContextStorage(() => {
 			let shouldKeepSpan = true;
 			require_trace_helpers.setSpanName(span, spanName);
-			if (sampleRate !== void 0 && sampleRate > 1) span.setAttribute(require_sampling.AUTOTEL_SAMPLING_RATE, sampleRate);
+			if (sampleRate !== void 0 && sampleRate > 1) span.setAttribute(require_init.AUTOTEL_SAMPLING_RATE, sampleRate);
 			const fn = fnFactory(require_init.createTraceContext(span));
 			const argsAttributes = {
 				...captureInputAttrs(args, options.captureInput),
 				...options.attributesFromArgs?.(args)
 			};
 			const handleTailSampling = (success, duration, error) => {
-				if (needsTailSampling && sampler.shouldKeepTrace) {
+				if (needsTailSampling && sampler.shouldKeepTrace && !require_init.isForceKept(span) && !require_init.debugCaptureRequested()) {
 					shouldKeepSpan = sampler.shouldKeepTrace(samplingContext, {
 						success,
 						duration,
 						error
 					});
-					span.setAttribute(require_sampling.AUTOTEL_SAMPLING_TAIL_KEEP, shouldKeepSpan);
-					span.setAttribute(require_sampling.AUTOTEL_SAMPLING_TAIL_EVALUATED, true);
+					span.setAttribute(require_init.AUTOTEL_SAMPLING_TAIL_KEEP, shouldKeepSpan);
+					span.setAttribute(require_init.AUTOTEL_SAMPLING_TAIL_EVALUATED, true);
 				}
 			};
 			const onSuccess = (result) => {
@@ -409,7 +444,8 @@ function wrapWithTracingSync(fnFactory, options, variableName) {
 					status: "started"
 				});
 				const result = _opentelemetry_api.context.with(_opentelemetry_api.trace.setSpan(require_init.getActiveContextWithBaggage(), span), () => fn.call(this, ...args));
-				if (result instanceof Promise) return result.then(async (value) => {
+				const resultPromise = promiseFromThenable(result);
+				if (resultPromise) return resultPromise.then(async (value) => {
 					const completed = onSuccess(value);
 					await flushRootTelemetry();
 					return completed;
@@ -659,7 +695,8 @@ function span(nameOrOptions, fn) {
 			try {
 				if (attributes) for (const [key, value] of Object.entries(attributes)) span.setAttribute(key, value);
 				const result = fn(span);
-				if (result instanceof Promise) return result.then((resolved) => {
+				const resultPromise = promiseFromThenable(result);
+				if (resultPromise) return resultPromise.then((resolved) => {
 					span.setStatus({ code: _opentelemetry_api.SpanStatusCode.OK });
 					span.end();
 					return resolved;

package/dist/{functional-DXy1cHFF.d.ts → functional-RFFpQzAV.d.ts}

@@ -1,5 +1,5 @@
 import { n as TraceContext } from "./trace-context-f89xPJbK.js";
-import { p as Sampler } from "./sampling-CM5S2Lp3.js";
+import { m as Sampler } from "./sampling-DBng4Foa.js";
 import { Attributes, Span, SpanKind } from "@opentelemetry/api";
 //#region src/functional-wrapper.d.ts
 type WrappedFunction<TArgs extends unknown[], TReturn> = (...args: TArgs) => TReturn | Promise<TReturn>;
@@ -192,7 +192,9 @@ declare const ctx: TraceContext;
  * })
  * ```
  */
+declare function traceImpl<TArgs extends unknown[], TReturn>(fn: (...args: TArgs) => PromiseLike<TReturn>): (...args: TArgs) => Promise<TReturn>;
 declare function traceImpl<TArgs extends unknown[], TReturn>(fn: (...args: TArgs) => TReturn): (...args: TArgs) => TReturn;
+declare function traceImpl<TArgs extends unknown[], TReturn>(name: string, fn: (...args: TArgs) => PromiseLike<TReturn>): (...args: TArgs) => Promise<TReturn>;
 declare function traceImpl<TArgs extends unknown[], TReturn>(name: string, fn: (...args: TArgs) => TReturn): (...args: TArgs) => TReturn;
 declare function traceImpl<TArgs extends unknown[], TReturn>(options: TracingOptions<TArgs, TReturn>, fn: (...args: TArgs) => TReturn): (...args: TArgs) => TReturn;
 declare function traceImpl(name: string): <TArgs extends unknown[], TReturn>(fn: (...args: TArgs) => TReturn) => (...args: TArgs) => TReturn;
@@ -258,7 +260,7 @@ declare const trace$1: typeof traceImpl & {
  * )(ctx => async (data) => { })
  * ```
  */
-declare function withTracing<TCfgArgs extends unknown[] = unknown[], TCfgReturn = unknown>(options?: TracingOptions<TCfgArgs, TCfgReturn>): <TArgs extends TCfgArgs, TReturn extends TCfgReturn>(fnFactory: (ctx: TraceContext) => (...args: TArgs) => TReturn | Promise<TReturn>) => WrappedFunction<TArgs, TReturn>;
+declare function withTracing<TCfgArgs extends unknown[] = unknown[], TCfgReturn = unknown>(options?: TracingOptions<TCfgArgs, TCfgReturn>): <TArgs extends TCfgArgs, TReturn extends TCfgReturn>(fnFactory: (ctx: TraceContext) => (...args: TArgs) => TReturn | PromiseLike<TReturn>) => WrappedFunction<TArgs, TReturn>;
 /**
  * Approach 3: instrument() - Batch auto-instrumentation
  *
@@ -309,8 +311,13 @@ declare function withTracing<TCfgArgs extends unknown[] = unknown[], TCfgReturn
  * })
  * ```
  */
+type InstrumentedFunction<TFunction extends AnyInstrumentable> = TFunction extends ((...args: infer TArgs) => infer TReturn) ? (...args: TArgs) => TReturn extends PromiseLike<infer TValue> ? Promise<TValue> : TReturn : never;
+type InstrumentedFunctions<T extends Record<string, AnyInstrumentable>> = { [TKey in keyof T]: InstrumentedFunction<T[TKey]>; };
+declare function instrument<TFunction extends AnyInstrumentable>(options: SingleInstrumentOptions<TFunction> & {
+  fn: TFunction & ((...args: Parameters<TFunction>) => PromiseLike<unknown>);
+}): (...args: Parameters<TFunction>) => Promise<Awaited<ReturnType<TFunction>>>;
 declare function instrument<TFunction extends AnyInstrumentable>(options: SingleInstrumentOptions<TFunction>): TFunction;
-declare function instrument<T extends Record<string, AnyInstrumentable>>(options: InstrumentOptions<T>): T;
+declare function instrument<T extends Record<string, AnyInstrumentable>>(options: InstrumentOptions<T>): InstrumentedFunctions<T>;
 /**
  * Options for span() function
  */
@@ -352,10 +359,10 @@ interface SpanOptions {
  * })
  * ```
  */
+declare function span<T = unknown>(name: string, fn: (span: Span) => PromiseLike<T>): Promise<T>;
+declare function span<T = unknown>(options: SpanOptions, fn: (span: Span) => PromiseLike<T>): Promise<T>;
 declare function span<T = unknown>(name: string, fn: (span: Span) => T): T;
-declare function span<T = unknown>(name: string, fn: (span: Span) => Promise<T>): Promise<T>;
 declare function span<T = unknown>(options: SpanOptions, fn: (span: Span) => T): T;
-declare function span<T = unknown>(options: SpanOptions, fn: (span: Span) => Promise<T>): Promise<T>;
 /**
  * Options for withNewContext() function
  */

package/dist/functional.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
-const require_functional = require('./functional-BhhP4Er_.cjs');
+const require_functional = require('./functional-DHlD7V0x.cjs');
 
 exports.ctx = require_functional.ctx;
 exports.getActiveTraceContext = require_functional.getActiveTraceContext;

package/dist/functional.d.cts

@@ -1,3 +1,3 @@
 import { n as TraceContext } from "./trace-context-C1Cn1q0y.cjs";
-import { a as getActiveTraceContext, c as trace, d as withTracing, f as InstrumentOptions, i as ctx, l as withBaggage, m as TracingOptions, n as WithBaggageOptions, o as instrument, p as SingleInstrumentOptions, r as WithNewContextOptions, s as span, t as SpanOptions, u as withNewContext } from "./functional-CJn2zY4o.cjs";
+import { a as getActiveTraceContext, c as trace, d as withTracing, f as InstrumentOptions, i as ctx, l as withBaggage, m as TracingOptions, n as WithBaggageOptions, o as instrument, p as SingleInstrumentOptions, r as WithNewContextOptions, s as span, t as SpanOptions, u as withNewContext } from "./functional-CgWcRXGX.cjs";
 export { type InstrumentOptions, type SingleInstrumentOptions, SpanOptions, type TraceContext, type TracingOptions, WithBaggageOptions, WithNewContextOptions, ctx, getActiveTraceContext, instrument, span, trace, withBaggage, withNewContext, withTracing };

package/dist/functional.d.ts

@@ -1,3 +1,3 @@
 import { n as TraceContext } from "./trace-context-f89xPJbK.js";
-import { a as getActiveTraceContext, c as trace, d as withTracing, f as InstrumentOptions, i as ctx, l as withBaggage, m as TracingOptions, n as WithBaggageOptions, o as instrument, p as SingleInstrumentOptions, r as WithNewContextOptions, s as span, t as SpanOptions, u as withNewContext } from "./functional-DXy1cHFF.js";
+import { a as getActiveTraceContext, c as trace, d as withTracing, f as InstrumentOptions, i as ctx, l as withBaggage, m as TracingOptions, n as WithBaggageOptions, o as instrument, p as SingleInstrumentOptions, r as WithNewContextOptions, s as span, t as SpanOptions, u as withNewContext } from "./functional-RFFpQzAV.js";
 export { type InstrumentOptions, type SingleInstrumentOptions, SpanOptions, type TraceContext, type TracingOptions, WithBaggageOptions, WithNewContextOptions, ctx, getActiveTraceContext, instrument, span, trace, withBaggage, withNewContext, withTracing };

package/dist/functional.js

@@ -1,3 +1,3 @@
-import { a as trace, c as withTracing, i as span, n as getActiveTraceContext, o as withBaggage, r as instrument, s as withNewContext, t as ctx } from "./functional-Cub1ku-C.js";
+import { a as trace, c as withTracing, i as span, n as getActiveTraceContext, o as withBaggage, r as instrument, s as withNewContext, t as ctx } from "./functional-C4js2vTM.js";
 
 export { ctx, getActiveTraceContext, instrument, span, trace, withBaggage, withNewContext, withTracing };

package/dist/http.cjs

@@ -1,7 +1,7 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 const require_values = require('./values-DApyq53y.cjs');
 const require_config = require('./config.cjs');
-const require_init = require('./init-C5aHRqi2.cjs');
+const require_init = require('./init-DmM7dprX.cjs');
 let _opentelemetry_api = require("@opentelemetry/api");
 
 //#region src/http.ts