Skip to content

Commit 392eb17

Browse files
Merge branch 'webb/litellm/close-spans' into webb/litellm/remove-mocks
2 parents 1b28574 + 8435f36 commit 392eb17

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

sentry_sdk/integrations/litellm.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,8 @@ def _success_callback(
178178
) -> None:
179179
"""Handle successful completion."""
180180

181-
span = _get_metadata_dict(kwargs).get("_sentry_span")
181+
metadata = _get_metadata_dict(kwargs)
182+
span = metadata.get("_sentry_span")
182183
if span is None:
183184
return
184185

@@ -234,7 +235,9 @@ def _success_callback(
234235
# Callback is fired multiple times when streaming a response.
235236
# Streaming flag checked at https://github.com/BerriAI/litellm/blob/33c3f13443eaf990ac8c6e3da78bddbc2b7d0e7a/litellm/litellm_core_utils/litellm_logging.py#L1603
236237
if is_streaming is not True or "complete_streaming_response" in kwargs:
237-
span.__exit__(None, None, None)
238+
span = metadata.pop("_sentry_span", None)
239+
if span is not None:
240+
span.__exit__(None, None, None)
238241

239242

240243
def _failure_callback(

0 commit comments

Comments
 (0)