spatialagents 0.1.0__tar.gz → 0.1.1__tar.gz

{spatialagents-0.1.0 → spatialagents-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: spatialagents
-Version: 0.1.0
+Version: 0.1.1
 Summary: The chat harness runtime SCIMAP Pro, Plexora and other BioCognia products share
 Author-email: Ajit Johnson Nirmal <ajitjohnson.n@gmail.com>
 License-Expression: LicenseRef-Proprietary

{spatialagents-0.1.0 → spatialagents-0.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "spatialagents"
-version = "0.1.0"
+version = "0.1.1"
 description = "The chat harness runtime SCIMAP Pro, Plexora and other BioCognia products share"
 readme = "README.md"
 authors = [

{spatialagents-0.1.0 → spatialagents-0.1.1}/src/spatialagents/__init__.py

@@ -43,7 +43,7 @@ from __future__ import annotations
 
 import importlib
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 _SUBMODULES = frozenset({
     "approvals", "backend", "cache_plan", "chat", "conversations", "errors", "gateway", "offload",

{spatialagents-0.1.0 → spatialagents-0.1.1}/src/spatialagents/gateway.py

@@ -22,11 +22,16 @@ Retries: a refusal that happened before anything reached a provider (rate
 limit, provider outage, unreachable gateway) is retried with the SAME
 idempotency key, which the gateway frees in exactly those cases; an expired
 token is refreshed once. Credit, cap and envelope refusals are never retried:
-they are surfaced so the run or conversation can pause (`PAUSE_CODES`).
+they are surfaced so the run or conversation can pause (`PAUSE_CODES`). A
+stream that breaks once the answer has begun -- the gateway stops, the
+connection drops, or nothing arrives for `CALL_TIMEOUT` seconds -- is
+`interrupted` and not retried: the call reached the provider, and the gateway
+may have settled it already.
 """
 
 from __future__ import annotations
 
+import http.client
 import json
 import time
 
@@ -61,6 +66,20 @@ __all__ = ["CALL_TIMEOUT", "CIRCUIT_ATTEMPTS", "CREDIT_CODES", "ENV_DEV", "ENV_G
            "gateway_url", "iter_sse"]
 
 
+def _events(response):
+    """`iter_sse`, with a connection that breaks mid-answer raised as
+    `interrupted`. Only the read is covered: what the caller does with an
+    event (`on_delta`) raises as itself."""
+    try:
+        yield from iter_sse(response)
+    except TimeoutError:
+        raise GatewayError(f"The gateway went silent mid-answer: nothing arrived for {CALL_TIMEOUT} s.",
+                           code="interrupted", detail={"transport": "TimeoutError"}) from None
+    except (OSError, http.client.HTTPException) as exc:
+        raise GatewayError(f"The connection to the gateway broke mid-answer ({type(exc).__name__}).",
+                           code="interrupted", detail={"transport": type(exc).__name__}) from None
+
+
 class GatewayClient(_shared.GatewayClient):
     """The shared transport, plus the harness's streamed model call.
 
@@ -126,7 +145,7 @@ class GatewayClient(_shared.GatewayClient):
         usage_event: dict = {}
         accepted: dict = {}
         stop = None
-        for event, data in iter_sse(response):
+        for event, data in _events(response):
             try:
                 payload = json.loads(data)
             except ValueError:

{spatialagents-0.1.0 → spatialagents-0.1.1}/tests/test_gateway.py

@@ -4,12 +4,14 @@ hand-written event streams."""
 
 from __future__ import annotations
 
+import contextlib
+import http.client
 import io
 import json
 
 import pytest
 
-from spatialagents.gateway import PAUSE_CODES, GatewayClient, GatewayError
+from spatialagents.gateway import CALL_TIMEOUT, PAUSE_CODES, GatewayClient, GatewayError
 from spatialagents.testing import FakeGateway, Reply, fixed_tokens
 from spatialagents.wire import ModelRequest, task_id
 
@@ -102,6 +104,56 @@ def test_a_stream_the_gateway_never_settled_is_interrupted():
     assert caught.value.code == "interrupted"
 
 
+HALF = ("content_block_delta", {"index": 0, "delta": {"type": "text_delta", "text": "half"}})
+
+
+def stalls(error, *events):
+    """A body that streams `events`, then fails its next read with `error`."""
+    def lines():
+        yield from sse(*events)
+        raise error
+    return lines()
+
+
+@pytest.mark.parametrize("error, said", [(TimeoutError("The read operation timed out"), "went silent"),
+                                         (ConnectionResetError(10054, "reset"), "ConnectionResetError"),
+                                         (http.client.IncompleteRead(b""), "IncompleteRead")])
+def test_a_connection_that_breaks_mid_answer_is_interrupted(error, said):
+    gc = GatewayClient(fixed_tokens(), "http://127.0.0.1:9")
+    deltas = []
+    with pytest.raises(GatewayError) as caught:
+        gc._read_stream(stalls(error, HALF), deltas.append)
+    assert caught.value.code == "interrupted" and said in str(caught.value)
+    assert deltas == ["half"]
+
+
+def test_a_call_cut_mid_answer_is_not_retried(monkeypatch):
+    gc = GatewayClient(fixed_tokens(), "http://127.0.0.1:9")
+    opened, retries = [], []
+
+    def open_(method, path, body=None, headers=None, timeout=60):
+        opened.append(timeout)
+        return contextlib.nullcontext(stalls(TimeoutError("The read operation timed out"), HALF))
+
+    monkeypatch.setattr(gc, "open", open_)
+    with pytest.raises(GatewayError) as caught:
+        gc.messages(request(), idempotency_key="stall-0001", on_retry=lambda n, exc: retries.append(n))
+    assert caught.value.code == "interrupted" and not caught.value.retryable
+    assert opened == [CALL_TIMEOUT] and retries == []
+
+
+def test_an_error_from_on_delta_is_not_taken_for_a_broken_connection():
+    class Refused(OSError):
+        pass
+
+    def on_delta(piece):
+        raise Refused(piece)
+
+    gc = GatewayClient(fixed_tokens(), "http://127.0.0.1:9")
+    with pytest.raises(Refused):
+        gc._read_stream(sse(HALF, USAGE), on_delta)
+
+
 def test_the_client_retries_an_outage_with_the_same_key_and_never_retries_credit():
     with FakeGateway(lambda packet, body: {"kind": "x"}) as gateway:
         gateway.fail_with = [(503, "provider_unavailable"), (429, "provider_rate_limited")]

{spatialagents-0.1.0 → spatialagents-0.1.1}/uv.lock

@@ -389,7 +389,7 @@ wheels = [
 
 [[package]]
 name = "spatialagents"
-version = "0.1.0"
+version = "0.1.1"
 source = { editable = "." }
 dependencies = [
     { name = "biocognia" },