@sentry/node 9.8.0 → 9.10.0

package/build/cjs/integrations/tracing/knex.js

@@ -12,21 +12,26 @@ const instrumentKnex = instrument.generateInstrumentOnce(
 );
 
 const _knexIntegration = (() => {
+  let instrumentationWrappedCallback;
+
   return {
     name: INTEGRATION_NAME,
     setupOnce() {
-      instrumentKnex();
+      const instrumentation = instrumentKnex();
+      instrumentationWrappedCallback = instrument.instrumentWhenWrapped(instrumentation);
     },
 
     setup(client) {
-      client.on('spanStart', span => {
-        const { data } = core.spanToJSON(span);
-        // knex.version is always set in the span data
-        // https://github.com/open-telemetry/opentelemetry-js-contrib/blob/0309caeafc44ac9cb13a3345b790b01b76d0497d/plugins/node/opentelemetry-instrumentation-knex/src/instrumentation.ts#L138
-        if ('knex.version' in data) {
-          span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.knex');
-        }
-      });
+      instrumentationWrappedCallback?.(() =>
+        client.on('spanStart', span => {
+          const { data } = core.spanToJSON(span);
+          // knex.version is always set in the span data
+          // https://github.com/open-telemetry/opentelemetry-js-contrib/blob/0309caeafc44ac9cb13a3345b790b01b76d0497d/plugins/node/opentelemetry-instrumentation-knex/src/instrumentation.ts#L138
+          if ('knex.version' in data) {
+            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.knex');
+          }
+        }),
+      );
     },
   };
 }) ;

package/build/cjs/integrations/tracing/knex.js.map

@@ -1 +1 @@
-{"version":3,"file":"knex.js","sources":["../../../../src/integrations/tracing/knex.ts"],"sourcesContent":["import { KnexInstrumentation } from '@opentelemetry/instrumentation-knex';\nimport { SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, defineIntegration, spanToJSON } from '@sentry/core';\nimport type { IntegrationFn } from '@sentry/core';\nimport { generateInstrumentOnce } from '../../otel/instrument';\n\nconst INTEGRATION_NAME = 'Knex';\n\nexport const instrumentKnex = generateInstrumentOnce(\n  INTEGRATION_NAME,\n  () => new KnexInstrumentation({ requireParentSpan: true }),\n);\n\nconst _knexIntegration = (() => {\n  return {\n    name: INTEGRATION_NAME,\n    setupOnce() {\n      instrumentKnex();\n    },\n\n    setup(client) {\n      client.on('spanStart', span => {\n        const { data } = spanToJSON(span);\n        // knex.version is always set in the span data\n        // https://github.com/open-telemetry/opentelemetry-js-contrib/blob/0309caeafc44ac9cb13a3345b790b01b76d0497d/plugins/node/opentelemetry-instrumentation-knex/src/instrumentation.ts#L138\n        if ('knex.version' in data) {\n          span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.knex');\n        }\n      });\n    },\n  };\n}) satisfies IntegrationFn;\n\n/**\n * Knex integration\n *\n * Capture tracing data for [Knex](https://knexjs.org/).\n *\n * @example\n * ```javascript\n * import * as Sentry from '@sentry/node';\n *\n * Sentry.init({\n *  integrations: [Sentry.knexIntegration()],\n * });\n * ```\n */\nexport const knexIntegration = defineIntegration(_knexIntegration);\n"],"names":["generateInstrumentOnce","KnexInstrumentation","spanToJSON","SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN","defineIntegration"],"mappings":";;;;;;AAKA,MAAM,gBAAA,GAAmB,MAAM;;AAExB,MAAM,cAAe,GAAEA,iCAAsB;AACpD,EAAE,gBAAgB;AAClB,EAAE,MAAM,IAAIC,uCAAmB,CAAC,EAAE,iBAAiB,EAAE,IAAK,EAAC,CAAC;AAC5D;;AAEA,MAAM,gBAAiB,IAAG,MAAM;AAChC,EAAE,OAAO;AACT,IAAI,IAAI,EAAE,gBAAgB;AAC1B,IAAI,SAAS,GAAG;AAChB,MAAM,cAAc,EAAE;AACtB,KAAK;;AAEL,IAAI,KAAK,CAAC,MAAM,EAAE;AAClB,MAAM,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,QAAQ;AACrC,QAAQ,MAAM,EAAE,IAAK,EAAA,GAAIC,eAAU,CAAC,IAAI,CAAC;AACzC;AACA;AACA,QAAQ,IAAI,cAAe,IAAG,IAAI,EAAE;AACpC,UAAU,IAAI,CAAC,YAAY,CAACC,qCAAgC,EAAE,mBAAmB,CAAC;AAClF;AACA,OAAO,CAAC;AACR,KAAK;AACL,GAAG;AACH,CAAC,CAAE;;AAEH;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;MACa,eAAgB,GAAEC,sBAAiB,CAAC,gBAAgB;;;;;"}
+{"version":3,"file":"knex.js","sources":["../../../../src/integrations/tracing/knex.ts"],"sourcesContent":["import { KnexInstrumentation } from '@opentelemetry/instrumentation-knex';\nimport { SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, defineIntegration, spanToJSON } from '@sentry/core';\nimport type { IntegrationFn } from '@sentry/core';\nimport { generateInstrumentOnce, instrumentWhenWrapped } from '../../otel/instrument';\n\nconst INTEGRATION_NAME = 'Knex';\n\nexport const instrumentKnex = generateInstrumentOnce(\n  INTEGRATION_NAME,\n  () => new KnexInstrumentation({ requireParentSpan: true }),\n);\n\nconst _knexIntegration = (() => {\n  let instrumentationWrappedCallback: undefined | ((callback: () => void) => void);\n\n  return {\n    name: INTEGRATION_NAME,\n    setupOnce() {\n      const instrumentation = instrumentKnex();\n      instrumentationWrappedCallback = instrumentWhenWrapped(instrumentation);\n    },\n\n    setup(client) {\n      instrumentationWrappedCallback?.(() =>\n        client.on('spanStart', span => {\n          const { data } = spanToJSON(span);\n          // knex.version is always set in the span data\n          // https://github.com/open-telemetry/opentelemetry-js-contrib/blob/0309caeafc44ac9cb13a3345b790b01b76d0497d/plugins/node/opentelemetry-instrumentation-knex/src/instrumentation.ts#L138\n          if ('knex.version' in data) {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.knex');\n          }\n        }),\n      );\n    },\n  };\n}) satisfies IntegrationFn;\n\n/**\n * Knex integration\n *\n * Capture tracing data for [Knex](https://knexjs.org/).\n *\n * @example\n * ```javascript\n * import * as Sentry from '@sentry/node';\n *\n * Sentry.init({\n *  integrations: [Sentry.knexIntegration()],\n * });\n * ```\n */\nexport const knexIntegration = defineIntegration(_knexIntegration);\n"],"names":["generateInstrumentOnce","KnexInstrumentation","instrumentWhenWrapped","spanToJSON","SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN","defineIntegration"],"mappings":";;;;;;AAKA,MAAM,gBAAA,GAAmB,MAAM;;AAExB,MAAM,cAAe,GAAEA,iCAAsB;AACpD,EAAE,gBAAgB;AAClB,EAAE,MAAM,IAAIC,uCAAmB,CAAC,EAAE,iBAAiB,EAAE,IAAK,EAAC,CAAC;AAC5D;;AAEA,MAAM,gBAAiB,IAAG,MAAM;AAChC,EAAE,IAAI,8BAA8B;;AAEpC,EAAE,OAAO;AACT,IAAI,IAAI,EAAE,gBAAgB;AAC1B,IAAI,SAAS,GAAG;AAChB,MAAM,MAAM,eAAA,GAAkB,cAAc,EAAE;AAC9C,MAAM,8BAA+B,GAAEC,gCAAqB,CAAC,eAAe,CAAC;AAC7E,KAAK;;AAEL,IAAI,KAAK,CAAC,MAAM,EAAE;AAClB,MAAM,8BAA8B,GAAG;AACvC,QAAQ,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,QAAQ;AACvC,UAAU,MAAM,EAAE,IAAK,EAAA,GAAIC,eAAU,CAAC,IAAI,CAAC;AAC3C;AACA;AACA,UAAU,IAAI,cAAe,IAAG,IAAI,EAAE;AACtC,YAAY,IAAI,CAAC,YAAY,CAACC,qCAAgC,EAAE,mBAAmB,CAAC;AACpF;AACA,SAAS,CAAC;AACV,OAAO;AACP,KAAK;AACL,GAAG;AACH,CAAC,CAAE;;AAEH;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;MACa,eAAgB,GAAEC,sBAAiB,CAAC,gBAAgB;;;;;"}

package/build/cjs/integrations/tracing/prisma.js

@@ -15,6 +15,18 @@ function isPrismaV6TracingHelper(helper) {
   return !!helper && typeof helper === 'object' && 'dispatchEngineSpans' in helper;
 }
 
