Skip to content

Commit 0dbcf09

Browse files
committed
test: fixed missing argument in test
1 parent 0cd60a0 commit 0dbcf09

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/test_llm_openai.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -222,14 +222,14 @@ def test_message_conversion_with_tool_and_non_tool():
222222
),
223223
]
224224

225-
set_default_model("openai/gpt-4o")
226-
227225
tool_save = get_tool("save")
228226
tool_shell = get_tool("shell")
229-
230227
assert tool_save and tool_shell
231228

232-
messages_dicts, _ = _prepare_messages_for_api(messages, [tool_save, tool_shell])
229+
model = get_model("openai/gpt-4o")
230+
messages_dicts, _ = _prepare_messages_for_api(
231+
messages, model.full, [tool_save, tool_shell]
232+
)
233233

234234
assert messages_dicts == [
235235
{"role": "user", "content": [{"type": "text", "text": "First user prompt"}]},

0 commit comments

Comments
 (0)