diff --git a/tests/test_prompts.py b/tests/test_prompts.py index 53001176..eea9ddb8 100644 --- a/tests/test_prompts.py +++ b/tests/test_prompts.py @@ -8,7 +8,7 @@ def test_get_prompt(): assert 1200 < len_tokens(prompt.content) < 3000 prompt = get_prompt("short") - assert 700 < len_tokens(prompt.content) < 900 + assert 700 < len_tokens(prompt.content) < 1200 prompt = get_prompt("Hello world!") assert prompt.content == "Hello world!"