dara-components 1.21.9__py3-none-any.whl → 1.21.23__py3-none-any.whl

dara/components/common/utils.py

@@ -17,7 +17,7 @@ limitations under the License.
 
 import os
 from enum import Enum
-from typing import Any, Optional, Union
+from typing import Any
 
 from dara.core.base_definitions import DaraBaseModel as BaseModel
 from dara.core.definitions import ComponentInstanceType
@@ -26,7 +26,7 @@ from dara.core.definitions import ComponentInstanceType
 class ItemBadge(BaseModel):
     """Add a badge to an item in a list"""
 
-    color: Optional[str] = None
+    color: str | None = None
     label: str
 
 
@@ -35,17 +35,17 @@ class Item(BaseModel):
     A class for serializing a list of options for the select to show
     """
 
-    badge: Optional[ItemBadge] = None
-    image: Optional[str] = None
+    badge: ItemBadge | None = None
+    image: str | None = None
     label: str
-    value: Union[str, int, float, BaseModel]
+    value: str | int | float | BaseModel
 
     def __init__(
         self,
-        value: Union[str, int, float, BaseModel],
+        value: str | int | float | BaseModel,
         label: str,
-        badge: Optional[ItemBadge] = None,
-        image: Optional[str] = None,
+        badge: ItemBadge | None = None,
+        image: str | None = None,
     ):
         # Handle general enums in the value field
         if isinstance(value, Enum):
@@ -98,23 +98,23 @@ class CarouselItem(BaseModel):
     :param image_width: Optional string containing the width the image should take
     """
 
