This commit is contained in:
Nuno Campos
2024-10-14 11:31:03 -07:00
parent bfe005fef0
commit dc083c6563
2 changed files with 3 additions and 2 deletions
+2 -2
View File
@@ -326,7 +326,7 @@ class RemoteGraph(PregelProtocol, Runnable):
response: dict = self.sync_client.threads.update_state( # type: ignore
thread_id=merged_config["configurable"]["thread_id"],
values=values, # type: ignore
values=values,
as_node=as_node,
checkpoint=self._get_checkpoint(merged_config),
)
@@ -342,7 +342,7 @@ class RemoteGraph(PregelProtocol, Runnable):
response: dict = await self.client.threads.update_state( # type: ignore
thread_id=merged_config["configurable"]["thread_id"],
values=values, # type: ignore
values=values,
as_node=as_node,
checkpoint=self._get_checkpoint(merged_config),
)
+1
View File
@@ -186,6 +186,7 @@ class ThreadTask(TypedDict):
interrupts: list[dict]
checkpoint: Optional[Checkpoint]
state: Optional["ThreadState"]
result: Optional[dict[str, Any]]
class ThreadState(TypedDict):