From bbc5fe094c4144c5a10260a601e6c02746af5309 Mon Sep 17 00:00:00 2001 From: Mason Daugherty Date: Mon, 21 Jul 2025 20:37:50 -0400 Subject: [PATCH] . --- .../language_models/chat_models/test_cache.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/libs/core/tests/unit_tests/language_models/chat_models/test_cache.py b/libs/core/tests/unit_tests/language_models/chat_models/test_cache.py index 27cd3b28c74..8bb057b6a05 100644 --- a/libs/core/tests/unit_tests/language_models/chat_models/test_cache.py +++ b/libs/core/tests/unit_tests/language_models/chat_models/test_cache.py @@ -375,12 +375,12 @@ def test_cache_with_generation_objects() -> None: # Replace with Generation objects (missing message field) corrupted_generations = [ - Generation( - text=gen.text, - generation_info=gen.generation_info, - type="Generation", # This is the key - wrong type - ) - for gen in cached_chat_generations + Generation( + text=gen.text, + generation_info=gen.generation_info, + type="Generation", # This is the key - wrong type + ) + for gen in cached_chat_generations ] cache._cache[cache_key] = corrupted_generations @@ -388,6 +388,8 @@ def test_cache_with_generation_objects() -> None: result2 = model.generate_response("test prompt") assert result2.generations[0].message.content == "hello" assert isinstance(result2.generations[0], ChatGeneration) + + def test_cleanup_serialized() -> None: cleanup_serialized = { "lc": 1,