From f0e86b5b0ff7bf8cc1926f6afaf04175db4402b0 Mon Sep 17 00:00:00 2001 From: Brandon Hancock Date: Tue, 3 Dec 2024 10:21:58 -0500 Subject: [PATCH] add in --version flag to cli. closes #1679. --- src/crewai/cli/cli.py | 6 +++++- tests/cli/cli_test.py | 7 +++++++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/src/crewai/cli/cli.py b/src/crewai/cli/cli.py index 600eb6142..703824798 100644 --- a/src/crewai/cli/cli.py +++ b/src/crewai/cli/cli.py @@ -26,6 +26,7 @@ from .update_crew import update_crew @click.group() +@click.version_option(pkg_resources.get_distribution("crewai").version) def crewai(): """Top-level command group for crewai.""" @@ -55,7 +56,10 @@ def create(type, name, provider, skip_provider=False): ) def version(tools): """Show the installed version of crewai.""" - crewai_version = pkg_resources.get_distribution("crewai").version + try: + crewai_version = pkg_resources.get_distribution("crewai").version + except Exception: + crewai_version = "unknown version" click.echo(f"crewai version: {crewai_version}") if tools: diff --git a/tests/cli/cli_test.py b/tests/cli/cli_test.py index 05e1cf03a..15ed81637 100644 --- a/tests/cli/cli_test.py +++ b/tests/cli/cli_test.py @@ -131,6 +131,13 @@ def test_reset_no_memory_flags(runner): ) +def test_version_flag(runner): + result = runner.invoke(version) + + assert result.exit_code == 0 + assert "crewai version:" in result.output + + def test_version_command(runner): result = runner.invoke(version)