From 88121b3e4eefbb918d571832d2f84b854fe0deb2 Mon Sep 17 00:00:00 2001 From: drbh Date: Mon, 30 Sep 2024 15:38:49 +0200 Subject: [PATCH] fix: expect simple message when no tool is selected --- integration-tests/models/test_tools_llama.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/integration-tests/models/test_tools_llama.py b/integration-tests/models/test_tools_llama.py index c337afa1..a03b0780 100644 --- a/integration-tests/models/test_tools_llama.py +++ b/integration-tests/models/test_tools_llama.py @@ -237,8 +237,8 @@ async def test_flash_llama_grammar_tools_insufficient_information( stream=False, ) - assert responses.choices[0].message.content is None assert ( - responses.choices[0].message.tool_calls[0]["function"]["name"] == "notify_error" + responses.choices[0].message.content + == "Cannot get current weather forecast from specified location and temperature unit. Please try again with different options." ) assert responses == response_snapshot