From 8c1a2509586f5524067163f7123992b36aa21bff Mon Sep 17 00:00:00 2001
From: liamhuber <liamhuber@greyhavensolutions.com>
Date: Fri, 20 Feb 2026 10:08:47 -0800
Subject: [PATCH 1/2] Allow immutable json primitives recursively

Signed-off-by: liamhuber <liamhuber@greyhavensolutions.com>
---
 src/python_workflow_definition/models.py | 19 ++++++++-
 tests/test_models.py                     | 50 ++++++++++++++++++++++++
 2 files changed, 67 insertions(+), 2 deletions(-)

diff --git a/src/python_workflow_definition/models.py b/src/python_workflow_definition/models.py
index 4980cfa..899ac01 100644
--- a/src/python_workflow_definition/models.py
+++ b/src/python_workflow_definition/models.py
@@ -1,5 +1,15 @@
 from pathlib import Path
-from typing import List, Union, Optional, Literal, Any, Annotated, Type, TypeVar
+from typing import (
+    List,
+    Union,
+    Optional,
+    Literal,
+    Any,
+    Annotated,
+    Type,
+    TypeAliasType,
+    TypeVar,
+)
 from pydantic import BaseModel, Field, field_validator, field_serializer
 from pydantic import ValidationError
 import json
@@ -18,6 +28,11 @@
     "PythonWorkflowDefinitionWorkflow",
 )
 
+JsonPrimitive = Union[str, int, float, bool, None]
+AllowableDefaults = TypeAliasType(
+    "AllowableDefaults", "Union[JsonPrimitive, tuple[AllowableDefaults, ...]]"
+)
+
 
 class PythonWorkflowDefinitionBaseNode(BaseModel):
     """Base model for all node types, containing common fields."""
@@ -33,7 +48,7 @@ class PythonWorkflowDefinitionInputNode(PythonWorkflowDefinitionBaseNode):
 
     type: Literal["input"]
     name: str
-    value: Optional[Any] = None
+    value: Optional[AllowableDefaults] = None
 
 
 class PythonWorkflowDefinitionOutputNode(PythonWorkflowDefinitionBaseNode):
diff --git a/tests/test_models.py b/tests/test_models.py
index 83f6066..adea6d6 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -12,6 +12,11 @@
     INTERNAL_DEFAULT_HANDLE,
 )
 
+
+class _NoTrivialSerialization:
+    pass
+
+
 class TestModels(unittest.TestCase):
     def setUp(self):
         self.valid_workflow_dict = {
@@ -40,6 +45,51 @@ def test_input_node(self):
         )
         self.assertEqual(node_with_value.value, 42)
 
+    def test_input_node_valid_values(self):
+        good_values = (
+            1,
+            1.1,
+            "string",
+            True,
+            None,
+            (1, 2),
+            (("recursive", "tuple"), (True, False)),
+        )
+        for value in good_values:
+            with self.subTest(value=value):
+                model = PythonWorkflowDefinitionInputNode.model_validate(
+                    {
+                        "id": 0,
+                        "type": "input",
+                        "name": "x",
+                        "value": value,
+                    }
+                )
+                self.assertEqual(
+                    value,
+                    PythonWorkflowDefinitionInputNode.model_validate(
+                        model.model_dump(mode="json")
+                    ).value
+                )
+
+
+    def test_input_node_invalid_value_raises(self):
+        bad_values = (
+            {"mutable": "thing"},
+            _NoTrivialSerialization(),
+        )
+        for value in bad_values:
+            with self.subTest(value=value):
+                with self.assertRaises(ValidationError):
+                    PythonWorkflowDefinitionInputNode.model_validate(
+                        {
+                            "id": 0,
+                            "type": "input",
+                            "name": "x",
+                            "value": value,
+                        }
+                    )
+
     def test_output_node(self):
         node = PythonWorkflowDefinitionOutputNode(id=1, type="output", name="test_output")
         self.assertEqual(node.id, 1)

From 5685e9fc673ecac5c9502da06a0e387f5188e126 Mon Sep 17 00:00:00 2001
From: liamhuber <liamhuber@greyhavensolutions.com>
Date: Fri, 20 Feb 2026 10:12:31 -0800
Subject: [PATCH 2/2] Import from typing_extensions

Signed-off-by: liamhuber <liamhuber@greyhavensolutions.com>
---
 src/python_workflow_definition/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/python_workflow_definition/models.py b/src/python_workflow_definition/models.py
index 899ac01..04567f3 100644
--- a/src/python_workflow_definition/models.py
+++ b/src/python_workflow_definition/models.py
@@ -7,9 +7,9 @@
     Any,
     Annotated,
     Type,
-    TypeAliasType,
     TypeVar,
 )
+from typing_extensions import TypeAliasType
 from pydantic import BaseModel, Field, field_validator, field_serializer
 from pydantic import ValidationError
 import json