Skip to content

Commit ca6541f

Browse files
authored
fix: fix create name in cli (#420)
* fix(cli_createname): avoid - * fix(tests): fix - and _
1 parent e4afab8 commit ca6541f

File tree

2 files changed

+17
-13
lines changed

2 files changed

+17
-13
lines changed

tests/cli/test_cli_create.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,10 @@
2020
def test_create_agent_with_options():
2121
runner = CliRunner()
2222
with runner.isolated_filesystem() as temp_dir:
23-
result = runner.invoke(create, ["test-agent", "--ark-api-key", "test-key"])
23+
result = runner.invoke(create, ["test_agent", "--ark-api-key", "test-key"])
2424
assert result.exit_code == 0
2525

26-
agent_folder = Path(temp_dir) / "test-agent"
26+
agent_folder = Path(temp_dir) / "test_agent"
2727
assert agent_folder.exists()
2828

2929
config_path = agent_folder / ".env"
@@ -42,31 +42,31 @@ def test_create_agent_overwrite_existing_directory():
4242
runner = CliRunner()
4343
with runner.isolated_filesystem() as temp_dir:
4444
# First, create the agent
45-
runner.invoke(create, ["test-agent", "--ark-api-key", "test-key"])
45+
runner.invoke(create, ["test_agent", "--ark-api-key", "test-key"])
4646

4747
# Attempt to create it again, but cancel the overwrite
4848
result = runner.invoke(
4949
create,
50-
["test-agent", "--ark-api-key", "test-key"],
50+
["test_agent", "--ark-api-key", "test-key"],
5151
input="n\n",
5252
)
5353
assert "Operation cancelled" in result.output
5454

5555
# Attempt to create it again, and confirm the overwrite
5656
result = runner.invoke(
5757
create,
58-
["test-agent", "--ark-api-key", "new-key"],
58+
["test_agent", "--ark-api-key", "new-key"],
5959
input="y\n",
6060
)
6161
assert result.exit_code == 0
62-
agent_folder = Path(temp_dir) / "test-agent"
62+
agent_folder = Path(temp_dir) / "test_agent"
6363
config_path = agent_folder / ".env"
6464
config_content = config_path.read_text()
6565
assert "MODEL_AGENT_API_KEY=new-key" in config_content
6666

6767

6868
def test_generate_files(tmp_path: Path):
69-
agent_name = "test-agent"
69+
agent_name = "test_agent"
7070
api_key = "test-key"
7171
target_dir = tmp_path / agent_name
7272

@@ -87,26 +87,26 @@ def test_generate_files(tmp_path: Path):
8787
def test_prompt_for_ark_api_key_enter_now():
8888
runner = CliRunner()
8989
with runner.isolated_filesystem():
90-
result = runner.invoke(create, input="test-agent\n1\nmy-secret-key\n")
90+
result = runner.invoke(create, input="test_agent\n1\nmy-secret-key\n")
9191
assert result.exit_code == 0
92-
assert "my-secret-key" in (Path("test-agent") / ".env").read_text()
92+
assert "my-secret-key" in (Path("test_agent") / ".env").read_text()
9393

9494

9595
def test_prompt_for_ark_api_key_configure_later():
9696
runner = CliRunner()
9797
with runner.isolated_filesystem():
98-
result = runner.invoke(create, input="test-agent\n2\n")
98+
result = runner.invoke(create, input="test_agent\n2\n")
9999
assert result.exit_code == 0
100-
assert "MODEL_AGENT_API_KEY=" in (Path("test-agent") / ".env").read_text()
100+
assert "MODEL_AGENT_API_KEY=" in (Path("test_agent") / ".env").read_text()
101101

102102

103103
def test_create_agent_with_prompts():
104104
runner = CliRunner()
105105
with runner.isolated_filesystem() as temp_dir:
106-
result = runner.invoke(create, input="test-agent\n1\ntest-key\n")
106+
result = runner.invoke(create, input="test_agent\n1\ntest-key\n")
107107
assert result.exit_code == 0
108108

109-
agent_folder = Path(temp_dir) / "test-agent"
109+
agent_folder = Path(temp_dir) / "test_agent"
110110
assert agent_folder.exists()
111111

112112
config_path = agent_folder / ".env"

veadk/cli/cli_create.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,10 @@ def create(agent_name: str, ark_api_key: str) -> None:
147147
"""
148148
if not agent_name:
149149
agent_name = click.prompt("Enter the agent name")
150+
151+
if "-" in agent_name:
152+
raise ValueError("Agent name cannot contain '-'. Use '_' instead.")
153+
150154
if not ark_api_key:
151155
ark_api_key = _prompt_for_ark_api_key()
152156

0 commit comments

Comments
 (0)