fix(langchain): Remove token checks and use correct API for embedding tests

src/runner/templates/embeddings/browser/langchain/config.json

@@ -11,5 +11,10 @@
     { "package": "@langchain/core", "version": "0.3.49" }
   ],
   "versions": ["0.3.24"],
-  "sentryVersions": ["latest"]
+  "sentryVersions": ["latest"],
+  "skip": {
+    "checks": {
+      "Basic Embeddings Test": ["checkEmbeddingTokenUsage"]
+    }
+  }
 }

src/runner/templates/embeddings/browser/langchain/template.njk

@@ -9,10 +9,10 @@
       throw new Error('OPENAI_API_KEY not provided');
     }
 
-    const embeddings = new OpenAIEmbeddings({
+    const embeddings = Sentry.instrumentLangChainEmbeddings(new OpenAIEmbeddings({
       modelName: "{{ inputs[0].model }}",
       openAIApiKey: OPENAI_API_KEY,
-    });
+    }));
 
     log('LangChain OpenAIEmbeddings initialized');
 {% endblock %}

src/runner/templates/embeddings/cloudflare/langchain/config.json

@@ -10,5 +10,10 @@
     { "package": "@langchain/core", "version": "0.3.49" }
   ],
   "versions": ["0.3.24"],
-  "sentryVersions": ["latest"]
+  "sentryVersions": ["latest"],
+  "skip": {
+    "checks": {
+      "Basic Embeddings Test": ["checkEmbeddingTokenUsage"]
+    }
+  }
 }

src/runner/templates/embeddings/cloudflare/langchain/template.njk

@@ -3,12 +3,10 @@
 {% block dynamic_imports %}
       const langchainOpenAI = await import("@langchain/openai");
       const OpenAIEmbeddings = langchainOpenAI.OpenAIEmbeddings;
-      const callbackHandler = Sentry.createLangChainCallbackHandler();
-      const embeddings = new OpenAIEmbeddings({
+      const embeddings = Sentry.instrumentLangChainEmbeddings(new OpenAIEmbeddings({
         modelName: "{{ inputs[0].model }}",
         openAIApiKey: env.OPENAI_API_KEY,
-        callbacks: [callbackHandler],
-      });
+      }));
 {% endblock %}
 
 {% block test %}

src/runner/templates/embeddings/nextjs/langchain/config.json

@@ -11,5 +11,10 @@
     { "package": "next", "version": "latest" }
   ],
   "versions": ["latest"],
-  "sentryVersions": ["latest"]
+  "sentryVersions": ["latest"],
+  "skip": {
+    "checks": {
+      "Basic Embeddings Test": ["checkEmbeddingTokenUsage"]
+    }
+  }
 }

src/runner/templates/embeddings/nextjs/langchain/template.njk

@@ -9,10 +9,10 @@ if (!OPENAI_API_KEY) {
   throw new Error('OPENAI_API_KEY not provided');
 }
 
-const embeddings = new OpenAIEmbeddings({
+const embeddings = Sentry.instrumentLangChainEmbeddings(new OpenAIEmbeddings({
   modelName: "{{ inputs[0].model }}",
   openAIApiKey: OPENAI_API_KEY,
-});
+}));
 console.log('LangChain OpenAIEmbeddings initialized');
 {% endblock %}
 

src/runner/templates/embeddings/node/langchain/config.json

@@ -10,5 +10,10 @@
     { "package": "@langchain/core", "version": "0.3.49" }
   ],
   "versions": ["0.3.24"],
-  "sentryVersions": ["latest"]
+  "sentryVersions": ["latest"],
+  "skip": {
+    "checks": {
+      "Basic Embeddings Test": ["checkEmbeddingTokenUsage"]
+    }
+  }
 }