steerable-agent-runtime 0.6.30__tar.gz → 0.6.32__tar.gz

{steerable_agent_runtime-0.6.30 → steerable_agent_runtime-0.6.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: steerable-agent-runtime
-Version: 0.6.30
+Version: 0.6.32
 Summary: Steerable agent runtime: LLM, tool, storage, and transport adapters.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{steerable_agent_runtime-0.6.30 → steerable_agent_runtime-0.6.32}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "steerable-agent-runtime"
-version = "0.6.30"
+version = "0.6.32"
 description = "Steerable agent runtime: LLM, tool, storage, and transport adapters."
 readme = "README.md"
 requires-python = ">=3.10"

{steerable_agent_runtime-0.6.30 → steerable_agent_runtime-0.6.32}/src/steerable_agent_runtime.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: steerable-agent-runtime
-Version: 0.6.30
+Version: 0.6.32
 Summary: Steerable agent runtime: LLM, tool, storage, and transport adapters.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{steerable_agent_runtime-0.6.30 → steerable_agent_runtime-0.6.32}/tests/test_hooks.py

@@ -429,6 +429,49 @@ async def test_on_request_error_retry_recovers() -> None:
     assert "".join(deltas) == "recovered after retry"
 
 
+@pytest.mark.asyncio
+async def test_on_request_error_retry_recovers_mid_stream(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.setenv("STEERABLE_RUST_CORELOOP", "0")
+
+    class _Provider:
+        name = "mid-stream-retry"
+        model = "mid-stream-retry"
+
+        def __init__(self) -> None:
+            self.attempts = 0
+
+        def stream(self, messages, *, tools=None, **kwargs):
+            async def _gen():
+                self.attempts += 1
+                if self.attempts == 1:
+                    yield LLMStreamChunk(content_delta="partial")
+                    raise ConnectionError("incomplete chunked read")
+                yield LLMStreamChunk(content_delta="recovered after retry")
+                yield LLMStreamChunk(finish_reason="stop")
+
+            return _gen()
+
+    class _Retry(NoopHooks):
+        def __init__(self) -> None:
+            self.calls = 0
+
+        async def on_request_error(self, error, transcript, ctx):
+            self.calls += 1
+            return RetryAction(kind="retry", delay_ms=0)
+
+    hooks = _Retry()
+    provider = _Provider()
+    loop = CoreLoop(provider, RouterToolExecutor(ToolRouter()), hooks=hooks)
+    events = await collect(loop.run([LLMMessage.text_of("user", "go")]))
+
+    assert provider.attempts == 2
+    assert hooks.calls == 1
+    assert final_completion(events)["status"] == "completed"
+    assert loop.history.projection[-1].content_text == "recovered after retry"
+
+
 @pytest.mark.asyncio
 async def test_on_request_error_fail_emits_error_event_and_ends() -> None:
     provider = make_provider([{"error": ConnectionError("stream dropped")}])