AI & Agent Dev Bug Sandbox Back to Radar

Tee Leaves Source Generator Open When A Child Iterator Is Never Started

The Tee context manager in langchain-core does not close the underlying source generator if one or more child iterators are never started, causing resource leaks.

highConfidence 95%Langchain Affected V1.6.6

Origin Analysis

Unstarted child iterators do not execute their finally blocks when closed, leaving their registered buffers active and preventing the source generator from being closed.

Run the provided sync and async reproduction code. Observe that 'sync source closed: False' and 'async source closed: False' are printed, indicating the source's finally block did not execute.

Fixing Code Block

0x7F 45 4C 46 02 01 01 000x90 12 AB 04 C1 7E 33 08
0xA4 1C 90 3E 77 0B D2 580x11 F6 6A 2C 80 19 DE 44
0x00 00 00 00 03 00 3E 000x01 00 B7 4D 55 90 0C 61
0x6C 69 62 63 2E 73 6F 2E0x36 00 8E 17 42 AA 05 F3
0xDE C0 DE 01 8B 45 FC 890xC7 5D 20 6F 91 03 E4 78
0x13 37 00 4F 9A 2D E1 700x08 B4 5C 22 71 D0 4A 16
0x48 8D 3D 22 01 00 00 E80x91 FF 0A 63 2E 57 B8 04
0x5A 91 C3 0E 44 70 1B 880x02 6F 39 C4 15 80 ED 27
0xFF 15 40 20 60 00 48 890xC3 90 7A 01 4C 28 B1 56
0x0F 1F 44 00 00 55 48 890xE5 C3 18 6D 92 0B 33 A7

Edge Case Audit

Ensure the source generator is not closed elsewhere before the Tee context exits, as this would cause a double-close. The fix may affect performance if many unstarted children exist. Rollback by reverting to the original __exit__/__aexit__ methods.