Restrict input values

To a subset of JSONifiable types

Signed-off-by: liamhuber <liamhuber@greyhavensolutions.com>

Author: liamhuber

src/python_workflow_definition/models.py

@@ -19,6 +19,9 @@
 )
 
 
+JsonPrimitive = Union[str, int, float, bool, None]
+
+
 class PythonWorkflowDefinitionBaseNode(BaseModel):
     """Base model for all node types, containing common fields."""
 
@@ -33,7 +36,7 @@ class PythonWorkflowDefinitionInputNode(PythonWorkflowDefinitionBaseNode):
 
     type: Literal["input"]
     name: str
-    value: Optional[Any] = None
+    value: Optional[JsonPrimitive] = None
 
 
 class PythonWorkflowDefinitionOutputNode(PythonWorkflowDefinitionBaseNode):

tests/test_models.py

@@ -4,6 +4,7 @@
 from unittest import mock
 from pydantic import ValidationError
 from python_workflow_definition.models import (
+    JsonPrimitive,
     PythonWorkflowDefinitionInputNode,
     PythonWorkflowDefinitionOutputNode,
     PythonWorkflowDefinitionFunctionNode,
@@ -40,6 +41,19 @@ def test_input_node(self):
         )
         self.assertEqual(node_with_value.value, 42)
 
+    def test_input_node_invalid_value_raises(self):
+        bad_value = (1, 2)
+        self.assertNotIsInstance(bad_value, JsonPrimitive)
+        with self.assertRaises(ValidationError):
+            PythonWorkflowDefinitionInputNode.model_validate(
+                {
+                    "id": 0,
+                    "type": "input",
+                    "name": "x",
+                    "value": bad_value,
+                }
+            )
+
     def test_output_node(self):
         node = PythonWorkflowDefinitionOutputNode(id=1, type="output", name="test_output")
         self.assertEqual(node.id, 1)