diff --git a/libs/core/langchain_core/messages/__init__.py b/libs/core/langchain_core/messages/__init__.py index c35dac72b98..86bea028051 100644 --- a/libs/core/langchain_core/messages/__init__.py +++ b/libs/core/langchain_core/messages/__init__.py @@ -92,6 +92,8 @@ def _message_from_dict(message: dict) -> BaseMessage: return ToolMessageChunk(**message["data"]) elif _type == "SystemMessageChunk": return SystemMessageChunk(**message["data"]) + elif _type == "ChatMessageChunk": + return ChatMessageChunk(**message["data"]) else: raise ValueError(f"Got unexpected message type: {_type}")