Ensure span is recording before ending as there are multiple paths where it can

end

Author: andreiborza

packages/core/src/utils/anthropic-ai/index.ts

@@ -226,7 +226,6 @@ function instrumentMethod<T extends unknown[], R>(
               const messageStream = target.apply(context, args);
               return instrumentStream(messageStream, span, options.recordOutputs ?? false);
             } catch (error) {
-              span.setStatus({ code: SPAN_STATUS_ERROR, message: 'internal_error' });
               captureException(error, {
                 mechanism: {
                   handled: false,
@@ -236,7 +235,12 @@ function instrumentMethod<T extends unknown[], R>(
                   },
                 },
               });
-              span.end();
+
+              if (span.isRecording()) {
+                span.setStatus({ code: SPAN_STATUS_ERROR, message: 'internal_error' });
+                span.end();
+              }
+
               throw error;
             }
           },

packages/core/src/utils/anthropic-ai/streaming.ts

@@ -254,7 +254,9 @@ function finalizeStreamSpan(state: StreamingState, span: Span, recordOutputs: bo
     });
   }
 
-  span.end();
+  if (span.isRecording()) {
+    span.end();
+  }
 }
 
 /**
@@ -289,14 +291,17 @@ export function instrumentStream<R extends { on: (...args: unknown[]) => void }>
   });
 
   stream.on('error', (error: unknown) => {
-    span.setStatus({ code: SPAN_STATUS_ERROR, message: 'stream_error' });
     captureException(error, {
       mechanism: {
         handled: false,
         type: 'auto.ai.anthropic.stream_error',
       },
     });
-    span.end();
+
+    if (span.isRecording()) {
+      span.setStatus({ code: SPAN_STATUS_ERROR, message: 'stream_error' });
+      span.end();
+    }
   });
 
   return stream;