langbot-plugin 0.6.3__tar.gz → 0.6.5__tar.gz

{langbot_plugin-0.6.3 → langbot_plugin-0.6.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: langbot-plugin
-Version: 0.6.3
+Version: 0.6.5
 Summary: This package contains the SDK, CLI for building plugins for LangBot, plus the runtime for hosting LangBot plugins
 Project-URL: Homepage, https://langbot.app
 Project-URL: Repository, https://github.com/langbot-app/langbot-plugin-sdk

{langbot_plugin-0.6.3 → langbot_plugin-0.6.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langbot-plugin"
-version = "0.6.3"
+version = "0.6.5"
 description = "This package contains the SDK, CLI for building plugins for LangBot, plus the runtime for hosting LangBot plugins"
 readme = "README.md"
 authors = [

{langbot_plugin-0.6.3 → langbot_plugin-0.6.5}/src/langbot_plugin/runtime/io/controllers/stdio/client.py

@@ -65,12 +65,14 @@ class StdioClientController(Controller):
         working_dir: str = ".",
         *,
         capture_stderr: bool = False,
+        close_process_on_callback_return: bool = True,
     ):
         self.command = command
         self.args = args
         self.env = env
         self.working_dir = working_dir
         self.capture_stderr = capture_stderr
+        self.close_process_on_callback_return = close_process_on_callback_return
 
     async def run(
         self,
@@ -95,7 +97,8 @@ class StdioClientController(Controller):
         try:
             await new_connection_callback(self.connection)
         finally:
-            await self.close()
+            if self.close_process_on_callback_return:
+                await self.close()
 
     async def close(self) -> None:
         """Close pipes and reap the owned subprocess."""

{langbot_plugin-0.6.3 → langbot_plugin-0.6.5}/src/langbot_plugin/runtime/plugin/dependency_environment.py

@@ -16,6 +16,7 @@ from dataclasses import dataclass
 
 from packaging.requirements import InvalidRequirement, Requirement
 from packaging.utils import canonicalize_name
+from packaging.version import InvalidVersion, Version
 
 from langbot_plugin.runtime.plugin.artifact import PluginArtifact
 
@@ -28,6 +29,7 @@ _MAX_ENVIRONMENT_ENTRIES = 100_000
 _MAX_ENVIRONMENT_TOTAL_BYTES = 2 * 1024 * 1024 * 1024
 _SHA256_PATTERN = re.compile(r"^[0-9a-f]{64}$")
 _RUNTIME_PROVIDED_DISTRIBUTIONS = frozenset({"langbot-plugin"})
+_RUNTIME_SDK_COMPATIBILITY_SERIES = (0, 6)
 
 
 class DependencyEnvironmentPreparationError(RuntimeError):
@@ -224,6 +226,39 @@ class PluginDependencyEnvironmentStore:
             shutil.rmtree(temporary_root, ignore_errors=True)
             raise
 
+    @staticmethod
+    def _runtime_sdk_requirement_satisfied(
+        requirement: Requirement,
+        runtime_version: str,
+    ) -> bool:
+        if not requirement.specifier or requirement.specifier.contains(
+            runtime_version,
+            prereleases=True,
+        ):
+            return True
+
+        # Published 0.6 plugins historically used exact SDK pins even though
+        # workers always import the Runtime-owned SDK. Preserve that deployed
+        # contract only within the reviewed 0.6 compatibility series: an older
+        # exact pin is a minimum ABI level, while every other constraint remains
+        # strict. Direct URLs are rejected before this helper is called.
+        specifiers = tuple(requirement.specifier)
+        if len(specifiers) != 1 or specifiers[0].operator != "==":
+            return False
+        requested_raw = specifiers[0].version
+        if "*" in requested_raw:
+            return False
+        try:
+            requested = Version(requested_raw)
+            runtime = Version(runtime_version)
+        except InvalidVersion:
+            return False
+        return (
+            requested.release[:2] == _RUNTIME_SDK_COMPATIBILITY_SERIES
+            and runtime.release[:2] == _RUNTIME_SDK_COMPATIBILITY_SERIES
+            and requested <= runtime
+        )
+
     @staticmethod
     def _read_requirements(
         artifact: PluginArtifact,
@@ -283,8 +318,9 @@ class PluginDependencyEnvironmentStore:
                     ) from exc
                 # Validate the already-installed Runtime SDK, including beta
                 # releases, rather than filtering candidates for installation.
-                if requirement.specifier and not requirement.specifier.contains(
-                    runtime_version, prereleases=True
+                if not PluginDependencyEnvironmentStore._runtime_sdk_requirement_satisfied(
+                    requirement,
+                    runtime_version,
                 ):
                     raise DependencyEnvironmentPreparationError(
                         "Plugin requires "

{langbot_plugin-0.6.3 → langbot_plugin-0.6.5}/src/langbot_plugin/runtime/plugin/mgr.py

@@ -26,6 +26,7 @@ from langbot_plugin.runtime.io.connection import Connection
 from langbot_plugin.runtime.io.controllers.stdio import (
     client as stdio_client_controller,
 )
+from langbot_plugin.runtime.io.controller import Controller
 from langbot_plugin.runtime.plugin.runner_service import RunnerRuntimeService
 from langbot_plugin.runtime.plugin import container as runtime_plugin_container
 from langbot_plugin.runtime.io.handlers import plugin as runtime_plugin_handler_cls
@@ -148,6 +149,7 @@ class SharedPluginWorkerRuntime:
         default_factory=dict
     )
     plugin_handler: runtime_plugin_handler_cls.PluginConnectionHandler | None = None
+    controller: Controller | None = None
     launch_task: asyncio.Task[None] | None = None
     transport_registered_event: asyncio.Event = field(default_factory=asyncio.Event)
     ready_event: asyncio.Event = field(default_factory=asyncio.Event)
@@ -1717,6 +1719,14 @@ class PluginManager:
             task.cancel()
             await asyncio.gather(task, return_exceptions=True)
         worker.launch_task = None
+        controller = worker.controller
+        if controller is not None:
+            close = getattr(controller, "close", None)
+            if close is not None:
+                with contextlib.suppress(Exception):
+                    await close()
+            if worker.controller is controller:
+                worker.controller = None
         if not worker.slots:
             transfer_root = (
                 self.artifact_store.base_path
@@ -1954,6 +1964,7 @@ class PluginManager:
                     dependency_environment=worker.dependency_environment,
                 )
             )
+            worker.controller = controller
 
             async def new_plugin_connection_callback(connection: Connection):
                 shared_transfer_root = (
@@ -1993,7 +2004,15 @@ class PluginManager:
                 finally:
                     await self.remove_plugin_handler(plugin_handler)
 
-            await controller.run(new_plugin_connection_callback)
+            try:
+                await controller.run(new_plugin_connection_callback)
+            finally:
+                close = getattr(controller, "close", None)
+                if close is not None:
+                    with contextlib.suppress(Exception):
+                        await close()
+                if worker.controller is controller:
+                    worker.controller = None
         finally:
             self._revoke_registration_capability(capability)
 

{langbot_plugin-0.6.3 → langbot_plugin-0.6.5}/src/langbot_plugin/runtime/plugin/worker_launcher.py

@@ -625,6 +625,7 @@ class PluginWorkerLauncher:
             args=self.build_shared_pool_nsjail_args(launch_spec),
             env={},
             working_dir="/",
+            close_process_on_callback_return=False,
         )
 
     def _append_readonly_runtime_mounts(

langbot_plugin-0.6.5/src/langbot_plugin/version.py

@@ -0,0 +1 @@
+__version__ = "0.6.5"

{langbot_plugin-0.6.3 → langbot_plugin-0.6.5}/tests/packaging/test_release_metadata.py

@@ -11,5 +11,5 @@ def test_stable_release_metadata_is_consistent() -> None:
         (root / "src/langbot_plugin/version.py").read_text(encoding="utf-8").strip()
     )
 
-    assert project["project"]["version"] == "0.6.3"
-    assert source_version == '__version__ = "0.6.3"'
+    assert project["project"]["version"] == "0.6.5"
+    assert source_version == '__version__ = "0.6.5"'

{langbot_plugin-0.6.3 → langbot_plugin-0.6.5}/tests/runtime/io/test_controllers.py

@@ -20,7 +20,7 @@ class FakeProcess:
     def __init__(self, stdin=object(), stdout=object()):
         self.stdin = stdin
         self.stdout = stdout
-        self.returncode = 0
+        self.returncode: int | None = 0
 
 
 class FakeWebSocket:
@@ -114,6 +114,33 @@ async def test_stdio_client_controller_captures_stderr_only_when_requested(
     assert captured["stderr"] is stdio_client.asyncio.subprocess.PIPE
 
 
+async def test_stdio_client_controller_can_transfer_process_lifetime_ownership(
+    monkeypatch,
+):
+    process = FakeProcess()
+    process.returncode = None
+
+    async def fake_create_subprocess_exec(*args, **kwargs):
+        return process
+
+    monkeypatch.setattr(
+        stdio_client.asyncio,
+        "create_subprocess_exec",
+        fake_create_subprocess_exec,
+    )
+    controller = stdio_client.StdioClientController(
+        "python",
+        [],
+        {},
+        close_process_on_callback_return=False,
+    )
+
+    await controller.run(lambda _connection: asyncio.sleep(0))
+
+    assert controller.process is process
+    assert controller.connection is not None
+
+
 async def test_stdio_client_controller_rejects_missing_pipes(monkeypatch):
     async def fake_create_subprocess_exec(*args, **kwargs):
         return FakeProcess(stdin=None, stdout=object())

{langbot_plugin-0.6.3 → langbot_plugin-0.6.5}/tests/runtime/plugin/test_dependency_environment.py

@@ -258,6 +258,63 @@ async def test_runtime_sdk_requirement_is_not_installed_into_plugin_environment(
     assert captured_requirements == ("third-party-demo==1.0.0",)
 
 
+@pytest.mark.parametrize("requirement", ["==0.6.0b5", "==0.6.1", "==0.6.2", "==0.6.3"])
+async def test_runtime_sdk_legacy_06_exact_pin_accepts_newer_compatible_06_runtime(
+    tmp_path,
+    monkeypatch,
+    requirement,
+):
+    artifact = _artifact(
+        tmp_path,
+        f"langbot-plugin{requirement}\nthird-party-demo==1.0.0\n",
+    )
+    store = PluginDependencyEnvironmentStore(tmp_path / "plugin-runtime")
+    captured_requirements = None
+    monkeypatch.setattr(
+        dependency_environment_module.importlib.metadata,
+        "version",
+        lambda name: "0.6.3" if name == "langbot-plugin" else "1.0.0",
+    )
+
+    async def installer(staging, requirements):
+        nonlocal captured_requirements
+        captured_requirements = requirements
+        _publish_fake_distribution(staging, requirements)
+
+    await store.prepare(
+        artifact,
+        runtime_fingerprint="runtime-v1",
+        installer=installer,
+    )
+
+    assert captured_requirements == ("third-party-demo==1.0.0",)
+
+
+@pytest.mark.parametrize("requirement", ["==0.5.5", "==0.7.0", ">=0.7.0", "!=0.6.3"])
+async def test_runtime_sdk_compatibility_does_not_relax_other_constraints(
+    tmp_path,
+    monkeypatch,
+    requirement,
+):
+    artifact = _artifact(tmp_path, f"langbot-plugin{requirement}\n")
+    store = PluginDependencyEnvironmentStore(tmp_path / "plugin-runtime")
+    monkeypatch.setattr(
+        dependency_environment_module.importlib.metadata,
+        "version",
+        lambda name: "0.6.3",
+    )
+
+    async def installer(staging, requirements):  # pragma: no cover - must not run
+        raise AssertionError("installer must not run")
+
+    with pytest.raises(DependencyEnvironmentPreparationError):
+        await store.prepare(
+            artifact,
+            runtime_fingerprint="runtime-v1",
+            installer=installer,
+        )
+
+
 @pytest.mark.parametrize(
     ("runtime_version", "requirement"),
     [

{langbot_plugin-0.6.3 → langbot_plugin-0.6.5}/tests/runtime/plugin/test_installation_manager.py

@@ -2082,6 +2082,7 @@ async def test_shared_worker_ready_timeout_cancels_hung_controller_and_records_f
     )
     worker = manager.installation_runtimes[binding].shared_worker
     cancelled = asyncio.Event()
+    closed = asyncio.Event()
 
     class NeverRegisterController:
         process = None
@@ -2092,6 +2093,9 @@ async def test_shared_worker_ready_timeout_cancels_hung_controller_and_records_f
             finally:
                 cancelled.set()
 
+        async def close(self):
+            closed.set()
+
     monkeypatch.setattr(
         manager.worker_launcher,
         "create_shared_pool_controller",
@@ -2110,6 +2114,8 @@ async def test_shared_worker_ready_timeout_cancels_hung_controller_and_records_f
     await asyncio.wait_for(wait_for_failure(), timeout=1)
 
     assert cancelled.is_set()
+    assert closed.is_set()
+    assert worker.controller is None
     assert manager.restart_coordinator.snapshot()["active_launches"] == 0
     assert manager.installation_runtimes[binding].state == "failed"
     await manager._stop_shared_worker(worker)

{langbot_plugin-0.6.3 → langbot_plugin-0.6.5}/tests/runtime/plugin/test_worker_launcher.py

@@ -277,6 +277,19 @@ def test_shared_launcher_absolutizes_runtime_paths_before_fixed_root_cwd(
     assert f"{staging.tmp_path.absolute()}:/tmp" in dependency_args
 
 
+def test_shared_pool_controller_transfers_process_lifetime_ownership(tmp_path):
+    launcher = PluginWorkerLauncher(
+        nsjail_path="/usr/bin/nsjail",
+        cgroup_v2_available=True,
+        platform="linux",
+    )
+    launcher.configure(_policy(), "shared")
+
+    controller = launcher.create_shared_pool_controller(_launch_spec(tmp_path))
+
+    assert getattr(controller, "close_process_on_callback_return") is False
+
+
 def test_shared_launcher_fails_without_nsjail_or_required_cgroup():
     no_nsjail = PluginWorkerLauncher(
         nsjail_path="",

langbot_plugin-0.6.3/src/langbot_plugin/version.py

@@ -1 +0,0 @@
-__version__ = "0.6.3"