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: adjust test expected content

Browse Source
This commit is contained in:
David Holtz 2024-10-09 23:21:43 +00:00
parent 93028b113a
commit 307c9ea371
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
integration-tests/models/__snapshots__/test_tools_llama/test_flash_llama_grammar_tools_insufficient_information.json
View File

@ -5,7 +5,7 @@
"index": 0,
"logprobs": null,
"message": {
"content": "I am a helpful assistant!",
"content": "I am an AI assistant",
"name": null,
"role": "assistant",
"tool_calls": null

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

@ -247,7 +247,7 @@ async def test_flash_llama_grammar_tools_insufficient_information(
)
assert responses.choices[0].message.tool_calls is None
assert responses.choices[0].message.content == "I am a helpful assistant!"
assert responses.choices[0].message.content == "I am an AI assistant"
assert responses == response_snapshot
@ -285,7 +285,7 @@ async def test_flash_llama_grammar_tools_insufficient_information_stream(
assert response.choices[0].delta.tool_calls is None
assert count == 5
assert content_generated == "I am a helpful assistant"
assert content_generated == "I am an AI assistant"
assert last_response == response_snapshot