feat: agent sandbox support HITL

This commit is contained in:
Harry
2026-02-09 22:20:30 +08:00
parent 5eeb6c56f0
commit 9614fe8e6e
3 changed files with 5 additions and 7 deletions

View File

@@ -119,6 +119,7 @@ class AdvancedChatAppRunner(WorkflowBasedAppRunner):
if resume_state is not None:
graph_runtime_state = resume_state
graph_runtime_state.set_sandbox(self._sandbox)
variable_pool = graph_runtime_state.variable_pool
graph = self._init_graph(
graph_config=self._workflow.graph_dict,
@@ -175,9 +176,7 @@ class AdvancedChatAppRunner(WorkflowBasedAppRunner):
# init graph
graph_runtime_state = GraphRuntimeState(variable_pool=variable_pool, start_at=time.time())
if self._sandbox:
graph_runtime_state.set_sandbox(self._sandbox)
graph_runtime_state.set_sandbox(self._sandbox)
graph = self._init_graph(
graph_config=self._workflow.graph_dict,

View File

@@ -78,6 +78,7 @@ class WorkflowAppRunner(WorkflowBasedAppRunner):
if resume_state is not None:
graph_runtime_state = resume_state
graph_runtime_state.set_sandbox(self._sandbox)
variable_pool = graph_runtime_state.variable_pool
graph = self._init_graph(
graph_config=self._workflow.graph_dict,
@@ -115,9 +116,7 @@ class WorkflowAppRunner(WorkflowBasedAppRunner):
)
graph_runtime_state = GraphRuntimeState(variable_pool=variable_pool, start_at=time.perf_counter())
if self._sandbox:
graph_runtime_state.set_sandbox(self._sandbox)
graph_runtime_state.set_sandbox(self._sandbox)
# init graph
graph = self._init_graph(

View File

@@ -355,7 +355,7 @@ class GraphRuntimeState:
def sandbox(self) -> Sandbox | None:
return self._sandbox
def set_sandbox(self, sandbox: Sandbox) -> None:
def set_sandbox(self, sandbox: Sandbox | None) -> None:
self._sandbox = sandbox
# ------------------------------------------------------------------