This commit is contained in:
Nuno Campos
2024-12-02 17:01:12 -08:00
parent d36e6ceaaf
commit 0071bd1e1c
+3 -2
View File
@@ -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