feat: persist and inspect run step budgets

This commit is contained in:
lda
2026-09-05 21:00:00 +07:00 Verified
parent 344902c17e
commit 7141a8818d
17 changed files with 707 additions and 8 deletions
+3
View File
@@ -15,6 +15,7 @@ from wf_artifacts import (
)
from wf_authoring import NodeSpec
from wf_core import (
RunLimits,
RunState,
Workflow,
)
@@ -343,6 +344,7 @@ class WfMcpService:
deployment: WorkflowDeployment | None = None,
artifact: WorkflowArtifact | None = None,
saved_subgraph_tree: SavedSubgraphTree | None = None,
limits: RunLimits | None = None,
):
return await self.workflow_runtime.run_workflow_from_plan(
plan,
@@ -350,6 +352,7 @@ class WfMcpService:
deployment=deployment,
artifact=artifact,
saved_subgraph_tree=saved_subgraph_tree,
limits=limits,
)
async def resume_workflow_from_plan(
@@ -13,6 +13,7 @@ from wf_api.operation_context import (
)
from wf_artifacts import DependencyDiagnostic, WorkflowArtifact, WorkflowDeployment
from wf_authoring import NodeSpec
from wf_core import RunLimits
from .core import WfMcpService
from .events import BrokerEventRecorder
@@ -70,6 +71,7 @@ class WfMcpWorkflowRuntimeRunner(WorkflowRuntimeRunner):
deployment=None,
artifact=None,
saved_subgraph_tree=None,
limits: RunLimits | None = None,
):
return await self.runtime.run_workflow_from_plan(
plan,
@@ -77,6 +79,7 @@ class WfMcpWorkflowRuntimeRunner(WorkflowRuntimeRunner):
deployment=deployment,
artifact=artifact,
saved_subgraph_tree=saved_subgraph_tree,
limits=limits,
)
async def resume_workflow_from_plan(
@@ -16,6 +16,7 @@ from wf_artifacts import WorkflowArtifact, WorkflowArtifactStore, WorkflowDeploy
from wf_authoring import NodeSpec
from wf_core import (
NodeUse,
RunLimits,
RunState,
RunStatus,
Workflow,
@@ -163,6 +164,7 @@ class WorkflowRuntimeService:
deployment: WorkflowDeployment | None = None,
artifact: WorkflowArtifact | None = None,
saved_subgraph_tree: SavedSubgraphTree | None = None,
limits: RunLimits | None = None,
) -> RunState:
self.emit_event(
make_event(
@@ -186,6 +188,7 @@ class WorkflowRuntimeService:
reducers=reducers,
subgraphs=prepared_subgraphs,
platform=platform_context,
limits=limits,
)
self.emit_event(
make_event(