File tree 1 file changed +2
-7
lines changed
sdk/python/src/datastar_py
1 file changed +2
-7
lines changed Original file line number Diff line number Diff line change 3
3
4
4
from django .http import StreamingHttpResponse as _StreamingHttpResponse
5
5
6
- from .sse import SSE_HEADERS , ServerSentEventGenerator , _async_map , _wrap_event
6
+ from .sse import SSE_HEADERS , ServerSentEventGenerator , _sse_iterable_wrapper
7
7
8
8
9
9
class DatastarStreamingHttpResponse (_StreamingHttpResponse ):
@@ -17,10 +17,5 @@ def sse_generator(generator_func):
17
17
@wraps (generator_func )
18
18
def _wrapper (* args , ** kwargs ):
19
19
content = generator_func (* args , ** kwargs )
20
- if isinstance (content , typing .AsyncIterable ):
21
- content = _async_map (_wrap_event , content )
22
- else :
23
- content = map (_wrap_event , content )
24
-
25
- return DatastarStreamingHttpResponse (content )
20
+ return DatastarStreamingHttpResponse (_sse_iterable_wrapper (content ))
26
21
return _wrapper
You can’t perform that action at this time.
0 commit comments