This commit is contained in:
Harrison Chase
2024-04-29 14:59:31 -07:00
parent d12c2bae6b
commit a32fe442f0
4 changed files with 15 additions and 8 deletions
+3 -1
View File
@@ -15,7 +15,9 @@ from typing import (
from langchain_core.runnables import Runnable
from langchain_core.runnables.base import RunnableLike
from langchain_core.runnables.config import RunnableConfig
from langchain_core.runnables.graph import Node as RunnableGraphNode
from langchain_core.runnables.graph import (
Node as RunnableGraphNode,
)
from langgraph.channels.ephemeral_value import EphemeralValue
from langgraph.checkpoint import BaseCheckpointSaver
+8 -4
View File
@@ -116,7 +116,8 @@ class Channel:
*,
key: Optional[str] = None,
tags: Optional[list[str]] = None,
) -> PregelNode: ...
) -> PregelNode:
...
@overload
@classmethod
@@ -126,7 +127,8 @@ class Channel:
*,
key: None = None,
tags: Optional[list[str]] = None,
) -> PregelNode: ...
) -> PregelNode:
...
@classmethod
def subscribe_to(
@@ -1210,7 +1212,8 @@ def _prepare_next_tasks(
processes: Mapping[str, PregelNode],
channels: Mapping[str, BaseChannel],
for_execution: Literal[False],
) -> tuple[Checkpoint, list[PregelTaskDescription]]: ...
) -> tuple[Checkpoint, list[PregelTaskDescription]]:
...
@overload
@@ -1219,7 +1222,8 @@ def _prepare_next_tasks(
processes: Mapping[str, PregelNode],
channels: Mapping[str, BaseChannel],
for_execution: Literal[True],
) -> tuple[Checkpoint, list[PregelExecutableTask]]: ...
) -> tuple[Checkpoint, list[PregelExecutableTask]]:
...
def _prepare_next_tasks(
+4 -2
View File
@@ -10,6 +10,8 @@ class SerializerProtocol(Protocol):
Valid implementations include the `pickle`, `json` and `orjson` modules.
"""
def dumps(self, obj: Any) -> bytes: ...
def dumps(self, obj: Any) -> bytes:
...
def loads(self, data: bytes) -> Any: ...
def loads(self, data: bytes) -> Any:
...
-1
View File
@@ -1,5 +1,4 @@
"""Main entrypoint into package."""
from importlib import metadata
try: