feat: persist and inspect run step budgets
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user