Job orchestrator

src/ess/livedata/config/workflow_spec.py

@@ -272,22 +272,21 @@ class WorkflowConfig(BaseModel):
     def from_params(
         cls,
         workflow_id: WorkflowId,
-        params: BaseModel | None = None,
-        aux_source_names: BaseModel | None = None,
+        params: dict | None = None,
+        aux_source_names: dict | None = None,
         job_number: JobNumber | None = None,
     ) -> WorkflowConfig:
         """
-        Create a WorkflowConfig from validated Pydantic models.
+        Create a WorkflowConfig from parameters.
 
         Parameters
         ----------
         workflow_id:
             Identifier for the workflow
         params:
-            Validated Pydantic model with workflow parameters, or None if no params
+            Workflow parameters as dict, or None if no params
         aux_source_names:
-            Validated Pydantic model with auxiliary source selections, or None if no
-            aux sources
+            Auxiliary source selections as dict, or None if no aux sources
         job_number:
             Optional job number (generated if not provided)
 
@@ -299,12 +298,8 @@ def from_params(
         return cls(
             identifier=workflow_id,
             job_number=job_number if job_number is not None else uuid.uuid4(),
-            aux_source_names=(
-                aux_source_names.model_dump(mode='json')
-                if aux_source_names is not None
-                else {}
-            ),
-            params=params.model_dump() if params is not None else {},
+            aux_source_names=aux_source_names or {},
+            params=params or {},
         )
 
 

src/ess/livedata/dashboard/configuration_adapter.py

@@ -17,6 +17,23 @@ class ConfigurationState(BaseModel):
     This model captures the user's configuration choices (sources, params,
     aux sources) that should be restored when reopening the dashboard.
     Used by both workflow and plotter configurations.
+
+    Schema Limitation
+    -----------------
+    This schema currently assumes all sources share the same `params` configuration,
+    with only `aux_source_names` varying per source. In reality, JobOrchestrator's
+    internal state (`staged_jobs`) allows different params per source via
+    `dict[SourceName, JobConfig]`.
+
+    For now, we expand on load: the single `params` dict is applied to all sources
+    in `source_names`, and `aux_source_names` is expanded per-source as needed.
+    This works because the current UI (WorkflowController.start_workflow) stages
+    the same params for all sources in a single operation.
+
+    Future work: If we support per-source params in the UI (e.g., "stage source1
+    with configA, stage source2 with configB"), this schema should be extended to:
+    `jobs: dict[str, JobConfigState]` where `JobConfigState` contains both params
+    and aux_source_names per source.
     """
 
     source_names: list[str] = Field(