-    title: Optional[str] = None
-    subtitle: Optional[str] = None
-    component: Optional[ComponentInstanceType] = None
-    image: Optional[str] = None
-    image_alt: Optional[str] = None
-    image_height: Optional[str] = None
-    image_width: Optional[str] = None
+    title: str | None = None
+    subtitle: str | None = None
+    component: ComponentInstanceType | None = None
+    image: str | None = None
+    image_alt: str | None = None
+    image_height: str | None = None
+    image_width: str | None = None
 
     def __init__(
         self,
-        title: Optional[str] = None,
-        subtitle: Optional[str] = None,
-        component: Optional[ComponentInstanceType] = None,
-        image: Optional[str] = None,
-        image_alt: Optional[str] = None,
-        image_height: Optional[str] = None,
-        image_width: Optional[str] = None,
+        title: str | None = None,
+        subtitle: str | None = None,
+        component: ComponentInstanceType | None = None,
+        image: str | None = None,
+        image_alt: str | None = None,
+        image_height: str | None = None,
+        image_width: str | None = None,
     ):
         super().__init__(
             title=title,

dara/components/graphs/components/base_graph_component.py

@@ -15,7 +15,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from typing import Any, Dict, List, Optional, Union
+from typing import Any
 
 from pydantic import ConfigDict, SerializerFunctionWrapHandler, field_serializer
 
@@ -57,24 +57,24 @@ class BaseGraphComponent(StyledComponentInstance):
     :param zoom_thresholds: Optional user-defined zoom thresholds. See [ZoomThresholds](../definitions/#zoomthresholds) for more details.
     """
 
-    additional_legends: Optional[List[GraphLegend]] = None
-    allow_selection_when_not_editable: Optional[bool] = False
-    available_inputs: Optional[List[str]] = None
-    default_legends: Dict[Union[EditorMode, str], List[GraphLegend]] = DEFAULT_LEGENDS
-    disable_edge_add: Optional[bool] = None
-    disable_latent_node_add: Optional[bool] = None
-    disable_node_removal: Optional[bool] = None
-    editable: Optional[bool] = False
-    graph_layout: Optional[GraphLayout] = FcoseLayout()
-    non_removable_nodes: Optional[List[str]] = None
-    on_click_edge: Optional[Action] = None
-    on_click_node: Optional[Action] = None
-    on_update: Optional[Action] = None
-    require_focus_to_zoom: Optional[bool] = True
-    simultaneous_edge_node_selection: Optional[bool] = False
-    tooltip_size: Optional[int] = None
-    verbose_descriptions: Optional[bool] = None
-    zoom_thresholds: Optional[ZoomThresholds] = None
+    additional_legends: list[GraphLegend] | None = None
+    allow_selection_when_not_editable: bool | None = False
+    available_inputs: list[str] | None = None
+    default_legends: dict[EditorMode | str, list[GraphLegend]] = DEFAULT_LEGENDS
+    disable_edge_add: bool | None = None
+    disable_latent_node_add: bool | None = None
+    disable_node_removal: bool | None = None
+    editable: bool | None = False
+    graph_layout: GraphLayout | None = FcoseLayout()
+    non_removable_nodes: list[str] | None = None
+    on_click_edge: Action | None = None
+    on_click_node: Action | None = None
+    on_update: Action | None = None
+    require_focus_to_zoom: bool | None = True
+    simultaneous_edge_node_selection: bool | None = False
+    tooltip_size: int | None = None
+    verbose_descriptions: bool | None = None
+    zoom_thresholds: ZoomThresholds | None = None
 
     model_config = ConfigDict(
         arbitrary_types_allowed=True,

dara/components/graphs/components/causal_graph_viewer.py

@@ -15,8 +15,6 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from typing import Optional, Union
-
 from pydantic import field_validator, model_validator
 
 from cai_causal_graph import CausalGraph, Skeleton
@@ -248,8 +246,8 @@ class CausalGraphViewer(BaseGraphComponent):
 
     js_module = '@darajs/components'
 
-    causal_graph: Union[CausalGraph, DerivedVariable, Variable, dict, Skeleton]
-    editor_mode: Optional[EditorMode] = None
+    causal_graph: CausalGraph | DerivedVariable | Variable | dict | Skeleton
+    editor_mode: EditorMode | None = None
 
     @field_validator('causal_graph')
     @classmethod

dara/components/graphs/components/edge_encoder.py

@@ -15,7 +15,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from typing import Any, Dict, List, Optional, Union
+from typing import Any
 
 from fastapi.encoders import jsonable_encoder
 from pydantic import (
@@ -129,18 +129,18 @@ class VisualEdgeEncoder(StyledComponentInstance):
 
     js_module = '@darajs/components'
 
-    additional_legends: Optional[List[GraphLegend]] = None
-    allow_selection_when_not_editable: Optional[bool] = False
-    default_legends: Dict[Union[EditorMode, str], List[GraphLegend]] = DEFAULT_LEGENDS
-    editable: Optional[bool] = False
-    graph_layout: Optional[GraphLayout] = MarketingLayout()
-    initial_constraints: Optional[Union[List[EdgeConstraint], ClientVariable]] = None
-    nodes: Union[List[str], Dict[str, Node], ClientVariable]
-    on_click_edge: Optional[Action] = None
-    on_click_node: Optional[Action] = None
-    on_update: Optional[Action] = None
-    require_focus_to_zoom: Optional[bool] = True
-    tooltip_size: Optional[str] = None
+    additional_legends: list[GraphLegend] | None = None
+    allow_selection_when_not_editable: bool | None = False
+    default_legends: dict[EditorMode | str, list[GraphLegend]] = DEFAULT_LEGENDS
+    editable: bool | None = False
+    graph_layout: GraphLayout | None = MarketingLayout()
+    initial_constraints: list[EdgeConstraint] | ClientVariable | None = None
+    nodes: list[str] | dict[str, Node] | ClientVariable
+    on_click_edge: Action | None = None
+    on_click_node: Action | None = None
+    on_update: Action | None = None
+    require_focus_to_zoom: bool | None = True
+    tooltip_size: str | None = None
 
     model_config = ConfigDict(
         arbitrary_types_allowed=True,

dara/components/graphs/components/node_hierarchy_builder.py

@@ -15,7 +15,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, ClassVar
 
 from pydantic import field_validator
 
@@ -26,17 +26,17 @@ from dara.core.interactivity import ClientVariable
 
 
 class NodeMeta(BaseModel):
-    label_size: Optional[int] = None
-    wrap_text: Optional[bool] = None
-    label: Optional[str] = None
-    tooltip: Optional[Union[str, Dict[str, str]]] = None
+    label_size: int | None = None
+    wrap_text: bool | None = None
+    label: str | None = None
+    tooltip: str | dict[str, str] | None = None
 
 
 class Node(BaseModel):
     name: str
-    meta: Optional[NodeMeta] = None
+    meta: NodeMeta | None = None
 
-    Meta = NodeMeta
+    Meta: ClassVar[type[NodeMeta]] = NodeMeta
 
 
 class NodeHierarchyBuilder(StyledComponentInstance):
@@ -109,15 +109,15 @@ class NodeHierarchyBuilder(StyledComponentInstance):
     js_module = '@darajs/components'
 
     editable: bool = True
-    nodes: Union[List[List[str]], List[str], List[Node], List[List[Node]], ClientVariable]
-    node_font_size: Optional[int] = None
-    node_size: Optional[int] = None
-    on_update: Optional[Action] = None
+    nodes: list[list[str]] | list[str] | list[Node] | list[list[Node]] | ClientVariable
+    node_font_size: int | None = None
+    node_size: int | None = None
+    on_update: Action | None = None
     wrap_node_text: bool = True
 
     @field_validator('nodes')
     @classmethod
-    def validate_nodes(cls, nodes: Any) -> Union[ClientVariable, List[List[str]], List[List[Node]]]:
+    def validate_nodes(cls, nodes: Any) -> ClientVariable | list[list[str]] | list[list[Node]]:
         if isinstance(nodes, ClientVariable):
             return nodes
         if not isinstance(nodes, list):

dara/components/graphs/definitions.py

@@ -16,7 +16,7 @@ limitations under the License.
 """
 
 from enum import Enum
-from typing import ClassVar, Dict, List, Literal, Optional, Type, Union
+from typing import ClassVar, Literal
 
 from pydantic import Field
 
@@ -49,12 +49,17 @@ class CenterSymbol(str, Enum):
     BIDIRECTED = 'bidirected'
 
 
+EdgeLegendType = type['EdgeLegend']
+SpacerLegendType = type['SpacerLegend']
+NodeLegendType = type['NodeLegend']
+
+
 class Legend(BaseModel):
     type: str
 
-    Edge: ClassVar[Type['EdgeLegend']]
-    Spacer: ClassVar[Type['SpacerLegend']]
-    Node: ClassVar[Type['NodeLegend']]
+    Edge: ClassVar[EdgeLegendType]
+    Spacer: ClassVar[SpacerLegendType]
+    Node: ClassVar[NodeLegendType]
 
 
 class SpacerLegend(Legend):
@@ -65,7 +70,7 @@ class SpacerLegend(Legend):
     """
 
     type: Literal['spacer'] = Field(default='spacer', frozen=True)  # type: ignore
-    label: Optional[str] = None
+    label: str | None = None
 
 
 class EdgeLegend(Legend):
@@ -80,11 +85,11 @@ class EdgeLegend(Legend):
     """
 
     type: Literal['edge'] = Field(default='edge', frozen=True)  # type: ignore
-    label: Optional[str] = None
-    arrow_type: Optional[ArrowType] = ArrowType.NORMAL
-    center_symbol: Optional[CenterSymbol] = CenterSymbol.NONE
-    color: Optional[str] = 'theme.grey5'
-    dash_array: Optional[str] = None
+    label: str | None = None
+    arrow_type: ArrowType | None = ArrowType.NORMAL
+    center_symbol: CenterSymbol | None = CenterSymbol.NONE
+    color: str | None = 'theme.grey5'
+    dash_array: str | None = None
 
 
 class NodeLegend(Legend):
@@ -97,25 +102,25 @@ class NodeLegend(Legend):
     """
 
     type: Literal['node'] = Field(default='node', frozen=True)  # type: ignore
-    label: Optional[str] = None
-    color: Optional[str] = 'theme.blue4'
-    highlight_color: Optional[str] = 'theme.primary'
+    label: str | None = None
+    color: str | None = 'theme.blue4'
+    highlight_color: str | None = 'theme.primary'
 
 
 Legend.Edge = EdgeLegend
 Legend.Spacer = SpacerLegend
 Legend.Node = NodeLegend
 
-GraphLegend = Union[EdgeLegend, SpacerLegend, NodeLegend]
+GraphLegend = EdgeLegend | SpacerLegend | NodeLegend
 
-DEFAULT_NODE_LEGENDS: List[GraphLegend] = [
+DEFAULT_NODE_LEGENDS: list[GraphLegend] = [
     Legend.Node(color='theme.blue1', label='Latent'),
     Legend.Node(color='theme.secondary', label='Target'),
     Legend.Node(label='Other'),
 ]
 
 # Default legends for each editor mode
-DEFAULT_LEGENDS: Dict[Union[EditorMode, str], List[GraphLegend]] = {
+DEFAULT_LEGENDS: dict[EditorMode | str, list[GraphLegend]] = {
     EditorMode.DEFAULT: DEFAULT_NODE_LEGENDS,
     EditorMode.PAG: [
         *DEFAULT_NODE_LEGENDS,

dara/components/graphs/graph_layout.py

@@ -17,7 +17,6 @@ limitations under the License.
 
 import abc
 from enum import Enum
-from typing import List, Optional, Union
 
 from pydantic import Field
 
@@ -49,7 +48,7 @@ class LayeringAlgorithm(Enum):
     SIMPLEX = 'simplex'
 
 
-Number = Union[int, float]
+Number = int | float
 
 
 class GraphLayout(BaseModel, abc.ABC):
@@ -60,8 +59,8 @@ class GraphLayout(BaseModel, abc.ABC):
     :param node_font_size: node font size in pixels
     """
 
-    node_size: Optional[int] = None
-    node_font_size: Optional[int] = None
+    node_size: int | None = None
+    node_font_size: int | None = None
 
 
 class TiersConfig(BaseModel):
@@ -77,8 +76,8 @@ class TiersConfig(BaseModel):
     """
 
     group: str
-    order_nodes_by: Optional[str] = None
-    rank: Optional[List[str]] = None
+    order_nodes_by: str | None = None
+    rank: list[str] | None = None
 
 
 class TieringLayout(BaseModel):
@@ -135,8 +134,8 @@ class TieringLayout(BaseModel):
     :param orientation: Orientation the tiers are displayed in default is horizontal
     """
 
-    tiers: Optional[Union[TiersConfig, List[List[str]]]] = None
-    orientation: Optional[DirectionType] = DirectionType.HORIZONTAL
+    tiers: TiersConfig | list[list[str]] | None = None
+    orientation: DirectionType | None = DirectionType.HORIZONTAL
 
 
 class GroupingLayout(BaseModel):
@@ -170,7 +169,7 @@ class GroupingLayout(BaseModel):
     :param group: Path within node to group property which defines the group it belong to,
     """
 
-    group: Optional[str] = None
+    group: str | None = None
 
 
 class CircularLayout(GraphLayout):
@@ -232,16 +231,16 @@ class FcoseLayout(GraphLayout, TieringLayout, GroupingLayout):
 
     layout_type: GraphLayoutType = Field(default=GraphLayoutType.FCOSE, frozen=True)
 
-    edge_elasticity: Optional[Number] = 0.45
-    edge_length: Optional[Number] = 3
-    energy: Optional[Number] = 0.1
-    gravity: Optional[Number] = 35
-    gravity_range: Optional[Number] = 80
-    high_quality: Optional[bool] = True
-    iterations: Optional[int] = 2500
-    node_repulsion: Optional[Number] = 6_500
-    node_separation: Optional[Number] = 75
-    tier_separation: Optional[Number] = 200
+    edge_elasticity: Number | None = 0.45
+    edge_length: Number | None = 3
+    energy: Number | None = 0.1
+    gravity: Number | None = 35
+    gravity_range: Number | None = 80
+    high_quality: bool | None = True
+    iterations: int | None = 2500
+    node_repulsion: Number | None = 6_500
+    node_separation: Number | None = 75
+    tier_separation: Number | None = 200
 
 
 class ForceAtlasLayout(GraphLayout):
@@ -270,14 +269,14 @@ class ForceAtlasLayout(GraphLayout):
 
     layout_type: GraphLayoutType = Field(default=GraphLayoutType.FORCE_ATLAS, frozen=True)
 
-    barnes_hut_optimize: Optional[bool] = False
-    edge_weight_influence: Optional[Number] = 1
-    gravity: Optional[Number] = 0.2
-    iterations: Optional[int] = 10_000
-    lin_log_mode: Optional[bool] = True
-    outbound_attraction_distribution: Optional[bool] = True
-    scaling_ratio: Optional[Number] = 8
-    strong_gravity_mode: Optional[bool] = False
+    barnes_hut_optimize: bool | None = False
+    edge_weight_influence: Number | None = 1
+    gravity: Number | None = 0.2
+    iterations: int | None = 10_000
+    lin_log_mode: bool | None = True
+    outbound_attraction_distribution: bool | None = True
+    scaling_ratio: Number | None = 8
+    strong_gravity_mode: bool | None = False
 
 
 class MarketingLayout(GraphLayout, TieringLayout):
@@ -291,8 +290,8 @@ class MarketingLayout(GraphLayout, TieringLayout):
     """
 
     layout_type: GraphLayoutType = Field(default=GraphLayoutType.MARKETING, frozen=True)
-    target_location: Optional[TargetLocation] = None
-    tier_separation: Optional[Number] = 300
+    target_location: TargetLocation | None = None
+    tier_separation: Number | None = 300
 
 
 class PlanarLayout(GraphLayout, TieringLayout):
@@ -308,8 +307,8 @@ class PlanarLayout(GraphLayout, TieringLayout):
     """
 
     layout_type: GraphLayoutType = Field(default=GraphLayoutType.PLANAR, frozen=True)
-    layering_algorithm: Optional[LayeringAlgorithm] = LayeringAlgorithm.SIMPLEX
-    orientation: Optional[DirectionType] = None
+    layering_algorithm: LayeringAlgorithm | None = LayeringAlgorithm.SIMPLEX
+    orientation: DirectionType | None = None
 
 
 class SpringLayout(GraphLayout, TieringLayout, GroupingLayout):
@@ -329,9 +328,9 @@ class SpringLayout(GraphLayout, TieringLayout, GroupingLayout):
 
     layout_type: GraphLayoutType = Field(default=GraphLayoutType.SPRING, frozen=True)
 
-    collision_force: Optional[Number] = 2
-    gravity: Optional[Number] = -50
-    link_force: Optional[Number] = 5
-    warmup_ticks: Optional[Number] = 100
-    tier_separation: Optional[Number] = 300
-    group_repel_strength: Optional[Number] = 2000
+    collision_force: Number | None = 2
+    gravity: Number | None = -50
+    link_force: Number | None = 5
+    warmup_ticks: Number | None = 100
+    tier_separation: Number | None = 300
+    group_repel_strength: Number | None = 2000

dara/components/plotting/bokeh/bokeh.py

@@ -16,7 +16,7 @@ limitations under the License.
 """
 
 from json import dumps
-from typing import Any, List, Optional, Tuple
+from typing import Any
 
 from bokeh.document import Document
 from bokeh.themes import Theme
@@ -29,7 +29,7 @@ from dara.core.definitions import StyledComponentInstance
 SETTINGS = {'THEME': light_theme}
 
 
-def _get_theme(theme_input: Optional[dict]):
+def _get_theme(theme_input: dict | None):
     if theme_input is not None:
         return Theme(json=theme_input)
     else:
@@ -52,7 +52,7 @@ class Bokeh(StyledComponentInstance):
     js_module = '@darajs/components'
 
     document: str
-    events: Optional[List[Tuple[str, Action]]] = None
+    events: list[tuple[str, Action]] | None = None
 
     model_config = ConfigDict(arbitrary_types_allowed=True, use_enum_values=True)
 
@@ -60,8 +60,8 @@ class Bokeh(StyledComponentInstance):
         self,
         figure: Any = None,
         document: Any = None,
-        theme: Optional[dict] = None,
-        events: Optional[List[Tuple[str, Action]]] = None,
+        theme: dict | None = None,
+        events: list[tuple[str, Action]] | None = None,
         **kwargs,
     ):
         """

dara/components/plotting/bokeh/utils.py

@@ -15,8 +15,6 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from typing import Union
-
 from bokeh.models import CustomJS
 from bokeh.plotting import figure
 
@@ -37,7 +35,7 @@ def figure_events(fig: figure):
         :param
         """
 
-        def generate_event(args: Union[dict, None] = None):
+        def generate_event(args: dict | None = None):
             """
             Generate a CustomJS event with the code, event name and arguments provided
 

dara/components/plotting/plotly/plotly.py

@@ -16,7 +16,7 @@ limitations under the License.
 """
 
 from enum import Enum
-from typing import Any, List, Optional
+from typing import Any, ClassVar
 
 import plotly.graph_objects as go
 import plotly.io as pio
@@ -80,8 +80,8 @@ class PlotlyEvent(BaseModel):
     """
 
     event_name: PlotlyEventName
-    actions: Optional[List[Action]] = None
-    custom_js: Optional[str] = None
+    actions: list[Action] | None = None
+    custom_js: str | None = None
 
 
 class Plotly(StyledComponentInstance):
@@ -112,10 +112,10 @@ class Plotly(StyledComponentInstance):
     js_module = '@darajs/components'
 
     figure: str
-    events: Optional[List[PlotlyEvent]] = None
+    events: list[PlotlyEvent] | None = None
 
-    EventName = PlotlyEventName
-    Event = PlotlyEvent
+    EventName: ClassVar[type[PlotlyEventName]] = PlotlyEventName
+    Event: ClassVar[type[PlotlyEvent]] = PlotlyEvent
 
     model_config = ConfigDict(
         arbitrary_types_allowed=True,
@@ -125,8 +125,8 @@ class Plotly(StyledComponentInstance):
     def __init__(
         self,
         figure: Any = None,
-        theme: Optional[dict] = None,
-        events: Optional[List[PlotlyEvent]] = None,
+        theme: dict | None = None,
+        events: list[PlotlyEvent] | None = None,
         **kwargs,
     ):
         if theme is None and figure is not None:

dara/components/plotting/plotly/themes.py

@@ -15,7 +15,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from typing import Any, Dict
+from typing import Any
 
 from dara.components.plotting.palettes import CategoricalDark10, CategoricalLight10
 from dara.core.visual.themes import dark, light
@@ -25,7 +25,7 @@ assert light_colors is not None
 dark_colors = dark.Dark.colors
 assert dark_colors is not None
 
-light_theme: Dict[str, Dict[str, Any]] = {
+light_theme: dict[str, dict[str, Any]] = {
     'layout': {
         'paper_bgcolor': 'rgba(255,255,255,0)',  # Set the background color of the plot
         'plot_bgcolor': 'rgba(255,255,255,0)',  # Set the background color of the plot area
@@ -85,7 +85,7 @@ light_theme: Dict[str, Dict[str, Any]] = {
     },
 }
 
-dark_theme: Dict[str, Dict[str, Any]] = {
+dark_theme: dict[str, dict[str, Any]] = {
     'layout': {
         'paper_bgcolor': 'rgba(255,255,255,0)',  # Set the background color of the plot
         'plot_bgcolor': 'rgba(255,255,255,0)',  # Set the background color of the plot area

dara/components/smart/chat/config.py

@@ -2,7 +2,7 @@
 Copyright (c) 2023 by Impulse Innovations Ltd. Private and confidential. Part of the causaLens product.
 """
 
-from typing import Callable
+from collections.abc import Callable
 
 from dara.components.smart.chat.types import NewMessageBody
 from dara.core.definitions import EndpointConfiguration

dara/components/smart/chat/types.py

@@ -2,8 +2,6 @@
 Copyright (c) 2023 by Impulse Innovations Ltd. Private and confidential. Part of the causaLens product.
 """
 
-from typing import List, Optional
-
 from dara.core.base_definitions import DaraBaseModel as BaseModel
 
 
@@ -17,9 +15,9 @@ class ChatUserData(BaseModel):
     :param groups: list of groups user belongs to
     """
 
-    id: Optional[str] = None
+    id: str | None = None
     name: str
-    email: Optional[str] = None
+    email: str | None = None
 
 
 class ChatMessage(BaseModel):
@@ -50,5 +48,5 @@ class NewMessageBody(BaseModel):
     """
 
     app_url: str
-    users: List[ChatUserData]
+    users: list[ChatUserData]
     content: ChatMessage

dara/components/smart/code_editor/code_editor.py

@@ -15,7 +15,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from typing import Literal, Optional, Union
+from typing import Literal
 
 from dara.core.definitions import StyledComponentInstance
 from dara.core.interactivity import ClientVariable
@@ -45,4 +45,4 @@ class CodeEditor(StyledComponentInstance):
 
     script: ClientVariable
 
-    language: Optional[Union[Literal['json'], Literal['python'], Literal['markdown'], Literal['sql']]] = None
+    language: Literal['json'] | Literal['python'] | Literal['markdown'] | Literal['sql'] | None = None