langbot-plugin 0.6.8__tar.gz → 0.6.9__tar.gz

{langbot_plugin-0.6.8 → langbot_plugin-0.6.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: langbot-plugin
-Version: 0.6.8
+Version: 0.6.9
 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.8 → langbot_plugin-0.6.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langbot-plugin"
-version = "0.6.8"
+version = "0.6.9"
 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.8 → langbot_plugin-0.6.9}/src/langbot_plugin/runtime/plugin/mgr.py

@@ -149,6 +149,9 @@ class SharedPluginWorkerRuntime:
         default_factory=dict
     )
     plugin_handler: runtime_plugin_handler_cls.PluginConnectionHandler | None = None
+    pending_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)
@@ -552,12 +555,23 @@ class PluginManager:
         *,
         plugin_author: str,
         plugin_name: str,
+        expected_shared_handler: runtime_plugin_handler_cls.PluginConnectionHandler
+        | None = None,
     ) -> _PendingPluginRegistration:
         key = self._find_pending_registration_key(capability)
         if key is None:
             raise ValueError(
                 "Plugin registration capability is invalid or already used"
             )
+        registration = self._pending_registrations[key]
+        if registration.shared_pool_digest is not None:
+            worker = self._shared_workers.get(registration.shared_pool_digest)
+            if (
+                expected_shared_handler is None
+                or worker is None
+                or worker.pending_plugin_handler is not expected_shared_handler
+            ):
+                raise ValueError("Shared plugin worker registration transport changed")
         registration = self._pending_registrations.pop(key)
         if (
             registration.plugin_author != plugin_author
@@ -1967,6 +1981,12 @@ class PluginManager:
             worker.controller = controller
 
             async def new_plugin_connection_callback(connection: Connection):
+                if (
+                    worker.pending_plugin_handler is not None
+                    or worker.plugin_handler is not None
+                ):
+                    await connection.close()
+                    return
                 shared_transfer_root = (
                     self.artifact_store.base_path
                     / "shared-transfers"
@@ -1991,12 +2011,7 @@ class PluginManager:
                         self.context.worker_policy.max_file_size_mb * 1024 * 1024
                     ),
                 )
-                if (
-                    worker.plugin_handler is not None
-                    and worker.plugin_handler is not plugin_handler
-                ):
-                    await connection.close()
-                    return
+                worker.pending_plugin_handler = plugin_handler
                 self.plugin_handlers.append(plugin_handler)
                 try:
                     await plugin_handler.run()
@@ -2172,6 +2187,9 @@ class PluginManager:
         if handler in self.plugin_handlers:
             self.plugin_handlers.remove(handler)
         for worker in self._shared_workers.values():
+            if worker.pending_plugin_handler is handler:
+                worker.pending_plugin_handler = None
+                return
             if worker.plugin_handler is handler:
                 async with worker.lifecycle_lock:
                     if worker.plugin_handler is not handler:
@@ -2590,6 +2608,7 @@ class PluginManager:
                 registration_capability or "",
                 plugin_author=plugin_author,
                 plugin_name=plugin_name,
+                expected_shared_handler=handler,
             )
             # From this point forward, use only the identity captured before the
             # child process was launched, never values supplied by plugin code.
@@ -2616,6 +2635,11 @@ class PluginManager:
                         registration.shared_pool_digest,
                         worker.slots,
                     )
+                    if worker.pending_plugin_handler is not handler:
+                        raise ValueError(
+                            "Shared plugin worker registration transport changed"
+                        )
+                    worker.pending_plugin_handler = None
                     worker.plugin_handler = handler
                     for slot_runtime in tuple(worker.slots.values()):
                         if (

langbot_plugin-0.6.9/src/langbot_plugin/version.py

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

{langbot_plugin-0.6.8 → langbot_plugin-0.6.9}/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.8"
-    assert source_version == '__version__ = "0.6.8"'
+    assert project["project"]["version"] == "0.6.9"
+    assert source_version == '__version__ = "0.6.9"'

{langbot_plugin-0.6.8 → langbot_plugin-0.6.9}/tests/runtime/plugin/test_installation_manager.py

@@ -441,6 +441,7 @@ async def test_failed_sole_shared_slot_retries_with_registered_handler(
 
     context.control_handler = Control()
     handler = Handler()
+    worker.pending_plugin_handler = handler
     capability = manager._issue_registration_capability(
         plugin_author="tester",
         plugin_name="demo",
@@ -448,6 +449,15 @@ async def test_failed_sole_shared_slot_retries_with_registered_handler(
         binding=binding,
         shared_pool_digest=digest,
     )
+    alt_handler = Handler()
+    with pytest.raises(ValueError, match="registration transport changed"):
+        await manager.register_plugin(
+            alt_handler,
+            container.model_dump(),
+            registration_capability=capability,
+        )
+    assert manager.is_registration_capability_pending(capability)
+    assert worker.pending_plugin_handler is handler
     await manager.register_plugin(
         handler,
         container.model_dump(),
@@ -545,6 +555,7 @@ async def test_failed_shared_slot_retry_keeps_healthy_sibling_and_worker(
 
     context.control_handler = Control()
     handler = Handler()
+    worker.pending_plugin_handler = handler
     capability = manager._issue_registration_capability(
         plugin_author="tester",
         plugin_name="demo",
@@ -771,6 +782,7 @@ async def test_shared_worker_registration_attaches_all_slots_with_exact_settings
     context.control_handler = Control()
     handler = Handler()
     worker = manager.installation_runtimes[binding_a].shared_worker
+    worker.pending_plugin_handler = handler
     capability = manager._issue_registration_capability(
         plugin_author="tester",
         plugin_name="demo",
@@ -865,6 +877,7 @@ async def test_shared_registration_waits_for_validated_slot_before_running(
 
     context.control_handler = Control()
     handler = Handler()
+    worker.pending_plugin_handler = handler
     capability = manager._issue_registration_capability(
         plugin_author="tester",
         plugin_name="demo",
@@ -972,6 +985,7 @@ async def test_shared_registration_reports_mixed_slot_attach_results(
 
     context.control_handler = Control()
     handler = Handler()
+    worker.pending_plugin_handler = handler
     capability = manager._issue_registration_capability(
         plugin_author="tester",
         plugin_name="demo",
@@ -1063,6 +1077,7 @@ async def test_shared_slot_uninitialized_container_is_failed(tmp_path, monkeypat
 
     context.control_handler = Control()
     handler = Handler()
+    worker.pending_plugin_handler = handler
     capability = manager._issue_registration_capability(
         plugin_author="tester",
         plugin_name="demo",
@@ -2132,6 +2147,7 @@ async def test_shared_connection_does_not_claim_registration_before_capability_v
     await asyncio.sleep(0)
 
     assert worker.plugin_handler is None
+    assert worker.pending_plugin_handler is handler
     assert handler in manager.plugin_handlers
 
     launch.cancel()

langbot_plugin-0.6.8/src/langbot_plugin/version.py

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