From d723a1e752017c4152652fde883d4249a77d4f68 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Thu, 2 May 2024 09:26:32 -0700 Subject: [PATCH] Add name to debug payload --- langgraph/pregel/debug.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/langgraph/pregel/debug.py b/langgraph/pregel/debug.py index 5fd548344..b444da9c1 100644 --- a/langgraph/pregel/debug.py +++ b/langgraph/pregel/debug.py @@ -23,6 +23,7 @@ class TaskPayload(TypedDict): class TaskResultPayload(TypedDict): id: str + name: str result: list[tuple[str, Any]] @@ -96,6 +97,7 @@ def map_debug_task_results( "step": step, "payload": { "id": str(uuid5(TASK_NAMESPACE, json.dumps((name, step)))), + "name": name, "result": [w for w in writes if w[0] in stream_channels_list], }, }