From b3c3b58f2c85ebdfd380c0c9b85d661757236887 Mon Sep 17 00:00:00 2001 From: Ryan Kraus Date: Mon, 5 Feb 2024 13:13:25 -0500 Subject: [PATCH] core[patch]: Fixed bug in dict to message conversion. (#17023) - **Description**: We discovered a bug converting dictionaries to messages where the ChatMessageChunk message type isn't handled. This PR adds support for that message type. - **Issue**: #17022 - **Dependencies**: None - **Twitter handle**: None --- libs/core/langchain_core/messages/__init__.py | 2 ++ 1 file changed, 2 insertions(+) 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}")