fix: body gen hangs on retry when inputs exhausted in append session (#13)

quettabit · web-flow · commit 043be930772f · 2026-04-08T01:19:28.000-06:00
diff --git a/src/s2_sdk/_s2s/_append_session.py b/src/s2_sdk/_s2s/_append_session.py
@@ -209,6 +209,7 @@ async def _body_gen(
     while True:
         inp = await input_queue.get()
         if inp is None:
+            await input_queue.put(None)
             return
         encoded = _encode_input(inp, compression)
         inflight_inputs.append(
