unique_toolkit 1.53.0__tar.gz → 1.53.1__tar.gz

{unique_toolkit-1.53.0 → unique_toolkit-1.53.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.53.0
+Version: 1.53.1
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>

{unique_toolkit-1.53.0 → unique_toolkit-1.53.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.53.0"
+version = "1.53.1"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"

{unique_toolkit-1.53.0 → unique_toolkit-1.53.1}/unique_toolkit/_common/pydantic/rjsf_tags.py

@@ -13,6 +13,7 @@ Key components:
 
 from __future__ import annotations
 
+import types
 from collections.abc import Callable
 from enum import StrEnum
 from typing import Annotated, Any, Union, get_args, get_origin
@@ -20,6 +21,8 @@ from typing import Annotated, Any, Union, get_args, get_origin
 from pydantic import BaseModel
 from typing_extensions import get_type_hints
 
+_UNION_ORIGINS = {Union, types.UnionType}
+
 
 class CustomWidgetName(StrEnum):
     """Mirrors TypeScript CustomWidgetName. Keep in sync when extending."""
@@ -774,7 +777,7 @@ def _unwrap_optional(ann: Any) -> Any:
         The unwrapped type if it was Optional/Union with None, otherwise
         the original type unchanged
     """
-    if get_origin(ann) is Union:
+    if get_origin(ann) in _UNION_ORIGINS:
         args = [a for a in get_args(ann) if a not in _NONE_TYPES]
         if len(args) == 1:
             return args[0]
@@ -921,7 +924,7 @@ def ui_schema_for_model(
             node["additionalProperties"] = val_node
 
         # Union -> anyOf branches
-        elif origin is Union:
+        elif origin in _UNION_ORIGINS:
             branches = []
             for alt in get_args(base):
                 if alt in _NONE_TYPES: