ai-agents-async-cancellation-h-when-a-long-running-async-graph-stream-is-cancelle-83d1f538

THEN In the on_done method of FuturesDict, defensively check that the callback is not None before calling it. This prevents the exception from being raised after task cancellation and ensures clean log output.

def on_done(self, task, fut):
    try:
        cb = self.callback()
        if cb is not None:
            cb(task, _exception(fut))
    finally:
        ...