Algorithm Plugin SDK

算法仓库只实现一个 Algorithm 类并声明 entry point。SDK 统一提供单次运行、 WebUI、HTTP service、release manifest、compute 注册和心跳。算法仓库不需要 Dockerfile 或 Web 框架代码。

算法侧最小接口

from algorithm_plugin_sdk import Algorithm

class MyAlgorithm(Algorithm):
    @classmethod
    def metadata(cls): ...
    def execute(self, request, context): ...

算法包需要声明 algorithm_plugin_sdk.algorithms entry point。同一个 execute() 会被所有运行方式复用。

单次运行

algorithm-plugin run camera-space-mano \
  --input '{"input_dataset":"/data/input1","output":"/data/output1"}' \
  --input '{"input_dataset":"/data/input2","output":"/data/output2"}' \
  --gpu-ids 0 \
  --parameters '{"batch_size":16}'

--gpu-ids 和 --parameters 可省略，也可用 --request-file 读取完整请求 JSON。

配置现有仓库

目标算法仓库必须满足：

• 仓库本身是 Git 工作树根目录，并且至少有一个 commit。
• 仓库包含自己的 .venv。
• <repository>/.venv/bin/algorithm-plugin 存在且可执行。
• 该启动器的 run --help 和 serve --help 均可正常加载。

建议将生成的 release-manifest.json 和 algorithm-plugin.json 加入算法仓库的 .gitignore，不要手动编辑或提交它们。

通过目标仓库的虚拟环境执行：

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 \
  --cluster production \
  --namespace camera-space-mano \
  --node-name 10.0.0.4 \
  --gpu-ids 0,1

service name 默认使用仓库目录名，也可以通过 --service-name 指定。 重复执行 configure 需要显式提供 --force。

确认输出后执行：

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

serve

/srv/camera-space-mano/.venv/bin/algorithm-plugin serve \
  --config /srv/camera-space-mano/algorithm-plugin.json

安装与测试

python -m pip install '.[service]'
python -m unittest discover -s tests -v