Skip to content

Commit 3f41578

Browse files
committed
Fix tests
1 parent 7688cfe commit 3f41578

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test_cli.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -227,21 +227,21 @@ def test_code_theme_unset(mocker: MockerFixture, env: TestEnv):
227227
env.set('OPENAI_API_KEY', 'test')
228228
mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat')
229229
cli([])
230-
mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai')
230+
mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai', history=None)
231231

232232

233233
def test_code_theme_light(mocker: MockerFixture, env: TestEnv):
234234
env.set('OPENAI_API_KEY', 'test')
235235
mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat')
236236
cli(['--code-theme=light'])
237-
mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'default', 'pai')
237+
mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'default', 'pai', history=None)
238238

239239

240240
def test_code_theme_dark(mocker: MockerFixture, env: TestEnv):
241241
env.set('OPENAI_API_KEY', 'test')
242242
mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat')
243243
cli(['--code-theme=dark'])
244-
mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai')
244+
mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai', history=None)
245245

246246

247247
def test_agent_to_cli_sync(mocker: MockerFixture, env: TestEnv):

0 commit comments

Comments
 (0)