{"version":3,"file":"instrumentation.js","sources":["../../../../../src/integrations/tracing/google-genai/instrumentation.ts"],"sourcesContent":["import type { InstrumentationConfig, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';\nimport { InstrumentationBase, InstrumentationNodeModuleDefinition } from '@opentelemetry/instrumentation';\nimport { InstrumentationNodeModuleFile } from '../InstrumentationNodeModuleFile';\nimport type { GoogleGenAIClient, GoogleGenAIOptions } from '@sentry/core';\nimport {\n  _INTERNAL_shouldSkipAiProviderWrapping,\n  GOOGLE_GENAI_INTEGRATION_NAME,\n  instrumentGoogleGenAIClient,\n  replaceExports,\n  SDK_VERSION,\n} from '@sentry/core';\n\nconst supportedVersions = ['>=0.10.0 <2'];\n\n/**\n * Represents the patched shape of the Google GenAI module export.\n */\ninterface PatchedModuleExports {\n  [key: string]: unknown;\n  GoogleGenAI?: unknown;\n}\n\ntype GoogleGenAIInstrumentationOptions = GoogleGenAIOptions & InstrumentationConfig;\n\n/**\n * Sentry Google GenAI instrumentation using OpenTelemetry.\n */\nexport class SentryGoogleGenAiInstrumentation extends InstrumentationBase<GoogleGenAIInstrumentationOptions> {\n  public constructor(config: GoogleGenAIInstrumentationOptions = {}) {\n    super('@sentry/instrumentation-google-genai', SDK_VERSION, config);\n  }\n\n  /**\n   * Initializes the instrumentation by defining the modules to be patched.\n   */\n  public init(): InstrumentationModuleDefinition {\n    const module = new InstrumentationNodeModuleDefinition(\n      '@google/genai',\n      supportedVersions,\n      exports => this._patch(exports),\n      exports => exports,\n      // In CJS, @google/genai re-exports from (dist/node/index.cjs) file.\n      // Patching only the root module sometimes misses the real implementation or\n      // gets overwritten when that file is loaded. We add a file-level patch so that\n      // _patch runs again on the concrete implementation\n      [\n        new InstrumentationNodeModuleFile(\n          '@google/genai/dist/node/index.cjs',\n          supportedVersions,\n          exports => this._patch(exports),\n          exports => exports,\n        ),\n      ],\n    );\n    return module;\n  }\n\n  /**\n   * Core patch logic applying instrumentation to the Google GenAI client constructor.\n   */\n  private _patch(exports: PatchedModuleExports): PatchedModuleExports | void {\n    const Original = exports.GoogleGenAI;\n    const config = this.getConfig();\n\n    if (typeof Original !== 'function') {\n      return exports;\n    }\n\n    const WrappedGoogleGenAI = function (this: unknown, ...args: unknown[]): GoogleGenAIClient {\n      // Check if wrapping should be skipped (e.g., when LangChain is handling instrumentation)\n      if (_INTERNAL_shouldSkipAiProviderWrapping(GOOGLE_GENAI_INTEGRATION_NAME)) {\n        return Reflect.construct(Original, args) as GoogleGenAIClient;\n      }\n\n      const instance = Reflect.construct(Original, args);\n\n      return instrumentGoogleGenAIClient(instance, config);\n    };\n\n    // Preserve static and prototype chains\n    Object.setPrototypeOf(WrappedGoogleGenAI, Original);\n    Object.setPrototypeOf(WrappedGoogleGenAI.prototype, Original.prototype);\n\n    for (const key of Object.getOwnPropertyNames(Original)) {\n      if (!['length', 'name', 'prototype'].includes(key)) {\n        const descriptor = Object.getOwnPropertyDescriptor(Original, key);\n        if (descriptor) {\n          Object.defineProperty(WrappedGoogleGenAI, key, descriptor);\n        }\n      }\n    }\n\n    // Replace google genai exports with the wrapped constructor\n    replaceExports(exports, 'GoogleGenAI', WrappedGoogleGenAI);\n\n    return exports;\n  }\n}\n"],"names":["InstrumentationBase","SDK_VERSION","InstrumentationNodeModuleDefinition","InstrumentationNodeModuleFile","_INTERNAL_shouldSkipAiProviderWrapping","GOOGLE_GENAI_INTEGRATION_NAME","instrumentGoogleGenAIClient","replaceExports"],"mappings":";;;;;;AAYA,MAAM,iBAAA,GAAoB,CAAC,aAAa,CAAA;AAejC,MAAM,yCAAyCA,mCAAA,CAAuD;AAAA,EACpG,WAAA,CAAY,MAAA,GAA4C,EAAC,EAAG;AACjE,IAAA,KAAA,CAAM,sCAAA,EAAwCC,kBAAa,MAAM,CAAA;AAAA,EACnE;AAAA;AAAA;AAAA;AAAA,EAKO,IAAA,GAAwC;AAC7C,IAAA,MAAM,SAAS,IAAIC,mDAAA;AAAA,MACjB,eAAA;AAAA,MACA,iBAAA;AAAA,MACA,CAAA,OAAA,KAAW,IAAA,CAAK,MAAA,CAAO,OAAO,CAAA;AAAA,MAC9B,CAAA,OAAA,KAAW,OAAA;AAAA;AAAA;AAAA;AAAA;AAAA,MAKX;AAAA,QACE,IAAIC,2DAAA;AAAA,UACF,mCAAA;AAAA,UACA,iBAAA;AAAA,UACA,CAAA,OAAA,KAAW,IAAA,CAAK,MAAA,CAAO,OAAO,CAAA;AAAA,UAC9B,CAAA,OAAA,KAAW;AAAA;AACb;AACF,KACF;AACA,IAAA,OAAO,MAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,OAAO,OAAA,EAA4D;AACzE,IAAA,MAAM,WAAW,OAAA,CAAQ,WAAA;AACzB,IAAA,MAAM,MAAA,GAAS,KAAK,SAAA,EAAU;AAE9B,IAAA,IAAI,OAAO,aAAa,UAAA,EAAY;AAClC,MAAA,OAAO,OAAA;AAAA,IACT;AAEA,IAAA,MAAM,kBAAA,GAAqB,YAA4B,IAAA,EAAoC;AAEzF,MAAA,IAAIC,2CAAA,CAAuCC,kCAA6B,CAAA,EAAG;AACzE,QAAA,OAAO,OAAA,CAAQ,SAAA,CAAU,QAAA,EAAU,IAAI,CAAA;AAAA,MACzC;AAEA,MAAA,MAAM,QAAA,GAAW,OAAA,CAAQ,SAAA,CAAU,QAAA,EAAU,IAAI,CAAA;AAEjD,MAAA,OAAOC,gCAAA,CAA4B,UAAU,MAAM,CAAA;AAAA,IACrD,CAAA;AAGA,IAAA,MAAA,CAAO,cAAA,CAAe,oBAAoB,QAAQ,CAAA;AAClD,IAAA,MAAA,CAAO,cAAA,CAAe,kBAAA,CAAmB,SAAA,EAAW,QAAA,CAAS,SAAS,CAAA;AAEtE,IAAA,KAAA,MAAW,GAAA,IAAO,MAAA,CAAO,mBAAA,CAAoB,QAAQ,CAAA,EAAG;AACtD,MAAA,IAAI,CAAC,CAAC,QAAA,EAAU,MAAA,EAAQ,WAAW,CAAA,CAAE,QAAA,CAAS,GAAG,CAAA,EAAG;AAClD,QAAA,MAAM,UAAA,GAAa,MAAA,CAAO,wBAAA,CAAyB,QAAA,EAAU,GAAG,CAAA;AAChE,QAAA,IAAI,UAAA,EAAY;AACd,UAAA,MAAA,CAAO,cAAA,CAAe,kBAAA,EAAoB,GAAA,EAAK,UAAU,CAAA;AAAA,QAC3D;AAAA,MACF;AAAA,IACF;AAGA,IAAAC,mBAAA,CAAe,OAAA,EAAS,eAAe,kBAAkB,CAAA;AAEzD,IAAA,OAAO,OAAA;AAAA,EACT;AACF;;;;"}