lfg

Author: nicohrubec

packages/core/src/index.ts

@@ -161,7 +161,7 @@ export { ANTHROPIC_AI_INTEGRATION_NAME } from './tracing/anthropic-ai/constants'
 export { instrumentGoogleGenAIClient } from './tracing/google-genai';
 export { GOOGLE_GENAI_INTEGRATION_NAME } from './tracing/google-genai/constants';
 export type { GoogleGenAIResponse } from './tracing/google-genai/types';
-export { createLangChainCallbackHandler, instrumentEmbeddingMethod, instrumentLangChainEmbeddings } from './tracing/langchain';
+export { createLangChainCallbackHandler, instrumentLangChainEmbeddings } from './tracing/langchain';
 export { LANGCHAIN_INTEGRATION_NAME } from './tracing/langchain/constants';
 export type { LangChainOptions, LangChainIntegration } from './tracing/langchain/types';
 export { instrumentStateGraphCompile, instrumentLangGraph } from './tracing/langgraph';

packages/core/src/tracing/langchain/index.ts

@@ -335,4 +335,4 @@ export function createLangChainCallbackHandler(options: LangChainOptions = {}):
   return handler;
 }
 
-export { instrumentEmbeddingMethod, instrumentLangChainEmbeddings } from './embeddings';
+export { instrumentLangChainEmbeddings } from './embeddings';

packages/node/src/integrations/tracing/langchain/instrumentation.ts

@@ -11,9 +11,9 @@ import {
   ANTHROPIC_AI_INTEGRATION_NAME,
   createLangChainCallbackHandler,
   GOOGLE_GENAI_INTEGRATION_NAME,
+  instrumentLangChainEmbeddings,
   OPENAI_INTEGRATION_NAME,
   SDK_VERSION,
-  instrumentEmbeddingMethod,
 } from '@sentry/core';
 
 const supportedVersions = ['>=0.1.0 <2.0.0'];
@@ -277,15 +277,6 @@ export class SentryLangChainInstrumentation extends InstrumentationBase<LangChai
     }
     targetProto.__sentry_patched_embeddings__ = true;
 
-    // Duck-type detection already verified these are functions
-    targetProto.embedQuery = instrumentEmbeddingMethod(
-      targetProto.embedQuery as (...args: unknown[]) => Promise<unknown>,
-      options,
-    );
-
-    targetProto.embedDocuments = instrumentEmbeddingMethod(
-      targetProto.embedDocuments as (...args: unknown[]) => Promise<unknown>,
-      options,
-    );
+    instrumentLangChainEmbeddings(targetProto, options);
   }
 }