Skip to content

Commit 1278090

Browse files
committed
Fix tests
1 parent bf5d828 commit 1278090

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
@@ -228,21 +228,21 @@ def test_code_theme_unset(mocker: MockerFixture, env: TestEnv):
228228
env.set('OPENAI_API_KEY', 'test')
229229
mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat')
230230
cli([])
231-
mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai')
231+
mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai', history=None)
232232

233233

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

240240

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

247247

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

0 commit comments

Comments
 (0)