From 94166f382d47c184f2435a3f6a1b0b3bb414a7ec 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 | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/ramalama/model.py b/ramalama/model.py index 1e3a6a45..768bf7d5 100644 --- a/ramalama/model.py +++ b/ramalama/model.py @@ -168,6 +168,19 @@ def setup_container(self, args): "--init", ] + container_labels = [] + 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) + if os.path.basename(args.engine) == "podman": conman_args += ["--pull=newer"] if args.podman_keep_groups: