feat(langchain): cover embeddings for all OTel provider packages

The OTel path covered 6 chat provider packages; chat is already handled by
the single @langchain/core BaseChatModel hook. For embeddings (the
per-package surface), wire the packages that ship an Embeddings subclass:
openai, google-genai, mistralai, and google-vertexai (whose methods live on
the shared @langchain/google-common base). anthropic and groq are chat-only.

The embeddings channel list is derived from the provider list so adding a
provider is a single edit.

Author: logaretm

packages/server-utils/src/integrations/tracing-channel/langchain.ts

@@ -16,6 +16,7 @@ import {
 } from '@sentry/core';
 import { DEBUG_BUILD } from '../../debug-build';
 import { CHANNELS } from '../../orchestrion/channels';
+import { langchainEmbeddingsChannels } from '../../orchestrion/config/langchain';
 import { bindTracingChannelToSpan } from '../../tracing-channel';
 
 // Same name as the OTel integration by design: when enabled, the OTel 'LangChain' integration is
@@ -89,7 +90,7 @@ const _langChainChannelIntegration = ((options: LangChainOptions = {}) => {
       // do the same here. `bindTracingChannelToSpan` needs the async-context binding that
       // `initOpenTelemetry()` registers after `setupOnce`, so wait for it before subscribing.
       waitForTracingChannelBinding(() => {
-        for (const channelName of [CHANNELS.LANGCHAIN_EMBED_QUERY, CHANNELS.LANGCHAIN_EMBED_DOCUMENTS]) {
+        for (const channelName of langchainEmbeddingsChannels) {
           DEBUG_BUILD && debug.log(`[orchestrion:langchain] subscribing to channel "${channelName}"`);
           bindTracingChannelToSpan(
             diagnosticsChannel.tracingChannel<EmbeddingsChannelContext>(channelName),

packages/server-utils/src/orchestrion/config/langchain.ts

@@ -29,23 +29,41 @@ const chatModelConfig = ['dist/language_models/chat_models.cjs', 'dist/language_
 );
 
 // Embeddings have no shared concrete method on the base class (each provider implements
-// `embedQuery`/`embedDocuments`), so they're hooked per provider. Only `@langchain/openai` is wired
-// for now; other providers follow the same shape (a class extending `Embeddings` with async
-// `embedQuery`/`embedDocuments`).
-const embeddingsConfig = ['dist/embeddings.cjs', 'dist/embeddings.js'].flatMap(filePath => {
-  const module = { name: '@langchain/openai', versionRange: '>=0.1.0 <2.0.0', filePath };
-
-  return [
-    { channelName: 'embedQuery', module, functionQuery: { methodName: 'embedQuery', kind: 'Async' as const } },
-    { channelName: 'embedDocuments', module, functionQuery: { methodName: 'embedDocuments', kind: 'Async' as const } },
-  ];
-});
+// `embedQuery`/`embedDocuments`), so they're hooked per package. These are the provider packages the
+// vendored OTel instrumentation covered that actually ship an `Embeddings` subclass: `@langchain/openai`,
+// `@langchain/google-genai` and `@langchain/mistralai` define the two methods on their own class, while
+// `@langchain/google-vertexai` inherits them from the shared `@langchain/google-common` base, so that base
+// is the module hooked for it. (anthropic and groq are chat-only — they ship no embeddings class.) The
+// `embedQuery`/`embedDocuments` channel names are per-method; orchestrion prefixes them with the module
+// name, so the full channel strings stay distinct across packages.
+const EMBEDDINGS_PROVIDERS = [
+  { name: '@langchain/openai', versionRange: '>=0.1.0 <2.0.0' },
+  { name: '@langchain/google-genai', versionRange: '>=0.1.0 <3.0.0' },
+  { name: '@langchain/mistralai', versionRange: '>=0.1.0 <2.0.0' },
+  { name: '@langchain/google-common', versionRange: '>=0.1.0 <3.0.0' },
+];
+
+const EMBEDDINGS_METHODS = ['embedQuery', 'embedDocuments'] as const;
+
+const embeddingsConfig = EMBEDDINGS_PROVIDERS.flatMap(({ name, versionRange }) =>
+  ['dist/embeddings.cjs', 'dist/embeddings.js'].flatMap(filePath =>
+    EMBEDDINGS_METHODS.map(method => ({
+      channelName: method,
+      module: { name, versionRange, filePath },
+      functionQuery: { methodName: method, kind: 'Async' as const },
+    })),
+  ),
+);
 
 export const langchainConfig = [...chatModelConfig, ...embeddingsConfig] satisfies InstrumentationConfig[];
 
+// The embeddings channel strings the subscriber binds to, derived from the provider list above so that
+// adding a provider is a single edit that both instruments it and subscribes the listener to it.
+export const langchainEmbeddingsChannels = EMBEDDINGS_PROVIDERS.flatMap(({ name }) =>
+  EMBEDDINGS_METHODS.map(method => `orchestrion:${name}:${method}`),
+);
+
 export const langchainChannels = {
   LANGCHAIN_CHAT_MODEL_INVOKE: 'orchestrion:@langchain/core:chatModelInvoke',
   LANGCHAIN_CHAT_MODEL_STREAM: 'orchestrion:@langchain/core:chatModelStream',
-  LANGCHAIN_EMBED_QUERY: 'orchestrion:@langchain/openai:embedQuery',
-  LANGCHAIN_EMBED_DOCUMENTS: 'orchestrion:@langchain/openai:embedDocuments',
 } as const;