From 6cdb8925c89dd1cbd6777cd6010cf6c3fd3e5a6f Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Thu, 21 Mar 2024 13:32:11 -0700 Subject: [PATCH] MessageGraph: convert message chunks to messages when added to graph --- langgraph/graph/message.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/langgraph/graph/message.py b/langgraph/graph/message.py index ad537b5c0..2863c6bd7 100644 --- a/langgraph/graph/message.py +++ b/langgraph/graph/message.py @@ -1,7 +1,7 @@ import uuid from typing import Annotated, Union -from langchain_core.messages import AnyMessage +from langchain_core.messages import AnyMessage, message_chunk_to_message from langgraph.graph.state import StateGraph @@ -11,9 +11,13 @@ Messages = Union[list[AnyMessage], AnyMessage] def add_messages(left: Messages, right: Messages) -> Messages: # coerce to list if not isinstance(left, list): - left = [left] + left = [message_chunk_to_message(left)] + else: + left = [message_chunk_to_message(m) for m in left] if not isinstance(right, list): - right = [right] + right = [message_chunk_to_message(right)] + else: + right = [message_chunk_to_message(m) for m in right] # assign missing ids for m in left: if m.id is None: