From 0071bd1e1cf91f84a9c364ee3eae8eacd9ad1001 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Mon, 2 Dec 2024 17:01:12 -0800 Subject: [PATCH] Lint --- libs/langgraph/langgraph/pregel/remote.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/remote.py b/libs/langgraph/langgraph/pregel/remote.py index 8ee7ec884..d45cdb310 100644 --- a/libs/langgraph/langgraph/pregel/remote.py +++ b/libs/langgraph/langgraph/pregel/remote.py @@ -28,6 +28,7 @@ from langgraph_sdk.client import ( get_sync_client, ) from langgraph_sdk.schema import Checkpoint, ThreadState +from langgraph_sdk.schema import Command as CommandSDK from langgraph_sdk.schema import StreamMode as StreamModeSDK from typing_extensions import Self @@ -599,7 +600,7 @@ class RemoteGraph(PregelProtocol): stream_mode, config ) if isinstance(input, Command): - command: dict[str, Any] = asdict(input) + command: Optional[CommandSDK] = cast(CommandSDK, asdict(input)) input = None else: command = None @@ -688,7 +689,7 @@ class RemoteGraph(PregelProtocol): stream_mode, config ) if isinstance(input, Command): - command: dict[str, Any] = asdict(input) + command: Optional[CommandSDK] = cast(CommandSDK, asdict(input)) input = None else: command = None