From e8a73e1505502b9c8a7e2984a59cd42d788f2648 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Wed, 15 Jan 2025 15:35:06 -0800 Subject: [PATCH] Remove flag --- libs/langgraph/langgraph/pregel/__init__.py | 1 - libs/langgraph/langgraph/pregel/loop.py | 5 ----- .../langgraph/scheduler/kafka/orchestrator.py | 2 -- 3 files changed, 8 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index 102e68be8..0a90aa10f 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -1642,7 +1642,6 @@ class Pregel(PregelProtocol): interrupt_after=interrupt_after_, manager=run_manager, debug=debug, - check_subgraphs=self.checkpointer is not True, ) as loop: # create runner runner = PregelRunner( diff --git a/libs/langgraph/langgraph/pregel/loop.py b/libs/langgraph/langgraph/pregel/loop.py index ff244b7de..0afe62658 100644 --- a/libs/langgraph/langgraph/pregel/loop.py +++ b/libs/langgraph/langgraph/pregel/loop.py @@ -202,7 +202,6 @@ class PregelLoop(LoopProtocol): interrupt_after: Union[All, Sequence[str]] = EMPTY_SEQ, interrupt_before: Union[All, Sequence[str]] = EMPTY_SEQ, manager: Union[None, AsyncParentRunManager, ParentRunManager] = None, - check_subgraphs: bool = True, debug: bool = False, ) -> None: super().__init__( @@ -822,7 +821,6 @@ class SyncPregelLoop(PregelLoop, ContextManager): interrupt_before: Union[All, Sequence[str]] = EMPTY_SEQ, output_keys: Union[str, Sequence[str]] = EMPTY_SEQ, stream_keys: Union[str, Sequence[str]] = EMPTY_SEQ, - check_subgraphs: bool = True, debug: bool = False, ) -> None: super().__init__( @@ -837,7 +835,6 @@ class SyncPregelLoop(PregelLoop, ContextManager): stream_keys=stream_keys, interrupt_after=interrupt_after, interrupt_before=interrupt_before, - check_subgraphs=check_subgraphs, manager=manager, debug=debug, ) @@ -959,7 +956,6 @@ class AsyncPregelLoop(PregelLoop, AsyncContextManager): manager: Union[None, AsyncParentRunManager, ParentRunManager] = None, output_keys: Union[str, Sequence[str]] = EMPTY_SEQ, stream_keys: Union[str, Sequence[str]] = EMPTY_SEQ, - check_subgraphs: bool = True, debug: bool = False, ) -> None: super().__init__( @@ -974,7 +970,6 @@ class AsyncPregelLoop(PregelLoop, AsyncContextManager): stream_keys=stream_keys, interrupt_after=interrupt_after, interrupt_before=interrupt_before, - check_subgraphs=check_subgraphs, manager=manager, debug=debug, ) diff --git a/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py b/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py index 4e5be8470..e3701b529 100644 --- a/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py +++ b/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py @@ -163,7 +163,6 @@ class AsyncKafkaOrchestrator(AbstractAsyncContextManager): stream_keys=graph.stream_channels, interrupt_after=graph.interrupt_after_nodes, interrupt_before=graph.interrupt_before_nodes, - check_subgraphs=False, ) as loop: if loop.tick(input_keys=graph.input_channels): # wait for checkpoint to be saved @@ -353,7 +352,6 @@ class KafkaOrchestrator(AbstractContextManager): stream_keys=graph.stream_channels, interrupt_after=graph.interrupt_after_nodes, interrupt_before=graph.interrupt_before_nodes, - check_subgraphs=False, ) as loop: if loop.tick(input_keys=graph.input_channels): # wait for checkpoint to be saved