From 3592a690ab089e239462ed160c9f37a92e2d823b Mon Sep 17 00:00:00 2001 From: Florent Benoit Date: Thu, 13 Feb 2025 19:52:47 +0100 Subject: [PATCH] feat: add ramalama labels about the execution on top of container fixes https://github.com/containers/ramalama/issues/798 Signed-off-by: Florent Benoit --- ramalama/model.py | 33 ++++++++++++++++++++++++++------- 1 file changed, 26 insertions(+), 7 deletions(-) diff --git a/ramalama/model.py b/ramalama/model.py index 20035e42..a91dd8dc 100644 --- a/ramalama/model.py +++ b/ramalama/model.py @@ -157,16 +157,35 @@ def setup_container(self, args): "run", "--rm", "-i", + ] + + container_labels = [ "--label", "RAMALAMA", - "--security-opt=label=disable", - "--name", - name, - "--env=HOME=/tmp", - "--cap-drop=all", - "--security-opt=no-new-privileges", - "--init", ] + if hasattr(args, "MODEL"): + container_labels += ["--label", f"ai.ramalama.model={args.MODEL}"] + if hasattr(args, "engine"): + container_labels += ["--label", f"ai.ramalama.engine={args.engine}"] + if hasattr(args, "runtime"): + container_labels += ["--label", f"ai.ramalama.runtime={args.runtime}"] + if hasattr(args, "port"): + container_labels += ["--label", f"ai.ramalama.port={args.port}"] + if hasattr(args, "subcommand"): + container_labels += ["--label", f"ai.ramalama.command={args.subcommand}"] + conman_args.extend(container_labels) + + conman_args.extend( + [ + "--security-opt=label=disable", + "--name", + name, + "--env=HOME=/tmp", + "--cap-drop=all", + "--security-opt=no-new-privileges", + "--init", + ] + ) if os.path.basename(args.engine) == "podman": conman_args += ["--pull=newer"]