mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +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 click
|
||||||
|
import pkg_resources
|
||||||
|
|
||||||
from .create_crew import create_crew
|
from .create_crew import create_crew
|
||||||
|
|
||||||
@@ -15,5 +16,22 @@ def create(project_name):
|
|||||||
create_crew(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__":
|
if __name__ == "__main__":
|
||||||
crewai()
|
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