From 9c513ee7e6d28ea88f6d0704a999fa40d6d953f7 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Fri, 31 May 2024 13:23:04 -0700 Subject: [PATCH] Lint --- langgraph/graph/graph.py | 4 +--- langgraph/pregel/debug.py | 13 ++++++------- langgraph/pregel/read.py | 2 +- 3 files changed, 8 insertions(+), 11 deletions(-) diff --git a/langgraph/graph/graph.py b/langgraph/graph/graph.py index b4fb5511e..8acc8f9b4 100644 --- a/langgraph/graph/graph.py +++ b/langgraph/graph/graph.py @@ -113,9 +113,7 @@ class Branch(NamedTuple): if any(dest is None or dest == START for dest in destinations): raise ValueError("Branch did not return a valid destination") if any(p.node == END for p in destinations if isinstance(p, Packet)): - raise InvalidUpdateError( - "Cannot send a packet with keyword arguments to the END node" - ) + raise InvalidUpdateError("Cannot send a packet to the END node") return writer(destinations) or input diff --git a/langgraph/pregel/debug.py b/langgraph/pregel/debug.py index f9cf7e59d..0fd00bdc8 100644 --- a/langgraph/pregel/debug.py +++ b/langgraph/pregel/debug.py @@ -68,17 +68,16 @@ def map_debug_tasks( if config is not None and TAG_HIDDEN in config.get("tags", []): continue - payload = { - "id": str(uuid5(TASK_NAMESPACE, json.dumps((name, step, idx)))), - "name": name, - "input": input, - "triggers": triggers, - } yield { "type": "task", "timestamp": ts, "step": step, - "payload": payload, + "payload": { + "id": str(uuid5(TASK_NAMESPACE, json.dumps((name, step, idx)))), + "name": name, + "input": input, + "triggers": triggers, + }, } diff --git a/langgraph/pregel/read.py b/langgraph/pregel/read.py index 902d96cf4..a14c704e4 100644 --- a/langgraph/pregel/read.py +++ b/langgraph/pregel/read.py @@ -1,6 +1,6 @@ from __future__ import annotations -from typing import Any, Callable, Dict, Mapping, Optional, Sequence, Union +from typing import Any, Callable, Mapping, Optional, Sequence, Union from langchain_core.pydantic_v1 import Field from langchain_core.runnables import (