Skip to content

Commit

Permalink
update tests
Browse files Browse the repository at this point in the history
  • Loading branch information
dillonalaird committed May 13, 2024
1 parent fd55636 commit 13df90f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ def test_call_with_mock(openai_llm_mock): # noqa: F811
response = llm([{"role": "user", "content": "test prompt"}])
assert response == "mocked response"
openai_llm_mock.chat.completions.create.assert_called_with(
model="gpt-4-turbo",
model="gpt-4o",
messages=[{"role": "user", "content": "test prompt"}],
)

Expand Down

0 comments on commit 13df90f

Please sign in to comment.