mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 15:48:29 +00:00
Add version command to CLI (#348)
* feat: add version command to cli with tools flag * test: check output of version and tools flag * fix: add version tool info to cli outputs
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
import click
|
||||
import pkg_resources
|
||||
|
||||
from .create_crew import create_crew
|
||||
|
||||
@@ -15,5 +16,22 @@ def create(project_name):
|
||||
create_crew(project_name)
|
||||
|
||||
|
||||
@crewai.command()
|
||||
@click.option(
|
||||
"--tools", is_flag=True, help="Show the installed version of crewai tools"
|
||||
)
|
||||
def version(tools):
|
||||
"""Show the installed version of crewai."""
|
||||
crewai_version = pkg_resources.get_distribution("crewai").version
|
||||
click.echo(f"crewai version: {crewai_version}")
|
||||
|
||||
if tools:
|
||||
try:
|
||||
tools_version = pkg_resources.get_distribution("crewai[tools]").version
|
||||
click.echo(f"crewai tools version: {tools_version}")
|
||||
except pkg_resources.DistributionNotFound:
|
||||
click.echo("crewai tools not installed")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
crewai()
|
||||
|
||||
20
tests/cli_test.py
Normal file
20
tests/cli_test.py
Normal file
@@ -0,0 +1,20 @@
|
||||
from click.testing import CliRunner
|
||||
from crewai.cli.cli import version
|
||||
|
||||
|
||||
def test_version_command():
|
||||
runner = CliRunner()
|
||||
result = runner.invoke(version)
|
||||
assert result.exit_code == 0
|
||||
assert "crewai version:" in result.output
|
||||
|
||||
|
||||
def test_version_command_with_tools():
|
||||
runner = CliRunner()
|
||||
result = runner.invoke(version, ["--tools"])
|
||||
assert result.exit_code == 0
|
||||
assert "crewai version:" in result.output
|
||||
assert (
|
||||
"crewai tools version:" in result.output
|
||||
or "crewai tools not installed" in result.output
|
||||
)
|
||||
Reference in New Issue
Block a user