projectdavid 1.32.14__py3-none-any.whl → 1.32.16__py3-none-any.whl

projectdavid/clients/synchronous_inference_wrapper.py

@@ -81,17 +81,23 @@ class SynchronousInferenceStream:
             def _filter_text(txt: str) -> str:  # noqa: D401
                 return txt
 
-        # helper – drain any leftover bytes still buffered
-        def _flush_tail() -> Optional[dict]:
-            """
-            Force-flush PeekGate / FunctionCallSuppressor so the last few
-            characters (kept for safety-margin) reach the frontend.
-            Returns a ready-to-stream chunk or None.
-            """
-            if suppress_fc:
-                tail = _filter_text("")  # empty feed ⇒ flush
-                if tail:
-                    return {"type": "content", "content": tail}
+        # helper – drain **all** residual bytes from the chain
+        def _drain_filters() -> Optional[dict]:
+            if not suppress_fc:
+                return None
+            parts: list[str] = []
+            # 1) normal draining loop
+            while True:
+                out = _filter_text("")
+                if not out:
+                    break
+                parts.append(out)
+            # 2) force-drain the ≤ 8 safety-margin tail
+            if not _peek_gate.suppressing and _peek_gate.buf:
+                parts.append(_peek_gate.buf)
+                _peek_gate.buf = ""
+            if parts:
+                return {"type": "content", "content": "".join(parts)}
             return None
 
         # ── main loop ─────────────────────────────────────────────
@@ -110,7 +116,6 @@ class SynchronousInferenceStream:
                 if chunk.get("type") in ("hot_code", "hot_code_output"):
                     yield chunk
                     continue
-
                 if (
                     chunk.get("stream_type") == "code_execution"
                     and chunk.get("chunk", {}).get("type") == "code_interpreter_stream"
@@ -127,21 +132,21 @@ class SynchronousInferenceStream:
                 # ④ everything else streams unchanged
                 yield chunk
 
-            # ─────────── graceful endings ───────────
+            # graceful endings ------------------------------------------------
             except StopAsyncIteration:
-                if tail := _flush_tail():
+                if tail := _drain_filters():
                     yield tail
                 LOG.info("Stream completed normally.")
                 break
 
             except asyncio.TimeoutError:
-                if tail := _flush_tail():
+                if tail := _drain_filters():
                     yield tail
                 LOG.error("[TimeoutError] Chunk wait expired – aborting stream.")
                 break
 
             except Exception as exc:  # noqa: BLE001
-                if tail := _flush_tail():
+                if tail := _drain_filters():
                     yield tail
                 LOG.error("Unexpected streaming error: %s", exc, exc_info=True)
                 break

{projectdavid-1.32.14.dist-info → projectdavid-1.32.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: projectdavid
-Version: 1.32.14
+Version: 1.32.16
 Summary: Python SDK for interacting with the Entities Assistant API.
 Author-email: Francis Neequaye Armah <francis.neequaye@projectdavid.co.uk>
 License: PolyForm Noncommercial License 1.0.0

{projectdavid-1.32.14.dist-info → projectdavid-1.32.16.dist-info}/RECORD

@@ -15,7 +15,7 @@ projectdavid/clients/files_client.py,sha256=XkIDzbQFGDrd88taf0Kouc_4YJOPIYEHiIyW
 projectdavid/clients/inference_client.py,sha256=xz4ACPv5Tkis604QxO5mJX1inH_TGDfQP-31geETYpE,6609
 projectdavid/clients/messages_client.py,sha256=467xeIt3VYs6cG8-bl-eDRi_auWOPmfd5tSJDmQSJUI,17232
 projectdavid/clients/runs.py,sha256=-fXOq5L9w2efDPmZkNxb0s2yjl6oN0XN4_aLXqaeceo,25270
-projectdavid/clients/synchronous_inference_wrapper.py,sha256=RuFOgvUar1p8xpklMZTMswDjZ8n2N3ghtjmDjSiI6Eo,6209
+projectdavid/clients/synchronous_inference_wrapper.py,sha256=JPOmEeF8ib5EwbH0QUvctyIYShpflZXOT-sMlhnFvdY,6373
 projectdavid/clients/threads_client.py,sha256=ekzU5w14zftmtmFkiec3NC90Of-_KVSUY1qH9cmfSFg,6771
 projectdavid/clients/tools_client.py,sha256=GkCVOmwpAoPqVt6aYmH0G1HIFha3iEwR9IIf9teR0j8,11487
 projectdavid/clients/users_client.py,sha256=eCuUb9qvyH1GUFhZu6TRL9zdoK-qzHSs8-Vmrk_0mmg,13729
@@ -34,8 +34,8 @@ projectdavid/utils/monitor_launcher.py,sha256=3YAgJdeuaUvq3JGvpA4ymqFsAnk29nH5q9
 projectdavid/utils/peek_gate.py,sha256=5whMRnDOQjATRpThWDJkvY9ScXuJ7Sd_-9rvGgXeTAQ,2532
 projectdavid/utils/run_monitor.py,sha256=F_WkqIP-qnWH-4llIbileWWLfRj2Q1Cg-ni23SR1rec,3786
 projectdavid/utils/vector_search_formatter.py,sha256=YTe3HPGec26qGY7uxY8_GS8lc4QaN6aNXMzkl29nZpI,1735
-projectdavid-1.32.14.dist-info/licenses/LICENSE,sha256=_8yjiEGttpS284BkfhXxfERqTRZW_tUaHiBB0GTJTMg,4563
-projectdavid-1.32.14.dist-info/METADATA,sha256=lwrij-9rEDGMo6elHx8cZyAmArJirSaL1Nit8177LeA,10782
-projectdavid-1.32.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-projectdavid-1.32.14.dist-info/top_level.txt,sha256=kil8GU4s7qYRfNnzGnFHhZnSNRSxgNG-J4HLgQMmMtw,13
-projectdavid-1.32.14.dist-info/RECORD,,
+projectdavid-1.32.16.dist-info/licenses/LICENSE,sha256=_8yjiEGttpS284BkfhXxfERqTRZW_tUaHiBB0GTJTMg,4563
+projectdavid-1.32.16.dist-info/METADATA,sha256=BOcsvrfPdNelHuFfb9u08dkcx6NxeyWEpoJcnqxaAEc,10782
+projectdavid-1.32.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+projectdavid-1.32.16.dist-info/top_level.txt,sha256=kil8GU4s7qYRfNnzGnFHhZnSNRSxgNG-J4HLgQMmMtw,13
+projectdavid-1.32.16.dist-info/RECORD,,