get custom transport names

Author: betegon

packages/core/src/integrations/mcp-server/attributeExtraction.ts

@@ -45,25 +45,16 @@ import type {
  * @returns Transport type mapping for span attributes
  */
 export function getTransportTypes(transport: MCPTransport): { mcpTransport: string; networkTransport: string } {
-  // Handle undefined transport gracefully while preserving type detection
   if (!transport?.constructor) {
     return { mcpTransport: 'unknown', networkTransport: 'unknown' };
   }
-  const transportName = transport.constructor.name?.toLowerCase() || '';
+  const transportName = transport.constructor.name?.toLowerCase() || 'unknown';
+  const networkTransport = transportName === 'stdio' ? 'pipe' : 'tcp';
 
-  if (transportName.includes('stdio')) {
-    return { mcpTransport: 'stdio', networkTransport: 'pipe' };
-  }
-
-  if (transportName.includes('streamablehttp') || transportName.includes('streamable')) {
-    return { mcpTransport: 'http', networkTransport: 'tcp' };
-  }
-
-  if (transportName.includes('sse')) {
-    return { mcpTransport: 'sse', networkTransport: 'tcp' };
-  }
-
-  return { mcpTransport: 'unknown', networkTransport: 'unknown' };
+  return {
+    mcpTransport: transportName,
+    networkTransport,
+  };
 }
 
 /**
@@ -264,17 +255,13 @@ export function extractClientInfo(extra: ExtraHandlerData): {
  * @param transport - MCP transport instance
  * @param extra - Optional extra handler data
  * @returns Transport attributes for span instrumentation
+ * @note sessionId may be undefined during initial setup - session should be established by client during initialize flow
  */
 export function buildTransportAttributes(
   transport: MCPTransport,
   extra?: ExtraHandlerData,
 ): Record<string, string | number> {
-  // Gracefully handle undefined sessionId during MCP initialization
-  // Respects client-provided sessions and waits for proper session establishment
   const sessionId = transport && 'sessionId' in transport ? transport.sessionId : undefined;
-
-  // Note: sessionId may be undefined during initial setup - this is expected behavior
-  // The actual session should be established by the client during the initialize flow
   const clientInfo = extra ? extractClientInfo(extra) : {};
   const { mcpTransport, networkTransport } = getTransportTypes(transport);
   const clientAttributes = getClientAttributes(transport);