Skip to content

Commit 4ddd181

Browse files
use context instead of settings (#7796)
1 parent 41db5c1 commit 4ddd181

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

dspy/utils/streaming.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -266,10 +266,9 @@ async def use_streaming():
266266
status_streaming_callback = StatusStreamingCallback(status_message_provider)
267267
if not any(isinstance(c, StatusStreamingCallback) for c in callbacks):
268268
callbacks.append(status_streaming_callback)
269-
settings.configure(callbacks=callbacks)
270269

271270
async def generator(args, kwargs, stream: MemoryObjectSendStream):
272-
with settings.context(send_stream=stream):
271+
with settings.context(send_stream=stream, callbacks=callbacks):
273272
prediction = await program(*args, **kwargs)
274273

275274
await stream.send(prediction)

0 commit comments

Comments
 (0)