+function getPrismaTracingHelper() {
+  const prismaInstrumentationObject = (globalThis ).PRISMA_INSTRUMENTATION;
+  const prismaTracingHelper =
+    prismaInstrumentationObject &&
+    typeof prismaInstrumentationObject === 'object' &&
+    'helper' in prismaInstrumentationObject
+      ? prismaInstrumentationObject.helper
+      : undefined;
+
+  return prismaTracingHelper;
+}
+
 class SentryPrismaInteropInstrumentation extends EsmInteropPrismaInstrumentation {
    constructor() {
     super();
@@ -26,13 +38,7 @@ class SentryPrismaInteropInstrumentation extends EsmInteropPrismaInstrumentation
     // The PrismaIntegration (super class) defines a global variable `global["PRISMA_INSTRUMENTATION"]` when `enable()` is called. This global variable holds a "TracingHelper" which Prisma uses internally to create tracing data. It's their way of not depending on OTEL with their main package. The sucky thing is, prisma broke the interface of the tracing helper with the v6 major update. This means that if you use Prisma 5 with the v6 instrumentation (or vice versa) Prisma just blows up, because tries to call methods on the helper that no longer exist.
     // Because we actually want to use the v6 instrumentation and not blow up in Prisma 5 user's faces, what we're doing here is backfilling the v5 method (`createEngineSpan`) with a noop so that no longer crashes when it attempts to call that function.
     // We still won't fully emit all the spans, but this could potentially be implemented in the future.
-    const prismaInstrumentationObject = (globalThis ).PRISMA_INSTRUMENTATION;
-    const prismaTracingHelper =
-      prismaInstrumentationObject &&
-      typeof prismaInstrumentationObject === 'object' &&
-      'helper' in prismaInstrumentationObject
-        ? prismaInstrumentationObject.helper
-        : undefined;
+    const prismaTracingHelper = getPrismaTracingHelper();
 
     let emittedWarning = false;
 
@@ -108,6 +114,12 @@ const prismaIntegration = core.defineIntegration(
         instrumentPrisma({ prismaInstrumentation });
       },
       setup(client) {
+        // If no tracing helper exists, we skip any work here
+        // this means that prisma is not being used
+        if (!getPrismaTracingHelper()) {
+          return;
+        }
+
         client.on('spanStart', span => {
           const spanJSON = core.spanToJSON(span);
           if (spanJSON.description?.startsWith('prisma:')) {

package/build/cjs/integrations/tracing/prisma.js.map

@@ -1 +1 @@
-{"version":3,"file":"prisma.js","sources":["../../../../src/integrations/tracing/prisma.ts"],"sourcesContent":["import type { Instrumentation } from '@opentelemetry/instrumentation';\n// When importing CJS modules into an ESM module, we cannot import the named exports directly.\nimport * as prismaInstrumentation from '@prisma/instrumentation';\nimport { SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, consoleSandbox, defineIntegration, spanToJSON } from '@sentry/core';\nimport { generateInstrumentOnce } from '../../otel/instrument';\nimport type { PrismaV5TracingHelper } from './prisma/vendor/v5-tracing-helper';\nimport type { PrismaV6TracingHelper } from './prisma/vendor/v6-tracing-helper';\n\nconst INTEGRATION_NAME = 'Prisma';\n\nconst EsmInteropPrismaInstrumentation: typeof prismaInstrumentation.PrismaInstrumentation =\n  // @ts-expect-error We need to do the following for interop reasons\n  // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n  prismaInstrumentation.default?.PrismaInstrumentation || prismaInstrumentation.PrismaInstrumentation;\n\ntype CompatibilityLayerTraceHelper = PrismaV5TracingHelper & PrismaV6TracingHelper;\n\nfunction isPrismaV6TracingHelper(helper: unknown): helper is PrismaV6TracingHelper {\n  return !!helper && typeof helper === 'object' && 'dispatchEngineSpans' in helper;\n}\n\nclass SentryPrismaInteropInstrumentation extends EsmInteropPrismaInstrumentation {\n  public constructor() {\n    super();\n  }\n\n  public enable(): void {\n    super.enable();\n\n    // The PrismaIntegration (super class) defines a global variable `global[\"PRISMA_INSTRUMENTATION\"]` when `enable()` is called. This global variable holds a \"TracingHelper\" which Prisma uses internally to create tracing data. It's their way of not depending on OTEL with their main package. The sucky thing is, prisma broke the interface of the tracing helper with the v6 major update. This means that if you use Prisma 5 with the v6 instrumentation (or vice versa) Prisma just blows up, because tries to call methods on the helper that no longer exist.\n    // Because we actually want to use the v6 instrumentation and not blow up in Prisma 5 user's faces, what we're doing here is backfilling the v5 method (`createEngineSpan`) with a noop so that no longer crashes when it attempts to call that function.\n    // We still won't fully emit all the spans, but this could potentially be implemented in the future.\n    const prismaInstrumentationObject = (globalThis as Record<string, unknown>).PRISMA_INSTRUMENTATION;\n    const prismaTracingHelper =\n      prismaInstrumentationObject &&\n      typeof prismaInstrumentationObject === 'object' &&\n      'helper' in prismaInstrumentationObject\n        ? prismaInstrumentationObject.helper\n        : undefined;\n\n    let emittedWarning = false;\n\n    if (isPrismaV6TracingHelper(prismaTracingHelper)) {\n      (prismaTracingHelper as CompatibilityLayerTraceHelper).createEngineSpan = () => {\n        consoleSandbox(() => {\n          if (!emittedWarning) {\n            emittedWarning = true;\n            // eslint-disable-next-line no-console\n            console.warn(\n              '[Sentry] The Sentry SDK supports tracing with Prisma version 5 only with limited capabilities. For full tracing capabilities pass `prismaInstrumentation` for version 5 to the Sentry `prismaIntegration`. Read more: https://docs.sentry.io/platforms/javascript/guides/node/configuration/integrations/prisma/',\n            );\n          }\n        });\n      };\n    }\n  }\n}\n\nexport const instrumentPrisma = generateInstrumentOnce<{ prismaInstrumentation?: Instrumentation }>(\n  INTEGRATION_NAME,\n  options => {\n    // Use a passed instrumentation instance to support older Prisma versions\n    if (options?.prismaInstrumentation) {\n      return options.prismaInstrumentation;\n    }\n\n    return new SentryPrismaInteropInstrumentation();\n  },\n);\n\n/**\n * Adds Sentry tracing instrumentation for the [prisma](https://www.npmjs.com/package/prisma) library.\n * For more information, see the [`prismaIntegration` documentation](https://docs.sentry.io/platforms/javascript/guides/node/configuration/integrations/prisma/).\n *\n * NOTE: By default, this integration works with Prisma version 6.\n * To get performance instrumentation for other Prisma versions,\n * 1. Install the `@prisma/instrumentation` package with the desired version.\n * 1. Pass a `new PrismaInstrumentation()` instance as exported from `@prisma/instrumentation` to the `prismaInstrumentation` option of this integration:\n *\n *    ```js\n *    import { PrismaInstrumentation } from '@prisma/instrumentation'\n *\n *    Sentry.init({\n *      integrations: [\n *        prismaIntegration({\n *          // Override the default instrumentation that Sentry uses\n *          prismaInstrumentation: new PrismaInstrumentation()\n *        })\n *      ]\n *    })\n *    ```\n *\n *    The passed instrumentation instance will override the default instrumentation instance the integration would use, while the `prismaIntegration` will still ensure data compatibility for the various Prisma versions.\n * 1. Depending on your Prisma version (prior to version 6), add `previewFeatures = [\"tracing\"]` to the client generator block of your Prisma schema:\n *\n *    ```\n *    generator client {\n *      provider = \"prisma-client-js\"\n *      previewFeatures = [\"tracing\"]\n *    }\n *    ```\n */\nexport const prismaIntegration = defineIntegration(\n  ({\n    prismaInstrumentation,\n  }: {\n    /**\n     * Overrides the instrumentation used by the Sentry SDK with the passed in instrumentation instance.\n     *\n     * NOTE: By default, the Sentry SDK uses the Prisma v6 instrumentation. Use this option if you need performance instrumentation different Prisma versions.\n     *\n     * For more information refer to the documentation of `prismaIntegration()` or see https://docs.sentry.io/platforms/javascript/guides/node/configuration/integrations/prisma/\n     */\n    prismaInstrumentation?: Instrumentation;\n  } = {}) => {\n    return {\n      name: INTEGRATION_NAME,\n      setupOnce() {\n        instrumentPrisma({ prismaInstrumentation });\n      },\n      setup(client) {\n        client.on('spanStart', span => {\n          const spanJSON = spanToJSON(span);\n          if (spanJSON.description?.startsWith('prisma:')) {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.prisma');\n          }\n\n          // Make sure we use the query text as the span name, for ex. SELECT * FROM \"User\" WHERE \"id\" = $1\n          if (spanJSON.description === 'prisma:engine:db_query' && spanJSON.data['db.query.text']) {\n            span.updateName(spanJSON.data['db.query.text'] as string);\n          }\n\n          // In Prisma v5.22+, the `db.system` attribute is automatically set\n          // On older versions, this is missing, so we add it here\n          if (spanJSON.description === 'prisma:engine:db_query' && !spanJSON.data['db.system']) {\n            span.setAttribute('db.system', 'prisma');\n          }\n        });\n      },\n    };\n  },\n);\n"],"names":["consoleSandbox","generateInstrumentOnce","defineIntegration","spanToJSON","SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN"],"mappings":";;;;;;AAQA,MAAM,gBAAA,GAAmB,QAAQ;;AAEjC,MAAM,+BAA+B;AACrC;AACA;AACA,EAAE,qBAAqB,CAAC,OAAO,EAAE,yBAAyB,qBAAqB,CAAC,qBAAqB;;AAIrG,SAAS,uBAAuB,CAAC,MAAM,EAA4C;AACnF,EAAE,OAAO,CAAC,CAAC,MAAA,IAAU,OAAO,MAAO,KAAI,QAAS,IAAG,qBAAsB,IAAG,MAAM;AAClF;;AAEA,MAAM,kCAAA,SAA2C,+BAAgC,CAAA;AACjF,GAAS,WAAW,GAAG;AACvB,IAAI,KAAK,EAAE;AACX;;AAEA,GAAS,MAAM,GAAS;AACxB,IAAI,KAAK,CAAC,MAAM,EAAE;;AAElB;AACA;AACA;AACA,IAAI,MAAM,2BAA4B,GAAE,CAAC,UAAW,GAA4B,sBAAsB;AACtG,IAAI,MAAM,mBAAoB;AAC9B,MAAM,2BAA4B;AAClC,MAAM,OAAO,2BAA4B,KAAI,QAAS;AACtD,MAAM,YAAY;AAClB,UAAU,2BAA2B,CAAC;AACtC,UAAU,SAAS;;AAEnB,IAAI,IAAI,cAAe,GAAE,KAAK;;AAE9B,IAAI,IAAI,uBAAuB,CAAC,mBAAmB,CAAC,EAAE;AACtD,MAAM,CAAC,sBAAsD,mBAAmB,MAAM;AACtF,QAAQA,mBAAc,CAAC,MAAM;AAC7B,UAAU,IAAI,CAAC,cAAc,EAAE;AAC/B,YAAY,cAAA,GAAiB,IAAI;AACjC;AACA,YAAY,OAAO,CAAC,IAAI;AACxB,cAAc,kTAAkT;AAChU,aAAa;AACb;AACA,SAAS,CAAC;AACV,OAAO;AACP;AACA;AACA;;AAEO,MAAM,gBAAiB,GAAEC,iCAAsB;AACtD,EAAE,gBAAgB;AAClB,EAAE,WAAW;AACb;AACA,IAAI,IAAI,OAAO,EAAE,qBAAqB,EAAE;AACxC,MAAM,OAAO,OAAO,CAAC,qBAAqB;AAC1C;;AAEA,IAAI,OAAO,IAAI,kCAAkC,EAAE;AACnD,GAAG;AACH;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACO,MAAM,iBAAkB,GAAEC,sBAAiB;AAClD,EAAE,CAAC;AACH,IAAI,qBAAqB;AACzB;;AASE,GAAI,EAAE,KAAK;AACb,IAAI,OAAO;AACX,MAAM,IAAI,EAAE,gBAAgB;AAC5B,MAAM,SAAS,GAAG;AAClB,QAAQ,gBAAgB,CAAC,EAAE,qBAAA,EAAuB,CAAC;AACnD,OAAO;AACP,MAAM,KAAK,CAAC,MAAM,EAAE;AACpB,QAAQ,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,QAAQ;AACvC,UAAU,MAAM,QAAS,GAAEC,eAAU,CAAC,IAAI,CAAC;AAC3C,UAAU,IAAI,QAAQ,CAAC,WAAW,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE;AAC3D,YAAY,IAAI,CAAC,YAAY,CAACC,qCAAgC,EAAE,qBAAqB,CAAC;AACtF;;AAEA;AACA,UAAU,IAAI,QAAQ,CAAC,gBAAgB,wBAAA,IAA4B,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE;AACnG,YAAY,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAA,EAAY;AACrE;;AAEA;AACA;AACA,UAAU,IAAI,QAAQ,CAAC,gBAAgB,wBAAA,IAA4B,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE;AAChG,YAAY,IAAI,CAAC,YAAY,CAAC,WAAW,EAAE,QAAQ,CAAC;AACpD;AACA,SAAS,CAAC;AACV,OAAO;AACP,KAAK;AACL,GAAG;AACH;;;;;"}
+{"version":3,"file":"prisma.js","sources":["../../../../src/integrations/tracing/prisma.ts"],"sourcesContent":["import type { Instrumentation } from '@opentelemetry/instrumentation';\n// When importing CJS modules into an ESM module, we cannot import the named exports directly.\nimport * as prismaInstrumentation from '@prisma/instrumentation';\nimport { SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, consoleSandbox, defineIntegration, spanToJSON } from '@sentry/core';\nimport { generateInstrumentOnce } from '../../otel/instrument';\nimport type { PrismaV5TracingHelper } from './prisma/vendor/v5-tracing-helper';\nimport type { PrismaV6TracingHelper } from './prisma/vendor/v6-tracing-helper';\n\nconst INTEGRATION_NAME = 'Prisma';\n\nconst EsmInteropPrismaInstrumentation: typeof prismaInstrumentation.PrismaInstrumentation =\n  // @ts-expect-error We need to do the following for interop reasons\n  // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n  prismaInstrumentation.default?.PrismaInstrumentation || prismaInstrumentation.PrismaInstrumentation;\n\ntype CompatibilityLayerTraceHelper = PrismaV5TracingHelper & PrismaV6TracingHelper;\n\nfunction isPrismaV6TracingHelper(helper: unknown): helper is PrismaV6TracingHelper {\n  return !!helper && typeof helper === 'object' && 'dispatchEngineSpans' in helper;\n}\n\nfunction getPrismaTracingHelper(): unknown | undefined {\n  const prismaInstrumentationObject = (globalThis as Record<string, unknown>).PRISMA_INSTRUMENTATION;\n  const prismaTracingHelper =\n    prismaInstrumentationObject &&\n    typeof prismaInstrumentationObject === 'object' &&\n    'helper' in prismaInstrumentationObject\n      ? prismaInstrumentationObject.helper\n      : undefined;\n\n  return prismaTracingHelper;\n}\n\nclass SentryPrismaInteropInstrumentation extends EsmInteropPrismaInstrumentation {\n  public constructor() {\n    super();\n  }\n\n  public enable(): void {\n    super.enable();\n\n    // The PrismaIntegration (super class) defines a global variable `global[\"PRISMA_INSTRUMENTATION\"]` when `enable()` is called. This global variable holds a \"TracingHelper\" which Prisma uses internally to create tracing data. It's their way of not depending on OTEL with their main package. The sucky thing is, prisma broke the interface of the tracing helper with the v6 major update. This means that if you use Prisma 5 with the v6 instrumentation (or vice versa) Prisma just blows up, because tries to call methods on the helper that no longer exist.\n    // Because we actually want to use the v6 instrumentation and not blow up in Prisma 5 user's faces, what we're doing here is backfilling the v5 method (`createEngineSpan`) with a noop so that no longer crashes when it attempts to call that function.\n    // We still won't fully emit all the spans, but this could potentially be implemented in the future.\n    const prismaTracingHelper = getPrismaTracingHelper();\n\n    let emittedWarning = false;\n\n    if (isPrismaV6TracingHelper(prismaTracingHelper)) {\n      (prismaTracingHelper as CompatibilityLayerTraceHelper).createEngineSpan = () => {\n        consoleSandbox(() => {\n          if (!emittedWarning) {\n            emittedWarning = true;\n            // eslint-disable-next-line no-console\n            console.warn(\n              '[Sentry] The Sentry SDK supports tracing with Prisma version 5 only with limited capabilities. For full tracing capabilities pass `prismaInstrumentation` for version 5 to the Sentry `prismaIntegration`. Read more: https://docs.sentry.io/platforms/javascript/guides/node/configuration/integrations/prisma/',\n            );\n          }\n        });\n      };\n    }\n  }\n}\n\nexport const instrumentPrisma = generateInstrumentOnce<{ prismaInstrumentation?: Instrumentation }>(\n  INTEGRATION_NAME,\n  options => {\n    // Use a passed instrumentation instance to support older Prisma versions\n    if (options?.prismaInstrumentation) {\n      return options.prismaInstrumentation;\n    }\n\n    return new SentryPrismaInteropInstrumentation();\n  },\n);\n\n/**\n * Adds Sentry tracing instrumentation for the [prisma](https://www.npmjs.com/package/prisma) library.\n * For more information, see the [`prismaIntegration` documentation](https://docs.sentry.io/platforms/javascript/guides/node/configuration/integrations/prisma/).\n *\n * NOTE: By default, this integration works with Prisma version 6.\n * To get performance instrumentation for other Prisma versions,\n * 1. Install the `@prisma/instrumentation` package with the desired version.\n * 1. Pass a `new PrismaInstrumentation()` instance as exported from `@prisma/instrumentation` to the `prismaInstrumentation` option of this integration:\n *\n *    ```js\n *    import { PrismaInstrumentation } from '@prisma/instrumentation'\n *\n *    Sentry.init({\n *      integrations: [\n *        prismaIntegration({\n *          // Override the default instrumentation that Sentry uses\n *          prismaInstrumentation: new PrismaInstrumentation()\n *        })\n *      ]\n *    })\n *    ```\n *\n *    The passed instrumentation instance will override the default instrumentation instance the integration would use, while the `prismaIntegration` will still ensure data compatibility for the various Prisma versions.\n * 1. Depending on your Prisma version (prior to version 6), add `previewFeatures = [\"tracing\"]` to the client generator block of your Prisma schema:\n *\n *    ```\n *    generator client {\n *      provider = \"prisma-client-js\"\n *      previewFeatures = [\"tracing\"]\n *    }\n *    ```\n */\nexport const prismaIntegration = defineIntegration(\n  ({\n    prismaInstrumentation,\n  }: {\n    /**\n     * Overrides the instrumentation used by the Sentry SDK with the passed in instrumentation instance.\n     *\n     * NOTE: By default, the Sentry SDK uses the Prisma v6 instrumentation. Use this option if you need performance instrumentation different Prisma versions.\n     *\n     * For more information refer to the documentation of `prismaIntegration()` or see https://docs.sentry.io/platforms/javascript/guides/node/configuration/integrations/prisma/\n     */\n    prismaInstrumentation?: Instrumentation;\n  } = {}) => {\n    return {\n      name: INTEGRATION_NAME,\n      setupOnce() {\n        instrumentPrisma({ prismaInstrumentation });\n      },\n      setup(client) {\n        // If no tracing helper exists, we skip any work here\n        // this means that prisma is not being used\n        if (!getPrismaTracingHelper()) {\n          return;\n        }\n\n        client.on('spanStart', span => {\n          const spanJSON = spanToJSON(span);\n          if (spanJSON.description?.startsWith('prisma:')) {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.prisma');\n          }\n\n          // Make sure we use the query text as the span name, for ex. SELECT * FROM \"User\" WHERE \"id\" = $1\n          if (spanJSON.description === 'prisma:engine:db_query' && spanJSON.data['db.query.text']) {\n            span.updateName(spanJSON.data['db.query.text'] as string);\n          }\n\n          // In Prisma v5.22+, the `db.system` attribute is automatically set\n          // On older versions, this is missing, so we add it here\n          if (spanJSON.description === 'prisma:engine:db_query' && !spanJSON.data['db.system']) {\n            span.setAttribute('db.system', 'prisma');\n          }\n        });\n      },\n    };\n  },\n);\n"],"names":["consoleSandbox","generateInstrumentOnce","defineIntegration","spanToJSON","SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN"],"mappings":";;;;;;AAQA,MAAM,gBAAA,GAAmB,QAAQ;;AAEjC,MAAM,+BAA+B;AACrC;AACA;AACA,EAAE,qBAAqB,CAAC,OAAO,EAAE,yBAAyB,qBAAqB,CAAC,qBAAqB;;AAIrG,SAAS,uBAAuB,CAAC,MAAM,EAA4C;AACnF,EAAE,OAAO,CAAC,CAAC,MAAA,IAAU,OAAO,MAAO,KAAI,QAAS,IAAG,qBAAsB,IAAG,MAAM;AAClF;;AAEA,SAAS,sBAAsB,GAAwB;AACvD,EAAE,MAAM,2BAA4B,GAAE,CAAC,UAAW,GAA4B,sBAAsB;AACpG,EAAE,MAAM,mBAAoB;AAC5B,IAAI,2BAA4B;AAChC,IAAI,OAAO,2BAA4B,KAAI,QAAS;AACpD,IAAI,YAAY;AAChB,QAAQ,2BAA2B,CAAC;AACpC,QAAQ,SAAS;;AAEjB,EAAE,OAAO,mBAAmB;AAC5B;;AAEA,MAAM,kCAAA,SAA2C,+BAAgC,CAAA;AACjF,GAAS,WAAW,GAAG;AACvB,IAAI,KAAK,EAAE;AACX;;AAEA,GAAS,MAAM,GAAS;AACxB,IAAI,KAAK,CAAC,MAAM,EAAE;;AAElB;AACA;AACA;AACA,IAAI,MAAM,mBAAA,GAAsB,sBAAsB,EAAE;;AAExD,IAAI,IAAI,cAAe,GAAE,KAAK;;AAE9B,IAAI,IAAI,uBAAuB,CAAC,mBAAmB,CAAC,EAAE;AACtD,MAAM,CAAC,sBAAsD,mBAAmB,MAAM;AACtF,QAAQA,mBAAc,CAAC,MAAM;AAC7B,UAAU,IAAI,CAAC,cAAc,EAAE;AAC/B,YAAY,cAAA,GAAiB,IAAI;AACjC;AACA,YAAY,OAAO,CAAC,IAAI;AACxB,cAAc,kTAAkT;AAChU,aAAa;AACb;AACA,SAAS,CAAC;AACV,OAAO;AACP;AACA;AACA;;AAEO,MAAM,gBAAiB,GAAEC,iCAAsB;AACtD,EAAE,gBAAgB;AAClB,EAAE,WAAW;AACb;AACA,IAAI,IAAI,OAAO,EAAE,qBAAqB,EAAE;AACxC,MAAM,OAAO,OAAO,CAAC,qBAAqB;AAC1C;;AAEA,IAAI,OAAO,IAAI,kCAAkC,EAAE;AACnD,GAAG;AACH;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACO,MAAM,iBAAkB,GAAEC,sBAAiB;AAClD,EAAE,CAAC;AACH,IAAI,qBAAqB;AACzB;;AASE,GAAI,EAAE,KAAK;AACb,IAAI,OAAO;AACX,MAAM,IAAI,EAAE,gBAAgB;AAC5B,MAAM,SAAS,GAAG;AAClB,QAAQ,gBAAgB,CAAC,EAAE,qBAAA,EAAuB,CAAC;AACnD,OAAO;AACP,MAAM,KAAK,CAAC,MAAM,EAAE;AACpB;AACA;AACA,QAAQ,IAAI,CAAC,sBAAsB,EAAE,EAAE;AACvC,UAAU;AACV;;AAEA,QAAQ,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,QAAQ;AACvC,UAAU,MAAM,QAAS,GAAEC,eAAU,CAAC,IAAI,CAAC;AAC3C,UAAU,IAAI,QAAQ,CAAC,WAAW,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE;AAC3D,YAAY,IAAI,CAAC,YAAY,CAACC,qCAAgC,EAAE,qBAAqB,CAAC;AACtF;;AAEA;AACA,UAAU,IAAI,QAAQ,CAAC,gBAAgB,wBAAA,IAA4B,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE;AACnG,YAAY,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAA,EAAY;AACrE;;AAEA;AACA;AACA,UAAU,IAAI,QAAQ,CAAC,gBAAgB,wBAAA,IAA4B,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE;AAChG,YAAY,IAAI,CAAC,YAAY,CAAC,WAAW,EAAE,QAAQ,CAAC;AACpD;AACA,SAAS,CAAC;AACV,OAAO;AACP,KAAK;AACL,GAAG;AACH;;;;;"}

package/build/cjs/integrations/tracing/tedious.js

@@ -18,25 +18,30 @@ const INTEGRATION_NAME = 'Tedious';
 const instrumentTedious = instrument.generateInstrumentOnce(INTEGRATION_NAME, () => new instrumentationTedious.TediousInstrumentation({}));
 
 const _tediousIntegration = (() => {
+  let instrumentationWrappedCallback;
+
   return {
     name: INTEGRATION_NAME,
     setupOnce() {
-      instrumentTedious();
+      const instrumentation = instrumentTedious();
+      instrumentationWrappedCallback = instrument.instrumentWhenWrapped(instrumentation);
     },
 
     setup(client) {
-      client.on('spanStart', span => {
-        const { description, data } = core.spanToJSON(span);
-        // Tedius integration always set a span name and `db.system` attribute to `mssql`.
-        if (!description || data['db.system'] !== 'mssql') {
-          return;
-        }
-
-        const operation = description.split(' ')[0] || '';
-        if (TEDIUS_INSTRUMENTED_METHODS.has(operation)) {
-          span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.tedious');
-        }
-      });
+      instrumentationWrappedCallback?.(() =>
+        client.on('spanStart', span => {
+          const { description, data } = core.spanToJSON(span);
+          // Tedius integration always set a span name and `db.system` attribute to `mssql`.
+          if (!description || data['db.system'] !== 'mssql') {
+            return;
+          }
+
+          const operation = description.split(' ')[0] || '';
+          if (TEDIUS_INSTRUMENTED_METHODS.has(operation)) {
+            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.tedious');
+          }
+        }),
+      );
     },
   };
 }) ;

package/build/cjs/integrations/tracing/tedious.js.map

@@ -1 +1 @@
-{"version":3,"file":"tedious.js","sources":["../../../../src/integrations/tracing/tedious.ts"],"sourcesContent":["import { TediousInstrumentation } from '@opentelemetry/instrumentation-tedious';\nimport { SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, defineIntegration, spanToJSON } from '@sentry/core';\nimport type { IntegrationFn } from '@sentry/core';\nimport { generateInstrumentOnce } from '../../otel/instrument';\n\nconst TEDIUS_INSTRUMENTED_METHODS = new Set([\n  'callProcedure',\n  'execSql',\n  'execSqlBatch',\n  'execBulkLoad',\n  'prepare',\n  'execute',\n]);\n\nconst INTEGRATION_NAME = 'Tedious';\n\nexport const instrumentTedious = generateInstrumentOnce(INTEGRATION_NAME, () => new TediousInstrumentation({}));\n\nconst _tediousIntegration = (() => {\n  return {\n    name: INTEGRATION_NAME,\n    setupOnce() {\n      instrumentTedious();\n    },\n\n    setup(client) {\n      client.on('spanStart', span => {\n        const { description, data } = spanToJSON(span);\n        // Tedius integration always set a span name and `db.system` attribute to `mssql`.\n        if (!description || data['db.system'] !== 'mssql') {\n          return;\n        }\n\n        const operation = description.split(' ')[0] || '';\n        if (TEDIUS_INSTRUMENTED_METHODS.has(operation)) {\n          span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.tedious');\n        }\n      });\n    },\n  };\n}) satisfies IntegrationFn;\n\n/**\n * Adds Sentry tracing instrumentation for the [tedious](https://www.npmjs.com/package/tedious) library.\n *\n * For more information, see the [`tediousIntegration` documentation](https://docs.sentry.io/platforms/javascript/guides/node/configuration/integrations/tedious/).\n *\n * @example\n * ```javascript\n * const Sentry = require('@sentry/node');\n *\n * Sentry.init({\n *  integrations: [Sentry.tediousIntegration()],\n * });\n * ```\n */\nexport const tediousIntegration = defineIntegration(_tediousIntegration);\n"],"names":["generateInstrumentOnce","TediousInstrumentation","spanToJSON","SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN","defineIntegration"],"mappings":";;;;;;AAKA,MAAM,2BAA4B,GAAE,IAAI,GAAG,CAAC;AAC5C,EAAE,eAAe;AACjB,EAAE,SAAS;AACX,EAAE,cAAc;AAChB,EAAE,cAAc;AAChB,EAAE,SAAS;AACX,EAAE,SAAS;AACX,CAAC,CAAC;;AAEF,MAAM,gBAAA,GAAmB,SAAS;;MAErB,iBAAkB,GAAEA,iCAAsB,CAAC,gBAAgB,EAAE,MAAM,IAAIC,6CAAsB,CAAC,EAAE,CAAC;;AAE9G,MAAM,mBAAoB,IAAG,MAAM;AACnC,EAAE,OAAO;AACT,IAAI,IAAI,EAAE,gBAAgB;AAC1B,IAAI,SAAS,GAAG;AAChB,MAAM,iBAAiB,EAAE;AACzB,KAAK;;AAEL,IAAI,KAAK,CAAC,MAAM,EAAE;AAClB,MAAM,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,QAAQ;AACrC,QAAQ,MAAM,EAAE,WAAW,EAAE,IAAA,KAASC,eAAU,CAAC,IAAI,CAAC;AACtD;AACA,QAAQ,IAAI,CAAC,WAAY,IAAG,IAAI,CAAC,WAAW,CAAA,KAAM,OAAO,EAAE;AAC3D,UAAU;AACV;;AAEA,QAAQ,MAAM,SAAA,GAAY,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAA,IAAK,EAAE;AACzD,QAAQ,IAAI,2BAA2B,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;AACxD,UAAU,IAAI,CAAC,YAAY,CAACC,qCAAgC,EAAE,sBAAsB,CAAC;AACrF;AACA,OAAO,CAAC;AACR,KAAK;AACL,GAAG;AACH,CAAC,CAAE;;AAEH;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;MACa,kBAAmB,GAAEC,sBAAiB,CAAC,mBAAmB;;;;;"}
+{"version":3,"file":"tedious.js","sources":["../../../../src/integrations/tracing/tedious.ts"],"sourcesContent":["import { TediousInstrumentation } from '@opentelemetry/instrumentation-tedious';\nimport { SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, defineIntegration, spanToJSON } from '@sentry/core';\nimport type { IntegrationFn } from '@sentry/core';\nimport { generateInstrumentOnce, instrumentWhenWrapped } from '../../otel/instrument';\n\nconst TEDIUS_INSTRUMENTED_METHODS = new Set([\n  'callProcedure',\n  'execSql',\n  'execSqlBatch',\n  'execBulkLoad',\n  'prepare',\n  'execute',\n]);\n\nconst INTEGRATION_NAME = 'Tedious';\n\nexport const instrumentTedious = generateInstrumentOnce(INTEGRATION_NAME, () => new TediousInstrumentation({}));\n\nconst _tediousIntegration = (() => {\n  let instrumentationWrappedCallback: undefined | ((callback: () => void) => void);\n\n  return {\n    name: INTEGRATION_NAME,\n    setupOnce() {\n      const instrumentation = instrumentTedious();\n      instrumentationWrappedCallback = instrumentWhenWrapped(instrumentation);\n    },\n\n    setup(client) {\n      instrumentationWrappedCallback?.(() =>\n        client.on('spanStart', span => {\n          const { description, data } = spanToJSON(span);\n          // Tedius integration always set a span name and `db.system` attribute to `mssql`.\n          if (!description || data['db.system'] !== 'mssql') {\n            return;\n          }\n\n          const operation = description.split(' ')[0] || '';\n          if (TEDIUS_INSTRUMENTED_METHODS.has(operation)) {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.tedious');\n          }\n        }),\n      );\n    },\n  };\n}) satisfies IntegrationFn;\n\n/**\n * Adds Sentry tracing instrumentation for the [tedious](https://www.npmjs.com/package/tedious) library.\n *\n * For more information, see the [`tediousIntegration` documentation](https://docs.sentry.io/platforms/javascript/guides/node/configuration/integrations/tedious/).\n *\n * @example\n * ```javascript\n * const Sentry = require('@sentry/node');\n *\n * Sentry.init({\n *  integrations: [Sentry.tediousIntegration()],\n * });\n * ```\n */\nexport const tediousIntegration = defineIntegration(_tediousIntegration);\n"],"names":["generateInstrumentOnce","TediousInstrumentation","instrumentWhenWrapped","spanToJSON","SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN","defineIntegration"],"mappings":";;;;;;AAKA,MAAM,2BAA4B,GAAE,IAAI,GAAG,CAAC;AAC5C,EAAE,eAAe;AACjB,EAAE,SAAS;AACX,EAAE,cAAc;AAChB,EAAE,cAAc;AAChB,EAAE,SAAS;AACX,EAAE,SAAS;AACX,CAAC,CAAC;;AAEF,MAAM,gBAAA,GAAmB,SAAS;;MAErB,iBAAkB,GAAEA,iCAAsB,CAAC,gBAAgB,EAAE,MAAM,IAAIC,6CAAsB,CAAC,EAAE,CAAC;;AAE9G,MAAM,mBAAoB,IAAG,MAAM;AACnC,EAAE,IAAI,8BAA8B;;AAEpC,EAAE,OAAO;AACT,IAAI,IAAI,EAAE,gBAAgB;AAC1B,IAAI,SAAS,GAAG;AAChB,MAAM,MAAM,eAAA,GAAkB,iBAAiB,EAAE;AACjD,MAAM,8BAA+B,GAAEC,gCAAqB,CAAC,eAAe,CAAC;AAC7E,KAAK;;AAEL,IAAI,KAAK,CAAC,MAAM,EAAE;AAClB,MAAM,8BAA8B,GAAG;AACvC,QAAQ,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,QAAQ;AACvC,UAAU,MAAM,EAAE,WAAW,EAAE,IAAA,KAASC,eAAU,CAAC,IAAI,CAAC;AACxD;AACA,UAAU,IAAI,CAAC,WAAY,IAAG,IAAI,CAAC,WAAW,CAAA,KAAM,OAAO,EAAE;AAC7D,YAAY;AACZ;;AAEA,UAAU,MAAM,SAAA,GAAY,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAA,IAAK,EAAE;AAC3D,UAAU,IAAI,2BAA2B,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;AAC1D,YAAY,IAAI,CAAC,YAAY,CAACC,qCAAgC,EAAE,sBAAsB,CAAC;AACvF;AACA,SAAS,CAAC;AACV,OAAO;AACP,KAAK;AACL,GAAG;AACH,CAAC,CAAE;;AAEH;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;MACa,kBAAmB,GAAEC,sBAAiB,CAAC,mBAAmB;;;;;"}

package/build/cjs/integrations/tracing/vercelai/index.js

@@ -12,145 +12,146 @@ const INTEGRATION_NAME = 'VercelAI';
 const instrumentVercelAi = instrument.generateInstrumentOnce(INTEGRATION_NAME, () => new instrumentation.SentryVercelAiInstrumentation({}));
 
 const _vercelAIIntegration = (() => {
+  let instrumentation;
+
   return {
     name: INTEGRATION_NAME,
     setupOnce() {
-      instrumentVercelAi();
-    },
-    processEvent(event) {
-      if (event.type === 'transaction' && event.spans?.length) {
-        for (const span of event.spans) {
-          const { data: attributes, description: name } = span;
-
-          if (!name || span.origin !== 'auto.vercelai.otel') {
-            continue;
-          }
-
-          if (attributes['ai.usage.completionTokens'] != undefined) {
-            attributes['ai.completion_tokens.used'] = attributes['ai.usage.completionTokens'];
-          }
-          if (attributes['ai.usage.promptTokens'] != undefined) {
-            attributes['ai.prompt_tokens.used'] = attributes['ai.usage.promptTokens'];
-          }
-          if (
-            typeof attributes['ai.usage.completionTokens'] == 'number' &&
-            typeof attributes['ai.usage.promptTokens'] == 'number'
-          ) {
-            attributes['ai.total_tokens.used'] =
-              attributes['ai.usage.completionTokens'] + attributes['ai.usage.promptTokens'];
-          }
-        }
-      }
-
-      return event;
+      instrumentation = instrumentVercelAi();
     },
     setup(client) {
-      client.on('spanStart', span => {
-        if (!instrumentation.sentryVercelAiPatched) {
-          return;
-        }
-
-        const { data: attributes, description: name } = core.spanToJSON(span);
-
-        if (!name) {
-          return;
-        }
-
-        // The id of the model
-        const aiModelId = attributes['ai.model.id'];
-
-        // the provider of the model
-        const aiModelProvider = attributes['ai.model.provider'];
-
-        // both of these must be defined for the integration to work
-        if (typeof aiModelId !== 'string' || typeof aiModelProvider !== 'string' || !aiModelId || !aiModelProvider) {
-          return;
-        }
-
-        let isPipelineSpan = false;
-
-        switch (name) {
-          case 'ai.generateText': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.generateText');
-            isPipelineSpan = true;
-            break;
-          }
-          case 'ai.generateText.doGenerate': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doGenerate');
-            break;
-          }
-          case 'ai.streamText': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.streamText');
-            isPipelineSpan = true;
-            break;
-          }
-          case 'ai.streamText.doStream': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doStream');
-            break;
-          }
-          case 'ai.generateObject': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.generateObject');
-            isPipelineSpan = true;
-            break;
-          }
-          case 'ai.generateObject.doGenerate': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doGenerate');
-            break;
-          }
-          case 'ai.streamObject': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.streamObject');
-            isPipelineSpan = true;
-            break;
-          }
-          case 'ai.streamObject.doStream': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doStream');
-            break;
-          }
-          case 'ai.embed': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.embed');
-            isPipelineSpan = true;
-            break;
-          }
-          case 'ai.embed.doEmbed': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.embeddings');
-            break;
-          }
-          case 'ai.embedMany': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.embedMany');
-            isPipelineSpan = true;
-            break;
-          }
-          case 'ai.embedMany.doEmbed': {
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.embeddings');
-            break;
-          }
-          case 'ai.toolCall':
-          case 'ai.stream.firstChunk':
-          case 'ai.stream.finish':
-            span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run');
-            break;
-        }
-
-        addOriginToSpan.addOriginToSpan(span, 'auto.vercelai.otel');
-
-        const nameWthoutAi = name.replace('ai.', '');
-        span.setAttribute('ai.pipeline.name', nameWthoutAi);
-        span.updateName(nameWthoutAi);
-
-        // If a Telemetry name is set and it is a pipeline span, use that as the operation name
-        const functionId = attributes['ai.telemetry.functionId'];
-        if (functionId && typeof functionId === 'string' && isPipelineSpan) {
-          span.updateName(functionId);
-          span.setAttribute('ai.pipeline.name', functionId);
-        }
-
-        if (attributes['ai.prompt']) {
-          span.setAttribute('ai.input_messages', attributes['ai.prompt']);
-        }
-        if (attributes['ai.model.id']) {
-          span.setAttribute('ai.model_id', attributes['ai.model.id']);
-        }
-        span.setAttribute('ai.streaming', name.includes('stream'));
+      instrumentation?.callWhenPatched(() => {
+        client.on('spanStart', span => {
+          const { data: attributes, description: name } = core.spanToJSON(span);
+
+          if (!name) {
+            return;
+          }
+
+          // The id of the model
+          const aiModelId = attributes['ai.model.id'];
+
+          // the provider of the model
+          const aiModelProvider = attributes['ai.model.provider'];
+
+          // both of these must be defined for the integration to work
+          if (typeof aiModelId !== 'string' || typeof aiModelProvider !== 'string' || !aiModelId || !aiModelProvider) {
+            return;
+          }
+
+          let isPipelineSpan = false;
+
+          switch (name) {
+            case 'ai.generateText': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.generateText');
+              isPipelineSpan = true;
+              break;
+            }
+            case 'ai.generateText.doGenerate': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doGenerate');
+              break;
+            }
+            case 'ai.streamText': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.streamText');
+              isPipelineSpan = true;
+              break;
+            }
+            case 'ai.streamText.doStream': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doStream');
+              break;
+            }
+            case 'ai.generateObject': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.generateObject');
+              isPipelineSpan = true;
+              break;
+            }
+            case 'ai.generateObject.doGenerate': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doGenerate');
+              break;
+            }
+            case 'ai.streamObject': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.streamObject');
+              isPipelineSpan = true;
+              break;
+            }
+            case 'ai.streamObject.doStream': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doStream');
+              break;
+            }
+            case 'ai.embed': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.embed');
+              isPipelineSpan = true;
+              break;
+            }
+            case 'ai.embed.doEmbed': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.embeddings');
+              break;
+            }
+            case 'ai.embedMany': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.embedMany');
+              isPipelineSpan = true;
+              break;
+            }
+            case 'ai.embedMany.doEmbed': {
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.embeddings');
+              break;
+            }
+            case 'ai.toolCall':
+            case 'ai.stream.firstChunk':
+            case 'ai.stream.finish':
+              span.setAttribute(core.SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run');
+              break;
+          }
+
+          addOriginToSpan.addOriginToSpan(span, 'auto.vercelai.otel');
+
+          const nameWthoutAi = name.replace('ai.', '');
+          span.setAttribute('ai.pipeline.name', nameWthoutAi);
+          span.updateName(nameWthoutAi);
+
+          // If a Telemetry name is set and it is a pipeline span, use that as the operation name
+          const functionId = attributes['ai.telemetry.functionId'];
+          if (functionId && typeof functionId === 'string' && isPipelineSpan) {
+            span.updateName(functionId);
+            span.setAttribute('ai.pipeline.name', functionId);
+          }
+
+          if (attributes['ai.prompt']) {
+            span.setAttribute('ai.input_messages', attributes['ai.prompt']);
+          }
+          if (attributes['ai.model.id']) {
+            span.setAttribute('ai.model_id', attributes['ai.model.id']);
+          }
+          span.setAttribute('ai.streaming', name.includes('stream'));
+        });
+
+        client.addEventProcessor(event => {
+          if (event.type === 'transaction' && event.spans?.length) {
+            for (const span of event.spans) {
+              const { data: attributes, description: name } = span;
+
+              if (!name || span.origin !== 'auto.vercelai.otel') {
+                continue;
+              }
+
+              if (attributes['ai.usage.completionTokens'] != undefined) {
+                attributes['ai.completion_tokens.used'] = attributes['ai.usage.completionTokens'];
+              }
+              if (attributes['ai.usage.promptTokens'] != undefined) {
+                attributes['ai.prompt_tokens.used'] = attributes['ai.usage.promptTokens'];
+              }
+              if (
+                typeof attributes['ai.usage.completionTokens'] == 'number' &&
+                typeof attributes['ai.usage.promptTokens'] == 'number'
+              ) {
+                attributes['ai.total_tokens.used'] =
+                  attributes['ai.usage.completionTokens'] + attributes['ai.usage.promptTokens'];
+              }
+            }
+          }
+
+          return event;
+        });
       });
     },
   };

