WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-06 17:57:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'jacob/dedupe' of github.com:langchain-ai/langgraph into jacob/dedupe

Browse Source
This commit is contained in:
jacoblee93
2025-02-06 11:52:29 -08:00
parent 32aa87d4b2 55a3352e9b
commit 871f15a630
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/langgraph/graph/message.py
+4 -5
View File
@@ -176,21 +176,19 @@ def add_messages(
for m in left:
if m.id is None:
m.id = str(uuid.uuid4())
dedupe_map = {}
for m in right:
if m.id is None:
m.id = str(uuid.uuid4())
dedupe_map[m.id] = m
right = list(dedupe_map.values())
# merge
left_idx_by_id = {m.id: i for i, m in enumerate(left)}
merged = left.copy()
merged_by_id = {m.id: i for i, m in enumerate(merged)}
ids_to_remove = set()
for m in right:
if (existing_idx := left_idx_by_id.get(m.id)) is not None:
if (existing_idx := merged_by_id.get(m.id)) is not None:
if isinstance(m, RemoveMessage):
ids_to_remove.add(m.id)
else:
ids_to_remove.discard(m.id)
merged[existing_idx] = m
else:
if isinstance(m, RemoveMessage):
@@ -198,6 +196,7 @@ def add_messages(
f"Attempting to delete a message with an ID that doesn't exist ('{m.id}')"
)
merged_by_id[m.id] = len(merged)
merged.append(m)
merged = [m for m in merged if m.id not in ids_to_remove]