diff --git a/src/hive_cli/config.py b/src/hive_cli/config.py index 6ecf338..6352230 100644 --- a/src/hive_cli/config.py +++ b/src/hive_cli/config.py @@ -199,6 +199,11 @@ class HiveConfig(BaseModel): description="The name of the project. Must be all lowercase.", ) + tags: Optional[list[str]] = Field( + default=None, + description="Tags to apply to the experiment, e.g. ['production', 'test'].", + ) + token_path: str = Field( default=os.path.expandvars("$HOME/.kube/config"), description="Path to the auth token file, default to ~/.kube/config", diff --git a/src/hive_cli/platform/k8s.py b/src/hive_cli/platform/k8s.py index bdf76df..7f7beac 100644 --- a/src/hive_cli/platform/k8s.py +++ b/src/hive_cli/platform/k8s.py @@ -290,6 +290,7 @@ def construct_experiment(name: str, namespace: str, config: HiveConfig) -> dict: }, "spec": { "projectName": config.project_name, + "tags": config.tags, "coordinatorConfigName": config.coordinator_config_name, "sandbox": { "image": config.sandbox.image,