huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-12 04:44:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: expect simple message when no tool is selected

Browse Source
This commit is contained in:
drbh 2024-09-30 15:38:49 +02:00
parent dfc4559017
commit 88121b3e4e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
integration-tests/models/test_tools_llama.py
View File

@ -237,8 +237,8 @@ async def test_flash_llama_grammar_tools_insufficient_information(
stream=False, stream=False,
) )
assert responses.choices[0].message.content is None
assert ( 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 assert responses == response_snapshot