From 8505341931670185ff950a21137436ff23c78d08 Mon Sep 17 00:00:00 2001 From: drbh Date: Mon, 25 Nov 2024 10:10:33 -0500 Subject: [PATCH] fix: replace expected output for continue test --- .../test_llama_completion_single_prompt_continue.json | 2 +- integration-tests/models/test_continue_final_message.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json b/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json index bd5cdd7c..100fb338 100644 --- a/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json +++ b/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json @@ -5,7 +5,7 @@ "index": 0, "logprobs": null, "message": { - "content": "the mouse is much smaller than an elephant. The average elephant weight is around 6,500 lbs (3,", + "content": " the royal mouse? It is a little more slender and only weighs around 1.5 pounds for males and 1.3 pounds", "role": "assistant" } } diff --git a/integration-tests/models/test_continue_final_message.py b/integration-tests/models/test_continue_final_message.py index 24ae720f..01c86dcd 100644 --- a/integration-tests/models/test_continue_final_message.py +++ b/integration-tests/models/test_continue_final_message.py @@ -71,6 +71,6 @@ def test_llama_completion_single_prompt_continue( content = response["choices"][0]["message"]["content"] assert ( content - == "the mouse is much smaller than an elephant. The average elephant weight is around 6,500 lbs (3," + == " the royal mouse? It is a little more slender and only weighs around 1.5 pounds for males and 1.3 pounds" ) assert response == response_snapshot