fix: preserve singleton conditioning payloads

2026-04-20 13:21:24 -07:00
parent ee5444336f
commit 2de907c8fa
6 changed files with 106 additions and 8 deletions
@@ -5082,6 +5082,43 @@ def test_apply_boundary_inputs_normalizes_wrapped_scalar_values(
    assert prompt["remote_1"]["inputs"]["value"] == 4


+@pytest.mark.parametrize(
+    ("module_fixture_name",),
+    [
+        ("remote_modal_app_module",),
+        ("modal_cloud_module",),
+    ],
+)
+def test_apply_boundary_inputs_preserves_singleton_conditioning_lists(
+    request: Any,
+    module_fixture_name: str,
+) -> None:
+    """Boundary input hydration must not flatten singleton CONDITIONING payloads."""
+    target_module = request.getfixturevalue(module_fixture_name)
+    conditioning = [["cond", {"pooled_output": None}]]
+    prompt = {
+        "remote_1": {
+            "class_type": "BoundarySource",
+            "inputs": {"value": 0},
+            "_meta": {},
+        }
+    }
+
+    target_module._apply_boundary_inputs(
+        prompt=prompt,
+        boundary_input_specs=[
+            {
+                "proxy_input_name": "remote_input_0",
+                "io_type": "CONDITIONING",
+                "targets": [{"node_id": "remote_1", "input_name": "value"}],
+            }
+        ],
+        hydrated_inputs={"remote_input_0": conditioning},
+    )
+
+    assert prompt["remote_1"]["inputs"]["value"] == conditioning
+
+
@pytest.mark.parametrize(
    ("module_fixture_name",),
    [