github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-25 13:07:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
Mason Daugherty
2025-07-21 20:37:50 -04:00
parent 246c23ccb1
commit bbc5fe094c
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
libs/core/tests/unit_tests/language_models/chat_models/test_cache.py
View File

@@ -375,12 +375,12 @@ def test_cache_with_generation_objects() -> None:
# Replace with Generation objects (missing message field) # Replace with Generation objects (missing message field)
corrupted_generations = [ corrupted_generations = [
Generation( Generation(
text=gen.text, text=gen.text,
generation_info=gen.generation_info, generation_info=gen.generation_info,
type="Generation", # This is the key - wrong type type="Generation", # This is the key - wrong type
) )
for gen in cached_chat_generations for gen in cached_chat_generations
] ]
cache._cache[cache_key] = corrupted_generations cache._cache[cache_key] = corrupted_generations
@@ -388,6 +388,8 @@ def test_cache_with_generation_objects() -> None:
result2 = model.generate_response("test prompt") result2 = model.generate_response("test prompt")
assert result2.generations[0].message.content == "hello" assert result2.generations[0].message.content == "hello"
assert isinstance(result2.generations[0], ChatGeneration) assert isinstance(result2.generations[0], ChatGeneration)
def test_cleanup_serialized() -> None: def test_cleanup_serialized() -> None:
cleanup_serialized = { cleanup_serialized = {
"lc": 1, "lc": 1,