fix: Fix cancellation flow in python component graph (#765)

420b7a82 · Pankaj Gupta · GitHub · 2972b7ed · 420b7a82
Unverified Commit 420b7a82 authored Apr 21, 2025 by Pankaj Gupta Committed by GitHub Apr 21, 2025
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

lib/bindings/python/rust/lib.rs lib/bindings/python/rust/lib.rs +1 -0

No files found.
--- a/lib/bindings/python/rust/lib.rs
+++ b/lib/bindings/python/rust/lib.rs
@@ -697,6 +697,7 @@ async fn process_stream(
        // Send the PyObject through the channel or log an error
        if let Err(e) = tx.send(annotated).await {
            tracing::error!("Failed to send response: {:?}", e);
+            break;
        }

        if is_error {