ref(node): Refactor orchestrion config into separate files

packages/server-utils/package.json

@@ -28,9 +28,9 @@
       "require": "./build/cjs/orchestrion/index.js"
     },
     "./orchestrion/config": {
-      "types": "./build/types/orchestrion/config.d.ts",
-      "import": "./build/esm/orchestrion/config.js",
-      "require": "./build/cjs/orchestrion/config.js"
+      "types": "./build/types/orchestrion/config/index.d.ts",
+      "import": "./build/esm/orchestrion/config/index.js",
+      "require": "./build/cjs/orchestrion/config/index.js"
     },
     "./orchestrion/register": {
       "types": "./build/types/orchestrion/runtime/register.d.ts",
@@ -54,7 +54,7 @@
         "build/types-ts3.8/orchestrion/index.d.ts"
       ],
       "orchestrion/config": [
-        "build/types-ts3.8/orchestrion/config.d.ts"
+        "build/types-ts3.8/orchestrion/config/index.d.ts"
       ],
       "orchestrion/register": [
         "build/types-ts3.8/orchestrion/runtime/register.d.ts"
@@ -68,7 +68,7 @@
         "build/types/orchestrion/index.d.ts"
       ],
       "orchestrion/config": [
-        "build/types/orchestrion/config.d.ts"
+        "build/types/orchestrion/config/index.d.ts"
       ],
       "orchestrion/register": [
         "build/types/orchestrion/runtime/register.d.ts"

packages/server-utils/rollup.npm.config.mjs

@@ -19,7 +19,7 @@ export default [
   ...orchestrionRuntimeHooks,
   ...makeNPMConfigVariants(
     makeBaseNPMConfig({
-      // `src/orchestrion/config.ts` and `src/orchestrion/bundler/vite.ts` are
+      // `src/orchestrion/config/index.ts` and `src/orchestrion/bundler/vite.ts` are
       // loaded via dedicated subpath exports (`.../orchestrion/config`,
       // `.../orchestrion/vite`) — neither is reachable from `src/index.ts`, so we
       // list them as separate entrypoints to guarantee they end up in build/esm
@@ -28,7 +28,7 @@ export default [
       entrypoints: [
         'src/index.ts',
         'src/orchestrion/index.ts',
-        'src/orchestrion/config.ts',
+        'src/orchestrion/config/index.ts',
         // `src/orchestrion/runtime/register.ts` backs the `./orchestrion/register`
         // subpath export; the Node SDK `require`s it synchronously from
         // `Sentry.init()` to install the channel-injection hooks.

packages/server-utils/src/orchestrion/channels.ts

@@ -1,3 +1,11 @@
+import { mysqlChannels } from './config/mysql';
+import { lruMemoizerChannels } from './config/lru-memoizer';
+import { ioredisChannels } from './config/ioredis';
+import { pgChannels } from './config/pg';
+import { openaiChannels } from './config/openai';
+import { anthropicAiChannels } from './config/anthropic-ai';
+import { vercelAiChannels } from './config/vercel-ai';
+
 /**
  * Fully-qualified `diagnostics_channel` names that orchestrion publishes to.
  *
@@ -12,33 +20,13 @@
  * they don't drift apart and silently stop firing.
  */
 export const CHANNELS = {
-  MYSQL_QUERY: 'orchestrion:mysql:query',
-  LRU_MEMOIZER_LOAD: 'orchestrion:lru-memoizer:load',
-  IOREDIS_COMMAND: 'orchestrion:ioredis:command',
-  IOREDIS_CONNECT: 'orchestrion:ioredis:connect',
-  PG_QUERY: 'orchestrion:pg:query',
-  PG_CONNECT: 'orchestrion:pg:connect',
-  PGPOOL_CONNECT: 'orchestrion:pg-pool:connect',
-  OPENAI_CHAT: 'orchestrion:openai:chat',
-  OPENAI_RESPONSES: 'orchestrion:openai:responses',
-  OPENAI_EMBEDDINGS: 'orchestrion:openai:embeddings',
-  OPENAI_CONVERSATIONS: 'orchestrion:openai:conversations',
-  ANTHROPIC_CHAT: 'orchestrion:@anthropic-ai/sdk:chat',
-  ANTHROPIC_MODELS: 'orchestrion:@anthropic-ai/sdk:models',
-  ANTHROPIC_MESSAGES_STREAM: 'orchestrion:@anthropic-ai/sdk:messages-stream',
-  // Vercel AI (`ai`) v6: orchestrion injects these so the same channel-based
-  // integration that consumes `ai`'s native `ai:telemetry` channel (v7) can
-  // also instrument v6. Each maps to a top-level function in `ai`'s bundle.
-  VERCEL_AI_GENERATE_TEXT: 'orchestrion:ai:generateText',
-  VERCEL_AI_STREAM_TEXT: 'orchestrion:ai:streamText',
-  VERCEL_AI_EMBED: 'orchestrion:ai:embed',
-  VERCEL_AI_EMBED_MANY: 'orchestrion:ai:embedMany',
-  VERCEL_AI_EXECUTE_TOOL_CALL: 'orchestrion:ai:executeToolCall',
-  // `resolveLanguageModel` is the single chokepoint every model call flows
-  // through; we wrap it to monkey-patch `doGenerate`/`doStream` on the returned
-  // model (the model-call site itself is an inline call with no injectable
-  // definition).
-  VERCEL_AI_RESOLVE_LANGUAGE_MODEL: 'orchestrion:ai:resolveLanguageModel',
+  ...mysqlChannels,
+  ...lruMemoizerChannels,
+  ...ioredisChannels,
+  ...pgChannels,
+  ...openaiChannels,
+  ...anthropicAiChannels,
+  ...vercelAiChannels,
 } as const;
 
 export type ChannelName = (typeof CHANNELS)[keyof typeof CHANNELS];

packages/server-utils/src/orchestrion/config/anthropic-ai.ts

@@ -0,0 +1,40 @@
+import type { InstrumentationConfig } from '@apm-js-collab/code-transformer';
+
+export const anthropicAiConfig = [
+  // One entry each for CJS/ESM
+  ...['resources/messages/messages.js', 'resources/messages/messages.mjs'].flatMap(filePath =>
+    (['create', 'countTokens'] as const).map(methodName => ({
+      channelName: 'chat',
+      module: { name: '@anthropic-ai/sdk', versionRange: '>=0.19.2 <1', filePath },
+      functionQuery: { className: 'Messages', methodName, kind: 'Auto' as const },
+    })),
+  ),
+  ...['resources/completions.js', 'resources/completions.mjs'].map(filePath => ({
+    channelName: 'chat',
+    module: { name: '@anthropic-ai/sdk', versionRange: '>=0.19.2 <1', filePath },
+    functionQuery: { className: 'Completions', methodName: 'create', kind: 'Auto' as const },
+  })),
+  ...['resources/beta/messages/messages.js', 'resources/beta/messages/messages.mjs'].map(filePath => ({
+    channelName: 'chat',
+    module: { name: '@anthropic-ai/sdk', versionRange: '>=0.19.2 <1', filePath },
+    functionQuery: { className: 'Messages', methodName: 'create', kind: 'Auto' as const },
+  })),
+  ...['resources/models.js', 'resources/models.mjs'].map(filePath => ({
+    channelName: 'models',
+    module: { name: '@anthropic-ai/sdk', versionRange: '>=0.19.2 <1', filePath },
+    functionQuery: { className: 'Models', methodName: 'retrieve', kind: 'Auto' as const },
+  })),
+  // `messages.stream()` returns a synchronous emitter, not a promise, so `kind: 'Sync'` is required:
+  // `Auto`'s promise wrapper never publishes `end` for a non-thenable return, so the span would never end.
+  ...['resources/messages/messages.js', 'resources/messages/messages.mjs'].map(filePath => ({
+    channelName: 'messages-stream',
+    module: { name: '@anthropic-ai/sdk', versionRange: '>=0.19.2 <1', filePath },
+    functionQuery: { className: 'Messages', methodName: 'stream', kind: 'Sync' as const },
+  })),
+] satisfies InstrumentationConfig[];
+
+export const anthropicAiChannels = {
+  ANTHROPIC_CHAT: 'orchestrion:@anthropic-ai/sdk:chat',
+  ANTHROPIC_MODELS: 'orchestrion:@anthropic-ai/sdk:models',
+  ANTHROPIC_MESSAGES_STREAM: 'orchestrion:@anthropic-ai/sdk:messages-stream',
+} as const;