File tree Expand file tree Collapse file tree 1 file changed +10
-0
lines changed
Expand file tree Collapse file tree 1 file changed +10
-0
lines changed Original file line number Diff line number Diff line change @@ -82,6 +82,11 @@ def setup_once() -> None:
8282def _wrap_stream_iter (
8383 f : "Callable[..., Iterator[RawMessageStreamEvent]]" ,
8484) -> "Callable[..., Iterator[RawMessageStreamEvent]]" :
85+ """
86+ Sets information received while iterating the response stream on the AI Client Span.
87+ Responsible for closing the AI Client Span.
88+ """
89+
8590 @wraps (f )
8691 def _patched_iter (self : "Stream" ) -> "Iterator[RawMessageStreamEvent]" :
8792 if not hasattr (self , "_sentry_span" ):
@@ -136,6 +141,11 @@ def _patched_iter(self: "Stream") -> "Iterator[RawMessageStreamEvent]":
136141def _wrap_async_stream_aiter (
137142 f : "Callable[..., AsyncIterator[RawMessageStreamEvent]]" ,
138143) -> "Callable[..., AsyncIterator[RawMessageStreamEvent]]" :
144+ """
145+ Sets information received while iterating the response stream on the AI Client Span.
146+ Responsible for closing the AI Client Span.
147+ """
148+
139149 @wraps (f )
140150 async def _patched_aiter (
141151 self : "AsyncStream" ,
You can’t perform that action at this time.
0 commit comments