Skip to content

Commit e5784a4

Browse files
authored
fix(sdk): manual instrumentation in all modules (#171)
1 parent eaaa3ef commit e5784a4

File tree

1 file changed

+6
-0
lines changed
  • packages/traceloop-sdk/src/lib/tracing

1 file changed

+6
-0
lines changed

packages/traceloop-sdk/src/lib/tracing/index.ts

+6
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,7 @@ export const manuallyInitInstrumentations = (
160160
} = require("@traceloop/instrumentation-pinecone");
161161
const instrumentation = new PineconeInstrumentation();
162162
instrumentations.push(instrumentation as Instrumentation);
163+
instrumentation.manuallyInstrument(instrumentModules.pinecone);
163164
}
164165

165166
if (instrumentModules?.google_vertexai) {
@@ -169,6 +170,7 @@ export const manuallyInitInstrumentations = (
169170
const instrumentation = new VertexAIInstrumentation();
170171
instrumentations.push(instrumentation);
171172
vertexaiInstrumentation = instrumentation;
173+
instrumentation.manuallyInstrument(instrumentModules.google_vertexai);
172174
}
173175

174176
if (instrumentModules?.google_aiplatform) {
@@ -178,6 +180,7 @@ export const manuallyInitInstrumentations = (
178180
const instrumentation = new AIPlatformInstrumentation();
179181
instrumentations.push(instrumentation as Instrumentation);
180182
aiplatformInstrumentation = instrumentation;
183+
instrumentation.manuallyInstrument(instrumentModules.google_aiplatform);
181184
}
182185

183186
if (instrumentModules?.bedrock) {
@@ -187,6 +190,7 @@ export const manuallyInitInstrumentations = (
187190
const instrumentation = new BedrockInstrumentation();
188191
instrumentations.push(instrumentation as Instrumentation);
189192
bedrockInstrumentation = instrumentation;
193+
instrumentation.manuallyInstrument(instrumentModules.bedrock);
190194
}
191195

192196
if (instrumentModules?.azureOpenAI) {
@@ -196,6 +200,7 @@ export const manuallyInitInstrumentations = (
196200
const instrumentation = new AzureOpenAIInstrumentation();
197201
instrumentations.push(instrumentation as Instrumentation);
198202
azureOpenAIInstrumentation = instrumentation;
203+
instrumentation.manuallyInstrument(instrumentModules.azureOpenAI);
199204
}
200205

201206
if (instrumentModules?.cohere) {
@@ -205,6 +210,7 @@ export const manuallyInitInstrumentations = (
205210
const instrumentation = new CohereInstrumentation();
206211
instrumentations.push(instrumentation as Instrumentation);
207212
cohereInstrumentation = instrumentation;
213+
instrumentation.manuallyInstrument(instrumentModules.cohere);
208214
}
209215
};
210216

0 commit comments

Comments
 (0)