Fix API deserialization if a list field is missing from the JSON

ai_diffusion/api.py

@@ -1,4 +1,4 @@
-from dataclasses import Field, dataclass, field, is_dataclass, fields
+from dataclasses import Field, dataclass, field, is_dataclass, fields, MISSING
 from copy import copy
 from enum import Enum
 from types import GenericAlias, UnionType
@@ -293,8 +293,12 @@ def _object(self, type, input: dict):
         return type(*values)
 
     def _field(self, field: Field, value):
-        if value is None:
+        if value is None and field.default is not MISSING:
             return field.default
+        elif value is None and field.default_factory is not MISSING:
+            return field.default_factory()
+        elif value is None:
+            return None
         field_type = field.type
         if isinstance(field_type, UnionType):
             field_type = get_args(field_type)[0]

tests/test_api.py

@@ -7,8 +7,9 @@
     ExtentInput,
     ImageInput,
     ConditioningInput,
+    RegionInput,
 )
-from ai_diffusion.image import Extent, Image, ImageFileFormat
+from ai_diffusion.image import Extent, Image, ImageFileFormat, Bounds
 from ai_diffusion.resources import ControlMode
 from ai_diffusion.util import ensure
 
@@ -58,3 +59,22 @@ def test_serialize():
     assert _ensure_cmp(result_control[1].image) == _ensure_cmp(input_control[1].image)
     assert result_control[2].image is None
     assert result == input
+
+
+def test_deserialize_list_default():
+    input = WorkflowInput(WorkflowKind.generate)
+    input.images = ImageInput(ExtentInput(Extent(1, 1), Extent(2, 2), Extent(3, 3), Extent(4, 4)))
+    input.conditioning = ConditioningInput(
+        "prompt",
+        regions=[
+            RegionInput(
+                Image.create(Extent(2, 2), Qt.GlobalColor.red), Bounds(0, 0, 2, 2), "positive", []
+            )
+        ],
+    )
+
+    data = input.to_dict()
+    del data["conditioning"]["regions"][0]["loras"]
+    result = WorkflowInput.from_dict(data)
+    assert result.conditioning is not None
+    assert len(result.conditioning.regions[0].loras) == 0