|
31 | 31 |
|
32 | 32 | def test_cli_version(capfd: CaptureFixture[str]): |
33 | 33 | assert cli(['--version']) == 0 |
34 | | - assert capfd.readouterr().out.startswith('pai - Pydantic AI CLI') |
| 34 | + assert capfd.readouterr().out.startswith('clai - Pydantic AI CLI') |
35 | 35 |
|
36 | 36 |
|
37 | 37 | def test_invalid_model(capfd: CaptureFixture[str]): |
@@ -141,7 +141,7 @@ def test_no_command_defaults_to_chat(mocker: MockerFixture): |
141 | 141 | def test_list_models(capfd: CaptureFixture[str]): |
142 | 142 | assert cli(['--list-models']) == 0 |
143 | 143 | output = capfd.readouterr().out.splitlines() |
144 | | - assert output[:3] == snapshot([IsStr(regex='pai - Pydantic AI CLI .*'), '', 'Available models:']) |
| 144 | + assert output[:3] == snapshot([IsStr(regex='clai - Pydantic AI CLI .*'), '', 'Available models:']) |
145 | 145 |
|
146 | 146 | providers = ( |
147 | 147 | 'openai', |
@@ -274,21 +274,21 @@ def test_code_theme_unset(mocker: MockerFixture, env: TestEnv): |
274 | 274 | env.set('OPENAI_API_KEY', 'test') |
275 | 275 | mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat') |
276 | 276 | cli([]) |
277 | | - mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai') |
| 277 | + mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'clai') |
278 | 278 |
|
279 | 279 |
|
280 | 280 | def test_code_theme_light(mocker: MockerFixture, env: TestEnv): |
281 | 281 | env.set('OPENAI_API_KEY', 'test') |
282 | 282 | mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat') |
283 | 283 | cli(['--code-theme=light']) |
284 | | - mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'default', 'pai') |
| 284 | + mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'default', 'clai') |
285 | 285 |
|
286 | 286 |
|
287 | 287 | def test_code_theme_dark(mocker: MockerFixture, env: TestEnv): |
288 | 288 | env.set('OPENAI_API_KEY', 'test') |
289 | 289 | mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat') |
290 | 290 | cli(['--code-theme=dark']) |
291 | | - mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai') |
| 291 | + mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'clai') |
292 | 292 |
|
293 | 293 |
|
294 | 294 | def test_agent_to_cli_sync(mocker: MockerFixture, env: TestEnv): |
|
0 commit comments