langbot-plugin 0.5.3__tar.gz → 0.5.4__tar.gz

{langbot_plugin-0.5.3 → langbot_plugin-0.5.4}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [project]
 name = "langbot-plugin"
-version = "0.5.3"
+version = "0.5.4"
 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.5.3 → langbot_plugin-0.5.4}/src/langbot_plugin/runtime/io/handler.py

@@ -34,7 +34,10 @@ from langbot_plugin.entities.io.errors import (
     ActionCallError,
 )
 from langbot_plugin.entities.io.actions.enums import ActionType, CommonAction
-from langbot_plugin.runtime.security import PLUGIN_RUNTIME_PROFILE_ENV
+from langbot_plugin.runtime.security import (
+    PLUGIN_FILE_STORAGE_DIR_ENV,
+    PLUGIN_RUNTIME_PROFILE_ENV,
+)
 from langbot_plugin.runtime.bounded_executor import blocking_work_scope
 
 logger = logging.getLogger(__name__)
@@ -124,7 +127,7 @@ class Handler(abc.ABC):
 
         if file_storage_dir is None:
             runtime_profile = os.environ.get(PLUGIN_RUNTIME_PROFILE_ENV, "oss_dev")
-            file_storage_dir = (
+            file_storage_dir = os.environ.get(PLUGIN_FILE_STORAGE_DIR_ENV) or (
                 SHARED_WORKER_FILE_STORAGE_DIR
                 if runtime_profile == "shared"
                 else FILE_STORAGE_DIR

{langbot_plugin-0.5.3 → langbot_plugin-0.5.4}/src/langbot_plugin/runtime/plugin/worker_launcher.py

@@ -28,6 +28,7 @@ from langbot_plugin.runtime.plugin.dependency_environment import (
 from langbot_plugin.runtime.helper import pkgmgr as pkgmgr_helper
 from langbot_plugin.runtime import bounded_executor
 from langbot_plugin.runtime.security import (
+    PLUGIN_FILE_STORAGE_DIR_ENV,
     PLUGIN_REGISTRATION_CAPABILITY_ENV,
     PLUGIN_RUNTIME_PROFILE_ENV,
 )
@@ -140,6 +141,9 @@ class PluginWorkerLauncher:
                 "--prod",
             ],
             env={
+                PLUGIN_FILE_STORAGE_DIR_ENV: str(
+                    launch_spec.paths.root_path / "rpc-transfer"
+                ),
                 PLUGIN_REGISTRATION_CAPABILITY_ENV: (
                     launch_spec.registration_capability
                 ),

{langbot_plugin-0.5.3 → langbot_plugin-0.5.4}/src/langbot_plugin/runtime/security.py

@@ -15,6 +15,7 @@ PLUGIN_DEBUG_KEY_HEADER = "X-LangBot-Plugin-Debug-Key"
 PLUGIN_REGISTRATION_CAPABILITY_ENV = "LANGBOT_PLUGIN_REGISTRATION_CAPABILITY"
 PLUGIN_REGISTRATION_CAPABILITY_HEADER = "X-LangBot-Plugin-Registration-Capability"
 PLUGIN_RUNTIME_PROFILE_ENV = "LANGBOT_PLUGIN_RUNTIME_PROFILE"
+PLUGIN_FILE_STORAGE_DIR_ENV = "LANGBOT_PLUGIN_FILE_STORAGE_DIR"
 RUNTIME_SECRET_MIN_LENGTH = 32
 WORKSPACE_DEBUG_TOKEN_TTL_SECONDS = 2 * 60 * 60
 

langbot_plugin-0.5.4/src/langbot_plugin/version.py

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

{langbot_plugin-0.5.3 → langbot_plugin-0.5.4}/tests/runtime/io/test_handler.py

@@ -511,6 +511,21 @@ def test_handler_file_storage_dir_is_created_for_instances(tmp_path, monkeypatch
     assert (tmp_path / "data" / "temp" / "lbp").is_dir()
 
 
+def test_managed_worker_file_storage_uses_launcher_directory(tmp_path, monkeypatch):
+    artifact = tmp_path / "artifact" / "code"
+    artifact.mkdir(parents=True)
+    artifact.chmod(0o555)
+    rpc_transfer = tmp_path / "installations" / "installation-a" / "rpc-transfer"
+    monkeypatch.chdir(artifact)
+    monkeypatch.setenv("LANGBOT_PLUGIN_FILE_STORAGE_DIR", str(rpc_transfer))
+
+    handler = Handler(QueueConnection())
+
+    assert handler.file_storage_dir == str(rpc_transfer)
+    assert rpc_transfer.is_dir()
+    assert not (artifact / "data").exists()
+
+
 def test_shared_worker_file_storage_uses_private_writable_tmp(tmp_path, monkeypatch):
     worker_tmp = tmp_path / "lbp-rpc"
     monkeypatch.setenv(PLUGIN_RUNTIME_PROFILE_ENV, "shared")

{langbot_plugin-0.5.3 → langbot_plugin-0.5.4}/tests/runtime/plugin/test_worker_launcher.py

@@ -22,6 +22,7 @@ from langbot_plugin.runtime.plugin.worker_launcher import (
     PluginWorkerLauncher,
 )
 from langbot_plugin.runtime.security import (
+    PLUGIN_FILE_STORAGE_DIR_ENV,
     PLUGIN_REGISTRATION_CAPABILITY_ENV,
     PLUGIN_RUNTIME_PROFILE_ENV,
 )
@@ -327,6 +328,7 @@ def test_launcher_builds_profile_specific_controllers(tmp_path):
     assert oss_controller.command == sys.executable
     assert oss_controller.working_dir == str(launch_spec.artifact.code_path)
     assert oss_controller.env == {
+        PLUGIN_FILE_STORAGE_DIR_ENV: str(launch_spec.paths.root_path / "rpc-transfer"),
         PLUGIN_REGISTRATION_CAPABILITY_ENV: "capability-value",
         PLUGIN_RUNTIME_PROFILE_ENV: "oss_dev",
     }

langbot_plugin-0.5.3/src/langbot_plugin/version.py

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