mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-08-20 22:52:29 +02:00
f87608a16f
Fixes #5554 This PR removes unused utility classes and functions from the prebuilt tests directory to clean up dead code. Changes include: - Removed unused classes from `libs/prebuilt/tests/any_str.py`: - Deleted FloatBetween, AnyDict, AnyVersion, and UnsortedSequence - Kept only AnyStr class - Removed unused functions from `libs/prebuilt/tests/messages.py`: - Deleted _AnyIdDocument and _AnyIdAIMessageChunk - Kept _AnyIdHumanMessage and _AnyIdToolMessage - Removed unused classes from `libs/prebuilt/tests/memory_assert.py`: - Deleted NoopSerializer, MemorySaverAssertCheckpointMetadata, and MemorySaverNoPending - Kept MemorySaverAssertImmutable Verification: - Manually checked for no remaining references to removed code - Maintained existing import structures - Preserved functionality of the prebuilt test suite The changes reduce code complexity and remove unnecessary utility classes that were not being used in the test suite. --------- Co-authored-by: open-swe-dev[bot] <open-swe-dev@users.noreply.github.com> Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com>
29 lines
842 B
Python
29 lines
842 B
Python
"""Redefined messages as a work-around for pydantic issue with AnyStr.
|
|
|
|
The code below creates version of pydantic models
|
|
that will work in unit tests with AnyStr as id field
|
|
Please note that the `id` field is assigned AFTER the model is created
|
|
to workaround an issue with pydantic ignoring the __eq__ method on
|
|
subclassed strings.
|
|
"""
|
|
|
|
from typing import Any
|
|
|
|
from langchain_core.messages import HumanMessage, ToolMessage
|
|
|
|
from tests.any_str import AnyStr
|
|
|
|
|
|
def _AnyIdHumanMessage(**kwargs: Any) -> HumanMessage:
|
|
"""Create a human message with an any id field."""
|
|
message = HumanMessage(**kwargs)
|
|
message.id = AnyStr()
|
|
return message
|
|
|
|
|
|
def _AnyIdToolMessage(**kwargs: Any) -> ToolMessage:
|
|
"""Create a tool message with an any id field."""
|
|
message = ToolMessage(**kwargs)
|
|
message.id = AnyStr()
|
|
return message
|