End unsampled Bedrock streaming spans instead of leaking them

Author: andreiborza

packages/server-utils/src/integrations/tracing-channel/aws-sdk/services/bedrock-runtime.ts

@@ -58,11 +58,19 @@ export class BedrockRuntimeServiceExtension implements ServiceExtension {
   }
 
   public responseHook(response: NormalizedResponse, span: Span): void {
+    const commandName = response.request.commandName;
+
     if (!span.isRecording()) {
+      // Streaming spans are ended by the wrapped stream (the subscriber's `deferSpanEnd` suppresses
+      // the helper's own `end()`). When the span isn't recording we skip wrapping, so end it here to
+      // avoid leaking an open span. Non-streaming commands are ended by the helper, so leave them.
+      if (commandName === 'ConverseStream' || commandName === 'InvokeModelWithResponseStream') {
+        span.end();
+      }
       return;
     }
 
-    switch (response.request.commandName) {
+    switch (commandName) {
       case 'Converse':
         return this._responseHookConverse(response, span);
       case 'ConverseStream':