Rewrite imports

langchain-ai · Sep 22, 2024 · 30b631a · 30b631a
1 parent d9854b3
commit 30b631a
Show file tree

Hide file tree

Showing 6 changed files with 286 additions and 286 deletions.
diff --git a/docs/docs/how-tos/map-reduce.ipynb b/docs/docs/how-tos/map-reduce.ipynb
diff --git a/docs/docs/reference/graphs.md b/docs/docs/reference/graphs.md
@@ -29,7 +29,7 @@ handler: python
 
 ## StreamMode
 
-::: langgraph.pregel.types.StreamMode
+::: langgraph.types.StreamMode
 
 ## Constants
 

diff --git a/libs/langgraph/langgraph/types.py b/libs/langgraph/langgraph/types.py
@@ -153,7 +153,7 @@ class Send:
         ...     subjects: list[str]
         ...     jokes: Annotated[list[str], operator.add]
         ...
-        >>> from langgraph.constants import Send
+        >>> from langgraph.types import Send
         >>> from langgraph.graph import END, START
         >>> def continue_to_jokes(state: OverallState):
         ...     return [Send("generate_joke", {"subject": s}) for s in state['subjects']]

diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py
@@ -52,7 +52,7 @@
     CheckpointTuple,
 )
 from langgraph.checkpoint.memory import MemorySaver
-from langgraph.constants import ERROR, PULL, PUSH, Interrupt, Send
+from langgraph.constants import ERROR, PULL, PUSH
 from langgraph.errors import InvalidUpdateError, NodeInterrupt
 from langgraph.graph import END, Graph
 from langgraph.graph.graph import START
@@ -71,7 +71,7 @@
 )
 from langgraph.pregel.retry import RetryPolicy
 from langgraph.store.memory import MemoryStore
-from langgraph.types import PregelTask, StreamWriter
+from langgraph.types import Interrupt, PregelTask, Send, StreamWriter
 from tests.any_str import AnyDict, AnyStr, AnyVersion, UnsortedSequence
 from tests.conftest import ALL_CHECKPOINTERS_SYNC, SHOULD_CHECK_SNAPSHOTS
 from tests.fake_chat import FakeChatModel

diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py
@@ -51,7 +51,7 @@
     CheckpointTuple,
 )
 from langgraph.checkpoint.memory import MemorySaver
-from langgraph.constants import ERROR, PULL, PUSH, Interrupt, Send
+from langgraph.constants import ERROR, PULL, PUSH
 from langgraph.errors import InvalidUpdateError, NodeInterrupt
 from langgraph.graph import END, Graph, StateGraph
 from langgraph.graph.graph import START
@@ -69,7 +69,7 @@
 )
 from langgraph.pregel.retry import RetryPolicy
 from langgraph.store.memory import MemoryStore
-from langgraph.types import PregelTask, StreamWriter
+from langgraph.types import Interrupt, PregelTask, Send, StreamWriter
 from tests.any_str import AnyDict, AnyStr, AnyVersion, UnsortedSequence
 from tests.conftest import (
     ALL_CHECKPOINTERS_ASYNC,

diff --git a/libs/scheduler-kafka/README.md b/libs/scheduler-kafka/README.md
@@ -95,7 +95,7 @@ You can pass any of the following values as `kwargs` to either `KafkaOrchestrato
 
 - batch_max_n (int): Maximum number of messages to include in a single batch. Default: 10.
 - batch_max_ms (int): Maximum time in milliseconds to wait for messages to include in a batch. Default: 1000.
-- retry_policy (langgraph.pregel.types.RetryPolicy): Controls which graph-level errors will be retried when processing messages. A good use for this is to retry database errors thrown by the checkpointer. Defaults to None.
+- retry_policy (langgraph.types.RetryPolicy): Controls which graph-level errors will be retried when processing messages. A good use for this is to retry database errors thrown by the checkpointer. Defaults to None.
 
 ### Connection settings