From efd33d860fdef71bc37f286aeab71fae30a0e607 Mon Sep 17 00:00:00 2001 From: vbarda Date: Wed, 14 May 2025 21:03:47 -0400 Subject: [PATCH] update --- libs/langgraph/langgraph/graph/state.py | 2 +- libs/langgraph/langgraph/pregel/draw.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/libs/langgraph/langgraph/graph/state.py b/libs/langgraph/langgraph/graph/state.py index 68a75fa8e..f6a29198b 100644 --- a/libs/langgraph/langgraph/graph/state.py +++ b/libs/langgraph/langgraph/graph/state.py @@ -1114,7 +1114,7 @@ def _get_root(input: Any) -> Optional[Sequence[tuple[str, Any]]]: else: updates.append(("__root__", i)) return updates - else: + elif input is not None: return [("__root__", input)] diff --git a/libs/langgraph/langgraph/pregel/draw.py b/libs/langgraph/langgraph/pregel/draw.py index f2d8e2631..edc689890 100644 --- a/libs/langgraph/langgraph/pregel/draw.py +++ b/libs/langgraph/langgraph/pregel/draw.py @@ -108,7 +108,8 @@ def draw_graph( for w in task.writers: # apply regular writes if isinstance(w, ChannelWrite): - w.invoke(None, task.config) + empty_input = "" if "__root__" in specs else {} + w.invoke(empty_input, task.config) # apply conditional writes declared for static analysis, only once if w not in static_seen: static_seen.add(w)