neatlogs 1.1.8 → 1.1.10

package/dist/ai-sdk.cjs

@@ -29,6 +29,7 @@ var import_api2 = require("@opentelemetry/api");
 // src/core/provider.ts
 var import_node_async_hooks = require("async_hooks");
 var import_api = require("@opentelemetry/api");
+var NEATLOGS_ROOT_SPAN_KEY = (0, import_api.createContextKey)("neatlogs.root_span");
 var PRIVATE_SPAN_STORAGE_KEY = /* @__PURE__ */ Symbol.for(
   "neatlogs.private_span_async_local_storage"
 );
@@ -87,7 +88,10 @@ function getNeatlogsParentContext(baseContext) {
 }
 function withNeatlogsSpan(span, fn, baseContext) {
   const base = baseContext ?? getNeatlogsActiveContext();
-  const ctx = import_api.trace.setSpan(base, span);
+  let ctx = import_api.trace.setSpan(base, span);
+  if (base.getValue(NEATLOGS_ROOT_SPAN_KEY) === void 0) {
+    ctx = ctx.setValue(NEATLOGS_ROOT_SPAN_KEY, span);
+  }
   return privateContextStorage.run(ctx, fn);
 }
 

package/dist/ai-sdk.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/ai-sdk.ts","../src/core/provider.ts"],"sourcesContent":["/**\n * Vercel AI SDK wrapper — inline implementation.\n *\n * Wraps `generateText`, `streamText`, `generateObject`, `streamObject` from\n * the `ai` package with OTel parent spans + forced telemetry. Static export,\n * no dynamic imports — bundler-friendly (works with Turbopack, webpack, esbuild).\n *\n * Usage:\n *   import { wrapAISDK } from 'neatlogs';\n *   import * as ai from 'ai';\n *   const { streamText, generateText } = wrapAISDK(ai);\n */\n\nimport { SpanStatusCode, type Span, type Tracer } from '@opentelemetry/api';\nimport { getNeatlogsTracer, getNeatlogsParentContext, isolateTracer, withNeatlogsSpan } from './core/provider.js';\n\nconst TRACER_NAME = 'neatlogs.ai-sdk';\n\n// -- Telemetry config --------------------------------------------------------\n\nexport interface CreateAITelemetryOptions {\n  metadata?: Record<string, unknown>;\n}\n\nexport interface AITelemetryConfig {\n  isEnabled: true;\n  recordInputs: true;\n  recordOutputs: true;\n  tracer: Tracer;\n  metadata: Record<string, unknown>;\n}\n\nexport function createAITelemetry(\n  opts: CreateAITelemetryOptions = {},\n): AITelemetryConfig {\n  const userMeta = opts.metadata ?? {};\n  return {\n    isEnabled: true,\n    recordInputs: true,\n    recordOutputs: true,\n    // Hand the AI SDK an isolation-aware tracer: it calls startActiveSpan()\n    // internally, which would otherwise parent its native spans from the foreign\n    // global context AND push them onto it (so a co-tenant's next span inherits\n    // ours). The facade routes both through the private Neatlogs context.\n    tracer: isolateTracer(getNeatlogsTracer(TRACER_NAME)),\n    metadata: { ...userMeta, neatlogsWrapped: true },\n  };\n}\n\n// -- Span attributes ---------------------------------------------------------\n\nfunction safeStringify(value: unknown): string {\n  try {\n    return JSON.stringify(value);\n  } catch {\n    return '';\n  }\n}\n\nfunction setInputValue(span: Span, opts: Record<string, unknown>): void {\n  // For generateText/streamText — only capture prompt or messages, not full model config\n  if (opts && ('prompt' in opts || 'messages' in opts)) {\n    const input = opts.messages ?? opts.prompt;\n    const stringified = safeStringify(input);\n    if (stringified) {\n      span.setAttribute('input.value', stringified);\n    }\n    return;\n  }\n  const stringified = safeStringify(opts);\n  if (stringified) {\n    span.setAttribute('input.value', stringified);\n  }\n}\n\nfunction setOutputValue(span: Span, result: unknown): void {\n  if (result && typeof result === 'object') {\n    const r = result as Record<string, unknown>;\n    // GenerateTextResult / StreamTextResult — extract meaningful fields\n    if ('text' in r && 'finishReason' in r) {\n      const text = String(r.text ?? '');\n      if (text) {\n        span.setAttribute('output.value', text);\n      }\n      if (r.finishReason) {\n        span.setAttribute('gen_ai.finish_reason', String(r.finishReason));\n      }\n      return;\n    }\n    // GenerateObjectResult — the structured object is the output, not `text`.\n    // Without this the envelope (object+usage+response+…) gets stringified whole.\n    if ('object' in r && 'finishReason' in r) {\n      if (r.object !== undefined) {\n        span.setAttribute('output.value', safeStringify(r.object));\n      }\n      if (r.finishReason) {\n        span.setAttribute('gen_ai.finish_reason', String(r.finishReason));\n      }\n      return;\n    }\n  }\n  const stringified = safeStringify(result);\n  if (stringified) {\n    span.setAttribute('output.value', stringified);\n  }\n}\n\n// Extract output from a streamText/streamObject `onFinish` event. The event\n// extends StepResult, so `text` (streamText) / `object` (streamObject) sit at\n// the top level alongside `finishReason` — but the event also carries `steps`,\n// `usage`, etc., so we pull only the meaningful fields instead of stringifying\n// the whole envelope.\nfunction setStreamOutputValue(span: Span, event: unknown): void {\n  if (!event || typeof event !== 'object') return;\n  const e = event as Record<string, unknown>;\n  if (typeof e.text === 'string' && e.text) {\n    span.setAttribute('output.value', e.text);\n  } else if ('object' in e && e.object !== undefined) {\n    const stringified = safeStringify(e.object);\n    if (stringified) span.setAttribute('output.value', stringified);\n  }\n  if (e.finishReason) {\n    span.setAttribute('gen_ai.finish_reason', String(e.finishReason));\n  }\n}\n\n// -- Wrapping ----------------------------------------------------------------\n\ntype WrappedFunctionName = 'generateText' | 'streamText' | 'generateObject' | 'streamObject' | 'embed' | 'embedMany' | 'rerank';\n\nconst WRAPPED_FUNCTIONS: readonly WrappedFunctionName[] = [\n  'generateText',\n  'streamText',\n  'generateObject',\n  'streamObject',\n  'embed',\n  'embedMany',\n  'rerank',\n] as const;\n\n/**\n * Wrap the `ai` module namespace so that every `generateText` / `streamText` /\n * `generateObject` / `streamObject` call:\n *\n *   1. Opens a parent OTel span on the active TracerProvider.\n *   2. Forces `experimental_telemetry: { isEnabled: true }`, merging user metadata.\n *   3. Records input/output on the parent span and propagates errors.\n *\n * Other exports (types, helpers) pass through unchanged.\n */\nexport function wrapAISDK<T extends Record<string, unknown>>(aiModule: T): T {\n  const wrapped: Record<string, unknown> = { ...aiModule };\n\n  for (const name of WRAPPED_FUNCTIONS) {\n    const original = aiModule[name];\n    if (typeof original !== 'function') continue;\n\n    if (name === 'streamText' || name === 'streamObject') {\n      wrapped[name] = createStreamWrapper(name, original as (opts: any) => unknown);\n    } else {\n      wrapped[name] = createAsyncWrapper(name, original as (opts: any) => Promise<unknown>);\n    }\n  }\n\n  return wrapped as T;\n}\n\nfunction rootSpanKind(name: WrappedFunctionName): string {\n  if (name === 'embed' || name === 'embedMany' || name === 'rerank') return 'CHAIN';\n  return 'WORKFLOW';\n}\n\n\nfunction getParentContext() {\n  // Our parent comes solely from the private span store; a\n  // foreign provider's active span must never become our ancestor.\n  return getNeatlogsParentContext();\n}\n\nfunction createAsyncWrapper(\n  name: WrappedFunctionName,\n  original: (opts: any) => Promise<unknown>,\n): (opts: any) => Promise<unknown> {\n  return async function wrappedAsyncFn(opts: any): Promise<unknown> {\n    const tracer = getNeatlogsTracer(TRACER_NAME);\n    // startSpan (NOT startActiveSpan) + withNeatlogsSpan: startActiveSpan would\n    // push our span onto the GLOBAL OTel context, so a foreign tracer's\n    // startSpan() inside generateText() would read it as parent and inherit our\n    // trace id. withNeatlogsSpan carries the parent in the private store in\n    // the private context, leaving the global context untouched.\n    const parentContext = getParentContext();\n    const span = tracer.startSpan(\n      `ai.${name}`,\n      { attributes: { 'openinference.span.kind': rootSpanKind(name) } },\n      parentContext,\n    );\n    return withNeatlogsSpan(\n      span,\n      async () => {\n        try {\n          const isEmbedOrRerank = name === 'embed' || name === 'embedMany' || name === 'rerank';\n          if (!isEmbedOrRerank) {\n            setInputValue(span, opts);\n          }\n          if (name === 'rerank' && opts?.query) {\n            span.setAttribute('ai.rerank.query', String(opts.query));\n          }\n          const merged = mergeTelemetry(opts);\n          const result = await original(merged);\n          if (!isEmbedOrRerank) {\n            setOutputValue(span, result);\n          }\n          return result;\n        } catch (err) {\n          recordSpanError(span, err);\n          throw err;\n        } finally {\n          span.end();\n        }\n      },\n      parentContext,\n    );\n  };\n}\n\n// streamText/streamObject return synchronously while the model keeps producing\n// tokens for seconds afterwards. Ending the span in a `finally` (as a plain sync\n// wrapper would) closes it in ~2ms with no output — the output only exists once\n// the stream finishes. Instead we keep the span open and end it from the AI SDK's\n// `onFinish` callback, where the final text/object is available. Any user-provided\n// `onFinish` is preserved and invoked first.\nfunction createStreamWrapper(\n  name: WrappedFunctionName,\n  original: (opts: any) => unknown,\n): (opts: any) => unknown {\n  return function wrappedStreamFn(opts: any): unknown {\n    const tracer = getNeatlogsTracer(TRACER_NAME);\n    // startSpan + withNeatlogsSpan (see createAsyncWrapper) so streamText's\n    // internals never see our span on the global OTel context. The span stays\n    // open past the run scope and is ended from onFinish/onError.\n    const parentContext = getParentContext();\n    const span = tracer.startSpan(\n      `ai.${name}`,\n      { attributes: { 'openinference.span.kind': rootSpanKind(name) } },\n      parentContext,\n    );\n    return withNeatlogsSpan(\n      span,\n      () => {\n        let spanEnded = false;\n        const endOnce = () => {\n          if (spanEnded) return;\n          spanEnded = true;\n          span.end();\n        };\n        try {\n          setInputValue(span, opts);\n          const merged = mergeTelemetry(opts);\n          const userOnFinish = opts?.onFinish;\n          const userOnError = opts?.onError;\n          const wrappedOpts = {\n            ...merged,\n            onFinish: async (event: any) => {\n              try {\n                setStreamOutputValue(span, event);\n              } finally {\n                endOnce();\n              }\n              if (typeof userOnFinish === 'function') {\n                return userOnFinish(event);\n              }\n            },\n            onError: (event: any) => {\n              recordSpanError(span, (event && event.error) ?? event);\n              endOnce();\n              if (typeof userOnError === 'function') {\n                return userOnError(event);\n              }\n            },\n          };\n          return original(wrappedOpts);\n        } catch (err) {\n          // Synchronous throw (e.g. bad arguments) — the stream never started.\n          recordSpanError(span, err);\n          endOnce();\n          throw err;\n        }\n      },\n      parentContext,\n    );\n  };\n}\n\nfunction mergeTelemetry(opts: any): any {\n  const baseTelemetry: AITelemetryConfig = createAITelemetry({\n    metadata: opts?.experimental_telemetry?.metadata,\n  });\n  return {\n    ...opts,\n    experimental_telemetry: {\n      ...opts?.experimental_telemetry,\n      ...baseTelemetry,\n    },\n  };\n}\n\n\nfunction recordSpanError(span: Span, err: unknown): void {\n  if (err instanceof Error) {\n    span.setStatus({ code: SpanStatusCode.ERROR, message: err.message });\n    span.recordException(err);\n  } else {\n    span.setStatus({ code: SpanStatusCode.ERROR, message: String(err) });\n  }\n}\n","/**\n * Neatlogs-owned tracing state.\n *\n * Spans are created by the private Neatlogs provider and their parent is carried\n * in a private context key. The process-global OpenTelemetry span is\n * deliberately left untouched so other observability SDKs cannot export or\n * become parents of Neatlogs spans.\n */\n\nimport { AsyncLocalStorage } from 'node:async_hooks';\nimport {\n  ROOT_CONTEXT,\n  INVALID_SPAN_CONTEXT,\n  trace as otelTrace,\n  type Context,\n  type Span,\n  type SpanOptions,\n  type Tracer,\n  type TracerProvider,\n} from '@opentelemetry/api';\n\n// Entry points are bundled independently (`neatlogs`, `neatlogs/openai`,\n// `neatlogs/ai`, `neatlogs/mastra`, … each in both CJS and ESM), so every piece\n// of shared tracing state — the private span store AND the resolved provider /\n// provider — must live on `globalThis` behind a `Symbol.for` key.\n// Otherwise `init()` (run from the `neatlogs` bundle) sets `_provider` in ITS\n// module copy while a wrapper imported from `neatlogs/openai` reads a different,\n// still-null copy and silently falls back to the foreign global provider.\nconst PRIVATE_SPAN_STORAGE_KEY = Symbol.for(\n  'neatlogs.private_span_async_local_storage',\n);\nconst PRIVATE_PROVIDER_STATE_KEY = Symbol.for(\n  'neatlogs.private_provider_state',\n);\ninterface PrivateProviderState {\n  provider: TracerProvider | null;\n}\ntype NeatlogsGlobal = typeof globalThis & {\n  [PRIVATE_SPAN_STORAGE_KEY]?: AsyncLocalStorage<Context>;\n  [PRIVATE_PROVIDER_STATE_KEY]?: PrivateProviderState;\n};\nconst neatlogsGlobal = globalThis as NeatlogsGlobal;\n// Stores the full Neatlogs Context (parent span PLUS any threaded values such as\n// trace()'s prompt-template keys), not just the span. We never\n// activate the OTel global context, so this private store is the ONLY channel\n// through which those values propagate down to descendant spans.\nconst privateContextStorage =\n  neatlogsGlobal[PRIVATE_SPAN_STORAGE_KEY] ??\n  (neatlogsGlobal[PRIVATE_SPAN_STORAGE_KEY] = new AsyncLocalStorage<Context>());\nconst providerState: PrivateProviderState =\n  neatlogsGlobal[PRIVATE_PROVIDER_STATE_KEY] ??\n  (neatlogsGlobal[PRIVATE_PROVIDER_STATE_KEY] = {\n    provider: null,\n  });\n// Wrappers may be constructed or accidentally invoked before init(). They must\n// never fall back to a foreign process-global provider, so pre-init calls use a\n// local no-op tracer and safely emit no exported spans.\nconst preInitTracer: Tracer = {\n  startSpan(): Span {\n    return otelTrace.wrapSpanContext(INVALID_SPAN_CONTEXT);\n  },\n  startActiveSpan<F extends (span: Span) => unknown>(\n    _name: string,\n    arg2?: SpanOptions | Context | F,\n    arg3?: Context | F,\n    arg4?: F,\n  ): ReturnType<F> {\n    const fn =\n      typeof arg2 === 'function'\n        ? arg2\n        : typeof arg3 === 'function'\n          ? arg3\n          : arg4;\n    return fn!(otelTrace.wrapSpanContext(INVALID_SPAN_CONTEXT)) as ReturnType<F>;\n  },\n};\n\n/** @internal Configure the provider used by Neatlogs-created spans. */\nexport function _setNeatlogsProvider(provider: TracerProvider | null): void {\n  providerState.provider = provider;\n}\n\n/** Resolve a tracer from the private provider when one is configured. */\nexport function getNeatlogsTracer(name: string): Tracer {\n  return providerState.provider?.getTracer(name) ?? preInitTracer;\n}\n\n/**\n * @internal The private Neatlogs provider, or null before\n * init(). Used by integrations that must repoint a self-instrumenting library's\n * captured provider onto ours.\n */\nexport function getNeatlogsProvider(): TracerProvider | null {\n  return providerState.provider;\n}\n\n/**\n * Wrap a tracer so that spans it creates parent from — and, for\n * `startActiveSpan`, activate on — the PRIVATE Neatlogs context instead of the\n * global one.\n *\n * We hand this to libraries that create their own spans off a tracer we give\n * them (the Vercel AI SDK's `experimental_telemetry.tracer`, which calls\n * `tracer.startActiveSpan()` internally). Without the facade the AI SDK's native\n * spans would parent from `context.active()` — the foreign co-tenant's context —\n * and `startActiveSpan` would push them onto the GLOBAL context, so a foreign\n * tracer's next span reads our native span as its parent. Both directions leak.\n *\n */\nexport function isolateTracer(tracer: Tracer): Tracer {\n  const facade: Tracer = {\n    startSpan(name: string, options?: SpanOptions, context?: Context): Span {\n      const parent = context ?? getNeatlogsActiveContext();\n      return tracer.startSpan(name, options, parent);\n    },\n    startActiveSpan<F extends (span: Span) => unknown>(\n      name: string,\n      arg2?: SpanOptions | Context | F,\n      arg3?: Context | F,\n      arg4?: F,\n    ): ReturnType<F> {\n      // Normalize the 2/3/4-arg overloads of startActiveSpan.\n      let options: SpanOptions | undefined;\n      let context: Context | undefined;\n      let fn: F;\n      if (typeof arg2 === 'function') {\n        fn = arg2 as F;\n      } else if (typeof arg3 === 'function') {\n        options = arg2 as SpanOptions;\n        fn = arg3 as F;\n      } else {\n        options = arg2 as SpanOptions;\n        context = arg3 as Context;\n        fn = arg4 as F;\n      }\n      const parent = context ?? getNeatlogsActiveContext();\n      const span = tracer.startSpan(name, options, parent);\n      return withNeatlogsSpan(span, () => fn(span), parent) as ReturnType<F>;\n    },\n  };\n  return facade;\n}\n\n/**\n * The base context to build new Neatlogs spans and values on. NEVER contains a\n * foreign provider's span: it reads our private store, or ROOT_CONTEXT when\n * nothing is active.\n */\nexport function getNeatlogsActiveContext(): Context {\n  return privateContextStorage.getStore() ?? ROOT_CONTEXT;\n}\n\n/** Return only the active Neatlogs span (never a foreign provider's span). */\nexport function getActiveNeatlogsSpan(): Span | undefined {\n  return otelTrace.getSpan(getNeatlogsActiveContext());\n}\n\n/**\n * Build a parent context that cannot contain a foreign provider's span.\n *\n * The active Neatlogs context already carries the parent span AND any values\n * the caller threaded in upstream (e.g. `trace()`'s prompt-template values), so\n * those values reach the span processor via `onStart(parentContext)`. An\n * explicit `baseContext` (a caller's own value-carrying context) is honored as-is.\n */\nexport function getNeatlogsParentContext(baseContext?: Context): Context {\n  return baseContext ?? getNeatlogsActiveContext();\n}\n\n/**\n * A base context for callers that thread parent linkage themselves\n * (callback/event handlers that keep their own run-id → span map: LangChain,\n * OpenAI-Agents, Claude Agent SDK).\n *\n * This is the ACTIVE Neatlogs context — the private store's\n * context if a Neatlogs `trace()`/`span()` encloses this call, else\n * ROOT_CONTEXT. So a handler's own root/entry span nests under an enclosing\n * Neatlogs trace (preserving its session + end-user id) when one exists, and\n * auto-roots cleanly when one doesn't. A foreign provider's active span can\n * never leak in as an ancestor because the global context is never read.\n */\nexport function getNeatlogsBaseContext(baseContext?: Context): Context {\n  return baseContext ?? getNeatlogsActiveContext();\n}\n\n/**\n * Build an execution context for a Neatlogs span while preserving the active\n * foreign context. The span rides our private context store.\n */\nexport function getNeatlogsExecutionContext(\n  span: Span,\n  baseContext: Context = ROOT_CONTEXT,\n): Context {\n  return otelTrace.setSpan(baseContext, span);\n}\n\n/**\n * Run a callback with a Neatlogs span active under the appropriate policy.\n *\n * The stored context is `setSpan(base, span)` — carrying the span PLUS whatever\n * values `base` holds — so threaded values (prompt templates)\n * propagate to descendant spans through our private store instead of the global\n * OTel context we deliberately never touch.\n */\nexport function withNeatlogsSpan<T>(\n  span: Span,\n  fn: () => T,\n  baseContext?: Context,\n): T {\n  const base = baseContext ?? getNeatlogsActiveContext();\n  const ctx = otelTrace.setSpan(base, span);\n  return privateContextStorage.run(ctx, fn);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAaA,IAAAA,cAAuD;;;ACJvD,8BAAkC;AAClC,iBASO;AASP,IAAM,2BAA2B,uBAAO;AAAA,EACtC;AACF;AACA,IAAM,6BAA6B,uBAAO;AAAA,EACxC;AACF;AAQA,IAAM,iBAAiB;AAKvB,IAAM,wBACJ,eAAe,wBAAwB,MACtC,eAAe,wBAAwB,IAAI,IAAI,0CAA2B;AAC7E,IAAM,gBACJ,eAAe,0BAA0B,MACxC,eAAe,0BAA0B,IAAI;AAAA,EAC5C,UAAU;AACZ;AAIF,IAAM,gBAAwB;AAAA,EAC5B,YAAkB;AAChB,WAAO,WAAAC,MAAU,gBAAgB,+BAAoB;AAAA,EACvD;AAAA,EACA,gBACE,OACA,MACA,MACA,MACe;AACf,UAAM,KACJ,OAAO,SAAS,aACZ,OACA,OAAO,SAAS,aACd,OACA;AACR,WAAO,GAAI,WAAAA,MAAU,gBAAgB,+BAAoB,CAAC;AAAA,EAC5D;AACF;AAQO,SAAS,kBAAkB,MAAsB;AACtD,SAAO,cAAc,UAAU,UAAU,IAAI,KAAK;AACpD;AAwBO,SAAS,cAAc,QAAwB;AACpD,QAAM,SAAiB;AAAA,IACrB,UAAU,MAAc,SAAuB,SAAyB;AACtE,YAAM,SAAS,WAAW,yBAAyB;AACnD,aAAO,OAAO,UAAU,MAAM,SAAS,MAAM;AAAA,IAC/C;AAAA,IACA,gBACE,MACA,MACA,MACA,MACe;AAEf,UAAI;AACJ,UAAI;AACJ,UAAI;AACJ,UAAI,OAAO,SAAS,YAAY;AAC9B,aAAK;AAAA,MACP,WAAW,OAAO,SAAS,YAAY;AACrC,kBAAU;AACV,aAAK;AAAA,MACP,OAAO;AACL,kBAAU;AACV,kBAAU;AACV,aAAK;AAAA,MACP;AACA,YAAM,SAAS,WAAW,yBAAyB;AACnD,YAAM,OAAO,OAAO,UAAU,MAAM,SAAS,MAAM;AACnD,aAAO,iBAAiB,MAAM,MAAM,GAAG,IAAI,GAAG,MAAM;AAAA,IACtD;AAAA,EACF;AACA,SAAO;AACT;AAOO,SAAS,2BAAoC;AAClD,SAAO,sBAAsB,SAAS,KAAK;AAC7C;AAeO,SAAS,yBAAyB,aAAgC;AACvE,SAAO,eAAe,yBAAyB;AACjD;AAqCO,SAAS,iBACd,MACA,IACA,aACG;AACH,QAAM,OAAO,eAAe,yBAAyB;AACrD,QAAM,MAAM,WAAAC,MAAU,QAAQ,MAAM,IAAI;AACxC,SAAO,sBAAsB,IAAI,KAAK,EAAE;AAC1C;;;ADpMA,IAAM,cAAc;AAgBb,SAAS,kBACd,OAAiC,CAAC,GACf;AACnB,QAAM,WAAW,KAAK,YAAY,CAAC;AACnC,SAAO;AAAA,IACL,WAAW;AAAA,IACX,cAAc;AAAA,IACd,eAAe;AAAA;AAAA;AAAA;AAAA;AAAA,IAKf,QAAQ,cAAc,kBAAkB,WAAW,CAAC;AAAA,IACpD,UAAU,EAAE,GAAG,UAAU,iBAAiB,KAAK;AAAA,EACjD;AACF;AAIA,SAAS,cAAc,OAAwB;AAC7C,MAAI;AACF,WAAO,KAAK,UAAU,KAAK;AAAA,EAC7B,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAEA,SAAS,cAAc,MAAY,MAAqC;AAEtE,MAAI,SAAS,YAAY,QAAQ,cAAc,OAAO;AACpD,UAAM,QAAQ,KAAK,YAAY,KAAK;AACpC,UAAMC,eAAc,cAAc,KAAK;AACvC,QAAIA,cAAa;AACf,WAAK,aAAa,eAAeA,YAAW;AAAA,IAC9C;AACA;AAAA,EACF;AACA,QAAM,cAAc,cAAc,IAAI;AACtC,MAAI,aAAa;AACf,SAAK,aAAa,eAAe,WAAW;AAAA,EAC9C;AACF;AAEA,SAAS,eAAe,MAAY,QAAuB;AACzD,MAAI,UAAU,OAAO,WAAW,UAAU;AACxC,UAAM,IAAI;AAEV,QAAI,UAAU,KAAK,kBAAkB,GAAG;AACtC,YAAM,OAAO,OAAO,EAAE,QAAQ,EAAE;AAChC,UAAI,MAAM;AACR,aAAK,aAAa,gBAAgB,IAAI;AAAA,MACxC;AACA,UAAI,EAAE,cAAc;AAClB,aAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,MAClE;AACA;AAAA,IACF;AAGA,QAAI,YAAY,KAAK,kBAAkB,GAAG;AACxC,UAAI,EAAE,WAAW,QAAW;AAC1B,aAAK,aAAa,gBAAgB,cAAc,EAAE,MAAM,CAAC;AAAA,MAC3D;AACA,UAAI,EAAE,cAAc;AAClB,aAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,MAClE;AACA;AAAA,IACF;AAAA,EACF;AACA,QAAM,cAAc,cAAc,MAAM;AACxC,MAAI,aAAa;AACf,SAAK,aAAa,gBAAgB,WAAW;AAAA,EAC/C;AACF;AAOA,SAAS,qBAAqB,MAAY,OAAsB;AAC9D,MAAI,CAAC,SAAS,OAAO,UAAU,SAAU;AACzC,QAAM,IAAI;AACV,MAAI,OAAO,EAAE,SAAS,YAAY,EAAE,MAAM;AACxC,SAAK,aAAa,gBAAgB,EAAE,IAAI;AAAA,EAC1C,WAAW,YAAY,KAAK,EAAE,WAAW,QAAW;AAClD,UAAM,cAAc,cAAc,EAAE,MAAM;AAC1C,QAAI,YAAa,MAAK,aAAa,gBAAgB,WAAW;AAAA,EAChE;AACA,MAAI,EAAE,cAAc;AAClB,SAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,EAClE;AACF;AAMA,IAAM,oBAAoD;AAAA,EACxD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAYO,SAAS,UAA6C,UAAgB;AAC3E,QAAM,UAAmC,EAAE,GAAG,SAAS;AAEvD,aAAW,QAAQ,mBAAmB;AACpC,UAAM,WAAW,SAAS,IAAI;AAC9B,QAAI,OAAO,aAAa,WAAY;AAEpC,QAAI,SAAS,gBAAgB,SAAS,gBAAgB;AACpD,cAAQ,IAAI,IAAI,oBAAoB,MAAM,QAAkC;AAAA,IAC9E,OAAO;AACL,cAAQ,IAAI,IAAI,mBAAmB,MAAM,QAA2C;AAAA,IACtF;AAAA,EACF;AAEA,SAAO;AACT;AAEA,SAAS,aAAa,MAAmC;AACvD,MAAI,SAAS,WAAW,SAAS,eAAe,SAAS,SAAU,QAAO;AAC1E,SAAO;AACT;AAGA,SAAS,mBAAmB;AAG1B,SAAO,yBAAyB;AAClC;AAEA,SAAS,mBACP,MACA,UACiC;AACjC,SAAO,eAAe,eAAe,MAA6B;AAChE,UAAM,SAAS,kBAAkB,WAAW;AAM5C,UAAM,gBAAgB,iBAAiB;AACvC,UAAM,OAAO,OAAO;AAAA,MAClB,MAAM,IAAI;AAAA,MACV,EAAE,YAAY,EAAE,2BAA2B,aAAa,IAAI,EAAE,EAAE;AAAA,MAChE;AAAA,IACF;AACA,WAAO;AAAA,MACL;AAAA,MACA,YAAY;AACV,YAAI;AACF,gBAAM,kBAAkB,SAAS,WAAW,SAAS,eAAe,SAAS;AAC7E,cAAI,CAAC,iBAAiB;AACpB,0BAAc,MAAM,IAAI;AAAA,UAC1B;AACA,cAAI,SAAS,YAAY,MAAM,OAAO;AACpC,iBAAK,aAAa,mBAAmB,OAAO,KAAK,KAAK,CAAC;AAAA,UACzD;AACA,gBAAM,SAAS,eAAe,IAAI;AAClC,gBAAM,SAAS,MAAM,SAAS,MAAM;AACpC,cAAI,CAAC,iBAAiB;AACpB,2BAAe,MAAM,MAAM;AAAA,UAC7B;AACA,iBAAO;AAAA,QACT,SAAS,KAAK;AACZ,0BAAgB,MAAM,GAAG;AACzB,gBAAM;AAAA,QACR,UAAE;AACA,eAAK,IAAI;AAAA,QACX;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAQA,SAAS,oBACP,MACA,UACwB;AACxB,SAAO,SAAS,gBAAgB,MAAoB;AAClD,UAAM,SAAS,kBAAkB,WAAW;AAI5C,UAAM,gBAAgB,iBAAiB;AACvC,UAAM,OAAO,OAAO;AAAA,MAClB,MAAM,IAAI;AAAA,MACV,EAAE,YAAY,EAAE,2BAA2B,aAAa,IAAI,EAAE,EAAE;AAAA,MAChE;AAAA,IACF;AACA,WAAO;AAAA,MACL;AAAA,MACA,MAAM;AACJ,YAAI,YAAY;AAChB,cAAM,UAAU,MAAM;AACpB,cAAI,UAAW;AACf,sBAAY;AACZ,eAAK,IAAI;AAAA,QACX;AACA,YAAI;AACF,wBAAc,MAAM,IAAI;AACxB,gBAAM,SAAS,eAAe,IAAI;AAClC,gBAAM,eAAe,MAAM;AAC3B,gBAAM,cAAc,MAAM;AAC1B,gBAAM,cAAc;AAAA,YAClB,GAAG;AAAA,YACH,UAAU,OAAO,UAAe;AAC9B,kBAAI;AACF,qCAAqB,MAAM,KAAK;AAAA,cAClC,UAAE;AACA,wBAAQ;AAAA,cACV;AACA,kBAAI,OAAO,iBAAiB,YAAY;AACtC,uBAAO,aAAa,KAAK;AAAA,cAC3B;AAAA,YACF;AAAA,YACA,SAAS,CAAC,UAAe;AACvB,8BAAgB,OAAO,SAAS,MAAM,UAAU,KAAK;AACrD,sBAAQ;AACR,kBAAI,OAAO,gBAAgB,YAAY;AACrC,uBAAO,YAAY,KAAK;AAAA,cAC1B;AAAA,YACF;AAAA,UACF;AACA,iBAAO,SAAS,WAAW;AAAA,QAC7B,SAAS,KAAK;AAEZ,0BAAgB,MAAM,GAAG;AACzB,kBAAQ;AACR,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,eAAe,MAAgB;AACtC,QAAM,gBAAmC,kBAAkB;AAAA,IACzD,UAAU,MAAM,wBAAwB;AAAA,EAC1C,CAAC;AACD,SAAO;AAAA,IACL,GAAG;AAAA,IACH,wBAAwB;AAAA,MACtB,GAAG,MAAM;AAAA,MACT,GAAG;AAAA,IACL;AAAA,EACF;AACF;AAGA,SAAS,gBAAgB,MAAY,KAAoB;AACvD,MAAI,eAAe,OAAO;AACxB,SAAK,UAAU,EAAE,MAAM,2BAAe,OAAO,SAAS,IAAI,QAAQ,CAAC;AACnE,SAAK,gBAAgB,GAAG;AAAA,EAC1B,OAAO;AACL,SAAK,UAAU,EAAE,MAAM,2BAAe,OAAO,SAAS,OAAO,GAAG,EAAE,CAAC;AAAA,EACrE;AACF;","names":["import_api","otelTrace","otelTrace","stringified"]}
+{"version":3,"sources":["../src/ai-sdk.ts","../src/core/provider.ts"],"sourcesContent":["/**\n * Vercel AI SDK wrapper — inline implementation.\n *\n * Wraps `generateText`, `streamText`, `generateObject`, `streamObject` from\n * the `ai` package with OTel parent spans + forced telemetry. Static export,\n * no dynamic imports — bundler-friendly (works with Turbopack, webpack, esbuild).\n *\n * Usage:\n *   import { wrapAISDK } from 'neatlogs';\n *   import * as ai from 'ai';\n *   const { streamText, generateText } = wrapAISDK(ai);\n */\n\nimport { SpanStatusCode, type Span, type Tracer } from '@opentelemetry/api';\nimport { getNeatlogsTracer, getNeatlogsParentContext, isolateTracer, withNeatlogsSpan } from './core/provider.js';\n\nconst TRACER_NAME = 'neatlogs.ai-sdk';\n\n// -- Telemetry config --------------------------------------------------------\n\nexport interface CreateAITelemetryOptions {\n  metadata?: Record<string, unknown>;\n}\n\nexport interface AITelemetryConfig {\n  isEnabled: true;\n  recordInputs: true;\n  recordOutputs: true;\n  tracer: Tracer;\n  metadata: Record<string, unknown>;\n}\n\nexport function createAITelemetry(\n  opts: CreateAITelemetryOptions = {},\n): AITelemetryConfig {\n  const userMeta = opts.metadata ?? {};\n  return {\n    isEnabled: true,\n    recordInputs: true,\n    recordOutputs: true,\n    // Hand the AI SDK an isolation-aware tracer: it calls startActiveSpan()\n    // internally, which would otherwise parent its native spans from the foreign\n    // global context AND push them onto it (so a co-tenant's next span inherits\n    // ours). The facade routes both through the private Neatlogs context.\n    tracer: isolateTracer(getNeatlogsTracer(TRACER_NAME)),\n    metadata: { ...userMeta, neatlogsWrapped: true },\n  };\n}\n\n// -- Span attributes ---------------------------------------------------------\n\nfunction safeStringify(value: unknown): string {\n  try {\n    return JSON.stringify(value);\n  } catch {\n    return '';\n  }\n}\n\nfunction setInputValue(span: Span, opts: Record<string, unknown>): void {\n  // For generateText/streamText — only capture prompt or messages, not full model config\n  if (opts && ('prompt' in opts || 'messages' in opts)) {\n    const input = opts.messages ?? opts.prompt;\n    const stringified = safeStringify(input);\n    if (stringified) {\n      span.setAttribute('input.value', stringified);\n    }\n    return;\n  }\n  const stringified = safeStringify(opts);\n  if (stringified) {\n    span.setAttribute('input.value', stringified);\n  }\n}\n\nfunction setOutputValue(span: Span, result: unknown): void {\n  if (result && typeof result === 'object') {\n    const r = result as Record<string, unknown>;\n    // GenerateTextResult / StreamTextResult — extract meaningful fields\n    if ('text' in r && 'finishReason' in r) {\n      const text = String(r.text ?? '');\n      if (text) {\n        span.setAttribute('output.value', text);\n      }\n      if (r.finishReason) {\n        span.setAttribute('gen_ai.finish_reason', String(r.finishReason));\n      }\n      return;\n    }\n    // GenerateObjectResult — the structured object is the output, not `text`.\n    // Without this the envelope (object+usage+response+…) gets stringified whole.\n    if ('object' in r && 'finishReason' in r) {\n      if (r.object !== undefined) {\n        span.setAttribute('output.value', safeStringify(r.object));\n      }\n      if (r.finishReason) {\n        span.setAttribute('gen_ai.finish_reason', String(r.finishReason));\n      }\n      return;\n    }\n  }\n  const stringified = safeStringify(result);\n  if (stringified) {\n    span.setAttribute('output.value', stringified);\n  }\n}\n\n// Extract output from a streamText/streamObject `onFinish` event. The event\n// extends StepResult, so `text` (streamText) / `object` (streamObject) sit at\n// the top level alongside `finishReason` — but the event also carries `steps`,\n// `usage`, etc., so we pull only the meaningful fields instead of stringifying\n// the whole envelope.\nfunction setStreamOutputValue(span: Span, event: unknown): void {\n  if (!event || typeof event !== 'object') return;\n  const e = event as Record<string, unknown>;\n  if (typeof e.text === 'string' && e.text) {\n    span.setAttribute('output.value', e.text);\n  } else if ('object' in e && e.object !== undefined) {\n    const stringified = safeStringify(e.object);\n    if (stringified) span.setAttribute('output.value', stringified);\n  }\n  if (e.finishReason) {\n    span.setAttribute('gen_ai.finish_reason', String(e.finishReason));\n  }\n}\n\n// -- Wrapping ----------------------------------------------------------------\n\ntype WrappedFunctionName = 'generateText' | 'streamText' | 'generateObject' | 'streamObject' | 'embed' | 'embedMany' | 'rerank';\n\nconst WRAPPED_FUNCTIONS: readonly WrappedFunctionName[] = [\n  'generateText',\n  'streamText',\n  'generateObject',\n  'streamObject',\n  'embed',\n  'embedMany',\n  'rerank',\n] as const;\n\n/**\n * Wrap the `ai` module namespace so that every `generateText` / `streamText` /\n * `generateObject` / `streamObject` call:\n *\n *   1. Opens a parent OTel span on the active TracerProvider.\n *   2. Forces `experimental_telemetry: { isEnabled: true }`, merging user metadata.\n *   3. Records input/output on the parent span and propagates errors.\n *\n * Other exports (types, helpers) pass through unchanged.\n */\nexport function wrapAISDK<T extends Record<string, unknown>>(aiModule: T): T {\n  const wrapped: Record<string, unknown> = { ...aiModule };\n\n  for (const name of WRAPPED_FUNCTIONS) {\n    const original = aiModule[name];\n    if (typeof original !== 'function') continue;\n\n    if (name === 'streamText' || name === 'streamObject') {\n      wrapped[name] = createStreamWrapper(name, original as (opts: any) => unknown);\n    } else {\n      wrapped[name] = createAsyncWrapper(name, original as (opts: any) => Promise<unknown>);\n    }\n  }\n\n  return wrapped as T;\n}\n\nfunction rootSpanKind(name: WrappedFunctionName): string {\n  if (name === 'embed' || name === 'embedMany' || name === 'rerank') return 'CHAIN';\n  return 'WORKFLOW';\n}\n\n\nfunction getParentContext() {\n  // Our parent comes solely from the private span store; a\n  // foreign provider's active span must never become our ancestor.\n  return getNeatlogsParentContext();\n}\n\nfunction createAsyncWrapper(\n  name: WrappedFunctionName,\n  original: (opts: any) => Promise<unknown>,\n): (opts: any) => Promise<unknown> {\n  return async function wrappedAsyncFn(opts: any): Promise<unknown> {\n    const tracer = getNeatlogsTracer(TRACER_NAME);\n    // startSpan (NOT startActiveSpan) + withNeatlogsSpan: startActiveSpan would\n    // push our span onto the GLOBAL OTel context, so a foreign tracer's\n    // startSpan() inside generateText() would read it as parent and inherit our\n    // trace id. withNeatlogsSpan carries the parent in the private store in\n    // the private context, leaving the global context untouched.\n    const parentContext = getParentContext();\n    const span = tracer.startSpan(\n      `ai.${name}`,\n      { attributes: { 'openinference.span.kind': rootSpanKind(name) } },\n      parentContext,\n    );\n    return withNeatlogsSpan(\n      span,\n      async () => {\n        try {\n          const isEmbedOrRerank = name === 'embed' || name === 'embedMany' || name === 'rerank';\n          if (!isEmbedOrRerank) {\n            setInputValue(span, opts);\n          }\n          if (name === 'rerank' && opts?.query) {\n            span.setAttribute('ai.rerank.query', String(opts.query));\n          }\n          const merged = mergeTelemetry(opts);\n          const result = await original(merged);\n          if (!isEmbedOrRerank) {\n            setOutputValue(span, result);\n          }\n          return result;\n        } catch (err) {\n          recordSpanError(span, err);\n          throw err;\n        } finally {\n          span.end();\n        }\n      },\n      parentContext,\n    );\n  };\n}\n\n// streamText/streamObject return synchronously while the model keeps producing\n// tokens for seconds afterwards. Ending the span in a `finally` (as a plain sync\n// wrapper would) closes it in ~2ms with no output — the output only exists once\n// the stream finishes. Instead we keep the span open and end it from the AI SDK's\n// `onFinish` callback, where the final text/object is available. Any user-provided\n// `onFinish` is preserved and invoked first.\nfunction createStreamWrapper(\n  name: WrappedFunctionName,\n  original: (opts: any) => unknown,\n): (opts: any) => unknown {\n  return function wrappedStreamFn(opts: any): unknown {\n    const tracer = getNeatlogsTracer(TRACER_NAME);\n    // startSpan + withNeatlogsSpan (see createAsyncWrapper) so streamText's\n    // internals never see our span on the global OTel context. The span stays\n    // open past the run scope and is ended from onFinish/onError.\n    const parentContext = getParentContext();\n    const span = tracer.startSpan(\n      `ai.${name}`,\n      { attributes: { 'openinference.span.kind': rootSpanKind(name) } },\n      parentContext,\n    );\n    return withNeatlogsSpan(\n      span,\n      () => {\n        let spanEnded = false;\n        const endOnce = () => {\n          if (spanEnded) return;\n          spanEnded = true;\n          span.end();\n        };\n        try {\n          setInputValue(span, opts);\n          const merged = mergeTelemetry(opts);\n          const userOnFinish = opts?.onFinish;\n          const userOnError = opts?.onError;\n          const wrappedOpts = {\n            ...merged,\n            onFinish: async (event: any) => {\n              try {\n                setStreamOutputValue(span, event);\n              } finally {\n                endOnce();\n              }\n              if (typeof userOnFinish === 'function') {\n                return userOnFinish(event);\n              }\n            },\n            onError: (event: any) => {\n              recordSpanError(span, (event && event.error) ?? event);\n              endOnce();\n              if (typeof userOnError === 'function') {\n                return userOnError(event);\n              }\n            },\n          };\n          return original(wrappedOpts);\n        } catch (err) {\n          // Synchronous throw (e.g. bad arguments) — the stream never started.\n          recordSpanError(span, err);\n          endOnce();\n          throw err;\n        }\n      },\n      parentContext,\n    );\n  };\n}\n\nfunction mergeTelemetry(opts: any): any {\n  const baseTelemetry: AITelemetryConfig = createAITelemetry({\n    metadata: opts?.experimental_telemetry?.metadata,\n  });\n  return {\n    ...opts,\n    experimental_telemetry: {\n      ...opts?.experimental_telemetry,\n      ...baseTelemetry,\n    },\n  };\n}\n\n\nfunction recordSpanError(span: Span, err: unknown): void {\n  if (err instanceof Error) {\n    span.setStatus({ code: SpanStatusCode.ERROR, message: err.message });\n    span.recordException(err);\n  } else {\n    span.setStatus({ code: SpanStatusCode.ERROR, message: String(err) });\n  }\n}\n","/**\n * Neatlogs-owned tracing state.\n *\n * Spans are created by the private Neatlogs provider and their parent is carried\n * in a private context key. The process-global OpenTelemetry span is\n * deliberately left untouched so other observability SDKs cannot export or\n * become parents of Neatlogs spans.\n */\n\nimport { AsyncLocalStorage } from 'node:async_hooks';\nimport {\n  ROOT_CONTEXT,\n  INVALID_SPAN_CONTEXT,\n  createContextKey,\n  trace as otelTrace,\n  type Context,\n  type Span,\n  type SpanOptions,\n  type Tracer,\n  type TracerProvider,\n} from '@opentelemetry/api';\n\n// Carries the trace-ROOT span down the private context so descendants can target\n// it (e.g. setTraceOutput). getActiveNeatlogsSpan() returns the innermost span,\n// which is not the root once nested; this key preserves the root reference.\nconst NEATLOGS_ROOT_SPAN_KEY = createContextKey('neatlogs.root_span');\n\n// Entry points are bundled independently (`neatlogs`, `neatlogs/openai`,\n// `neatlogs/ai`, `neatlogs/mastra`, … each in both CJS and ESM), so every piece\n// of shared tracing state — the private span store AND the resolved provider /\n// provider — must live on `globalThis` behind a `Symbol.for` key.\n// Otherwise `init()` (run from the `neatlogs` bundle) sets `_provider` in ITS\n// module copy while a wrapper imported from `neatlogs/openai` reads a different,\n// still-null copy and silently falls back to the foreign global provider.\nconst PRIVATE_SPAN_STORAGE_KEY = Symbol.for(\n  'neatlogs.private_span_async_local_storage',\n);\nconst PRIVATE_PROVIDER_STATE_KEY = Symbol.for(\n  'neatlogs.private_provider_state',\n);\ninterface PrivateProviderState {\n  provider: TracerProvider | null;\n}\ntype NeatlogsGlobal = typeof globalThis & {\n  [PRIVATE_SPAN_STORAGE_KEY]?: AsyncLocalStorage<Context>;\n  [PRIVATE_PROVIDER_STATE_KEY]?: PrivateProviderState;\n};\nconst neatlogsGlobal = globalThis as NeatlogsGlobal;\n// Stores the full Neatlogs Context (parent span PLUS any threaded values such as\n// trace()'s prompt-template keys), not just the span. We never\n// activate the OTel global context, so this private store is the ONLY channel\n// through which those values propagate down to descendant spans.\nconst privateContextStorage =\n  neatlogsGlobal[PRIVATE_SPAN_STORAGE_KEY] ??\n  (neatlogsGlobal[PRIVATE_SPAN_STORAGE_KEY] = new AsyncLocalStorage<Context>());\nconst providerState: PrivateProviderState =\n  neatlogsGlobal[PRIVATE_PROVIDER_STATE_KEY] ??\n  (neatlogsGlobal[PRIVATE_PROVIDER_STATE_KEY] = {\n    provider: null,\n  });\n// Wrappers may be constructed or accidentally invoked before init(). They must\n// never fall back to a foreign process-global provider, so pre-init calls use a\n// local no-op tracer and safely emit no exported spans.\nconst preInitTracer: Tracer = {\n  startSpan(): Span {\n    return otelTrace.wrapSpanContext(INVALID_SPAN_CONTEXT);\n  },\n  startActiveSpan<F extends (span: Span) => unknown>(\n    _name: string,\n    arg2?: SpanOptions | Context | F,\n    arg3?: Context | F,\n    arg4?: F,\n  ): ReturnType<F> {\n    const fn =\n      typeof arg2 === 'function'\n        ? arg2\n        : typeof arg3 === 'function'\n          ? arg3\n          : arg4;\n    return fn!(otelTrace.wrapSpanContext(INVALID_SPAN_CONTEXT)) as ReturnType<F>;\n  },\n};\n\n/** @internal Configure the provider used by Neatlogs-created spans. */\nexport function _setNeatlogsProvider(provider: TracerProvider | null): void {\n  providerState.provider = provider;\n}\n\n/** Resolve a tracer from the private provider when one is configured. */\nexport function getNeatlogsTracer(name: string): Tracer {\n  return providerState.provider?.getTracer(name) ?? preInitTracer;\n}\n\n/**\n * @internal The private Neatlogs provider, or null before\n * init(). Used by integrations that must repoint a self-instrumenting library's\n * captured provider onto ours.\n */\nexport function getNeatlogsProvider(): TracerProvider | null {\n  return providerState.provider;\n}\n\n/**\n * Wrap a tracer so that spans it creates parent from — and, for\n * `startActiveSpan`, activate on — the PRIVATE Neatlogs context instead of the\n * global one.\n *\n * We hand this to libraries that create their own spans off a tracer we give\n * them (the Vercel AI SDK's `experimental_telemetry.tracer`, which calls\n * `tracer.startActiveSpan()` internally). Without the facade the AI SDK's native\n * spans would parent from `context.active()` — the foreign co-tenant's context —\n * and `startActiveSpan` would push them onto the GLOBAL context, so a foreign\n * tracer's next span reads our native span as its parent. Both directions leak.\n *\n */\nexport function isolateTracer(tracer: Tracer): Tracer {\n  const facade: Tracer = {\n    startSpan(name: string, options?: SpanOptions, context?: Context): Span {\n      const parent = context ?? getNeatlogsActiveContext();\n      return tracer.startSpan(name, options, parent);\n    },\n    startActiveSpan<F extends (span: Span) => unknown>(\n      name: string,\n      arg2?: SpanOptions | Context | F,\n      arg3?: Context | F,\n      arg4?: F,\n    ): ReturnType<F> {\n      // Normalize the 2/3/4-arg overloads of startActiveSpan.\n      let options: SpanOptions | undefined;\n      let context: Context | undefined;\n      let fn: F;\n      if (typeof arg2 === 'function') {\n        fn = arg2 as F;\n      } else if (typeof arg3 === 'function') {\n        options = arg2 as SpanOptions;\n        fn = arg3 as F;\n      } else {\n        options = arg2 as SpanOptions;\n        context = arg3 as Context;\n        fn = arg4 as F;\n      }\n      const parent = context ?? getNeatlogsActiveContext();\n      const span = tracer.startSpan(name, options, parent);\n      return withNeatlogsSpan(span, () => fn(span), parent) as ReturnType<F>;\n    },\n  };\n  return facade;\n}\n\n/**\n * The base context to build new Neatlogs spans and values on. NEVER contains a\n * foreign provider's span: it reads our private store, or ROOT_CONTEXT when\n * nothing is active.\n */\nexport function getNeatlogsActiveContext(): Context {\n  return privateContextStorage.getStore() ?? ROOT_CONTEXT;\n}\n\n/** Return only the active Neatlogs span (never a foreign provider's span). */\nexport function getActiveNeatlogsSpan(): Span | undefined {\n  return otelTrace.getSpan(getNeatlogsActiveContext());\n}\n\n/**\n * Return the trace-ROOT Neatlogs span, or undefined when no trace is active.\n *\n * Unlike {@link getActiveNeatlogsSpan} (innermost), this is the outermost span\n * of the current trace — the one the backend derives trace-level output from\n * (`parent_span_id=''`). It is stashed on the private context the first time a\n * span becomes active, so nested calls still resolve to the root.\n */\nexport function getNeatlogsRootSpan(): Span | undefined {\n  return getNeatlogsActiveContext().getValue(NEATLOGS_ROOT_SPAN_KEY) as\n    | Span\n    | undefined;\n}\n\n/**\n * Build a parent context that cannot contain a foreign provider's span.\n *\n * The active Neatlogs context already carries the parent span AND any values\n * the caller threaded in upstream (e.g. `trace()`'s prompt-template values), so\n * those values reach the span processor via `onStart(parentContext)`. An\n * explicit `baseContext` (a caller's own value-carrying context) is honored as-is.\n */\nexport function getNeatlogsParentContext(baseContext?: Context): Context {\n  return baseContext ?? getNeatlogsActiveContext();\n}\n\n/**\n * A base context for callers that thread parent linkage themselves\n * (callback/event handlers that keep their own run-id → span map: LangChain,\n * OpenAI-Agents, Claude Agent SDK).\n *\n * This is the ACTIVE Neatlogs context — the private store's\n * context if a Neatlogs `trace()`/`span()` encloses this call, else\n * ROOT_CONTEXT. So a handler's own root/entry span nests under an enclosing\n * Neatlogs trace (preserving its session + end-user id) when one exists, and\n * auto-roots cleanly when one doesn't. A foreign provider's active span can\n * never leak in as an ancestor because the global context is never read.\n */\nexport function getNeatlogsBaseContext(baseContext?: Context): Context {\n  return baseContext ?? getNeatlogsActiveContext();\n}\n\n/**\n * Build an execution context for a Neatlogs span while preserving the active\n * foreign context. The span rides our private context store.\n */\nexport function getNeatlogsExecutionContext(\n  span: Span,\n  baseContext: Context = ROOT_CONTEXT,\n): Context {\n  return otelTrace.setSpan(baseContext, span);\n}\n\n/**\n * Run a callback with a Neatlogs span active under the appropriate policy.\n *\n * The stored context is `setSpan(base, span)` — carrying the span PLUS whatever\n * values `base` holds — so threaded values (prompt templates)\n * propagate to descendant spans through our private store instead of the global\n * OTel context we deliberately never touch.\n */\nexport function withNeatlogsSpan<T>(\n  span: Span,\n  fn: () => T,\n  baseContext?: Context,\n): T {\n  const base = baseContext ?? getNeatlogsActiveContext();\n  let ctx = otelTrace.setSpan(base, span);\n  // The first span activated in a context with no root recorded IS the root of\n  // this trace; remember it so descendants (setTraceOutput) can target it.\n  if (base.getValue(NEATLOGS_ROOT_SPAN_KEY) === undefined) {\n    ctx = ctx.setValue(NEATLOGS_ROOT_SPAN_KEY, span);\n  }\n  return privateContextStorage.run(ctx, fn);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAaA,IAAAA,cAAuD;;;ACJvD,8BAAkC;AAClC,iBAUO;AAKP,IAAM,6BAAyB,6BAAiB,oBAAoB;AASpE,IAAM,2BAA2B,uBAAO;AAAA,EACtC;AACF;AACA,IAAM,6BAA6B,uBAAO;AAAA,EACxC;AACF;AAQA,IAAM,iBAAiB;AAKvB,IAAM,wBACJ,eAAe,wBAAwB,MACtC,eAAe,wBAAwB,IAAI,IAAI,0CAA2B;AAC7E,IAAM,gBACJ,eAAe,0BAA0B,MACxC,eAAe,0BAA0B,IAAI;AAAA,EAC5C,UAAU;AACZ;AAIF,IAAM,gBAAwB;AAAA,EAC5B,YAAkB;AAChB,WAAO,WAAAC,MAAU,gBAAgB,+BAAoB;AAAA,EACvD;AAAA,EACA,gBACE,OACA,MACA,MACA,MACe;AACf,UAAM,KACJ,OAAO,SAAS,aACZ,OACA,OAAO,SAAS,aACd,OACA;AACR,WAAO,GAAI,WAAAA,MAAU,gBAAgB,+BAAoB,CAAC;AAAA,EAC5D;AACF;AAQO,SAAS,kBAAkB,MAAsB;AACtD,SAAO,cAAc,UAAU,UAAU,IAAI,KAAK;AACpD;AAwBO,SAAS,cAAc,QAAwB;AACpD,QAAM,SAAiB;AAAA,IACrB,UAAU,MAAc,SAAuB,SAAyB;AACtE,YAAM,SAAS,WAAW,yBAAyB;AACnD,aAAO,OAAO,UAAU,MAAM,SAAS,MAAM;AAAA,IAC/C;AAAA,IACA,gBACE,MACA,MACA,MACA,MACe;AAEf,UAAI;AACJ,UAAI;AACJ,UAAI;AACJ,UAAI,OAAO,SAAS,YAAY;AAC9B,aAAK;AAAA,MACP,WAAW,OAAO,SAAS,YAAY;AACrC,kBAAU;AACV,aAAK;AAAA,MACP,OAAO;AACL,kBAAU;AACV,kBAAU;AACV,aAAK;AAAA,MACP;AACA,YAAM,SAAS,WAAW,yBAAyB;AACnD,YAAM,OAAO,OAAO,UAAU,MAAM,SAAS,MAAM;AACnD,aAAO,iBAAiB,MAAM,MAAM,GAAG,IAAI,GAAG,MAAM;AAAA,IACtD;AAAA,EACF;AACA,SAAO;AACT;AAOO,SAAS,2BAAoC;AAClD,SAAO,sBAAsB,SAAS,KAAK;AAC7C;AA6BO,SAAS,yBAAyB,aAAgC;AACvE,SAAO,eAAe,yBAAyB;AACjD;AAqCO,SAAS,iBACd,MACA,IACA,aACG;AACH,QAAM,OAAO,eAAe,yBAAyB;AACrD,MAAI,MAAM,WAAAC,MAAU,QAAQ,MAAM,IAAI;AAGtC,MAAI,KAAK,SAAS,sBAAsB,MAAM,QAAW;AACvD,UAAM,IAAI,SAAS,wBAAwB,IAAI;AAAA,EACjD;AACA,SAAO,sBAAsB,IAAI,KAAK,EAAE;AAC1C;;;AD7NA,IAAM,cAAc;AAgBb,SAAS,kBACd,OAAiC,CAAC,GACf;AACnB,QAAM,WAAW,KAAK,YAAY,CAAC;AACnC,SAAO;AAAA,IACL,WAAW;AAAA,IACX,cAAc;AAAA,IACd,eAAe;AAAA;AAAA;AAAA;AAAA;AAAA,IAKf,QAAQ,cAAc,kBAAkB,WAAW,CAAC;AAAA,IACpD,UAAU,EAAE,GAAG,UAAU,iBAAiB,KAAK;AAAA,EACjD;AACF;AAIA,SAAS,cAAc,OAAwB;AAC7C,MAAI;AACF,WAAO,KAAK,UAAU,KAAK;AAAA,EAC7B,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAEA,SAAS,cAAc,MAAY,MAAqC;AAEtE,MAAI,SAAS,YAAY,QAAQ,cAAc,OAAO;AACpD,UAAM,QAAQ,KAAK,YAAY,KAAK;AACpC,UAAMC,eAAc,cAAc,KAAK;AACvC,QAAIA,cAAa;AACf,WAAK,aAAa,eAAeA,YAAW;AAAA,IAC9C;AACA;AAAA,EACF;AACA,QAAM,cAAc,cAAc,IAAI;AACtC,MAAI,aAAa;AACf,SAAK,aAAa,eAAe,WAAW;AAAA,EAC9C;AACF;AAEA,SAAS,eAAe,MAAY,QAAuB;AACzD,MAAI,UAAU,OAAO,WAAW,UAAU;AACxC,UAAM,IAAI;AAEV,QAAI,UAAU,KAAK,kBAAkB,GAAG;AACtC,YAAM,OAAO,OAAO,EAAE,QAAQ,EAAE;AAChC,UAAI,MAAM;AACR,aAAK,aAAa,gBAAgB,IAAI;AAAA,MACxC;AACA,UAAI,EAAE,cAAc;AAClB,aAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,MAClE;AACA;AAAA,IACF;AAGA,QAAI,YAAY,KAAK,kBAAkB,GAAG;AACxC,UAAI,EAAE,WAAW,QAAW;AAC1B,aAAK,aAAa,gBAAgB,cAAc,EAAE,MAAM,CAAC;AAAA,MAC3D;AACA,UAAI,EAAE,cAAc;AAClB,aAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,MAClE;AACA;AAAA,IACF;AAAA,EACF;AACA,QAAM,cAAc,cAAc,MAAM;AACxC,MAAI,aAAa;AACf,SAAK,aAAa,gBAAgB,WAAW;AAAA,EAC/C;AACF;AAOA,SAAS,qBAAqB,MAAY,OAAsB;AAC9D,MAAI,CAAC,SAAS,OAAO,UAAU,SAAU;AACzC,QAAM,IAAI;AACV,MAAI,OAAO,EAAE,SAAS,YAAY,EAAE,MAAM;AACxC,SAAK,aAAa,gBAAgB,EAAE,IAAI;AAAA,EAC1C,WAAW,YAAY,KAAK,EAAE,WAAW,QAAW;AAClD,UAAM,cAAc,cAAc,EAAE,MAAM;AAC1C,QAAI,YAAa,MAAK,aAAa,gBAAgB,WAAW;AAAA,EAChE;AACA,MAAI,EAAE,cAAc;AAClB,SAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,EAClE;AACF;AAMA,IAAM,oBAAoD;AAAA,EACxD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAYO,SAAS,UAA6C,UAAgB;AAC3E,QAAM,UAAmC,EAAE,GAAG,SAAS;AAEvD,aAAW,QAAQ,mBAAmB;AACpC,UAAM,WAAW,SAAS,IAAI;AAC9B,QAAI,OAAO,aAAa,WAAY;AAEpC,QAAI,SAAS,gBAAgB,SAAS,gBAAgB;AACpD,cAAQ,IAAI,IAAI,oBAAoB,MAAM,QAAkC;AAAA,IAC9E,OAAO;AACL,cAAQ,IAAI,IAAI,mBAAmB,MAAM,QAA2C;AAAA,IACtF;AAAA,EACF;AAEA,SAAO;AACT;AAEA,SAAS,aAAa,MAAmC;AACvD,MAAI,SAAS,WAAW,SAAS,eAAe,SAAS,SAAU,QAAO;AAC1E,SAAO;AACT;AAGA,SAAS,mBAAmB;AAG1B,SAAO,yBAAyB;AAClC;AAEA,SAAS,mBACP,MACA,UACiC;AACjC,SAAO,eAAe,eAAe,MAA6B;AAChE,UAAM,SAAS,kBAAkB,WAAW;AAM5C,UAAM,gBAAgB,iBAAiB;AACvC,UAAM,OAAO,OAAO;AAAA,MAClB,MAAM,IAAI;AAAA,MACV,EAAE,YAAY,EAAE,2BAA2B,aAAa,IAAI,EAAE,EAAE;AAAA,MAChE;AAAA,IACF;AACA,WAAO;AAAA,MACL;AAAA,MACA,YAAY;AACV,YAAI;AACF,gBAAM,kBAAkB,SAAS,WAAW,SAAS,eAAe,SAAS;AAC7E,cAAI,CAAC,iBAAiB;AACpB,0BAAc,MAAM,IAAI;AAAA,UAC1B;AACA,cAAI,SAAS,YAAY,MAAM,OAAO;AACpC,iBAAK,aAAa,mBAAmB,OAAO,KAAK,KAAK,CAAC;AAAA,UACzD;AACA,gBAAM,SAAS,eAAe,IAAI;AAClC,gBAAM,SAAS,MAAM,SAAS,MAAM;AACpC,cAAI,CAAC,iBAAiB;AACpB,2BAAe,MAAM,MAAM;AAAA,UAC7B;AACA,iBAAO;AAAA,QACT,SAAS,KAAK;AACZ,0BAAgB,MAAM,GAAG;AACzB,gBAAM;AAAA,QACR,UAAE;AACA,eAAK,IAAI;AAAA,QACX;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAQA,SAAS,oBACP,MACA,UACwB;AACxB,SAAO,SAAS,gBAAgB,MAAoB;AAClD,UAAM,SAAS,kBAAkB,WAAW;AAI5C,UAAM,gBAAgB,iBAAiB;AACvC,UAAM,OAAO,OAAO;AAAA,MAClB,MAAM,IAAI;AAAA,MACV,EAAE,YAAY,EAAE,2BAA2B,aAAa,IAAI,EAAE,EAAE;AAAA,MAChE;AAAA,IACF;AACA,WAAO;AAAA,MACL;AAAA,MACA,MAAM;AACJ,YAAI,YAAY;AAChB,cAAM,UAAU,MAAM;AACpB,cAAI,UAAW;AACf,sBAAY;AACZ,eAAK,IAAI;AAAA,QACX;AACA,YAAI;AACF,wBAAc,MAAM,IAAI;AACxB,gBAAM,SAAS,eAAe,IAAI;AAClC,gBAAM,eAAe,MAAM;AAC3B,gBAAM,cAAc,MAAM;AAC1B,gBAAM,cAAc;AAAA,YAClB,GAAG;AAAA,YACH,UAAU,OAAO,UAAe;AAC9B,kBAAI;AACF,qCAAqB,MAAM,KAAK;AAAA,cAClC,UAAE;AACA,wBAAQ;AAAA,cACV;AACA,kBAAI,OAAO,iBAAiB,YAAY;AACtC,uBAAO,aAAa,KAAK;AAAA,cAC3B;AAAA,YACF;AAAA,YACA,SAAS,CAAC,UAAe;AACvB,8BAAgB,OAAO,SAAS,MAAM,UAAU,KAAK;AACrD,sBAAQ;AACR,kBAAI,OAAO,gBAAgB,YAAY;AACrC,uBAAO,YAAY,KAAK;AAAA,cAC1B;AAAA,YACF;AAAA,UACF;AACA,iBAAO,SAAS,WAAW;AAAA,QAC7B,SAAS,KAAK;AAEZ,0BAAgB,MAAM,GAAG;AACzB,kBAAQ;AACR,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,eAAe,MAAgB;AACtC,QAAM,gBAAmC,kBAAkB;AAAA,IACzD,UAAU,MAAM,wBAAwB;AAAA,EAC1C,CAAC;AACD,SAAO;AAAA,IACL,GAAG;AAAA,IACH,wBAAwB;AAAA,MACtB,GAAG,MAAM;AAAA,MACT,GAAG;AAAA,IACL;AAAA,EACF;AACF;AAGA,SAAS,gBAAgB,MAAY,KAAoB;AACvD,MAAI,eAAe,OAAO;AACxB,SAAK,UAAU,EAAE,MAAM,2BAAe,OAAO,SAAS,IAAI,QAAQ,CAAC;AACnE,SAAK,gBAAgB,GAAG;AAAA,EAC1B,OAAO;AACL,SAAK,UAAU,EAAE,MAAM,2BAAe,OAAO,SAAS,OAAO,GAAG,EAAE,CAAC;AAAA,EACrE;AACF;","names":["import_api","otelTrace","otelTrace","stringified"]}

package/dist/ai-sdk.mjs

@@ -6,8 +6,10 @@ import { AsyncLocalStorage } from "async_hooks";
 import {
   ROOT_CONTEXT,
   INVALID_SPAN_CONTEXT,
+  createContextKey,
   trace as otelTrace
 } from "@opentelemetry/api";
+var NEATLOGS_ROOT_SPAN_KEY = createContextKey("neatlogs.root_span");
 var PRIVATE_SPAN_STORAGE_KEY = /* @__PURE__ */ Symbol.for(
   "neatlogs.private_span_async_local_storage"
 );
@@ -66,7 +68,10 @@ function getNeatlogsParentContext(baseContext) {
 }
 function withNeatlogsSpan(span, fn, baseContext) {
   const base = baseContext ?? getNeatlogsActiveContext();
-  const ctx = otelTrace.setSpan(base, span);
+  let ctx = otelTrace.setSpan(base, span);
+  if (base.getValue(NEATLOGS_ROOT_SPAN_KEY) === void 0) {
+    ctx = ctx.setValue(NEATLOGS_ROOT_SPAN_KEY, span);
+  }
   return privateContextStorage.run(ctx, fn);
 }
 

package/dist/ai-sdk.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/ai-sdk.ts","../src/core/provider.ts"],"sourcesContent":["/**\n * Vercel AI SDK wrapper — inline implementation.\n *\n * Wraps `generateText`, `streamText`, `generateObject`, `streamObject` from\n * the `ai` package with OTel parent spans + forced telemetry. Static export,\n * no dynamic imports — bundler-friendly (works with Turbopack, webpack, esbuild).\n *\n * Usage:\n *   import { wrapAISDK } from 'neatlogs';\n *   import * as ai from 'ai';\n *   const { streamText, generateText } = wrapAISDK(ai);\n */\n\nimport { SpanStatusCode, type Span, type Tracer } from '@opentelemetry/api';\nimport { getNeatlogsTracer, getNeatlogsParentContext, isolateTracer, withNeatlogsSpan } from './core/provider.js';\n\nconst TRACER_NAME = 'neatlogs.ai-sdk';\n\n// -- Telemetry config --------------------------------------------------------\n\nexport interface CreateAITelemetryOptions {\n  metadata?: Record<string, unknown>;\n}\n\nexport interface AITelemetryConfig {\n  isEnabled: true;\n  recordInputs: true;\n  recordOutputs: true;\n  tracer: Tracer;\n  metadata: Record<string, unknown>;\n}\n\nexport function createAITelemetry(\n  opts: CreateAITelemetryOptions = {},\n): AITelemetryConfig {\n  const userMeta = opts.metadata ?? {};\n  return {\n    isEnabled: true,\n    recordInputs: true,\n    recordOutputs: true,\n    // Hand the AI SDK an isolation-aware tracer: it calls startActiveSpan()\n    // internally, which would otherwise parent its native spans from the foreign\n    // global context AND push them onto it (so a co-tenant's next span inherits\n    // ours). The facade routes both through the private Neatlogs context.\n    tracer: isolateTracer(getNeatlogsTracer(TRACER_NAME)),\n    metadata: { ...userMeta, neatlogsWrapped: true },\n  };\n}\n\n// -- Span attributes ---------------------------------------------------------\n\nfunction safeStringify(value: unknown): string {\n  try {\n    return JSON.stringify(value);\n  } catch {\n    return '';\n  }\n}\n\nfunction setInputValue(span: Span, opts: Record<string, unknown>): void {\n  // For generateText/streamText — only capture prompt or messages, not full model config\n  if (opts && ('prompt' in opts || 'messages' in opts)) {\n    const input = opts.messages ?? opts.prompt;\n    const stringified = safeStringify(input);\n    if (stringified) {\n      span.setAttribute('input.value', stringified);\n    }\n    return;\n  }\n  const stringified = safeStringify(opts);\n  if (stringified) {\n    span.setAttribute('input.value', stringified);\n  }\n}\n\nfunction setOutputValue(span: Span, result: unknown): void {\n  if (result && typeof result === 'object') {\n    const r = result as Record<string, unknown>;\n    // GenerateTextResult / StreamTextResult — extract meaningful fields\n    if ('text' in r && 'finishReason' in r) {\n      const text = String(r.text ?? '');\n      if (text) {\n        span.setAttribute('output.value', text);\n      }\n      if (r.finishReason) {\n        span.setAttribute('gen_ai.finish_reason', String(r.finishReason));\n      }\n      return;\n    }\n    // GenerateObjectResult — the structured object is the output, not `text`.\n    // Without this the envelope (object+usage+response+…) gets stringified whole.\n    if ('object' in r && 'finishReason' in r) {\n      if (r.object !== undefined) {\n        span.setAttribute('output.value', safeStringify(r.object));\n      }\n      if (r.finishReason) {\n        span.setAttribute('gen_ai.finish_reason', String(r.finishReason));\n      }\n      return;\n    }\n  }\n  const stringified = safeStringify(result);\n  if (stringified) {\n    span.setAttribute('output.value', stringified);\n  }\n}\n\n// Extract output from a streamText/streamObject `onFinish` event. The event\n// extends StepResult, so `text` (streamText) / `object` (streamObject) sit at\n// the top level alongside `finishReason` — but the event also carries `steps`,\n// `usage`, etc., so we pull only the meaningful fields instead of stringifying\n// the whole envelope.\nfunction setStreamOutputValue(span: Span, event: unknown): void {\n  if (!event || typeof event !== 'object') return;\n  const e = event as Record<string, unknown>;\n  if (typeof e.text === 'string' && e.text) {\n    span.setAttribute('output.value', e.text);\n  } else if ('object' in e && e.object !== undefined) {\n    const stringified = safeStringify(e.object);\n    if (stringified) span.setAttribute('output.value', stringified);\n  }\n  if (e.finishReason) {\n    span.setAttribute('gen_ai.finish_reason', String(e.finishReason));\n  }\n}\n\n// -- Wrapping ----------------------------------------------------------------\n\ntype WrappedFunctionName = 'generateText' | 'streamText' | 'generateObject' | 'streamObject' | 'embed' | 'embedMany' | 'rerank';\n\nconst WRAPPED_FUNCTIONS: readonly WrappedFunctionName[] = [\n  'generateText',\n  'streamText',\n  'generateObject',\n  'streamObject',\n  'embed',\n  'embedMany',\n  'rerank',\n] as const;\n\n/**\n * Wrap the `ai` module namespace so that every `generateText` / `streamText` /\n * `generateObject` / `streamObject` call:\n *\n *   1. Opens a parent OTel span on the active TracerProvider.\n *   2. Forces `experimental_telemetry: { isEnabled: true }`, merging user metadata.\n *   3. Records input/output on the parent span and propagates errors.\n *\n * Other exports (types, helpers) pass through unchanged.\n */\nexport function wrapAISDK<T extends Record<string, unknown>>(aiModule: T): T {\n  const wrapped: Record<string, unknown> = { ...aiModule };\n\n  for (const name of WRAPPED_FUNCTIONS) {\n    const original = aiModule[name];\n    if (typeof original !== 'function') continue;\n\n    if (name === 'streamText' || name === 'streamObject') {\n      wrapped[name] = createStreamWrapper(name, original as (opts: any) => unknown);\n    } else {\n      wrapped[name] = createAsyncWrapper(name, original as (opts: any) => Promise<unknown>);\n    }\n  }\n\n  return wrapped as T;\n}\n\nfunction rootSpanKind(name: WrappedFunctionName): string {\n  if (name === 'embed' || name === 'embedMany' || name === 'rerank') return 'CHAIN';\n  return 'WORKFLOW';\n}\n\n\nfunction getParentContext() {\n  // Our parent comes solely from the private span store; a\n  // foreign provider's active span must never become our ancestor.\n  return getNeatlogsParentContext();\n}\n\nfunction createAsyncWrapper(\n  name: WrappedFunctionName,\n  original: (opts: any) => Promise<unknown>,\n): (opts: any) => Promise<unknown> {\n  return async function wrappedAsyncFn(opts: any): Promise<unknown> {\n    const tracer = getNeatlogsTracer(TRACER_NAME);\n    // startSpan (NOT startActiveSpan) + withNeatlogsSpan: startActiveSpan would\n    // push our span onto the GLOBAL OTel context, so a foreign tracer's\n    // startSpan() inside generateText() would read it as parent and inherit our\n    // trace id. withNeatlogsSpan carries the parent in the private store in\n    // the private context, leaving the global context untouched.\n    const parentContext = getParentContext();\n    const span = tracer.startSpan(\n      `ai.${name}`,\n      { attributes: { 'openinference.span.kind': rootSpanKind(name) } },\n      parentContext,\n    );\n    return withNeatlogsSpan(\n      span,\n      async () => {\n        try {\n          const isEmbedOrRerank = name === 'embed' || name === 'embedMany' || name === 'rerank';\n          if (!isEmbedOrRerank) {\n            setInputValue(span, opts);\n          }\n          if (name === 'rerank' && opts?.query) {\n            span.setAttribute('ai.rerank.query', String(opts.query));\n          }\n          const merged = mergeTelemetry(opts);\n          const result = await original(merged);\n          if (!isEmbedOrRerank) {\n            setOutputValue(span, result);\n          }\n          return result;\n        } catch (err) {\n          recordSpanError(span, err);\n          throw err;\n        } finally {\n          span.end();\n        }\n      },\n      parentContext,\n    );\n  };\n}\n\n// streamText/streamObject return synchronously while the model keeps producing\n// tokens for seconds afterwards. Ending the span in a `finally` (as a plain sync\n// wrapper would) closes it in ~2ms with no output — the output only exists once\n// the stream finishes. Instead we keep the span open and end it from the AI SDK's\n// `onFinish` callback, where the final text/object is available. Any user-provided\n// `onFinish` is preserved and invoked first.\nfunction createStreamWrapper(\n  name: WrappedFunctionName,\n  original: (opts: any) => unknown,\n): (opts: any) => unknown {\n  return function wrappedStreamFn(opts: any): unknown {\n    const tracer = getNeatlogsTracer(TRACER_NAME);\n    // startSpan + withNeatlogsSpan (see createAsyncWrapper) so streamText's\n    // internals never see our span on the global OTel context. The span stays\n    // open past the run scope and is ended from onFinish/onError.\n    const parentContext = getParentContext();\n    const span = tracer.startSpan(\n      `ai.${name}`,\n      { attributes: { 'openinference.span.kind': rootSpanKind(name) } },\n      parentContext,\n    );\n    return withNeatlogsSpan(\n      span,\n      () => {\n        let spanEnded = false;\n        const endOnce = () => {\n          if (spanEnded) return;\n          spanEnded = true;\n          span.end();\n        };\n        try {\n          setInputValue(span, opts);\n          const merged = mergeTelemetry(opts);\n          const userOnFinish = opts?.onFinish;\n          const userOnError = opts?.onError;\n          const wrappedOpts = {\n            ...merged,\n            onFinish: async (event: any) => {\n              try {\n                setStreamOutputValue(span, event);\n              } finally {\n                endOnce();\n              }\n              if (typeof userOnFinish === 'function') {\n                return userOnFinish(event);\n              }\n            },\n            onError: (event: any) => {\n              recordSpanError(span, (event && event.error) ?? event);\n              endOnce();\n              if (typeof userOnError === 'function') {\n                return userOnError(event);\n              }\n            },\n          };\n          return original(wrappedOpts);\n        } catch (err) {\n          // Synchronous throw (e.g. bad arguments) — the stream never started.\n          recordSpanError(span, err);\n          endOnce();\n          throw err;\n        }\n      },\n      parentContext,\n    );\n  };\n}\n\nfunction mergeTelemetry(opts: any): any {\n  const baseTelemetry: AITelemetryConfig = createAITelemetry({\n    metadata: opts?.experimental_telemetry?.metadata,\n  });\n  return {\n    ...opts,\n    experimental_telemetry: {\n      ...opts?.experimental_telemetry,\n      ...baseTelemetry,\n    },\n  };\n}\n\n\nfunction recordSpanError(span: Span, err: unknown): void {\n  if (err instanceof Error) {\n    span.setStatus({ code: SpanStatusCode.ERROR, message: err.message });\n    span.recordException(err);\n  } else {\n    span.setStatus({ code: SpanStatusCode.ERROR, message: String(err) });\n  }\n}\n","/**\n * Neatlogs-owned tracing state.\n *\n * Spans are created by the private Neatlogs provider and their parent is carried\n * in a private context key. The process-global OpenTelemetry span is\n * deliberately left untouched so other observability SDKs cannot export or\n * become parents of Neatlogs spans.\n */\n\nimport { AsyncLocalStorage } from 'node:async_hooks';\nimport {\n  ROOT_CONTEXT,\n  INVALID_SPAN_CONTEXT,\n  trace as otelTrace,\n  type Context,\n  type Span,\n  type SpanOptions,\n  type Tracer,\n  type TracerProvider,\n} from '@opentelemetry/api';\n\n// Entry points are bundled independently (`neatlogs`, `neatlogs/openai`,\n// `neatlogs/ai`, `neatlogs/mastra`, … each in both CJS and ESM), so every piece\n// of shared tracing state — the private span store AND the resolved provider /\n// provider — must live on `globalThis` behind a `Symbol.for` key.\n// Otherwise `init()` (run from the `neatlogs` bundle) sets `_provider` in ITS\n// module copy while a wrapper imported from `neatlogs/openai` reads a different,\n// still-null copy and silently falls back to the foreign global provider.\nconst PRIVATE_SPAN_STORAGE_KEY = Symbol.for(\n  'neatlogs.private_span_async_local_storage',\n);\nconst PRIVATE_PROVIDER_STATE_KEY = Symbol.for(\n  'neatlogs.private_provider_state',\n);\ninterface PrivateProviderState {\n  provider: TracerProvider | null;\n}\ntype NeatlogsGlobal = typeof globalThis & {\n  [PRIVATE_SPAN_STORAGE_KEY]?: AsyncLocalStorage<Context>;\n  [PRIVATE_PROVIDER_STATE_KEY]?: PrivateProviderState;\n};\nconst neatlogsGlobal = globalThis as NeatlogsGlobal;\n// Stores the full Neatlogs Context (parent span PLUS any threaded values such as\n// trace()'s prompt-template keys), not just the span. We never\n// activate the OTel global context, so this private store is the ONLY channel\n// through which those values propagate down to descendant spans.\nconst privateContextStorage =\n  neatlogsGlobal[PRIVATE_SPAN_STORAGE_KEY] ??\n  (neatlogsGlobal[PRIVATE_SPAN_STORAGE_KEY] = new AsyncLocalStorage<Context>());\nconst providerState: PrivateProviderState =\n  neatlogsGlobal[PRIVATE_PROVIDER_STATE_KEY] ??\n  (neatlogsGlobal[PRIVATE_PROVIDER_STATE_KEY] = {\n    provider: null,\n  });\n// Wrappers may be constructed or accidentally invoked before init(). They must\n// never fall back to a foreign process-global provider, so pre-init calls use a\n// local no-op tracer and safely emit no exported spans.\nconst preInitTracer: Tracer = {\n  startSpan(): Span {\n    return otelTrace.wrapSpanContext(INVALID_SPAN_CONTEXT);\n  },\n  startActiveSpan<F extends (span: Span) => unknown>(\n    _name: string,\n    arg2?: SpanOptions | Context | F,\n    arg3?: Context | F,\n    arg4?: F,\n  ): ReturnType<F> {\n    const fn =\n      typeof arg2 === 'function'\n        ? arg2\n        : typeof arg3 === 'function'\n          ? arg3\n          : arg4;\n    return fn!(otelTrace.wrapSpanContext(INVALID_SPAN_CONTEXT)) as ReturnType<F>;\n  },\n};\n\n/** @internal Configure the provider used by Neatlogs-created spans. */\nexport function _setNeatlogsProvider(provider: TracerProvider | null): void {\n  providerState.provider = provider;\n}\n\n/** Resolve a tracer from the private provider when one is configured. */\nexport function getNeatlogsTracer(name: string): Tracer {\n  return providerState.provider?.getTracer(name) ?? preInitTracer;\n}\n\n/**\n * @internal The private Neatlogs provider, or null before\n * init(). Used by integrations that must repoint a self-instrumenting library's\n * captured provider onto ours.\n */\nexport function getNeatlogsProvider(): TracerProvider | null {\n  return providerState.provider;\n}\n\n/**\n * Wrap a tracer so that spans it creates parent from — and, for\n * `startActiveSpan`, activate on — the PRIVATE Neatlogs context instead of the\n * global one.\n *\n * We hand this to libraries that create their own spans off a tracer we give\n * them (the Vercel AI SDK's `experimental_telemetry.tracer`, which calls\n * `tracer.startActiveSpan()` internally). Without the facade the AI SDK's native\n * spans would parent from `context.active()` — the foreign co-tenant's context —\n * and `startActiveSpan` would push them onto the GLOBAL context, so a foreign\n * tracer's next span reads our native span as its parent. Both directions leak.\n *\n */\nexport function isolateTracer(tracer: Tracer): Tracer {\n  const facade: Tracer = {\n    startSpan(name: string, options?: SpanOptions, context?: Context): Span {\n      const parent = context ?? getNeatlogsActiveContext();\n      return tracer.startSpan(name, options, parent);\n    },\n    startActiveSpan<F extends (span: Span) => unknown>(\n      name: string,\n      arg2?: SpanOptions | Context | F,\n      arg3?: Context | F,\n      arg4?: F,\n    ): ReturnType<F> {\n      // Normalize the 2/3/4-arg overloads of startActiveSpan.\n      let options: SpanOptions | undefined;\n      let context: Context | undefined;\n      let fn: F;\n      if (typeof arg2 === 'function') {\n        fn = arg2 as F;\n      } else if (typeof arg3 === 'function') {\n        options = arg2 as SpanOptions;\n        fn = arg3 as F;\n      } else {\n        options = arg2 as SpanOptions;\n        context = arg3 as Context;\n        fn = arg4 as F;\n      }\n      const parent = context ?? getNeatlogsActiveContext();\n      const span = tracer.startSpan(name, options, parent);\n      return withNeatlogsSpan(span, () => fn(span), parent) as ReturnType<F>;\n    },\n  };\n  return facade;\n}\n\n/**\n * The base context to build new Neatlogs spans and values on. NEVER contains a\n * foreign provider's span: it reads our private store, or ROOT_CONTEXT when\n * nothing is active.\n */\nexport function getNeatlogsActiveContext(): Context {\n  return privateContextStorage.getStore() ?? ROOT_CONTEXT;\n}\n\n/** Return only the active Neatlogs span (never a foreign provider's span). */\nexport function getActiveNeatlogsSpan(): Span | undefined {\n  return otelTrace.getSpan(getNeatlogsActiveContext());\n}\n\n/**\n * Build a parent context that cannot contain a foreign provider's span.\n *\n * The active Neatlogs context already carries the parent span AND any values\n * the caller threaded in upstream (e.g. `trace()`'s prompt-template values), so\n * those values reach the span processor via `onStart(parentContext)`. An\n * explicit `baseContext` (a caller's own value-carrying context) is honored as-is.\n */\nexport function getNeatlogsParentContext(baseContext?: Context): Context {\n  return baseContext ?? getNeatlogsActiveContext();\n}\n\n/**\n * A base context for callers that thread parent linkage themselves\n * (callback/event handlers that keep their own run-id → span map: LangChain,\n * OpenAI-Agents, Claude Agent SDK).\n *\n * This is the ACTIVE Neatlogs context — the private store's\n * context if a Neatlogs `trace()`/`span()` encloses this call, else\n * ROOT_CONTEXT. So a handler's own root/entry span nests under an enclosing\n * Neatlogs trace (preserving its session + end-user id) when one exists, and\n * auto-roots cleanly when one doesn't. A foreign provider's active span can\n * never leak in as an ancestor because the global context is never read.\n */\nexport function getNeatlogsBaseContext(baseContext?: Context): Context {\n  return baseContext ?? getNeatlogsActiveContext();\n}\n\n/**\n * Build an execution context for a Neatlogs span while preserving the active\n * foreign context. The span rides our private context store.\n */\nexport function getNeatlogsExecutionContext(\n  span: Span,\n  baseContext: Context = ROOT_CONTEXT,\n): Context {\n  return otelTrace.setSpan(baseContext, span);\n}\n\n/**\n * Run a callback with a Neatlogs span active under the appropriate policy.\n *\n * The stored context is `setSpan(base, span)` — carrying the span PLUS whatever\n * values `base` holds — so threaded values (prompt templates)\n * propagate to descendant spans through our private store instead of the global\n * OTel context we deliberately never touch.\n */\nexport function withNeatlogsSpan<T>(\n  span: Span,\n  fn: () => T,\n  baseContext?: Context,\n): T {\n  const base = baseContext ?? getNeatlogsActiveContext();\n  const ctx = otelTrace.setSpan(base, span);\n  return privateContextStorage.run(ctx, fn);\n}\n"],"mappings":";AAaA,SAAS,sBAA8C;;;ACJvD,SAAS,yBAAyB;AAClC;AAAA,EACE;AAAA,EACA;AAAA,EACA,SAAS;AAAA,OAMJ;AASP,IAAM,2BAA2B,uBAAO;AAAA,EACtC;AACF;AACA,IAAM,6BAA6B,uBAAO;AAAA,EACxC;AACF;AAQA,IAAM,iBAAiB;AAKvB,IAAM,wBACJ,eAAe,wBAAwB,MACtC,eAAe,wBAAwB,IAAI,IAAI,kBAA2B;AAC7E,IAAM,gBACJ,eAAe,0BAA0B,MACxC,eAAe,0BAA0B,IAAI;AAAA,EAC5C,UAAU;AACZ;AAIF,IAAM,gBAAwB;AAAA,EAC5B,YAAkB;AAChB,WAAO,UAAU,gBAAgB,oBAAoB;AAAA,EACvD;AAAA,EACA,gBACE,OACA,MACA,MACA,MACe;AACf,UAAM,KACJ,OAAO,SAAS,aACZ,OACA,OAAO,SAAS,aACd,OACA;AACR,WAAO,GAAI,UAAU,gBAAgB,oBAAoB,CAAC;AAAA,EAC5D;AACF;AAQO,SAAS,kBAAkB,MAAsB;AACtD,SAAO,cAAc,UAAU,UAAU,IAAI,KAAK;AACpD;AAwBO,SAAS,cAAc,QAAwB;AACpD,QAAM,SAAiB;AAAA,IACrB,UAAU,MAAc,SAAuB,SAAyB;AACtE,YAAM,SAAS,WAAW,yBAAyB;AACnD,aAAO,OAAO,UAAU,MAAM,SAAS,MAAM;AAAA,IAC/C;AAAA,IACA,gBACE,MACA,MACA,MACA,MACe;AAEf,UAAI;AACJ,UAAI;AACJ,UAAI;AACJ,UAAI,OAAO,SAAS,YAAY;AAC9B,aAAK;AAAA,MACP,WAAW,OAAO,SAAS,YAAY;AACrC,kBAAU;AACV,aAAK;AAAA,MACP,OAAO;AACL,kBAAU;AACV,kBAAU;AACV,aAAK;AAAA,MACP;AACA,YAAM,SAAS,WAAW,yBAAyB;AACnD,YAAM,OAAO,OAAO,UAAU,MAAM,SAAS,MAAM;AACnD,aAAO,iBAAiB,MAAM,MAAM,GAAG,IAAI,GAAG,MAAM;AAAA,IACtD;AAAA,EACF;AACA,SAAO;AACT;AAOO,SAAS,2BAAoC;AAClD,SAAO,sBAAsB,SAAS,KAAK;AAC7C;AAeO,SAAS,yBAAyB,aAAgC;AACvE,SAAO,eAAe,yBAAyB;AACjD;AAqCO,SAAS,iBACd,MACA,IACA,aACG;AACH,QAAM,OAAO,eAAe,yBAAyB;AACrD,QAAM,MAAM,UAAU,QAAQ,MAAM,IAAI;AACxC,SAAO,sBAAsB,IAAI,KAAK,EAAE;AAC1C;;;ADpMA,IAAM,cAAc;AAgBb,SAAS,kBACd,OAAiC,CAAC,GACf;AACnB,QAAM,WAAW,KAAK,YAAY,CAAC;AACnC,SAAO;AAAA,IACL,WAAW;AAAA,IACX,cAAc;AAAA,IACd,eAAe;AAAA;AAAA;AAAA;AAAA;AAAA,IAKf,QAAQ,cAAc,kBAAkB,WAAW,CAAC;AAAA,IACpD,UAAU,EAAE,GAAG,UAAU,iBAAiB,KAAK;AAAA,EACjD;AACF;AAIA,SAAS,cAAc,OAAwB;AAC7C,MAAI;AACF,WAAO,KAAK,UAAU,KAAK;AAAA,EAC7B,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAEA,SAAS,cAAc,MAAY,MAAqC;AAEtE,MAAI,SAAS,YAAY,QAAQ,cAAc,OAAO;AACpD,UAAM,QAAQ,KAAK,YAAY,KAAK;AACpC,UAAMA,eAAc,cAAc,KAAK;AACvC,QAAIA,cAAa;AACf,WAAK,aAAa,eAAeA,YAAW;AAAA,IAC9C;AACA;AAAA,EACF;AACA,QAAM,cAAc,cAAc,IAAI;AACtC,MAAI,aAAa;AACf,SAAK,aAAa,eAAe,WAAW;AAAA,EAC9C;AACF;AAEA,SAAS,eAAe,MAAY,QAAuB;AACzD,MAAI,UAAU,OAAO,WAAW,UAAU;AACxC,UAAM,IAAI;AAEV,QAAI,UAAU,KAAK,kBAAkB,GAAG;AACtC,YAAM,OAAO,OAAO,EAAE,QAAQ,EAAE;AAChC,UAAI,MAAM;AACR,aAAK,aAAa,gBAAgB,IAAI;AAAA,MACxC;AACA,UAAI,EAAE,cAAc;AAClB,aAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,MAClE;AACA;AAAA,IACF;AAGA,QAAI,YAAY,KAAK,kBAAkB,GAAG;AACxC,UAAI,EAAE,WAAW,QAAW;AAC1B,aAAK,aAAa,gBAAgB,cAAc,EAAE,MAAM,CAAC;AAAA,MAC3D;AACA,UAAI,EAAE,cAAc;AAClB,aAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,MAClE;AACA;AAAA,IACF;AAAA,EACF;AACA,QAAM,cAAc,cAAc,MAAM;AACxC,MAAI,aAAa;AACf,SAAK,aAAa,gBAAgB,WAAW;AAAA,EAC/C;AACF;AAOA,SAAS,qBAAqB,MAAY,OAAsB;AAC9D,MAAI,CAAC,SAAS,OAAO,UAAU,SAAU;AACzC,QAAM,IAAI;AACV,MAAI,OAAO,EAAE,SAAS,YAAY,EAAE,MAAM;AACxC,SAAK,aAAa,gBAAgB,EAAE,IAAI;AAAA,EAC1C,WAAW,YAAY,KAAK,EAAE,WAAW,QAAW;AAClD,UAAM,cAAc,cAAc,EAAE,MAAM;AAC1C,QAAI,YAAa,MAAK,aAAa,gBAAgB,WAAW;AAAA,EAChE;AACA,MAAI,EAAE,cAAc;AAClB,SAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,EAClE;AACF;AAMA,IAAM,oBAAoD;AAAA,EACxD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAYO,SAAS,UAA6C,UAAgB;AAC3E,QAAM,UAAmC,EAAE,GAAG,SAAS;AAEvD,aAAW,QAAQ,mBAAmB;AACpC,UAAM,WAAW,SAAS,IAAI;AAC9B,QAAI,OAAO,aAAa,WAAY;AAEpC,QAAI,SAAS,gBAAgB,SAAS,gBAAgB;AACpD,cAAQ,IAAI,IAAI,oBAAoB,MAAM,QAAkC;AAAA,IAC9E,OAAO;AACL,cAAQ,IAAI,IAAI,mBAAmB,MAAM,QAA2C;AAAA,IACtF;AAAA,EACF;AAEA,SAAO;AACT;AAEA,SAAS,aAAa,MAAmC;AACvD,MAAI,SAAS,WAAW,SAAS,eAAe,SAAS,SAAU,QAAO;AAC1E,SAAO;AACT;AAGA,SAAS,mBAAmB;AAG1B,SAAO,yBAAyB;AAClC;AAEA,SAAS,mBACP,MACA,UACiC;AACjC,SAAO,eAAe,eAAe,MAA6B;AAChE,UAAM,SAAS,kBAAkB,WAAW;AAM5C,UAAM,gBAAgB,iBAAiB;AACvC,UAAM,OAAO,OAAO;AAAA,MAClB,MAAM,IAAI;AAAA,MACV,EAAE,YAAY,EAAE,2BAA2B,aAAa,IAAI,EAAE,EAAE;AAAA,MAChE;AAAA,IACF;AACA,WAAO;AAAA,MACL;AAAA,MACA,YAAY;AACV,YAAI;AACF,gBAAM,kBAAkB,SAAS,WAAW,SAAS,eAAe,SAAS;AAC7E,cAAI,CAAC,iBAAiB;AACpB,0BAAc,MAAM,IAAI;AAAA,UAC1B;AACA,cAAI,SAAS,YAAY,MAAM,OAAO;AACpC,iBAAK,aAAa,mBAAmB,OAAO,KAAK,KAAK,CAAC;AAAA,UACzD;AACA,gBAAM,SAAS,eAAe,IAAI;AAClC,gBAAM,SAAS,MAAM,SAAS,MAAM;AACpC,cAAI,CAAC,iBAAiB;AACpB,2BAAe,MAAM,MAAM;AAAA,UAC7B;AACA,iBAAO;AAAA,QACT,SAAS,KAAK;AACZ,0BAAgB,MAAM,GAAG;AACzB,gBAAM;AAAA,QACR,UAAE;AACA,eAAK,IAAI;AAAA,QACX;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAQA,SAAS,oBACP,MACA,UACwB;AACxB,SAAO,SAAS,gBAAgB,MAAoB;AAClD,UAAM,SAAS,kBAAkB,WAAW;AAI5C,UAAM,gBAAgB,iBAAiB;AACvC,UAAM,OAAO,OAAO;AAAA,MAClB,MAAM,IAAI;AAAA,MACV,EAAE,YAAY,EAAE,2BAA2B,aAAa,IAAI,EAAE,EAAE;AAAA,MAChE;AAAA,IACF;AACA,WAAO;AAAA,MACL;AAAA,MACA,MAAM;AACJ,YAAI,YAAY;AAChB,cAAM,UAAU,MAAM;AACpB,cAAI,UAAW;AACf,sBAAY;AACZ,eAAK,IAAI;AAAA,QACX;AACA,YAAI;AACF,wBAAc,MAAM,IAAI;AACxB,gBAAM,SAAS,eAAe,IAAI;AAClC,gBAAM,eAAe,MAAM;AAC3B,gBAAM,cAAc,MAAM;AAC1B,gBAAM,cAAc;AAAA,YAClB,GAAG;AAAA,YACH,UAAU,OAAO,UAAe;AAC9B,kBAAI;AACF,qCAAqB,MAAM,KAAK;AAAA,cAClC,UAAE;AACA,wBAAQ;AAAA,cACV;AACA,kBAAI,OAAO,iBAAiB,YAAY;AACtC,uBAAO,aAAa,KAAK;AAAA,cAC3B;AAAA,YACF;AAAA,YACA,SAAS,CAAC,UAAe;AACvB,8BAAgB,OAAO,SAAS,MAAM,UAAU,KAAK;AACrD,sBAAQ;AACR,kBAAI,OAAO,gBAAgB,YAAY;AACrC,uBAAO,YAAY,KAAK;AAAA,cAC1B;AAAA,YACF;AAAA,UACF;AACA,iBAAO,SAAS,WAAW;AAAA,QAC7B,SAAS,KAAK;AAEZ,0BAAgB,MAAM,GAAG;AACzB,kBAAQ;AACR,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,eAAe,MAAgB;AACtC,QAAM,gBAAmC,kBAAkB;AAAA,IACzD,UAAU,MAAM,wBAAwB;AAAA,EAC1C,CAAC;AACD,SAAO;AAAA,IACL,GAAG;AAAA,IACH,wBAAwB;AAAA,MACtB,GAAG,MAAM;AAAA,MACT,GAAG;AAAA,IACL;AAAA,EACF;AACF;AAGA,SAAS,gBAAgB,MAAY,KAAoB;AACvD,MAAI,eAAe,OAAO;AACxB,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,SAAS,IAAI,QAAQ,CAAC;AACnE,SAAK,gBAAgB,GAAG;AAAA,EAC1B,OAAO;AACL,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,SAAS,OAAO,GAAG,EAAE,CAAC;AAAA,EACrE;AACF;","names":["stringified"]}
+{"version":3,"sources":["../src/ai-sdk.ts","../src/core/provider.ts"],"sourcesContent":["/**\n * Vercel AI SDK wrapper — inline implementation.\n *\n * Wraps `generateText`, `streamText`, `generateObject`, `streamObject` from\n * the `ai` package with OTel parent spans + forced telemetry. Static export,\n * no dynamic imports — bundler-friendly (works with Turbopack, webpack, esbuild).\n *\n * Usage:\n *   import { wrapAISDK } from 'neatlogs';\n *   import * as ai from 'ai';\n *   const { streamText, generateText } = wrapAISDK(ai);\n */\n\nimport { SpanStatusCode, type Span, type Tracer } from '@opentelemetry/api';\nimport { getNeatlogsTracer, getNeatlogsParentContext, isolateTracer, withNeatlogsSpan } from './core/provider.js';\n\nconst TRACER_NAME = 'neatlogs.ai-sdk';\n\n// -- Telemetry config --------------------------------------------------------\n\nexport interface CreateAITelemetryOptions {\n  metadata?: Record<string, unknown>;\n}\n\nexport interface AITelemetryConfig {\n  isEnabled: true;\n  recordInputs: true;\n  recordOutputs: true;\n  tracer: Tracer;\n  metadata: Record<string, unknown>;\n}\n\nexport function createAITelemetry(\n  opts: CreateAITelemetryOptions = {},\n): AITelemetryConfig {\n  const userMeta = opts.metadata ?? {};\n  return {\n    isEnabled: true,\n    recordInputs: true,\n    recordOutputs: true,\n    // Hand the AI SDK an isolation-aware tracer: it calls startActiveSpan()\n    // internally, which would otherwise parent its native spans from the foreign\n    // global context AND push them onto it (so a co-tenant's next span inherits\n    // ours). The facade routes both through the private Neatlogs context.\n    tracer: isolateTracer(getNeatlogsTracer(TRACER_NAME)),\n    metadata: { ...userMeta, neatlogsWrapped: true },\n  };\n}\n\n// -- Span attributes ---------------------------------------------------------\n\nfunction safeStringify(value: unknown): string {\n  try {\n    return JSON.stringify(value);\n  } catch {\n    return '';\n  }\n}\n\nfunction setInputValue(span: Span, opts: Record<string, unknown>): void {\n  // For generateText/streamText — only capture prompt or messages, not full model config\n  if (opts && ('prompt' in opts || 'messages' in opts)) {\n    const input = opts.messages ?? opts.prompt;\n    const stringified = safeStringify(input);\n    if (stringified) {\n      span.setAttribute('input.value', stringified);\n    }\n    return;\n  }\n  const stringified = safeStringify(opts);\n  if (stringified) {\n    span.setAttribute('input.value', stringified);\n  }\n}\n\nfunction setOutputValue(span: Span, result: unknown): void {\n  if (result && typeof result === 'object') {\n    const r = result as Record<string, unknown>;\n    // GenerateTextResult / StreamTextResult — extract meaningful fields\n    if ('text' in r && 'finishReason' in r) {\n      const text = String(r.text ?? '');\n      if (text) {\n        span.setAttribute('output.value', text);\n      }\n      if (r.finishReason) {\n        span.setAttribute('gen_ai.finish_reason', String(r.finishReason));\n      }\n      return;\n    }\n    // GenerateObjectResult — the structured object is the output, not `text`.\n    // Without this the envelope (object+usage+response+…) gets stringified whole.\n    if ('object' in r && 'finishReason' in r) {\n      if (r.object !== undefined) {\n        span.setAttribute('output.value', safeStringify(r.object));\n      }\n      if (r.finishReason) {\n        span.setAttribute('gen_ai.finish_reason', String(r.finishReason));\n      }\n      return;\n    }\n  }\n  const stringified = safeStringify(result);\n  if (stringified) {\n    span.setAttribute('output.value', stringified);\n  }\n}\n\n// Extract output from a streamText/streamObject `onFinish` event. The event\n// extends StepResult, so `text` (streamText) / `object` (streamObject) sit at\n// the top level alongside `finishReason` — but the event also carries `steps`,\n// `usage`, etc., so we pull only the meaningful fields instead of stringifying\n// the whole envelope.\nfunction setStreamOutputValue(span: Span, event: unknown): void {\n  if (!event || typeof event !== 'object') return;\n  const e = event as Record<string, unknown>;\n  if (typeof e.text === 'string' && e.text) {\n    span.setAttribute('output.value', e.text);\n  } else if ('object' in e && e.object !== undefined) {\n    const stringified = safeStringify(e.object);\n    if (stringified) span.setAttribute('output.value', stringified);\n  }\n  if (e.finishReason) {\n    span.setAttribute('gen_ai.finish_reason', String(e.finishReason));\n  }\n}\n\n// -- Wrapping ----------------------------------------------------------------\n\ntype WrappedFunctionName = 'generateText' | 'streamText' | 'generateObject' | 'streamObject' | 'embed' | 'embedMany' | 'rerank';\n\nconst WRAPPED_FUNCTIONS: readonly WrappedFunctionName[] = [\n  'generateText',\n  'streamText',\n  'generateObject',\n  'streamObject',\n  'embed',\n  'embedMany',\n  'rerank',\n] as const;\n\n/**\n * Wrap the `ai` module namespace so that every `generateText` / `streamText` /\n * `generateObject` / `streamObject` call:\n *\n *   1. Opens a parent OTel span on the active TracerProvider.\n *   2. Forces `experimental_telemetry: { isEnabled: true }`, merging user metadata.\n *   3. Records input/output on the parent span and propagates errors.\n *\n * Other exports (types, helpers) pass through unchanged.\n */\nexport function wrapAISDK<T extends Record<string, unknown>>(aiModule: T): T {\n  const wrapped: Record<string, unknown> = { ...aiModule };\n\n  for (const name of WRAPPED_FUNCTIONS) {\n    const original = aiModule[name];\n    if (typeof original !== 'function') continue;\n\n    if (name === 'streamText' || name === 'streamObject') {\n      wrapped[name] = createStreamWrapper(name, original as (opts: any) => unknown);\n    } else {\n      wrapped[name] = createAsyncWrapper(name, original as (opts: any) => Promise<unknown>);\n    }\n  }\n\n  return wrapped as T;\n}\n\nfunction rootSpanKind(name: WrappedFunctionName): string {\n  if (name === 'embed' || name === 'embedMany' || name === 'rerank') return 'CHAIN';\n  return 'WORKFLOW';\n}\n\n\nfunction getParentContext() {\n  // Our parent comes solely from the private span store; a\n  // foreign provider's active span must never become our ancestor.\n  return getNeatlogsParentContext();\n}\n\nfunction createAsyncWrapper(\n  name: WrappedFunctionName,\n  original: (opts: any) => Promise<unknown>,\n): (opts: any) => Promise<unknown> {\n  return async function wrappedAsyncFn(opts: any): Promise<unknown> {\n    const tracer = getNeatlogsTracer(TRACER_NAME);\n    // startSpan (NOT startActiveSpan) + withNeatlogsSpan: startActiveSpan would\n    // push our span onto the GLOBAL OTel context, so a foreign tracer's\n    // startSpan() inside generateText() would read it as parent and inherit our\n    // trace id. withNeatlogsSpan carries the parent in the private store in\n    // the private context, leaving the global context untouched.\n    const parentContext = getParentContext();\n    const span = tracer.startSpan(\n      `ai.${name}`,\n      { attributes: { 'openinference.span.kind': rootSpanKind(name) } },\n      parentContext,\n    );\n    return withNeatlogsSpan(\n      span,\n      async () => {\n        try {\n          const isEmbedOrRerank = name === 'embed' || name === 'embedMany' || name === 'rerank';\n          if (!isEmbedOrRerank) {\n            setInputValue(span, opts);\n          }\n          if (name === 'rerank' && opts?.query) {\n            span.setAttribute('ai.rerank.query', String(opts.query));\n          }\n          const merged = mergeTelemetry(opts);\n          const result = await original(merged);\n          if (!isEmbedOrRerank) {\n            setOutputValue(span, result);\n          }\n          return result;\n        } catch (err) {\n          recordSpanError(span, err);\n          throw err;\n        } finally {\n          span.end();\n        }\n      },\n      parentContext,\n    );\n  };\n}\n\n// streamText/streamObject return synchronously while the model keeps producing\n// tokens for seconds afterwards. Ending the span in a `finally` (as a plain sync\n// wrapper would) closes it in ~2ms with no output — the output only exists once\n// the stream finishes. Instead we keep the span open and end it from the AI SDK's\n// `onFinish` callback, where the final text/object is available. Any user-provided\n// `onFinish` is preserved and invoked first.\nfunction createStreamWrapper(\n  name: WrappedFunctionName,\n  original: (opts: any) => unknown,\n): (opts: any) => unknown {\n  return function wrappedStreamFn(opts: any): unknown {\n    const tracer = getNeatlogsTracer(TRACER_NAME);\n    // startSpan + withNeatlogsSpan (see createAsyncWrapper) so streamText's\n    // internals never see our span on the global OTel context. The span stays\n    // open past the run scope and is ended from onFinish/onError.\n    const parentContext = getParentContext();\n    const span = tracer.startSpan(\n      `ai.${name}`,\n      { attributes: { 'openinference.span.kind': rootSpanKind(name) } },\n      parentContext,\n    );\n    return withNeatlogsSpan(\n      span,\n      () => {\n        let spanEnded = false;\n        const endOnce = () => {\n          if (spanEnded) return;\n          spanEnded = true;\n          span.end();\n        };\n        try {\n          setInputValue(span, opts);\n          const merged = mergeTelemetry(opts);\n          const userOnFinish = opts?.onFinish;\n          const userOnError = opts?.onError;\n          const wrappedOpts = {\n            ...merged,\n            onFinish: async (event: any) => {\n              try {\n                setStreamOutputValue(span, event);\n              } finally {\n                endOnce();\n              }\n              if (typeof userOnFinish === 'function') {\n                return userOnFinish(event);\n              }\n            },\n            onError: (event: any) => {\n              recordSpanError(span, (event && event.error) ?? event);\n              endOnce();\n              if (typeof userOnError === 'function') {\n                return userOnError(event);\n              }\n            },\n          };\n          return original(wrappedOpts);\n        } catch (err) {\n          // Synchronous throw (e.g. bad arguments) — the stream never started.\n          recordSpanError(span, err);\n          endOnce();\n          throw err;\n        }\n      },\n      parentContext,\n    );\n  };\n}\n\nfunction mergeTelemetry(opts: any): any {\n  const baseTelemetry: AITelemetryConfig = createAITelemetry({\n    metadata: opts?.experimental_telemetry?.metadata,\n  });\n  return {\n    ...opts,\n    experimental_telemetry: {\n      ...opts?.experimental_telemetry,\n      ...baseTelemetry,\n    },\n  };\n}\n\n\nfunction recordSpanError(span: Span, err: unknown): void {\n  if (err instanceof Error) {\n    span.setStatus({ code: SpanStatusCode.ERROR, message: err.message });\n    span.recordException(err);\n  } else {\n    span.setStatus({ code: SpanStatusCode.ERROR, message: String(err) });\n  }\n}\n","/**\n * Neatlogs-owned tracing state.\n *\n * Spans are created by the private Neatlogs provider and their parent is carried\n * in a private context key. The process-global OpenTelemetry span is\n * deliberately left untouched so other observability SDKs cannot export or\n * become parents of Neatlogs spans.\n */\n\nimport { AsyncLocalStorage } from 'node:async_hooks';\nimport {\n  ROOT_CONTEXT,\n  INVALID_SPAN_CONTEXT,\n  createContextKey,\n  trace as otelTrace,\n  type Context,\n  type Span,\n  type SpanOptions,\n  type Tracer,\n  type TracerProvider,\n} from '@opentelemetry/api';\n\n// Carries the trace-ROOT span down the private context so descendants can target\n// it (e.g. setTraceOutput). getActiveNeatlogsSpan() returns the innermost span,\n// which is not the root once nested; this key preserves the root reference.\nconst NEATLOGS_ROOT_SPAN_KEY = createContextKey('neatlogs.root_span');\n\n// Entry points are bundled independently (`neatlogs`, `neatlogs/openai`,\n// `neatlogs/ai`, `neatlogs/mastra`, … each in both CJS and ESM), so every piece\n// of shared tracing state — the private span store AND the resolved provider /\n// provider — must live on `globalThis` behind a `Symbol.for` key.\n// Otherwise `init()` (run from the `neatlogs` bundle) sets `_provider` in ITS\n// module copy while a wrapper imported from `neatlogs/openai` reads a different,\n// still-null copy and silently falls back to the foreign global provider.\nconst PRIVATE_SPAN_STORAGE_KEY = Symbol.for(\n  'neatlogs.private_span_async_local_storage',\n);\nconst PRIVATE_PROVIDER_STATE_KEY = Symbol.for(\n  'neatlogs.private_provider_state',\n);\ninterface PrivateProviderState {\n  provider: TracerProvider | null;\n}\ntype NeatlogsGlobal = typeof globalThis & {\n  [PRIVATE_SPAN_STORAGE_KEY]?: AsyncLocalStorage<Context>;\n  [PRIVATE_PROVIDER_STATE_KEY]?: PrivateProviderState;\n};\nconst neatlogsGlobal = globalThis as NeatlogsGlobal;\n// Stores the full Neatlogs Context (parent span PLUS any threaded values such as\n// trace()'s prompt-template keys), not just the span. We never\n// activate the OTel global context, so this private store is the ONLY channel\n// through which those values propagate down to descendant spans.\nconst privateContextStorage =\n  neatlogsGlobal[PRIVATE_SPAN_STORAGE_KEY] ??\n  (neatlogsGlobal[PRIVATE_SPAN_STORAGE_KEY] = new AsyncLocalStorage<Context>());\nconst providerState: PrivateProviderState =\n  neatlogsGlobal[PRIVATE_PROVIDER_STATE_KEY] ??\n  (neatlogsGlobal[PRIVATE_PROVIDER_STATE_KEY] = {\n    provider: null,\n  });\n// Wrappers may be constructed or accidentally invoked before init(). They must\n// never fall back to a foreign process-global provider, so pre-init calls use a\n// local no-op tracer and safely emit no exported spans.\nconst preInitTracer: Tracer = {\n  startSpan(): Span {\n    return otelTrace.wrapSpanContext(INVALID_SPAN_CONTEXT);\n  },\n  startActiveSpan<F extends (span: Span) => unknown>(\n    _name: string,\n    arg2?: SpanOptions | Context | F,\n    arg3?: Context | F,\n    arg4?: F,\n  ): ReturnType<F> {\n    const fn =\n      typeof arg2 === 'function'\n        ? arg2\n        : typeof arg3 === 'function'\n          ? arg3\n          : arg4;\n    return fn!(otelTrace.wrapSpanContext(INVALID_SPAN_CONTEXT)) as ReturnType<F>;\n  },\n};\n\n/** @internal Configure the provider used by Neatlogs-created spans. */\nexport function _setNeatlogsProvider(provider: TracerProvider | null): void {\n  providerState.provider = provider;\n}\n\n/** Resolve a tracer from the private provider when one is configured. */\nexport function getNeatlogsTracer(name: string): Tracer {\n  return providerState.provider?.getTracer(name) ?? preInitTracer;\n}\n\n/**\n * @internal The private Neatlogs provider, or null before\n * init(). Used by integrations that must repoint a self-instrumenting library's\n * captured provider onto ours.\n */\nexport function getNeatlogsProvider(): TracerProvider | null {\n  return providerState.provider;\n}\n\n/**\n * Wrap a tracer so that spans it creates parent from — and, for\n * `startActiveSpan`, activate on — the PRIVATE Neatlogs context instead of the\n * global one.\n *\n * We hand this to libraries that create their own spans off a tracer we give\n * them (the Vercel AI SDK's `experimental_telemetry.tracer`, which calls\n * `tracer.startActiveSpan()` internally). Without the facade the AI SDK's native\n * spans would parent from `context.active()` — the foreign co-tenant's context —\n * and `startActiveSpan` would push them onto the GLOBAL context, so a foreign\n * tracer's next span reads our native span as its parent. Both directions leak.\n *\n */\nexport function isolateTracer(tracer: Tracer): Tracer {\n  const facade: Tracer = {\n    startSpan(name: string, options?: SpanOptions, context?: Context): Span {\n      const parent = context ?? getNeatlogsActiveContext();\n      return tracer.startSpan(name, options, parent);\n    },\n    startActiveSpan<F extends (span: Span) => unknown>(\n      name: string,\n      arg2?: SpanOptions | Context | F,\n      arg3?: Context | F,\n      arg4?: F,\n    ): ReturnType<F> {\n      // Normalize the 2/3/4-arg overloads of startActiveSpan.\n      let options: SpanOptions | undefined;\n      let context: Context | undefined;\n      let fn: F;\n      if (typeof arg2 === 'function') {\n        fn = arg2 as F;\n      } else if (typeof arg3 === 'function') {\n        options = arg2 as SpanOptions;\n        fn = arg3 as F;\n      } else {\n        options = arg2 as SpanOptions;\n        context = arg3 as Context;\n        fn = arg4 as F;\n      }\n      const parent = context ?? getNeatlogsActiveContext();\n      const span = tracer.startSpan(name, options, parent);\n      return withNeatlogsSpan(span, () => fn(span), parent) as ReturnType<F>;\n    },\n  };\n  return facade;\n}\n\n/**\n * The base context to build new Neatlogs spans and values on. NEVER contains a\n * foreign provider's span: it reads our private store, or ROOT_CONTEXT when\n * nothing is active.\n */\nexport function getNeatlogsActiveContext(): Context {\n  return privateContextStorage.getStore() ?? ROOT_CONTEXT;\n}\n\n/** Return only the active Neatlogs span (never a foreign provider's span). */\nexport function getActiveNeatlogsSpan(): Span | undefined {\n  return otelTrace.getSpan(getNeatlogsActiveContext());\n}\n\n/**\n * Return the trace-ROOT Neatlogs span, or undefined when no trace is active.\n *\n * Unlike {@link getActiveNeatlogsSpan} (innermost), this is the outermost span\n * of the current trace — the one the backend derives trace-level output from\n * (`parent_span_id=''`). It is stashed on the private context the first time a\n * span becomes active, so nested calls still resolve to the root.\n */\nexport function getNeatlogsRootSpan(): Span | undefined {\n  return getNeatlogsActiveContext().getValue(NEATLOGS_ROOT_SPAN_KEY) as\n    | Span\n    | undefined;\n}\n\n/**\n * Build a parent context that cannot contain a foreign provider's span.\n *\n * The active Neatlogs context already carries the parent span AND any values\n * the caller threaded in upstream (e.g. `trace()`'s prompt-template values), so\n * those values reach the span processor via `onStart(parentContext)`. An\n * explicit `baseContext` (a caller's own value-carrying context) is honored as-is.\n */\nexport function getNeatlogsParentContext(baseContext?: Context): Context {\n  return baseContext ?? getNeatlogsActiveContext();\n}\n\n/**\n * A base context for callers that thread parent linkage themselves\n * (callback/event handlers that keep their own run-id → span map: LangChain,\n * OpenAI-Agents, Claude Agent SDK).\n *\n * This is the ACTIVE Neatlogs context — the private store's\n * context if a Neatlogs `trace()`/`span()` encloses this call, else\n * ROOT_CONTEXT. So a handler's own root/entry span nests under an enclosing\n * Neatlogs trace (preserving its session + end-user id) when one exists, and\n * auto-roots cleanly when one doesn't. A foreign provider's active span can\n * never leak in as an ancestor because the global context is never read.\n */\nexport function getNeatlogsBaseContext(baseContext?: Context): Context {\n  return baseContext ?? getNeatlogsActiveContext();\n}\n\n/**\n * Build an execution context for a Neatlogs span while preserving the active\n * foreign context. The span rides our private context store.\n */\nexport function getNeatlogsExecutionContext(\n  span: Span,\n  baseContext: Context = ROOT_CONTEXT,\n): Context {\n  return otelTrace.setSpan(baseContext, span);\n}\n\n/**\n * Run a callback with a Neatlogs span active under the appropriate policy.\n *\n * The stored context is `setSpan(base, span)` — carrying the span PLUS whatever\n * values `base` holds — so threaded values (prompt templates)\n * propagate to descendant spans through our private store instead of the global\n * OTel context we deliberately never touch.\n */\nexport function withNeatlogsSpan<T>(\n  span: Span,\n  fn: () => T,\n  baseContext?: Context,\n): T {\n  const base = baseContext ?? getNeatlogsActiveContext();\n  let ctx = otelTrace.setSpan(base, span);\n  // The first span activated in a context with no root recorded IS the root of\n  // this trace; remember it so descendants (setTraceOutput) can target it.\n  if (base.getValue(NEATLOGS_ROOT_SPAN_KEY) === undefined) {\n    ctx = ctx.setValue(NEATLOGS_ROOT_SPAN_KEY, span);\n  }\n  return privateContextStorage.run(ctx, fn);\n}\n"],"mappings":";AAaA,SAAS,sBAA8C;;;ACJvD,SAAS,yBAAyB;AAClC;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA,SAAS;AAAA,OAMJ;AAKP,IAAM,yBAAyB,iBAAiB,oBAAoB;AASpE,IAAM,2BAA2B,uBAAO;AAAA,EACtC;AACF;AACA,IAAM,6BAA6B,uBAAO;AAAA,EACxC;AACF;AAQA,IAAM,iBAAiB;AAKvB,IAAM,wBACJ,eAAe,wBAAwB,MACtC,eAAe,wBAAwB,IAAI,IAAI,kBAA2B;AAC7E,IAAM,gBACJ,eAAe,0BAA0B,MACxC,eAAe,0BAA0B,IAAI;AAAA,EAC5C,UAAU;AACZ;AAIF,IAAM,gBAAwB;AAAA,EAC5B,YAAkB;AAChB,WAAO,UAAU,gBAAgB,oBAAoB;AAAA,EACvD;AAAA,EACA,gBACE,OACA,MACA,MACA,MACe;AACf,UAAM,KACJ,OAAO,SAAS,aACZ,OACA,OAAO,SAAS,aACd,OACA;AACR,WAAO,GAAI,UAAU,gBAAgB,oBAAoB,CAAC;AAAA,EAC5D;AACF;AAQO,SAAS,kBAAkB,MAAsB;AACtD,SAAO,cAAc,UAAU,UAAU,IAAI,KAAK;AACpD;AAwBO,SAAS,cAAc,QAAwB;AACpD,QAAM,SAAiB;AAAA,IACrB,UAAU,MAAc,SAAuB,SAAyB;AACtE,YAAM,SAAS,WAAW,yBAAyB;AACnD,aAAO,OAAO,UAAU,MAAM,SAAS,MAAM;AAAA,IAC/C;AAAA,IACA,gBACE,MACA,MACA,MACA,MACe;AAEf,UAAI;AACJ,UAAI;AACJ,UAAI;AACJ,UAAI,OAAO,SAAS,YAAY;AAC9B,aAAK;AAAA,MACP,WAAW,OAAO,SAAS,YAAY;AACrC,kBAAU;AACV,aAAK;AAAA,MACP,OAAO;AACL,kBAAU;AACV,kBAAU;AACV,aAAK;AAAA,MACP;AACA,YAAM,SAAS,WAAW,yBAAyB;AACnD,YAAM,OAAO,OAAO,UAAU,MAAM,SAAS,MAAM;AACnD,aAAO,iBAAiB,MAAM,MAAM,GAAG,IAAI,GAAG,MAAM;AAAA,IACtD;AAAA,EACF;AACA,SAAO;AACT;AAOO,SAAS,2BAAoC;AAClD,SAAO,sBAAsB,SAAS,KAAK;AAC7C;AA6BO,SAAS,yBAAyB,aAAgC;AACvE,SAAO,eAAe,yBAAyB;AACjD;AAqCO,SAAS,iBACd,MACA,IACA,aACG;AACH,QAAM,OAAO,eAAe,yBAAyB;AACrD,MAAI,MAAM,UAAU,QAAQ,MAAM,IAAI;AAGtC,MAAI,KAAK,SAAS,sBAAsB,MAAM,QAAW;AACvD,UAAM,IAAI,SAAS,wBAAwB,IAAI;AAAA,EACjD;AACA,SAAO,sBAAsB,IAAI,KAAK,EAAE;AAC1C;;;AD7NA,IAAM,cAAc;AAgBb,SAAS,kBACd,OAAiC,CAAC,GACf;AACnB,QAAM,WAAW,KAAK,YAAY,CAAC;AACnC,SAAO;AAAA,IACL,WAAW;AAAA,IACX,cAAc;AAAA,IACd,eAAe;AAAA;AAAA;AAAA;AAAA;AAAA,IAKf,QAAQ,cAAc,kBAAkB,WAAW,CAAC;AAAA,IACpD,UAAU,EAAE,GAAG,UAAU,iBAAiB,KAAK;AAAA,EACjD;AACF;AAIA,SAAS,cAAc,OAAwB;AAC7C,MAAI;AACF,WAAO,KAAK,UAAU,KAAK;AAAA,EAC7B,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAEA,SAAS,cAAc,MAAY,MAAqC;AAEtE,MAAI,SAAS,YAAY,QAAQ,cAAc,OAAO;AACpD,UAAM,QAAQ,KAAK,YAAY,KAAK;AACpC,UAAMA,eAAc,cAAc,KAAK;AACvC,QAAIA,cAAa;AACf,WAAK,aAAa,eAAeA,YAAW;AAAA,IAC9C;AACA;AAAA,EACF;AACA,QAAM,cAAc,cAAc,IAAI;AACtC,MAAI,aAAa;AACf,SAAK,aAAa,eAAe,WAAW;AAAA,EAC9C;AACF;AAEA,SAAS,eAAe,MAAY,QAAuB;AACzD,MAAI,UAAU,OAAO,WAAW,UAAU;AACxC,UAAM,IAAI;AAEV,QAAI,UAAU,KAAK,kBAAkB,GAAG;AACtC,YAAM,OAAO,OAAO,EAAE,QAAQ,EAAE;AAChC,UAAI,MAAM;AACR,aAAK,aAAa,gBAAgB,IAAI;AAAA,MACxC;AACA,UAAI,EAAE,cAAc;AAClB,aAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,MAClE;AACA;AAAA,IACF;AAGA,QAAI,YAAY,KAAK,kBAAkB,GAAG;AACxC,UAAI,EAAE,WAAW,QAAW;AAC1B,aAAK,aAAa,gBAAgB,cAAc,EAAE,MAAM,CAAC;AAAA,MAC3D;AACA,UAAI,EAAE,cAAc;AAClB,aAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,MAClE;AACA;AAAA,IACF;AAAA,EACF;AACA,QAAM,cAAc,cAAc,MAAM;AACxC,MAAI,aAAa;AACf,SAAK,aAAa,gBAAgB,WAAW;AAAA,EAC/C;AACF;AAOA,SAAS,qBAAqB,MAAY,OAAsB;AAC9D,MAAI,CAAC,SAAS,OAAO,UAAU,SAAU;AACzC,QAAM,IAAI;AACV,MAAI,OAAO,EAAE,SAAS,YAAY,EAAE,MAAM;AACxC,SAAK,aAAa,gBAAgB,EAAE,IAAI;AAAA,EAC1C,WAAW,YAAY,KAAK,EAAE,WAAW,QAAW;AAClD,UAAM,cAAc,cAAc,EAAE,MAAM;AAC1C,QAAI,YAAa,MAAK,aAAa,gBAAgB,WAAW;AAAA,EAChE;AACA,MAAI,EAAE,cAAc;AAClB,SAAK,aAAa,wBAAwB,OAAO,EAAE,YAAY,CAAC;AAAA,EAClE;AACF;AAMA,IAAM,oBAAoD;AAAA,EACxD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAYO,SAAS,UAA6C,UAAgB;AAC3E,QAAM,UAAmC,EAAE,GAAG,SAAS;AAEvD,aAAW,QAAQ,mBAAmB;AACpC,UAAM,WAAW,SAAS,IAAI;AAC9B,QAAI,OAAO,aAAa,WAAY;AAEpC,QAAI,SAAS,gBAAgB,SAAS,gBAAgB;AACpD,cAAQ,IAAI,IAAI,oBAAoB,MAAM,QAAkC;AAAA,IAC9E,OAAO;AACL,cAAQ,IAAI,IAAI,mBAAmB,MAAM,QAA2C;AAAA,IACtF;AAAA,EACF;AAEA,SAAO;AACT;AAEA,SAAS,aAAa,MAAmC;AACvD,MAAI,SAAS,WAAW,SAAS,eAAe,SAAS,SAAU,QAAO;AAC1E,SAAO;AACT;AAGA,SAAS,mBAAmB;AAG1B,SAAO,yBAAyB;AAClC;AAEA,SAAS,mBACP,MACA,UACiC;AACjC,SAAO,eAAe,eAAe,MAA6B;AAChE,UAAM,SAAS,kBAAkB,WAAW;AAM5C,UAAM,gBAAgB,iBAAiB;AACvC,UAAM,OAAO,OAAO;AAAA,MAClB,MAAM,IAAI;AAAA,MACV,EAAE,YAAY,EAAE,2BAA2B,aAAa,IAAI,EAAE,EAAE;AAAA,MAChE;AAAA,IACF;AACA,WAAO;AAAA,MACL;AAAA,MACA,YAAY;AACV,YAAI;AACF,gBAAM,kBAAkB,SAAS,WAAW,SAAS,eAAe,SAAS;AAC7E,cAAI,CAAC,iBAAiB;AACpB,0BAAc,MAAM,IAAI;AAAA,UAC1B;AACA,cAAI,SAAS,YAAY,MAAM,OAAO;AACpC,iBAAK,aAAa,mBAAmB,OAAO,KAAK,KAAK,CAAC;AAAA,UACzD;AACA,gBAAM,SAAS,eAAe,IAAI;AAClC,gBAAM,SAAS,MAAM,SAAS,MAAM;AACpC,cAAI,CAAC,iBAAiB;AACpB,2BAAe,MAAM,MAAM;AAAA,UAC7B;AACA,iBAAO;AAAA,QACT,SAAS,KAAK;AACZ,0BAAgB,MAAM,GAAG;AACzB,gBAAM;AAAA,QACR,UAAE;AACA,eAAK,IAAI;AAAA,QACX;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAQA,SAAS,oBACP,MACA,UACwB;AACxB,SAAO,SAAS,gBAAgB,MAAoB;AAClD,UAAM,SAAS,kBAAkB,WAAW;AAI5C,UAAM,gBAAgB,iBAAiB;AACvC,UAAM,OAAO,OAAO;AAAA,MAClB,MAAM,IAAI;AAAA,MACV,EAAE,YAAY,EAAE,2BAA2B,aAAa,IAAI,EAAE,EAAE;AAAA,MAChE;AAAA,IACF;AACA,WAAO;AAAA,MACL;AAAA,MACA,MAAM;AACJ,YAAI,YAAY;AAChB,cAAM,UAAU,MAAM;AACpB,cAAI,UAAW;AACf,sBAAY;AACZ,eAAK,IAAI;AAAA,QACX;AACA,YAAI;AACF,wBAAc,MAAM,IAAI;AACxB,gBAAM,SAAS,eAAe,IAAI;AAClC,gBAAM,eAAe,MAAM;AAC3B,gBAAM,cAAc,MAAM;AAC1B,gBAAM,cAAc;AAAA,YAClB,GAAG;AAAA,YACH,UAAU,OAAO,UAAe;AAC9B,kBAAI;AACF,qCAAqB,MAAM,KAAK;AAAA,cAClC,UAAE;AACA,wBAAQ;AAAA,cACV;AACA,kBAAI,OAAO,iBAAiB,YAAY;AACtC,uBAAO,aAAa,KAAK;AAAA,cAC3B;AAAA,YACF;AAAA,YACA,SAAS,CAAC,UAAe;AACvB,8BAAgB,OAAO,SAAS,MAAM,UAAU,KAAK;AACrD,sBAAQ;AACR,kBAAI,OAAO,gBAAgB,YAAY;AACrC,uBAAO,YAAY,KAAK;AAAA,cAC1B;AAAA,YACF;AAAA,UACF;AACA,iBAAO,SAAS,WAAW;AAAA,QAC7B,SAAS,KAAK;AAEZ,0BAAgB,MAAM,GAAG;AACzB,kBAAQ;AACR,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,eAAe,MAAgB;AACtC,QAAM,gBAAmC,kBAAkB;AAAA,IACzD,UAAU,MAAM,wBAAwB;AAAA,EAC1C,CAAC;AACD,SAAO;AAAA,IACL,GAAG;AAAA,IACH,wBAAwB;AAAA,MACtB,GAAG,MAAM;AAAA,MACT,GAAG;AAAA,IACL;AAAA,EACF;AACF;AAGA,SAAS,gBAAgB,MAAY,KAAoB;AACvD,MAAI,eAAe,OAAO;AACxB,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,SAAS,IAAI,QAAQ,CAAC;AACnE,SAAK,gBAAgB,GAAG;AAAA,EAC1B,OAAO;AACL,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,SAAS,OAAO,GAAG,EAAE,CAAC;AAAA,EACrE;AACF;","names":["stringified"]}

package/dist/anthropic.cjs

@@ -101,6 +101,7 @@ function currentEndUserMetadata() {
 // src/core/provider.ts
 var import_node_async_hooks3 = require("async_hooks");
 var import_api = require("@opentelemetry/api");
+var NEATLOGS_ROOT_SPAN_KEY = (0, import_api.createContextKey)("neatlogs.root_span");
 var PRIVATE_SPAN_STORAGE_KEY = /* @__PURE__ */ Symbol.for(
   "neatlogs.private_span_async_local_storage"
 );
@@ -135,7 +136,10 @@ function getNeatlogsParentContext(baseContext) {
 }
 function withNeatlogsSpan(span, fn, baseContext) {
   const base = baseContext ?? getNeatlogsActiveContext();
-  const ctx = import_api.trace.setSpan(base, span);
+  let ctx = import_api.trace.setSpan(base, span);
+  if (base.getValue(NEATLOGS_ROOT_SPAN_KEY) === void 0) {
+    ctx = ctx.setValue(NEATLOGS_ROOT_SPAN_KEY, span);
+  }
   return privateContextStorage.run(ctx, fn);
 }