@sentry/node
Version:
Sentry Node SDK using OpenTelemetry for performance instrumentation
108 lines (94 loc) • 3.57 kB
JavaScript
import { InstrumentationBase, InstrumentationNodeModuleDefinition } from '@opentelemetry/instrumentation';
import { SDK_VERSION, getCurrentScope, ANTHROPIC_AI_INTEGRATION_NAME, instrumentAnthropicAiClient } from '@sentry/core';
const supportedVersions = ['>=0.19.2 <1.0.0'];
/**
* Determines telemetry recording settings.
*/
function determineRecordingSettings(
integrationOptions,
defaultEnabled,
) {
const recordInputs = integrationOptions?.recordInputs ?? defaultEnabled;
const recordOutputs = integrationOptions?.recordOutputs ?? defaultEnabled;
return { recordInputs, recordOutputs };
}
/**
* Sentry Anthropic AI instrumentation using OpenTelemetry.
*/
class SentryAnthropicAiInstrumentation extends InstrumentationBase {
constructor(config = {}) {
super('@sentry/instrumentation-anthropic-ai', SDK_VERSION, config);
}
/**
* Initializes the instrumentation by defining the modules to be patched.
*/
init() {
const module = new InstrumentationNodeModuleDefinition(
'@anthropic-ai/sdk',
supportedVersions,
this._patch.bind(this),
);
return module;
}
/**
* Core patch logic applying instrumentation to the Anthropic AI client constructor.
*/
_patch(exports) {
const Original = exports.Anthropic;
const WrappedAnthropic = function ( ...args) {
const instance = Reflect.construct(Original, args);
const scopeClient = getCurrentScope().getClient();
const integration = scopeClient?.getIntegrationByName(ANTHROPIC_AI_INTEGRATION_NAME);
const integrationOpts = integration?.options;
const defaultPii = Boolean(scopeClient?.getOptions().sendDefaultPii);
const { recordInputs, recordOutputs } = determineRecordingSettings(integrationOpts, defaultPii);
return instrumentAnthropicAiClient(instance , {
recordInputs,
recordOutputs,
});
} ;
// Preserve static and prototype chains
Object.setPrototypeOf(WrappedAnthropic, Original);
Object.setPrototypeOf(WrappedAnthropic.prototype, Original.prototype);
for (const key of Object.getOwnPropertyNames(Original)) {
if (!['length', 'name', 'prototype'].includes(key)) {
const descriptor = Object.getOwnPropertyDescriptor(Original, key);
if (descriptor) {
Object.defineProperty(WrappedAnthropic, key, descriptor);
}
}
}
// Constructor replacement - handle read-only properties
// The Anthropic property might have only a getter, so use defineProperty
try {
exports.Anthropic = WrappedAnthropic;
} catch (error) {
// If direct assignment fails, override the property descriptor
Object.defineProperty(exports, 'Anthropic', {
value: WrappedAnthropic,
writable: true,
configurable: true,
enumerable: true,
});
}
// Wrap the default export if it points to the original constructor
// Constructor replacement - handle read-only properties
// The Anthropic property might have only a getter, so use defineProperty
if (exports.default === Original) {
try {
exports.default = WrappedAnthropic;
} catch (error) {
// If direct assignment fails, override the property descriptor
Object.defineProperty(exports, 'default', {
value: WrappedAnthropic,
writable: true,
configurable: true,
enumerable: true,
});
}
}
return exports;
}
}
export { SentryAnthropicAiInstrumentation };
//# sourceMappingURL=instrumentation.js.map