revert to disciminated union for AnyComponent

pydantic · Dec 13, 2023 · 2d34524 · 2d34524
1 parent bbb5d61
commit 2d34524
Showing 1 changed file with 29 additions and 26 deletions.
diff --git a/src/python-fastui/fastui/components/__init__.py b/src/python-fastui/fastui/components/__init__.py
@@ -231,30 +231,33 @@ class Custom(_p.BaseModel, extra='forbid'):
     type: _t.Literal['Custom'] = 'Custom'
 
 
-AnyComponent = _t.Union[
-    Text,
-    Paragraph,
-    PageTitle,
-    Div,
-    Page,
-    Heading,
-    Markdown,
-    Code,
-    Json,
-    Button,
-    Link,
-    LinkList,
-    Navbar,
-    Modal,
-    ServerLoad,
-    Image,
-    Iframe,
-    Custom,
-    Table,
-    Pagination,
-    Display,
-    Details,
-    Form,
-    FormField,
-    ModelForm,
+AnyComponent = _te.Annotated[
+    _t.Union[
+        Text,
+        Paragraph,
+        PageTitle,
+        Div,
+        Page,
+        Heading,
+        Markdown,
+        Code,
+        Json,
+        Button,
+        Link,
+        LinkList,
+        Navbar,
+        Modal,
+        ServerLoad,
+        Image,
+        Iframe,
+        Custom,
+        Table,
+        Pagination,
+        Display,
+        Details,
+        Form,
+        FormField,
+        ModelForm,
+    ],
+    _p.Field(discriminator='type'),
 ]