github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-26 16:43:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[patch]: dont deep copy merge_message_runs (#28454)

Browse Source 
afaict no need to deep copy here, if we merge messages then we convert
them to chunks first anyways
This commit is contained in:
Bagatur 2025-02-22 13:56:45 -08:00 committed by GitHub
parent afa94e5bf7
commit 979a991dc2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libs/core/langchain_core/messages/utils.py
View File

@ -557,15 +557,14 @@ def merge_message_runs(
messages = convert_to_messages(messages) messages = convert_to_messages(messages)
merged: list[BaseMessage] = [] merged: list[BaseMessage] = []
for msg in messages: for msg in messages:
curr = msg.model_copy(deep=True)
last = merged.pop() if merged else None last = merged.pop() if merged else None
if not last: if not last:
merged.append(curr) merged.append(msg)
elif isinstance(curr, ToolMessage) or not isinstance(curr, last.__class__): elif isinstance(msg, ToolMessage) or not isinstance(msg, last.__class__):
merged.extend([last, curr]) merged.extend([last, msg])
else: else:
last_chunk = _msg_to_chunk(last) last_chunk = _msg_to_chunk(last)
curr_chunk = _msg_to_chunk(curr) curr_chunk = _msg_to_chunk(msg)
if curr_chunk.response_metadata: if curr_chunk.response_metadata:
curr_chunk.response_metadata.clear() curr_chunk.response_metadata.clear()
if ( if (