diff options
Diffstat (limited to 'tests/test_clear_messages.py')
| -rw-r--r-- | tests/test_clear_messages.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tests/test_clear_messages.py b/tests/test_clear_messages.py index c3f3595..a0653ca 100644 --- a/tests/test_clear_messages.py +++ b/tests/test_clear_messages.py @@ -82,6 +82,7 @@ def test_add_message_stores_token_count_and_clears_it(): assert reloaded["token_count"] == { "prompt_eval_count": 120, "eval_count": 80, + "num_ctx": 8192, "total_count": 200, } |