package/build/cjs/integrations/tracing/vercelai/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sources":["../../../../../src/integrations/tracing/vercelai/index.ts"],"sourcesContent":["/* eslint-disable complexity */\nimport { SEMANTIC_ATTRIBUTE_SENTRY_OP, defineIntegration, spanToJSON } from '@sentry/core';\nimport type { IntegrationFn } from '@sentry/core';\nimport { generateInstrumentOnce } from '../../../otel/instrument';\nimport { addOriginToSpan } from '../../../utils/addOriginToSpan';\nimport { SentryVercelAiInstrumentation, sentryVercelAiPatched } from './instrumentation';\n\nconst INTEGRATION_NAME = 'VercelAI';\n\nexport const instrumentVercelAi = generateInstrumentOnce(INTEGRATION_NAME, () => new SentryVercelAiInstrumentation({}));\n\nconst _vercelAIIntegration = (() => {\n  return {\n    name: INTEGRATION_NAME,\n    setupOnce() {\n      instrumentVercelAi();\n    },\n    processEvent(event) {\n      if (event.type === 'transaction' && event.spans?.length) {\n        for (const span of event.spans) {\n          const { data: attributes, description: name } = span;\n\n          if (!name || span.origin !== 'auto.vercelai.otel') {\n            continue;\n          }\n\n          if (attributes['ai.usage.completionTokens'] != undefined) {\n            attributes['ai.completion_tokens.used'] = attributes['ai.usage.completionTokens'];\n          }\n          if (attributes['ai.usage.promptTokens'] != undefined) {\n            attributes['ai.prompt_tokens.used'] = attributes['ai.usage.promptTokens'];\n          }\n          if (\n            typeof attributes['ai.usage.completionTokens'] == 'number' &&\n            typeof attributes['ai.usage.promptTokens'] == 'number'\n          ) {\n            attributes['ai.total_tokens.used'] =\n              attributes['ai.usage.completionTokens'] + attributes['ai.usage.promptTokens'];\n          }\n        }\n      }\n\n      return event;\n    },\n    setup(client) {\n      client.on('spanStart', span => {\n        if (!sentryVercelAiPatched) {\n          return;\n        }\n\n        const { data: attributes, description: name } = spanToJSON(span);\n\n        if (!name) {\n          return;\n        }\n\n        // The id of the model\n        const aiModelId = attributes['ai.model.id'];\n\n        // the provider of the model\n        const aiModelProvider = attributes['ai.model.provider'];\n\n        // both of these must be defined for the integration to work\n        if (typeof aiModelId !== 'string' || typeof aiModelProvider !== 'string' || !aiModelId || !aiModelProvider) {\n          return;\n        }\n\n        let isPipelineSpan = false;\n\n        switch (name) {\n          case 'ai.generateText': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.generateText');\n            isPipelineSpan = true;\n            break;\n          }\n          case 'ai.generateText.doGenerate': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doGenerate');\n            break;\n          }\n          case 'ai.streamText': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.streamText');\n            isPipelineSpan = true;\n            break;\n          }\n          case 'ai.streamText.doStream': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doStream');\n            break;\n          }\n          case 'ai.generateObject': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.generateObject');\n            isPipelineSpan = true;\n            break;\n          }\n          case 'ai.generateObject.doGenerate': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doGenerate');\n            break;\n          }\n          case 'ai.streamObject': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.streamObject');\n            isPipelineSpan = true;\n            break;\n          }\n          case 'ai.streamObject.doStream': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doStream');\n            break;\n          }\n          case 'ai.embed': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.embed');\n            isPipelineSpan = true;\n            break;\n          }\n          case 'ai.embed.doEmbed': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.embeddings');\n            break;\n          }\n          case 'ai.embedMany': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.embedMany');\n            isPipelineSpan = true;\n            break;\n          }\n          case 'ai.embedMany.doEmbed': {\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.embeddings');\n            break;\n          }\n          case 'ai.toolCall':\n          case 'ai.stream.firstChunk':\n          case 'ai.stream.finish':\n            span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run');\n            break;\n        }\n\n        addOriginToSpan(span, 'auto.vercelai.otel');\n\n        const nameWthoutAi = name.replace('ai.', '');\n        span.setAttribute('ai.pipeline.name', nameWthoutAi);\n        span.updateName(nameWthoutAi);\n\n        // If a Telemetry name is set and it is a pipeline span, use that as the operation name\n        const functionId = attributes['ai.telemetry.functionId'];\n        if (functionId && typeof functionId === 'string' && isPipelineSpan) {\n          span.updateName(functionId);\n          span.setAttribute('ai.pipeline.name', functionId);\n        }\n\n        if (attributes['ai.prompt']) {\n          span.setAttribute('ai.input_messages', attributes['ai.prompt']);\n        }\n        if (attributes['ai.model.id']) {\n          span.setAttribute('ai.model_id', attributes['ai.model.id']);\n        }\n        span.setAttribute('ai.streaming', name.includes('stream'));\n      });\n    },\n  };\n}) satisfies IntegrationFn;\n\n/**\n * Adds Sentry tracing instrumentation for the [ai](https://www.npmjs.com/package/ai) library.\n *\n * For more information, see the [`ai` documentation](https://sdk.vercel.ai/docs/ai-sdk-core/telemetry).\n *\n * @example\n * ```javascript\n * const Sentry = require('@sentry/node');\n *\n * Sentry.init({\n *  integrations: [Sentry.vercelAIIntegration()],\n * });\n * ```\n *\n * By default this integration adds tracing support to all `ai` function calls. If you need to disable\n * collecting spans for a specific call, you can do so by setting `experimental_telemetry.isEnabled` to\n * `false` in the first argument of the function call.\n *\n * ```javascript\n * const result = await generateText({\n *   model: openai('gpt-4-turbo'),\n *   experimental_telemetry: { isEnabled: false },\n * });\n * ```\n *\n * If you want to collect inputs and outputs for a specific call, you must specifically opt-in to each\n * function call by setting `experimental_telemetry.recordInputs` and `experimental_telemetry.recordOutputs`\n * to `true`.\n *\n * ```javascript\n * const result = await generateText({\n *  model: openai('gpt-4-turbo'),\n *  experimental_telemetry: { isEnabled: true, recordInputs: true, recordOutputs: true },\n * });\n */\nexport const vercelAIIntegration = defineIntegration(_vercelAIIntegration);\n"],"names":["generateInstrumentOnce","SentryVercelAiInstrumentation","sentryVercelAiPatched","spanToJSON","SEMANTIC_ATTRIBUTE_SENTRY_OP","addOriginToSpan","defineIntegration"],"mappings":";;;;;;;AAAA;;AAOA,MAAM,gBAAA,GAAmB,UAAU;;MAEtB,kBAAmB,GAAEA,iCAAsB,CAAC,gBAAgB,EAAE,MAAM,IAAIC,6CAA6B,CAAC,EAAE,CAAC;;AAEtH,MAAM,oBAAqB,IAAG,MAAM;AACpC,EAAE,OAAO;AACT,IAAI,IAAI,EAAE,gBAAgB;AAC1B,IAAI,SAAS,GAAG;AAChB,MAAM,kBAAkB,EAAE;AAC1B,KAAK;AACL,IAAI,YAAY,CAAC,KAAK,EAAE;AACxB,MAAM,IAAI,KAAK,CAAC,IAAK,KAAI,aAAc,IAAG,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE;AAC/D,QAAQ,KAAK,MAAM,IAAA,IAAQ,KAAK,CAAC,KAAK,EAAE;AACxC,UAAU,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,IAAA,EAAO,GAAE,IAAI;;AAE9D,UAAU,IAAI,CAAC,IAAK,IAAG,IAAI,CAAC,MAAA,KAAW,oBAAoB,EAAE;AAC7D,YAAY;AACZ;;AAEA,UAAU,IAAI,UAAU,CAAC,2BAA2B,CAAE,IAAG,SAAS,EAAE;AACpE,YAAY,UAAU,CAAC,2BAA2B,CAAA,GAAI,UAAU,CAAC,2BAA2B,CAAC;AAC7F;AACA,UAAU,IAAI,UAAU,CAAC,uBAAuB,CAAE,IAAG,SAAS,EAAE;AAChE,YAAY,UAAU,CAAC,uBAAuB,CAAA,GAAI,UAAU,CAAC,uBAAuB,CAAC;AACrF;AACA,UAAU;AACV,YAAY,OAAO,UAAU,CAAC,2BAA2B,CAAA,IAAK,QAAS;AACvE,YAAY,OAAO,UAAU,CAAC,uBAAuB,KAAK;AAC1D,YAAY;AACZ,YAAY,UAAU,CAAC,sBAAsB,CAAE;AAC/C,cAAc,UAAU,CAAC,2BAA2B,CAAA,GAAI,UAAU,CAAC,uBAAuB,CAAC;AAC3F;AACA;AACA;;AAEA,MAAM,OAAO,KAAK;AAClB,KAAK;AACL,IAAI,KAAK,CAAC,MAAM,EAAE;AAClB,MAAM,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,QAAQ;AACrC,QAAQ,IAAI,CAACC,qCAAqB,EAAE;AACpC,UAAU;AACV;;AAEA,QAAQ,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,IAAA,EAAO,GAAEC,eAAU,CAAC,IAAI,CAAC;;AAExE,QAAQ,IAAI,CAAC,IAAI,EAAE;AACnB,UAAU;AACV;;AAEA;AACA,QAAQ,MAAM,SAAU,GAAE,UAAU,CAAC,aAAa,CAAC;;AAEnD;AACA,QAAQ,MAAM,eAAgB,GAAE,UAAU,CAAC,mBAAmB,CAAC;;AAE/D;AACA,QAAQ,IAAI,OAAO,SAAU,KAAI,YAAY,OAAO,eAAgB,KAAI,YAAY,CAAC,aAAa,CAAC,eAAe,EAAE;AACpH,UAAU;AACV;;AAEA,QAAQ,IAAI,cAAe,GAAE,KAAK;;AAElC,QAAQ,QAAQ,IAAI;AACpB,UAAU,KAAK,iBAAiB,EAAE;AAClC,YAAY,IAAI,CAAC,YAAY,CAACC,iCAA4B,EAAE,0BAA0B,CAAC;AACvF,YAAY,cAAA,GAAiB,IAAI;AACjC,YAAY;AACZ;AACA,UAAU,KAAK,4BAA4B,EAAE;AAC7C,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,mBAAmB,CAAC;AAChF,YAAY;AACZ;AACA,UAAU,KAAK,eAAe,EAAE;AAChC,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,wBAAwB,CAAC;AACrF,YAAY,cAAA,GAAiB,IAAI;AACjC,YAAY;AACZ;AACA,UAAU,KAAK,wBAAwB,EAAE;AACzC,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,iBAAiB,CAAC;AAC9E,YAAY;AACZ;AACA,UAAU,KAAK,mBAAmB,EAAE;AACpC,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,4BAA4B,CAAC;AACzF,YAAY,cAAA,GAAiB,IAAI;AACjC,YAAY;AACZ;AACA,UAAU,KAAK,8BAA8B,EAAE;AAC/C,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,mBAAmB,CAAC;AAChF,YAAY;AACZ;AACA,UAAU,KAAK,iBAAiB,EAAE;AAClC,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,0BAA0B,CAAC;AACvF,YAAY,cAAA,GAAiB,IAAI;AACjC,YAAY;AACZ;AACA,UAAU,KAAK,0BAA0B,EAAE;AAC3C,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,iBAAiB,CAAC;AAC9E,YAAY;AACZ;AACA,UAAU,KAAK,UAAU,EAAE;AAC3B,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,mBAAmB,CAAC;AAChF,YAAY,cAAA,GAAiB,IAAI;AACjC,YAAY;AACZ;AACA,UAAU,KAAK,kBAAkB,EAAE;AACnC,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,eAAe,CAAC;AAC5E,YAAY;AACZ;AACA,UAAU,KAAK,cAAc,EAAE;AAC/B,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,uBAAuB,CAAC;AACpF,YAAY,cAAA,GAAiB,IAAI;AACjC,YAAY;AACZ;AACA,UAAU,KAAK,sBAAsB,EAAE;AACvC,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,eAAe,CAAC;AAC5E,YAAY;AACZ;AACA,UAAU,KAAK,aAAa;AAC5B,UAAU,KAAK,sBAAsB;AACrC,UAAU,KAAK,kBAAkB;AACjC,YAAY,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,QAAQ,CAAC;AACrE,YAAY;AACZ;;AAEA,QAAQC,+BAAe,CAAC,IAAI,EAAE,oBAAoB,CAAC;;AAEnD,QAAQ,MAAM,YAAa,GAAE,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;AACpD,QAAQ,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,YAAY,CAAC;AAC3D,QAAQ,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC;;AAErC;AACA,QAAQ,MAAM,UAAW,GAAE,UAAU,CAAC,yBAAyB,CAAC;AAChE,QAAQ,IAAI,UAAA,IAAc,OAAO,eAAe,QAAA,IAAY,cAAc,EAAE;AAC5E,UAAU,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;AACrC,UAAU,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,UAAU,CAAC;AAC3D;;AAEA,QAAQ,IAAI,UAAU,CAAC,WAAW,CAAC,EAAE;AACrC,UAAU,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,UAAU,CAAC,WAAW,CAAC,CAAC;AACzE;AACA,QAAQ,IAAI,UAAU,CAAC,aAAa,CAAC,EAAE;AACvC,UAAU,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC;AACrE;AACA,QAAQ,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AAClE,OAAO,CAAC;AACR,KAAK;AACL,GAAG;AACH,CAAC,CAAE;;AAEH;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;MACa,mBAAoB,GAAEC,sBAAiB,CAAC,oBAAoB;;;;;"}
+{"version":3,"file":"index.js","sources":["../../../../../src/integrations/tracing/vercelai/index.ts"],"sourcesContent":["/* eslint-disable complexity */\nimport { SEMANTIC_ATTRIBUTE_SENTRY_OP, defineIntegration, spanToJSON } from '@sentry/core';\nimport type { IntegrationFn } from '@sentry/core';\nimport { generateInstrumentOnce } from '../../../otel/instrument';\nimport { addOriginToSpan } from '../../../utils/addOriginToSpan';\nimport { SentryVercelAiInstrumentation } from './instrumentation';\n\nconst INTEGRATION_NAME = 'VercelAI';\n\nexport const instrumentVercelAi = generateInstrumentOnce(INTEGRATION_NAME, () => new SentryVercelAiInstrumentation({}));\n\nconst _vercelAIIntegration = (() => {\n  let instrumentation: undefined | SentryVercelAiInstrumentation;\n\n  return {\n    name: INTEGRATION_NAME,\n    setupOnce() {\n      instrumentation = instrumentVercelAi();\n    },\n    setup(client) {\n      instrumentation?.callWhenPatched(() => {\n        client.on('spanStart', span => {\n          const { data: attributes, description: name } = spanToJSON(span);\n\n          if (!name) {\n            return;\n          }\n\n          // The id of the model\n          const aiModelId = attributes['ai.model.id'];\n\n          // the provider of the model\n          const aiModelProvider = attributes['ai.model.provider'];\n\n          // both of these must be defined for the integration to work\n          if (typeof aiModelId !== 'string' || typeof aiModelProvider !== 'string' || !aiModelId || !aiModelProvider) {\n            return;\n          }\n\n          let isPipelineSpan = false;\n\n          switch (name) {\n            case 'ai.generateText': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.generateText');\n              isPipelineSpan = true;\n              break;\n            }\n            case 'ai.generateText.doGenerate': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doGenerate');\n              break;\n            }\n            case 'ai.streamText': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.streamText');\n              isPipelineSpan = true;\n              break;\n            }\n            case 'ai.streamText.doStream': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doStream');\n              break;\n            }\n            case 'ai.generateObject': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.generateObject');\n              isPipelineSpan = true;\n              break;\n            }\n            case 'ai.generateObject.doGenerate': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doGenerate');\n              break;\n            }\n            case 'ai.streamObject': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.streamObject');\n              isPipelineSpan = true;\n              break;\n            }\n            case 'ai.streamObject.doStream': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run.doStream');\n              break;\n            }\n            case 'ai.embed': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.embed');\n              isPipelineSpan = true;\n              break;\n            }\n            case 'ai.embed.doEmbed': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.embeddings');\n              break;\n            }\n            case 'ai.embedMany': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.pipeline.embedMany');\n              isPipelineSpan = true;\n              break;\n            }\n            case 'ai.embedMany.doEmbed': {\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.embeddings');\n              break;\n            }\n            case 'ai.toolCall':\n            case 'ai.stream.firstChunk':\n            case 'ai.stream.finish':\n              span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'ai.run');\n              break;\n          }\n\n          addOriginToSpan(span, 'auto.vercelai.otel');\n\n          const nameWthoutAi = name.replace('ai.', '');\n          span.setAttribute('ai.pipeline.name', nameWthoutAi);\n          span.updateName(nameWthoutAi);\n\n          // If a Telemetry name is set and it is a pipeline span, use that as the operation name\n          const functionId = attributes['ai.telemetry.functionId'];\n          if (functionId && typeof functionId === 'string' && isPipelineSpan) {\n            span.updateName(functionId);\n            span.setAttribute('ai.pipeline.name', functionId);\n          }\n\n          if (attributes['ai.prompt']) {\n            span.setAttribute('ai.input_messages', attributes['ai.prompt']);\n          }\n          if (attributes['ai.model.id']) {\n            span.setAttribute('ai.model_id', attributes['ai.model.id']);\n          }\n          span.setAttribute('ai.streaming', name.includes('stream'));\n        });\n\n        client.addEventProcessor(event => {\n          if (event.type === 'transaction' && event.spans?.length) {\n            for (const span of event.spans) {\n              const { data: attributes, description: name } = span;\n\n              if (!name || span.origin !== 'auto.vercelai.otel') {\n                continue;\n              }\n\n              if (attributes['ai.usage.completionTokens'] != undefined) {\n                attributes['ai.completion_tokens.used'] = attributes['ai.usage.completionTokens'];\n              }\n              if (attributes['ai.usage.promptTokens'] != undefined) {\n                attributes['ai.prompt_tokens.used'] = attributes['ai.usage.promptTokens'];\n              }\n              if (\n                typeof attributes['ai.usage.completionTokens'] == 'number' &&\n                typeof attributes['ai.usage.promptTokens'] == 'number'\n              ) {\n                attributes['ai.total_tokens.used'] =\n                  attributes['ai.usage.completionTokens'] + attributes['ai.usage.promptTokens'];\n              }\n            }\n          }\n\n          return event;\n        });\n      });\n    },\n  };\n}) satisfies IntegrationFn;\n\n/**\n * Adds Sentry tracing instrumentation for the [ai](https://www.npmjs.com/package/ai) library.\n *\n * For more information, see the [`ai` documentation](https://sdk.vercel.ai/docs/ai-sdk-core/telemetry).\n *\n * @example\n * ```javascript\n * const Sentry = require('@sentry/node');\n *\n * Sentry.init({\n *  integrations: [Sentry.vercelAIIntegration()],\n * });\n * ```\n *\n * By default this integration adds tracing support to all `ai` function calls. If you need to disable\n * collecting spans for a specific call, you can do so by setting `experimental_telemetry.isEnabled` to\n * `false` in the first argument of the function call.\n *\n * ```javascript\n * const result = await generateText({\n *   model: openai('gpt-4-turbo'),\n *   experimental_telemetry: { isEnabled: false },\n * });\n * ```\n *\n * If you want to collect inputs and outputs for a specific call, you must specifically opt-in to each\n * function call by setting `experimental_telemetry.recordInputs` and `experimental_telemetry.recordOutputs`\n * to `true`.\n *\n * ```javascript\n * const result = await generateText({\n *  model: openai('gpt-4-turbo'),\n *  experimental_telemetry: { isEnabled: true, recordInputs: true, recordOutputs: true },\n * });\n */\nexport const vercelAIIntegration = defineIntegration(_vercelAIIntegration);\n"],"names":["generateInstrumentOnce","SentryVercelAiInstrumentation","spanToJSON","SEMANTIC_ATTRIBUTE_SENTRY_OP","addOriginToSpan","defineIntegration"],"mappings":";;;;;;;AAAA;;AAOA,MAAM,gBAAA,GAAmB,UAAU;;MAEtB,kBAAmB,GAAEA,iCAAsB,CAAC,gBAAgB,EAAE,MAAM,IAAIC,6CAA6B,CAAC,EAAE,CAAC;;AAEtH,MAAM,oBAAqB,IAAG,MAAM;AACpC,EAAE,IAAI,eAAe;;AAErB,EAAE,OAAO;AACT,IAAI,IAAI,EAAE,gBAAgB;AAC1B,IAAI,SAAS,GAAG;AAChB,MAAM,eAAgB,GAAE,kBAAkB,EAAE;AAC5C,KAAK;AACL,IAAI,KAAK,CAAC,MAAM,EAAE;AAClB,MAAM,eAAe,EAAE,eAAe,CAAC,MAAM;AAC7C,QAAQ,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,QAAQ;AACvC,UAAU,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,IAAA,EAAO,GAAEC,eAAU,CAAC,IAAI,CAAC;;AAE1E,UAAU,IAAI,CAAC,IAAI,EAAE;AACrB,YAAY;AACZ;;AAEA;AACA,UAAU,MAAM,SAAU,GAAE,UAAU,CAAC,aAAa,CAAC;;AAErD;AACA,UAAU,MAAM,eAAgB,GAAE,UAAU,CAAC,mBAAmB,CAAC;;AAEjE;AACA,UAAU,IAAI,OAAO,SAAU,KAAI,YAAY,OAAO,eAAgB,KAAI,YAAY,CAAC,aAAa,CAAC,eAAe,EAAE;AACtH,YAAY;AACZ;;AAEA,UAAU,IAAI,cAAe,GAAE,KAAK;;AAEpC,UAAU,QAAQ,IAAI;AACtB,YAAY,KAAK,iBAAiB,EAAE;AACpC,cAAc,IAAI,CAAC,YAAY,CAACC,iCAA4B,EAAE,0BAA0B,CAAC;AACzF,cAAc,cAAA,GAAiB,IAAI;AACnC,cAAc;AACd;AACA,YAAY,KAAK,4BAA4B,EAAE;AAC/C,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,mBAAmB,CAAC;AAClF,cAAc;AACd;AACA,YAAY,KAAK,eAAe,EAAE;AAClC,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,wBAAwB,CAAC;AACvF,cAAc,cAAA,GAAiB,IAAI;AACnC,cAAc;AACd;AACA,YAAY,KAAK,wBAAwB,EAAE;AAC3C,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,iBAAiB,CAAC;AAChF,cAAc;AACd;AACA,YAAY,KAAK,mBAAmB,EAAE;AACtC,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,4BAA4B,CAAC;AAC3F,cAAc,cAAA,GAAiB,IAAI;AACnC,cAAc;AACd;AACA,YAAY,KAAK,8BAA8B,EAAE;AACjD,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,mBAAmB,CAAC;AAClF,cAAc;AACd;AACA,YAAY,KAAK,iBAAiB,EAAE;AACpC,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,0BAA0B,CAAC;AACzF,cAAc,cAAA,GAAiB,IAAI;AACnC,cAAc;AACd;AACA,YAAY,KAAK,0BAA0B,EAAE;AAC7C,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,iBAAiB,CAAC;AAChF,cAAc;AACd;AACA,YAAY,KAAK,UAAU,EAAE;AAC7B,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,mBAAmB,CAAC;AAClF,cAAc,cAAA,GAAiB,IAAI;AACnC,cAAc;AACd;AACA,YAAY,KAAK,kBAAkB,EAAE;AACrC,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,eAAe,CAAC;AAC9E,cAAc;AACd;AACA,YAAY,KAAK,cAAc,EAAE;AACjC,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,uBAAuB,CAAC;AACtF,cAAc,cAAA,GAAiB,IAAI;AACnC,cAAc;AACd;AACA,YAAY,KAAK,sBAAsB,EAAE;AACzC,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,eAAe,CAAC;AAC9E,cAAc;AACd;AACA,YAAY,KAAK,aAAa;AAC9B,YAAY,KAAK,sBAAsB;AACvC,YAAY,KAAK,kBAAkB;AACnC,cAAc,IAAI,CAAC,YAAY,CAACA,iCAA4B,EAAE,QAAQ,CAAC;AACvE,cAAc;AACd;;AAEA,UAAUC,+BAAe,CAAC,IAAI,EAAE,oBAAoB,CAAC;;AAErD,UAAU,MAAM,YAAa,GAAE,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;AACtD,UAAU,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,YAAY,CAAC;AAC7D,UAAU,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC;;AAEvC;AACA,UAAU,MAAM,UAAW,GAAE,UAAU,CAAC,yBAAyB,CAAC;AAClE,UAAU,IAAI,UAAA,IAAc,OAAO,eAAe,QAAA,IAAY,cAAc,EAAE;AAC9E,YAAY,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;AACvC,YAAY,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,UAAU,CAAC;AAC7D;;AAEA,UAAU,IAAI,UAAU,CAAC,WAAW,CAAC,EAAE;AACvC,YAAY,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,UAAU,CAAC,WAAW,CAAC,CAAC;AAC3E;AACA,UAAU,IAAI,UAAU,CAAC,aAAa,CAAC,EAAE;AACzC,YAAY,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC;AACvE;AACA,UAAU,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AACpE,SAAS,CAAC;;AAEV,QAAQ,MAAM,CAAC,iBAAiB,CAAC,SAAS;AAC1C,UAAU,IAAI,KAAK,CAAC,IAAK,KAAI,aAAc,IAAG,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE;AACnE,YAAY,KAAK,MAAM,IAAA,IAAQ,KAAK,CAAC,KAAK,EAAE;AAC5C,cAAc,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,IAAA,EAAO,GAAE,IAAI;;AAElE,cAAc,IAAI,CAAC,IAAK,IAAG,IAAI,CAAC,MAAA,KAAW,oBAAoB,EAAE;AACjE,gBAAgB;AAChB;;AAEA,cAAc,IAAI,UAAU,CAAC,2BAA2B,CAAE,IAAG,SAAS,EAAE;AACxE,gBAAgB,UAAU,CAAC,2BAA2B,CAAA,GAAI,UAAU,CAAC,2BAA2B,CAAC;AACjG;AACA,cAAc,IAAI,UAAU,CAAC,uBAAuB,CAAE,IAAG,SAAS,EAAE;AACpE,gBAAgB,UAAU,CAAC,uBAAuB,CAAA,GAAI,UAAU,CAAC,uBAAuB,CAAC;AACzF;AACA,cAAc;AACd,gBAAgB,OAAO,UAAU,CAAC,2BAA2B,CAAA,IAAK,QAAS;AAC3E,gBAAgB,OAAO,UAAU,CAAC,uBAAuB,KAAK;AAC9D,gBAAgB;AAChB,gBAAgB,UAAU,CAAC,sBAAsB,CAAE;AACnD,kBAAkB,UAAU,CAAC,2BAA2B,CAAA,GAAI,UAAU,CAAC,uBAAuB,CAAC;AAC/F;AACA;AACA;;AAEA,UAAU,OAAO,KAAK;AACtB,SAAS,CAAC;AACV,OAAO,CAAC;AACR,KAAK;AACL,GAAG;AACH,CAAC,CAAE;;AAEH;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;MACa,mBAAoB,GAAEC,sBAAiB,CAAC,oBAAoB;;;;;"}

package/build/cjs/integrations/tracing/vercelai/instrumentation.js

@@ -14,8 +14,6 @@ const INSTRUMENTED_METHODS = [
   'embedMany',
 ] ;
 
-exports.sentryVercelAiPatched = false;
-
 /**
  * This detects is added by the Sentry Vercel AI Integration to detect if the integration should
  * be enabled.
@@ -23,9 +21,11 @@ exports.sentryVercelAiPatched = false;
  * It also patches the `ai` module to enable Vercel AI telemetry automatically for all methods.
  */
 class SentryVercelAiInstrumentation extends instrumentation.InstrumentationBase {
+   __init() {this._isPatched = false;}
+   __init2() {this._callbacks = [];}
+
    constructor(config = {}) {
-    super('@sentry/instrumentation-vercel-ai', core.SDK_VERSION, config);
-  }
+    super('@sentry/instrumentation-vercel-ai', core.SDK_VERSION, config);SentryVercelAiInstrumentation.prototype.__init.call(this);SentryVercelAiInstrumentation.prototype.__init2.call(this);  }
 
   /**
    * Initializes the instrumentation by defining the modules to be patched.
@@ -35,11 +35,26 @@ class SentryVercelAiInstrumentation extends instrumentation.InstrumentationBase
     return module;
   }
 
+  /**
+   * Call the provided callback when the module is patched.
+   * If it has already been patched, the callback will be called immediately.
+   */
+   callWhenPatched(callback) {
+    if (this._isPatched) {
+      callback();
+    } else {
+      this._callbacks.push(callback);
+    }
+  }
+
   /**
    * Patches module exports to enable Vercel AI telemetry.
    */
    _patch(moduleExports) {
-    exports.sentryVercelAiPatched = true;
+    this._isPatched = true;
+
+    this._callbacks.forEach(callback => callback());
+    this._callbacks = [];
 
     function generatePatch(name) {
       return (...args) => {