langbot-plugin 0.7.0__tar.gz → 0.7.1__tar.gz

{langbot_plugin-0.7.0 → langbot_plugin-0.7.1}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [project]
 name = "langbot-plugin"
-version = "0.7.0"
+version = "0.7.1"
 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.7.0 → langbot_plugin-0.7.1}/src/langbot_plugin/cli/run/handler.py

@@ -827,6 +827,17 @@ class PluginRuntimeHandler(Handler):
             with bind_invocation(self, config={}, binding=action_context):
                 yield
             return
+        # Dedicated workers are bound to one installation during initialization;
+        # they have no shared slot. Keep their action scope usable while fencing
+        # all other installation bindings to an attached shared slot.
+        if self.bound_action_context == action_context:
+            with bind_invocation(
+                self,
+                config=self.plugin_container.plugin_config,
+                binding=action_context,
+            ):
+                yield
+            return
         slot = self._slot_containers.get(action_context.installation_uuid)
         if slot is None or slot.binding != action_context:
             raise ValueError("Shared plugin slot is not attached")

{langbot_plugin-0.7.0 → langbot_plugin-0.7.1}/tests/cli/run/test_runtime_handler.py

@@ -275,6 +275,54 @@ async def test_plugin_runtime_handler_binds_context_from_initialize_envelope():
     assert handler.bound_action_context == action_context
 
 
+async def test_dedicated_initialize_uses_bound_installation_without_shared_slot():
+    from langbot_plugin.api.proxies.invocation import invocation_capability
+
+    handler, initialized = _handler()
+    binding = InstallationBinding(
+        instance_uuid="instance-1",
+        workspace_uuid="workspace-a",
+        placement_generation=2,
+        installation_uuid="dedicated-1",
+        runtime_revision=1,
+        artifact_digest="a" * 64,
+    )
+    handler.plugin_container.plugin_config = {"dedicated": True}
+    handler.bind_action_context(binding)
+    with handler.action_invocation_scope(
+        RuntimeToPluginAction.INITIALIZE_PLUGIN.value, binding
+    ):
+        capability = invocation_capability(handler)
+        assert capability is not None
+        assert capability.binding == binding
+    async with ProtocolSession(handler) as session:
+        response = await session.request(
+            RuntimeToPluginAction.INITIALIZE_PLUGIN.value,
+            {"plugin_settings": {"enabled": True}},
+            action_context=binding,
+        )
+    assert response["code"] == 0
+    assert initialized == [{"enabled": True}]
+    assert handler._slot_containers == {}
+
+
+async def test_unattached_shared_slot_remains_rejected():
+    handler, _initialized = _handler()
+    binding = InstallationBinding(
+        instance_uuid="instance-1",
+        workspace_uuid="workspace-a",
+        placement_generation=2,
+        installation_uuid="shared-1",
+        runtime_revision=1,
+        artifact_digest="a" * 64,
+    )
+    with pytest.raises(ValueError, match="Shared plugin slot is not attached"):
+        with handler.action_invocation_scope(
+            RuntimeToPluginAction.INITIALIZE_PLUGIN.value, binding
+        ):
+            pass
+
+
 async def test_attach_scope_revokes_authority_copied_by_initialize_task():
     from langbot_plugin.api.proxies.invocation import invocation_capability