mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-09-01 04:39:01 +02:00
Nuno Campos
d22fe60d4c
- any invalid utf-8 chars now removed on dumps - fix serialization of Send - remove serialization of NamedTuple, which doesn't work - add test for custom serde passed to memory saver - add test using Send and JsonPlus serde
34 lines
745 B
Python
34 lines
745 B
Python
from typing import Any
|
|
|
|
CONFIG_KEY_SEND = "__pregel_send"
|
|
CONFIG_KEY_READ = "__pregel_read"
|
|
|
|
INTERRUPT = "__interrupt__"
|
|
TASKS = "__pregel_tasks"
|
|
|
|
RESERVED = {INTERRUPT, TASKS, CONFIG_KEY_SEND, CONFIG_KEY_READ}
|
|
|
|
TAG_HIDDEN = "langsmith:hidden"
|
|
|
|
|
|
class Send:
|
|
node: str
|
|
arg: Any
|
|
|
|
def __init__(self, /, node: str, arg: Any) -> None:
|
|
self.node = node
|
|
self.arg = arg
|
|
|
|
def __hash__(self) -> int:
|
|
return hash((self.node, self.arg))
|
|
|
|
def __repr__(self) -> str:
|
|
return f"Send(node={self.node!r}, arg={self.arg!r})"
|
|
|
|
def __eq__(self, value: object) -> bool:
|
|
return (
|
|
isinstance(value, Send)
|
|
and self.node == value.node
|
|
and self.arg == value.arg
|
|
)
|