This commit is contained in:
Shuchang Zheng 2024-12-03 23:41:05 -08:00 committed by GitHub
parent 3b2e2de197
commit 521e355591
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 16 additions and 3 deletions

View file

@ -510,3 +510,19 @@ class WorkflowContextManager:
await self.workflow_run_contexts[workflow_run_id].register_block_parameters(
self.aws_client, parameters, organization
)
def add_context_parameter(self, workflow_run_id: str, context_parameter: ContextParameter) -> None:
self._validate_workflow_run_context(workflow_run_id)
self.workflow_run_contexts[workflow_run_id].parameters[context_parameter.key] = context_parameter
async def set_parameter_values_for_output_parameter_dependent_blocks(
self,
workflow_run_id: str,
output_parameter: OutputParameter,
value: dict[str, Any] | list | str | None,
) -> None:
self._validate_workflow_run_context(workflow_run_id)
await self.workflow_run_contexts[workflow_run_id].set_parameter_values_for_output_parameter_dependent_blocks(
output_parameter,
value,
)