Skip to content

Commit feef692

Browse files
committed
fix: update cli init test for Z.ai provider insertion
Ollama moved from position 11 to 12 after Z.ai was added. Also add missing tushare token prompt return value.
1 parent eb969e7 commit feef692

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

agent/tests/test_cli_init.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,14 +81,15 @@ def test_cmd_init_ollama_skips_api_key(self, tmp_path: Path) -> None:
8181
env_path = tmp_path / ".env"
8282

8383
with patch.object(cli, "_INIT_ENV_PATH", env_path), \
84-
patch.object(cli.IntPrompt, "ask", return_value=11), \
84+
patch.object(cli.IntPrompt, "ask", return_value=12), \
8585
patch.object(
8686
cli.Prompt,
8787
"ask",
8888
side_effect=[
8989
"http://localhost:11434/v1",
9090
"qwen2.5:32b",
9191
"",
92+
"",
9293
],
9394
):
9495
result = cli.cmd_init()

0 commit comments

Comments
 (0)