safe exit pattern

Author: alexander-alderman-webb

sentry_sdk/integrations/litellm.py

@@ -235,8 +235,9 @@ def _success_callback(
         # Callback is fired multiple times when streaming a response.
         # Streaming flag checked at https://github.com/BerriAI/litellm/blob/33c3f13443eaf990ac8c6e3da78bddbc2b7d0e7a/litellm/litellm_core_utils/litellm_logging.py#L1603
         if is_streaming is not True or "complete_streaming_response" in kwargs:
-            span.__exit__(None, None, None)
-            del metadata["_sentry_span"]
+            span = metadata.pop("_sentry_span", None)
+            if span is not None:
+                span.__exit__(None, None, None)
 
 
 def _failure_callback(