ls-algorithm-plugin-sdk 0.3.0__tar.gz → 0.3.1__tar.gz

{ls_algorithm_plugin_sdk-0.3.0 → ls_algorithm_plugin_sdk-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ls-algorithm-plugin-sdk
-Version: 0.3.0
+Version: 0.3.1
 Summary: Protocol-independent runtime SDK for dataset algorithms
 Author: Ling Robotics
 License: Proprietary
@@ -15,10 +15,6 @@ Requires-Dist: build>=1.2; extra == "dev"
 
 # Algorithm Plugin SDK
 
-算法仓库只实现一个 `Algorithm` 类并声明 entry point。SDK 统一提供单次运行、
-WebUI、HTTP service、release manifest、compute 注册和心跳。算法仓库不需要
-Dockerfile 或 Web 框架代码。
-
 ## 算法侧最小接口
 
 ```python
@@ -30,8 +26,7 @@ class MyAlgorithm(Algorithm):
     def execute(self, request, context): ...
 ```
 
-算法包需要声明 `algorithm_plugin_sdk.algorithms` entry point。同一个 `execute()`
-会被所有运行方式复用。
+算法包需要声明 `algorithm_plugin_sdk.algorithms` entry point。
 
 ## 单次运行
 
@@ -49,10 +44,9 @@ algorithm-plugin run camera-space-mano \
 
 目标算法仓库必须满足：
 
-- 仓库本身是 Git 工作树根目录，并且至少有一个 commit。
+- 仓库本身是 Git 工作树根目录。
 - 仓库包含自己的 `.venv`。
 - `<repository>/.venv/bin/algorithm-plugin` 存在且可执行。
-- 该启动器的 `run --help` 和 `serve --help` 均可正常加载。
 
 建议将生成的 `release-manifest.json` 和 `algorithm-plugin.json` 加入算法仓库的
 `.gitignore`，不要手动编辑或提交它们。
@@ -64,16 +58,11 @@ sudo /srv/camera-space-mano/.venv/bin/algorithm-plugin configure \
   --repository /srv/camera-space-mano \
   --compute-url http://compute:5180 \
   --api-key "$LDP_INTERNAL_API_KEY" \
-  --public-url http://10.0.0.4:9000 \
-  --input-root /data/jobs/input \
-  --workspace-root /data/jobs/workspace \
+  --port 9030 \
   --cluster production \
-  --namespace camera-space-mano \
-  --node-name 10.0.0.4 \
   --gpu-ids 0,1
 ```
 
-service name 默认使用仓库目录名，也可以通过 `--service-name` 指定。
 重复执行 configure 需要显式提供 `--force`。
 
 确认输出后执行：
@@ -82,7 +71,7 @@ service name 默认使用仓库目录名，也可以通过 `--service-name` 指
 sudo systemctl daemon-reload
 sudo systemctl enable camera-space-mano.service
 sudo systemctl restart camera-space-mano.service
-sudo systemctl status camera-space-mano.service
+sudo systemctl status camera-space-mano.service --no-pager
 ```
 
 ## serve
@@ -95,6 +84,6 @@ sudo systemctl status camera-space-mano.service
 ## 安装与测试
 
 ```bash
-python -m pip install '.[service]'
+python -m pip install .
 python -m unittest discover -s tests -v
 ```

{ls_algorithm_plugin_sdk-0.3.0 → ls_algorithm_plugin_sdk-0.3.1}/README.md

@@ -1,9 +1,5 @@
 # Algorithm Plugin SDK
 
-算法仓库只实现一个 `Algorithm` 类并声明 entry point。SDK 统一提供单次运行、
-WebUI、HTTP service、release manifest、compute 注册和心跳。算法仓库不需要
-Dockerfile 或 Web 框架代码。
-
 ## 算法侧最小接口
 
 ```python
@@ -15,8 +11,7 @@ class MyAlgorithm(Algorithm):
     def execute(self, request, context): ...
 ```
 
-算法包需要声明 `algorithm_plugin_sdk.algorithms` entry point。同一个 `execute()`
-会被所有运行方式复用。
+算法包需要声明 `algorithm_plugin_sdk.algorithms` entry point。
 
 ## 单次运行
 
@@ -34,10 +29,9 @@ algorithm-plugin run camera-space-mano \
 
 目标算法仓库必须满足：
 
-- 仓库本身是 Git 工作树根目录，并且至少有一个 commit。
+- 仓库本身是 Git 工作树根目录。
 - 仓库包含自己的 `.venv`。
 - `<repository>/.venv/bin/algorithm-plugin` 存在且可执行。
-- 该启动器的 `run --help` 和 `serve --help` 均可正常加载。
 
 建议将生成的 `release-manifest.json` 和 `algorithm-plugin.json` 加入算法仓库的
 `.gitignore`，不要手动编辑或提交它们。
@@ -49,16 +43,11 @@ sudo /srv/camera-space-mano/.venv/bin/algorithm-plugin configure \
   --repository /srv/camera-space-mano \
   --compute-url http://compute:5180 \
   --api-key "$LDP_INTERNAL_API_KEY" \
-  --public-url http://10.0.0.4:9000 \
-  --input-root /data/jobs/input \
-  --workspace-root /data/jobs/workspace \
+  --port 9030 \
   --cluster production \
-  --namespace camera-space-mano \
-  --node-name 10.0.0.4 \
   --gpu-ids 0,1
 ```
 
-service name 默认使用仓库目录名，也可以通过 `--service-name` 指定。
 重复执行 configure 需要显式提供 `--force`。
 
 确认输出后执行：
@@ -67,7 +56,7 @@ service name 默认使用仓库目录名，也可以通过 `--service-name` 指
 sudo systemctl daemon-reload
 sudo systemctl enable camera-space-mano.service
 sudo systemctl restart camera-space-mano.service
-sudo systemctl status camera-space-mano.service
+sudo systemctl status camera-space-mano.service --no-pager
 ```
 
 ## serve
@@ -80,6 +69,6 @@ sudo systemctl status camera-space-mano.service
 ## 安装与测试
 
 ```bash
-python -m pip install '.[service]'
+python -m pip install .
 python -m unittest discover -s tests -v
 ```

{ls_algorithm_plugin_sdk-0.3.0 → ls_algorithm_plugin_sdk-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ls-algorithm-plugin-sdk"
-version = "0.3.0"
+version = "0.3.1"
 description = "Protocol-independent runtime SDK for dataset algorithms"
 readme = "README.md"
 requires-python = ">=3.10"

{ls_algorithm_plugin_sdk-0.3.0 → ls_algorithm_plugin_sdk-0.3.1}/src/algorithm_plugin_sdk/cli_impl/configure.py

@@ -1,11 +1,14 @@
 from __future__ import annotations
 
 import argparse
+import ipaddress
 import os
 import re
+import socket
 import subprocess
 import sys
 from pathlib import Path
+from urllib.parse import urlparse
 
 from ..deployment import (
     DEFAULT_CONFIG_NAME,
@@ -17,6 +20,11 @@ from .parsing import environment, gpu_ids
 
 
 SYSTEMD_UNIT_DIR = Path("/etc/systemd/system")
+INPUT_ROOT = Path("/mnt/ldp_uploads/ldp-uploads")
+PRIVATE_IPV4_NETWORKS = tuple(
+    ipaddress.ip_network(network)
+    for network in ("10.0.0.0/8", "172.16.0.0/12", "192.168.0.0/16")
+)
 
 SERVICE_NAME_PATTERN = re.compile(
     r"^[A-Za-z0-9][A-Za-z0-9_.@-]*$"
@@ -65,18 +73,6 @@ def configure_parser(parser: argparse.ArgumentParser) -> None:
         "--api-key",
         default=os.getenv("LDP_INTERNAL_API_KEY"),
     )
-    parser.add_argument(
-        "--public-url",
-        default=os.getenv("LDP_PLUGIN_PUBLIC_URL"),
-    )
-    parser.add_argument(
-        "--input-root",
-        default=os.getenv("LDP_JOB_INPUT_ROOT"),
-    )
-    parser.add_argument(
-        "--workspace-root",
-        default=os.getenv("LDP_JOB_WORKSPACE_ROOT"),
-    )
     parser.add_argument(
         "--instance-key",
         default=os.getenv("LDP_PLUGIN_INSTANCE_KEY"),
@@ -85,14 +81,6 @@ def configure_parser(parser: argparse.ArgumentParser) -> None:
         "--cluster",
         default=os.getenv("LDP_CLUSTER"),
     )
-    parser.add_argument(
-        "--namespace",
-        default=os.getenv("POD_NAMESPACE"),
-    )
-    parser.add_argument(
-        "--node-name",
-        default=os.getenv("NODE_NAME"),
-    )
     parser.add_argument(
         "--local",
         action="store_true",
@@ -105,10 +93,6 @@ def configure_parser(parser: argparse.ArgumentParser) -> None:
         default=[],
         metavar="NAME=VALUE",
     )
-    parser.add_argument(
-        "--service-name",
-        help="systemd unit name without .service; defaults to repository name",
-    )
     parser.add_argument(
         "--force",
         action="store_true",
@@ -119,33 +103,35 @@ def configure_parser(parser: argparse.ArgumentParser) -> None:
 
 def resolve_service_name(
     repository: Path,
-    requested: str | None,
 ) -> str:
-    name = requested or repository.name
+    name = repository.name
     if name.endswith(".service"):
         raise ValueError(
-            "service name must not include the .service suffix"
+            "repository name must not include the .service suffix"
         )
     if not SERVICE_NAME_PATTERN.fullmatch(name):
         raise ValueError(
-            "service name may contain only letters, numbers, "
+            "repository name may contain only letters, numbers, "
             "underscore, dot, @, and hyphen"
         )
     return name
 
 
-def _quote_systemd_path(value: Path) -> str:
+def _escape_systemd_path(value: Path) -> str:
     text = str(value)
     if "\n" in text or "\r" in text or "\0" in text:
         raise ValueError(
             f"systemd path contains control characters: {value}"
         )
-    escaped = (
+    return (
         text.replace("\\", "\\\\")
         .replace('"', '\\"')
         .replace("%", "%%")
     )
-    return f'"{escaped}"'
+
+
+def _quote_systemd_path(value: Path) -> str:
+    return f'"{_escape_systemd_path(value)}"'
 
 
 def render_systemd_unit(
@@ -167,7 +153,7 @@ def render_systemd_unit(
             "[Service]",
             "Type=simple",
             "User=root",
-            f"WorkingDirectory={_quote_systemd_path(root)}",
+            f"WorkingDirectory={_escape_systemd_path(root)}",
             (
                 f"ExecStart={_quote_systemd_path(launcher)} "
                 f"serve --config {_quote_systemd_path(config)}"
@@ -317,6 +303,35 @@ def resolve_config_path(
     return path.resolve()
 
 
+def resolve_private_ipv4(compute_url: str) -> str:
+    parsed = urlparse(compute_url)
+    if parsed.scheme not in {"http", "https"} or not parsed.hostname:
+        raise ValueError("compute URL must be an http or https URL")
+    try:
+        port = parsed.port or (443 if parsed.scheme == "https" else 80)
+    except ValueError as exc:
+        raise ValueError("compute URL has an invalid port") from exc
+
+    try:
+        with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as probe:
+            probe.connect((parsed.hostname, port))
+            value = probe.getsockname()[0]
+    except OSError as exc:
+        raise RuntimeError(
+            f"cannot resolve the local IP used to reach {parsed.hostname}: {exc}"
+        ) from exc
+
+    address = ipaddress.ip_address(value)
+    if not isinstance(address, ipaddress.IPv4Address) or not any(
+        address in network for network in PRIVATE_IPV4_NETWORKS
+    ):
+        raise RuntimeError(
+            f"the local IP used to reach {parsed.hostname} is not a private IPv4: "
+            f"{address}"
+        )
+    return str(address)
+
+
 def registration_from_args(
     args: argparse.Namespace,
     repository: Path,
@@ -327,12 +342,9 @@ def registration_from_args(
     supplied = {
         "computeUrl": args.compute_url,
         "apiKey": args.api_key,
-        "publicUrl": args.public_url,
-        "inputRoot": args.input_root,
-        "workspaceRoot": args.workspace_root,
+        "inputRoot": str(INPUT_ROOT),
+        "workspaceRoot": str(repository.resolve()),
         "cluster": args.cluster,
-        "namespace": args.namespace,
-        "nodeName": args.node_name,
     }
     missing = [
         name
@@ -350,6 +362,9 @@ def registration_from_args(
         name: str(value)
         for name, value in supplied.items()
     }
+    private_ipv4 = resolve_private_ipv4(registration["computeUrl"])
+    registration["publicUrl"] = f"http://{private_ipv4}:{args.port}"
+    registration["nodeName"] = private_ipv4
     registration["instanceKey"] = stable_instance_key(
         repository / ".algorithm-plugin" / "instance-key",
         requested=args.instance_key,
@@ -383,10 +398,7 @@ def execute(args: argparse.Namespace) -> int:
     require_root()
     repository = validate_repository(args.repository)
     validate_configure_runtime(repository)
-    service_name = resolve_service_name(
-        repository,
-        args.service_name,
-    )
+    service_name = resolve_service_name(repository)
     config_path = resolve_config_path(
         repository,
         args.config_output,
@@ -418,4 +430,5 @@ def execute(args: argparse.Namespace) -> int:
     print("  sudo systemctl daemon-reload")
     print(f"  sudo systemctl enable {service_name}.service")
     print(f"  sudo systemctl restart {service_name}.service")
-    return 0
+    print(f"  sudo systemctl status {service_name}.service --no-pager")
+    return 0

{ls_algorithm_plugin_sdk-0.3.0 → ls_algorithm_plugin_sdk-0.3.1}/src/algorithm_plugin_sdk/deployment.py

@@ -154,6 +154,8 @@ class DeploymentConfig:
             instance,
             repository=repository,
         )
+        if registration is not None:
+            registration = {**registration, "namespace": release.name}
         release_path = release.save(
             repository / "release-manifest.json"
         )

{ls_algorithm_plugin_sdk-0.3.0 → ls_algorithm_plugin_sdk-0.3.1}/src/ls_algorithm_plugin_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ls-algorithm-plugin-sdk
-Version: 0.3.0
+Version: 0.3.1
 Summary: Protocol-independent runtime SDK for dataset algorithms
 Author: Ling Robotics
 License: Proprietary
@@ -15,10 +15,6 @@ Requires-Dist: build>=1.2; extra == "dev"
 
 # Algorithm Plugin SDK
 
-算法仓库只实现一个 `Algorithm` 类并声明 entry point。SDK 统一提供单次运行、
-WebUI、HTTP service、release manifest、compute 注册和心跳。算法仓库不需要
-Dockerfile 或 Web 框架代码。
-
 ## 算法侧最小接口
 
 ```python
@@ -30,8 +26,7 @@ class MyAlgorithm(Algorithm):
     def execute(self, request, context): ...
 ```
 
-算法包需要声明 `algorithm_plugin_sdk.algorithms` entry point。同一个 `execute()`
-会被所有运行方式复用。
+算法包需要声明 `algorithm_plugin_sdk.algorithms` entry point。
 
 ## 单次运行
 
@@ -49,10 +44,9 @@ algorithm-plugin run camera-space-mano \
 
 目标算法仓库必须满足：
 
-- 仓库本身是 Git 工作树根目录，并且至少有一个 commit。
+- 仓库本身是 Git 工作树根目录。
 - 仓库包含自己的 `.venv`。
 - `<repository>/.venv/bin/algorithm-plugin` 存在且可执行。
-- 该启动器的 `run --help` 和 `serve --help` 均可正常加载。
 
 建议将生成的 `release-manifest.json` 和 `algorithm-plugin.json` 加入算法仓库的
 `.gitignore`，不要手动编辑或提交它们。
@@ -64,16 +58,11 @@ sudo /srv/camera-space-mano/.venv/bin/algorithm-plugin configure \
   --repository /srv/camera-space-mano \
   --compute-url http://compute:5180 \
   --api-key "$LDP_INTERNAL_API_KEY" \
-  --public-url http://10.0.0.4:9000 \
-  --input-root /data/jobs/input \
-  --workspace-root /data/jobs/workspace \
+  --port 9030 \
   --cluster production \
-  --namespace camera-space-mano \
-  --node-name 10.0.0.4 \
   --gpu-ids 0,1
 ```
 
-service name 默认使用仓库目录名，也可以通过 `--service-name` 指定。
 重复执行 configure 需要显式提供 `--force`。
 
 确认输出后执行：
@@ -82,7 +71,7 @@ service name 默认使用仓库目录名，也可以通过 `--service-name` 指
 sudo systemctl daemon-reload
 sudo systemctl enable camera-space-mano.service
 sudo systemctl restart camera-space-mano.service
-sudo systemctl status camera-space-mano.service
+sudo systemctl status camera-space-mano.service --no-pager
 ```
 
 ## serve
@@ -95,6 +84,6 @@ sudo systemctl status camera-space-mano.service
 ## 安装与测试
 
 ```bash
-python -m pip install '.[service]'
+python -m pip install .
 python -m unittest discover -s tests -v
 ```

{ls_algorithm_plugin_sdk-0.3.0 → ls_algorithm_plugin_sdk-0.3.1}/tests/test_configure.py

@@ -14,6 +14,7 @@ from algorithm_plugin_sdk.cli_impl.configure import (
     generate_config,
     registration_from_args,
     render_systemd_unit,
+    resolve_private_ipv4,
     resolve_service_name,
     validate_configure_runtime,
     validate_repository,
@@ -142,7 +143,6 @@ class ConfigureTests(unittest.TestCase):
                 "--gpu-ids", "2,0",
                 "--service-token", "secret",
                 "--env", "DEMO_MODE=fast",
-                "--service-name", "demo-service",
                 "--force",
             ]
         )
@@ -157,9 +157,22 @@ class ConfigureTests(unittest.TestCase):
         self.assertEqual(args.gpu_ids, [2, 0])
         self.assertEqual(args.service_token, "secret")
         self.assertEqual(args.env, [("DEMO_MODE", "fast")])
-        self.assertEqual(args.service_name, "demo-service")
         self.assertTrue(args.force)
 
+    def test_rejects_derived_configure_options(self) -> None:
+        for option in (
+            "--input-root",
+            "--workspace-root",
+            "--public-url",
+            "--namespace",
+            "--node-name",
+            "--service-name",
+        ):
+            with self.subTest(option=option), self.assertRaises(SystemExit):
+                build_parser().parse_args(
+                    ["configure", option, "custom-value"]
+                )
+
     def test_generates_and_saves_config_for_the_existing_repository(self) -> None:
         repository = Path("/srv/demo").resolve()
         args = build_parser().parse_args(
@@ -202,17 +215,13 @@ class ConfigureTests(unittest.TestCase):
 
         with self.assertRaisesRegex(
             ValueError,
-            "computeUrl.*apiKey.*publicUrl",
+            "computeUrl.*apiKey.*cluster",
         ):
             registration_from_args(args, Path("/srv/demo"))
 
     def test_builds_registration_and_reuses_instance_key(self) -> None:
         with tempfile.TemporaryDirectory() as directory:
             repository = Path(directory)
-            input_root = repository / "input"
-            workspace_root = repository / "workspace"
-            input_root.mkdir()
-            workspace_root.mkdir()
             args = build_parser().parse_args(
                 [
                     "configure",
@@ -220,26 +229,30 @@ class ConfigureTests(unittest.TestCase):
                     "http://compute:5180",
                     "--api-key",
                     "secret",
-                    "--public-url",
-                    "http://127.0.0.1:9000",
-                    "--input-root",
-                    str(input_root),
-                    "--workspace-root",
-                    str(workspace_root),
                     "--cluster",
                     "production",
-                    "--namespace",
-                    "demo",
-                    "--node-name",
-                    "node-1",
+                    "--port",
+                    "9030",
                 ]
             )
 
-            first = registration_from_args(args, repository)
-            second = registration_from_args(args, repository)
+            with patch(
+                "algorithm_plugin_sdk.cli_impl.configure.resolve_private_ipv4",
+                return_value="172.21.32.10",
+            ):
+                first = registration_from_args(args, repository)
+                second = registration_from_args(args, repository)
 
             self.assertEqual(first, second)
             self.assertEqual(first["computeUrl"], "http://compute:5180")
+            self.assertEqual(first["nodeName"], "172.21.32.10")
+            self.assertEqual(first["publicUrl"], "http://172.21.32.10:9030")
+            self.assertEqual(
+                first["inputRoot"],
+                "/mnt/ldp_uploads/ldp-uploads",
+            )
+            self.assertEqual(first["workspaceRoot"], str(repository.resolve()))
+            self.assertNotIn("namespace", first)
             self.assertTrue(first["instanceKey"])
             self.assertEqual(
                 (
@@ -250,24 +263,48 @@ class ConfigureTests(unittest.TestCase):
                 first["instanceKey"],
             )
 
+    def test_resolves_private_ipv4_from_compute_route(self) -> None:
+        probe = Mock()
+        probe.__enter__ = Mock(return_value=probe)
+        probe.__exit__ = Mock(return_value=None)
+        probe.getsockname.return_value = ("172.21.32.10", 42000)
+
+        with patch(
+            "algorithm_plugin_sdk.cli_impl.configure.socket.socket",
+            return_value=probe,
+        ):
+            actual = resolve_private_ipv4("http://compute.internal:5180")
+
+        self.assertEqual(actual, "172.21.32.10")
+        probe.connect.assert_called_once_with(("compute.internal", 5180))
+
+    def test_rejects_non_private_route_ipv4(self) -> None:
+        probe = Mock()
+        probe.__enter__ = Mock(return_value=probe)
+        probe.__exit__ = Mock(return_value=None)
+        probe.getsockname.return_value = ("62.234.57.176", 42000)
+
+        with (
+            patch(
+                "algorithm_plugin_sdk.cli_impl.configure.socket.socket",
+                return_value=probe,
+            ),
+            self.assertRaisesRegex(RuntimeError, "not a private IPv4"),
+        ):
+            resolve_private_ipv4("https://compute.example.com")
+
     def test_defaults_service_name_to_repository_directory(self) -> None:
         repository = Path("/srv/demo-algorithm")
 
         self.assertEqual(
-            resolve_service_name(repository, None),
+            resolve_service_name(repository),
             "demo-algorithm",
         )
-        self.assertEqual(
-            resolve_service_name(repository, "custom-name"),
-            "custom-name",
-        )
-
-    def test_rejects_unsafe_or_suffixed_service_names(self) -> None:
-        repository = Path("/srv/demo")
 
-        for name in ("bad name", "../demo", "demo.service"):
+    def test_rejects_unsafe_repository_names(self) -> None:
+        for name in ("bad name", "..", "demo.service"):
             with self.subTest(name=name), self.assertRaises(ValueError):
-                resolve_service_name(repository, name)
+                resolve_service_name(Path("/srv") / name)
 
     def test_renders_root_service_from_repository_virtualenv(self) -> None:
         unit = render_systemd_unit(
@@ -286,7 +323,7 @@ class ConfigureTests(unittest.TestCase):
             "[Service]\n"
             "Type=simple\n"
             "User=root\n"
-            'WorkingDirectory="/srv/demo algorithm"\n'
+            "WorkingDirectory=/srv/demo algorithm\n"
             'ExecStart="/srv/demo algorithm/.venv/bin/algorithm-plugin" '
             'serve --config "/srv/demo algorithm/algorithm-plugin.json"\n'
             "Environment=PYTHONUNBUFFERED=1\n"
@@ -432,8 +469,6 @@ class ConfigureTests(unittest.TestCase):
                 "configure",
                 "--repository",
                 str(repository),
-                "--service-name",
-                "demo",
                 "--local",
             ]
         )
@@ -483,7 +518,8 @@ class ConfigureTests(unittest.TestCase):
             "Next commands:\n"
             "  sudo systemctl daemon-reload\n"
             "  sudo systemctl enable demo.service\n"
-            "  sudo systemctl restart demo.service\n",
+            "  sudo systemctl restart demo.service\n"
+            "  sudo systemctl status demo.service --no-pager\n",
         )
 
     def test_execute_rejects_unit_conflict_before_config_generation(self) -> None:
@@ -493,8 +529,6 @@ class ConfigureTests(unittest.TestCase):
                 "configure",
                 "--repository",
                 str(repository),
-                "--service-name",
-                "demo",
                 "--local",
             ]
         )
@@ -529,4 +563,4 @@ class ConfigureTests(unittest.TestCase):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    unittest.main()

{ls_algorithm_plugin_sdk-0.3.0 → ls_algorithm_plugin_sdk-0.3.1}/tests/test_deployment.py

@@ -95,6 +95,44 @@ class DeploymentConfigTests(unittest.TestCase):
             )
             self.assertEqual(stat.S_IMODE(config.path.stat().st_mode), 0o600)
 
+    def test_uses_metadata_name_as_registration_namespace(self) -> None:
+        with tempfile.TemporaryDirectory() as directory:
+            root = Path(directory)
+            self._repository(root)
+            registration = {
+                "computeUrl": "http://compute:5180",
+                "apiKey": "secret",
+                "publicUrl": "http://127.0.0.1:9000",
+                "inputRoot": str(root),
+                "workspaceRoot": str(root),
+                "cluster": "production",
+                "namespace": "ignored",
+                "nodeName": "node-1",
+            }
+
+            with patch(
+                "algorithm_plugin_sdk.deployment.load_algorithm",
+                return_value=FakeAlgorithm(),
+            ):
+                config = DeploymentConfig.generate(
+                    output=root / "algorithm-plugin.json",
+                    root=root,
+                    algorithm=None,
+                    host="0.0.0.0",
+                    port=9000,
+                    webui=True,
+                    max_concurrency=1,
+                    scratch_dir=None,
+                    token=None,
+                    registration=registration,
+                )
+
+            self.assertEqual(
+                config.payload["registration"]["namespace"],
+                "demo-algorithm",
+            )
+            self.assertEqual(registration["namespace"], "ignored")
+
     def test_repository_is_importable_during_generate_and_load(self) -> None:
         with tempfile.TemporaryDirectory() as directory:
             root = Path(directory)