Fix dynamic flag binding and bump version to 0.0.5

Author: luocfprime

envlit/cli.py

@@ -120,12 +120,14 @@ def parse_args(self, ctx: click.Context, args: list):
 
                         # Get flag aliases
                         flag_aliases = flag_config.get("flag", [f"--{flag_name}"])
+                        if isinstance(flag_aliases, str):
+                            flag_aliases = [flag_aliases]
                         default_value = flag_config.get("default", None)
                         target = flag_config.get("target", flag_name.upper())
 
                         # Add option to command
                         option = click.Option(
-                            param_decls=flag_aliases,
+                            param_decls=[*flag_aliases, flag_name],
                             default=default_value,
                             help=f"Set {target} (default: {default_value})",
                         )

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "envlit"
-version = "0.0.4"
+version = "0.0.5"
 description = "A minimal CLI tool to organize, load, and switch between your project's environment contexts."
 authors = [{ name = "Chaofan Luo" }]
 readme = "README.md"

tests/test_cli_init.py

@@ -166,3 +166,42 @@ def test_init_includes_error_handling(runner):
     assert "if envlit unload" in output
     assert "else" in output
     assert "Error: Failed to generate" in output
+
+
+def test_load_binds_dynamic_flag_value_to_config_key(runner, tmp_path):
+    """Test dynamic flags still work when config key differs from long option name."""
+    config_file = tmp_path / "devices.yaml"
+    config_file.write_text("""
+flags:
+  num_devices:
+    flag: ["--num"]
+    default: "8"
+    target: "FSDP_DEVICES"
+hooks:
+  post_load:
+    - name: "Show devices"
+      script: "echo $FSDP_DEVICES"
+""")
+
+    result = runner.invoke(cli, ["load", "--config", str(config_file), "--num", "4"])
+
+    assert result.exit_code == 0
+    assert 'export FSDP_DEVICES="4"' in result.output
+    assert "echo $FSDP_DEVICES" in result.output
+
+
+def test_load_preserves_existing_dynamic_flag_behavior(runner, tmp_path):
+    """Test dynamic flags continue to work when config key matches option name."""
+    config_file = tmp_path / "cuda.yaml"
+    config_file.write_text("""
+flags:
+  cuda:
+    flag: ["--cuda", "-g"]
+    default: "0"
+    target: "CUDA_VISIBLE_DEVICES"
+""")
+
+    result = runner.invoke(cli, ["load", "--config", str(config_file), "-g", "0,1"])
+
+    assert result.exit_code == 0
+    assert 'export CUDA_VISIBLE_DEVICES="0,1"' in result.output