use click to get package version

Signed-off-by: Harshith-umesh <[email protected]>

Author: Harshith-umesh

src/guidellm/__main__.py

@@ -19,33 +19,13 @@
 from guidellm.utils import DefaultGroupHandler
 from guidellm.utils import cli as cli_tools
 
-# Import version information
-try:
-    from guidellm.version import version
-except ImportError:
-    version = "unknown"
-
 STRATEGY_PROFILE_CHOICES = list(
     set(list(get_args(ProfileType)) + list(get_args(StrategyType)))
 )
 
 
-def _version_callback(ctx: click.Context, _param: click.Parameter, value: bool) -> None:
-    """Callback for --version flag."""
-    if value:
-        click.echo(f"guidellm version: {version}")
-        ctx.exit()
-
-
 @click.group()
-@click.option(
-    "--version",
-    is_flag=True,
-    expose_value=False,
-    is_eager=True,
-    help="Show the version and exit.",
-    callback=_version_callback,
-)
+@click.version_option(package_name="guidellm", message="guidellm version: %(version)s")
 def cli():
     pass
 

tests/unit/test_cli.py

@@ -83,21 +83,23 @@ def test_version_flag_case_sensitivity():
 
 @pytest.mark.integration
 def test_version_integration_with_actual_version():
-    """Integration test to verify version matches what's in version.py."""
+    """Integration test to verify version matches importlib.metadata."""
+    import importlib.metadata
+
     try:
-        from guidellm.version import (
-            version as actual_version,
-        )
+        actual_version = importlib.metadata.version("guidellm")
 
         runner = CliRunner()
         result = runner.invoke(cli, ["--version"])
 
         assert result.exit_code == 0
         expected_output = f"guidellm version: {actual_version}"
         assert expected_output in result.output
-    except ImportError:
+    except importlib.metadata.PackageNotFoundError:
+        # If package is not installed, the CLI should show an error
+        # This is expected behavior when the package isn't properly installed
         runner = CliRunner()
         result = runner.invoke(cli, ["--version"])
 
-        assert result.exit_code == 0
-        assert "guidellm version: unknown" in result.output
+        # Click will handle the error when package is not found
+        assert result.exit_code != 0