refactor(antdx): components migration

Author: Col0ring

backend/modelscope_studio/components/antdx/__init__.py

@@ -21,9 +21,11 @@
 from .prompts.item import AntdXPromptsItem as PromptsItem
 from .sender import AntdXSender as Sender
 from .sender.header import AntdXSenderHeader as SenderHeader
+from .sender.switch import AntdXSenderSwitch as SenderSwitch
 from .suggestion import AntdXSuggestion as Suggestion
 from .suggestion.item import AntdXSuggestionItem as SuggestionItem
 from .thought_chain import AntdXThoughtChain as ThoughtChain
 from .thought_chain.item import AntdXThoughtChainItem as ThoughtChainItem
+from .thought_chain.thought_chain_item import AntdXThoughtChainThoughtChainItem as ThoughtChainThoughtChainItem
 from .welcome import AntdXWelcome as Welcome
 from .x_provider import AntdXXProvider as XProvider

backend/modelscope_studio/components/antdx/components.py

@@ -20,9 +20,11 @@
 from .prompts.item import AntdXPromptsItem
 from .sender import AntdXSender
 from .sender.header import AntdXSenderHeader
+from .sender.switch import AntdXSenderSwitch
 from .suggestion import AntdXSuggestion
 from .suggestion.item import AntdXSuggestionItem
 from .thought_chain import AntdXThoughtChain
 from .thought_chain.item import AntdXThoughtChainItem
+from .thought_chain.thought_chain_item import AntdXThoughtChainThoughtChainItem
 from .welcome import AntdXWelcome
 from .x_provider import AntdXXProvider

backend/modelscope_studio/components/antdx/conversations/__init__.py

@@ -32,12 +32,18 @@ class AntdXConversations(ModelScopeLayoutComponent):
         EventListener("menu_select",
                       callback=lambda block: block._internal.update(
                           bind_menu_select_event=True)),
+        EventListener("groupable_expand",
+                      callback=lambda block: block._internal.update(
+                          bind_groupable_expand_event=True)),
+        EventListener("creation_click",
+                      callback=lambda block: block._internal.update(
+                          bind_creation_click_event=True)),
     ]
 
     # supported slots
     SLOTS = [
         'menu.expandIcon', 'menu.overflowedIndicator', 'menu.trigger',
-        'groupable.title', 'items'
+        'groupable.label', 'items', 'creation.icon', 'creation.label'
     ]
 
     def __init__(
@@ -49,8 +55,10 @@ def __init__(
             items: list[dict] | None = None,
             menu: str | dict | None = None,
             groupable: bool | dict | None = None,
-            styles: dict | str | None= None,
-            class_names: dict | str | None= None,
+            shortcut_keys: dict | None = None,
+            creation: dict | None = None,
+            styles: dict | str | None = None,
+            class_names: dict | str | None = None,
             root_class_name: str | None = None,
             as_item: str | None = None,
             _internal: None = None,
@@ -74,6 +82,8 @@ def __init__(
         self.items = items
         self.menu = menu
         self.groupable = groupable
+        self.shortcut_keys = shortcut_keys
+        self.creation = creation
         self.styles = styles
         self.class_names = class_names
         self.root_class_name = root_class_name

backend/modelscope_studio/components/antdx/conversations/item/__init__.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any
+from typing import Any, Literal
 
 from .....utils.dev import ModelScopeLayoutComponent, resolve_frontend_dir
 
@@ -21,13 +21,14 @@ class AntdXConversationsItem(ModelScopeLayoutComponent):
     def __init__(
             self,
             label: str | None = None,
-            additional_props: dict | None = None,
             *,
             key: str | None = None,
-            timestamp: int | float | None = None,
+            type: Literal['divider'] | None = None,
             group: str | None = None,
             icon: str | None = None,
             disabled: bool | None = None,
+            dashed: bool | None = None,
+            additional_props: dict | None = None,
             as_item: str | None = None,
             _internal: None = None,
             # gradio properties
@@ -46,7 +47,8 @@ def __init__(
                          **kwargs)
         self.additional_props = additional_props
         self.label = label
-        self.timestamp = timestamp
+        self.type = type
+        self.dashed = dashed
         self.group = group
         self.icon = icon
         self.disabled = disabled

backend/modelscope_studio/components/antdx/prompts/__init__.py

@@ -27,16 +27,18 @@ class AntdXPrompts(ModelScopeLayoutComponent):
 
     def __init__(
             self,
-            additional_props: dict | None = None,
             *,
             items: list[dict] | None = None,
             prefix_cls: str | None = None,
             title: str | None = None,
             vertical: bool | None = None,
+            fade_in: bool | None = None,
+            fade_in_left: bool | None = None,
             wrap: bool | None = None,
-            styles: dict | str | None= None,
-            class_names: dict | str | None= None,
+            styles: dict | str | None = None,
+            class_names: dict | str | None = None,
             root_class_name: str | None = None,
+            additional_props: dict | None = None,
             as_item: str | None = None,
             _internal: None = None,
             # gradio properties
@@ -58,6 +60,8 @@ def __init__(
         self.prefix_cls = prefix_cls
         self.title = title
         self.vertical = vertical
+        self.fade_in = fade_in
+        self.fade_in_left = fade_in_left
         self.wrap = wrap
         self.styles = styles
         self.class_names = class_names

backend/modelscope_studio/components/antdx/prompts/item/__init__.py

@@ -18,12 +18,12 @@ class AntdXPromptsItem(ModelScopeLayoutComponent):
     def __init__(
             self,
             label: str | None = None,
-            additional_props: dict | None = None,
             *,
             key: str | None = None,
             description: str | None = None,
             icon: str | None = None,
             disabled: bool | None = None,
+            additional_props: dict | None = None,
             as_item: str | None = None,
             _internal: None = None,
             # gradio properties

backend/modelscope_studio/components/antdx/sender/__init__.py

@@ -7,6 +7,7 @@
 
 from ....utils.dev import ModelScopeDataLayoutComponent, resolve_frontend_dir
 from .header import AntdXSenderHeader
+from .switch import AntdXSenderSwitch
 
 
 # as inputs, outputs
@@ -15,6 +16,7 @@ class AntdXSender(ModelScopeDataLayoutComponent):
     Ant Design X: https://x.ant.design/components/sender
     """
     Header = AntdXSenderHeader
+    Switch = AntdXSenderSwitch
 
     EVENTS = [
         EventListener("change",
@@ -51,32 +53,41 @@ class AntdXSender(ModelScopeDataLayoutComponent):
         EventListener("paste_file",
                       callback=lambda block: block._internal.update(
                           bind_pasteFile_event=True)),
+        EventListener("skill_closable_close",
+                      callback=lambda block: block._internal.update(
+                          bind_skill_closable_close_event=True)),
     ]
 
     # supported slots
-    SLOTS = ['actions', 'header', 'prefix', 'footer']
+    SLOTS = [
+        'suffix', 'header', 'prefix', 'footer', 'skill.title',
+        'skill.toolTip.title'
+        'skill.closable.closeIcon'
+    ]
 
     def __init__(
             self,
             value: str | None = None,
-            additional_props: dict | None = None,
             *,
-            auto_size: bool | dict | None = None,
-            footer: str | None = None,
-            actions: str | bool | None = None,
             allow_speech: bool | dict | None = None,
-            class_names: dict | str | None= None,
+            class_names: dict | str | None = None,
             components: dict | None = None,
             default_value: str | None = None,
-            loading: bool | None = None,
             disabled: bool | None = None,
+            auto_size: bool | dict | None = None,
+            loading: bool | None = None,
+            suffix: str | bool | None = None,
+            footer: str | None = None,
             header: str | None = None,
             prefix: str | None = None,
             read_only: bool | None = None,
-            styles: dict | str | None= None,
+            styles: dict | str | None = None,
             submit_type: Literal['enter', 'shiftEnter'] | None = None,
             placeholder: str | None = None,
+            slot_config: list[dict] | None = None,
+            skill: dict | None = None,
             root_class_name: str | None = None,
+            additional_props: dict | None = None,
             as_item: str | None = None,
             _internal: None = None,
             # gradio properties
@@ -108,7 +119,9 @@ def __init__(
         self.read_only = read_only
         self.loading = loading
         self.header = header
-        self.actions = actions
+        self.suffix = suffix
+        self.slot_config = slot_config
+        self.skill = skill
         self.placeholder = placeholder
         self.root_class_name = root_class_name
 

backend/modelscope_studio/components/antdx/sender/header/__init__.py

@@ -21,14 +21,14 @@ class AntdXSenderHeader(ModelScopeLayoutComponent):
 
     def __init__(
             self,
-            additional_props: dict | None = None,
             *,
             closable: bool | None = None,
             force_render: bool | None = None,
-            styles: dict | str | None= None,
-            class_names: dict | str | None= None,
+            styles: dict | str | None = None,
+            class_names: dict | str | None = None,
             open: bool | None = None,
             title: str | None = None,
+            additional_props: dict | None = None,
             as_item: str | None = None,
             _internal: None = None,
             # gradio properties

backend/modelscope_studio/components/antdx/sender/switch/__init__.py

@@ -0,0 +1,80 @@
+from __future__ import annotations
+
+from gradio.events import EventListener
+
+from .....utils.dev import ModelScopeDataLayoutComponent, resolve_frontend_dir
+
+
+class AntdXSenderSwitch(ModelScopeDataLayoutComponent):
+    """
+    Ant Design X: https://x.ant.design/components/sender
+    """
+    EVENTS = [
+        EventListener("change",
+                      callback=lambda block: block._internal.update(
+                          bind_change_event=True)),
+    ]
+
+    # supported slots
+    SLOTS = ['checkedChildren', 'unCheckedChildren', 'icon']
+
+    def __init__(
+            self,
+            value: bool | None = None,
+            *,
+            checked_children: str | None = None,
+            un_checked_children: str | None = None,
+            icon: str | None = None,
+            disabled: bool | None = None,
+            loading: bool | None = None,
+            default_value: bool | None = None,
+            styles: dict | str | None = None,
+            class_names: dict | str | None = None,
+            root_class_name: str | None = None,
+            additional_props: dict | None = None,
+            as_item: str | None = None,
+            _internal: None = None,
+            # gradio properties
+            visible: bool = True,
+            elem_id: str | None = None,
+            elem_classes: list[str] | str | None = None,
+            elem_style: dict | None = None,
+            render: bool = True,
+            **kwargs):
+        super().__init__(value=value,
+                         visible=visible,
+                         elem_id=elem_id,
+                         elem_classes=elem_classes,
+                         render=render,
+                         as_item=as_item,
+                         elem_style=elem_style,
+                         **kwargs)
+        self.additional_props = additional_props
+        self.styles = styles
+        self.class_names = class_names
+        self.root_class_name = root_class_name
+        self.checked_children = checked_children
+        self.un_checked_children = un_checked_children
+        self.icon = icon
+        self.disabled = disabled
+        self.loading = loading
+        self.default_value = default_value
+
+    FRONTEND_DIR = resolve_frontend_dir("sender", 'switch', type="antdx")
+
+    @property
+    def skip_api(self):
+        return True
+
+    def preprocess(self, payload: None) -> None:
+        return payload
+
+    def postprocess(self, value: None) -> None:
+
+        return value
+
+    def example_payload(self) -> None:
+        return None
+
+    def example_value(self) -> None:
+        return None

backend/modelscope_studio/components/antdx/thought_chain/__init__.py

@@ -6,6 +6,7 @@
 
 from ....utils.dev import ModelScopeLayoutComponent, resolve_frontend_dir
 from .item import AntdXThoughtChainItem
+from .thought_chain_item import AntdXThoughtChainThoughtChainItem
 
 
 class AntdXThoughtChain(ModelScopeLayoutComponent):
@@ -14,11 +15,12 @@ class AntdXThoughtChain(ModelScopeLayoutComponent):
     """
 
     Item = AntdXThoughtChainItem
+    ThoughtChainItem = AntdXThoughtChainThoughtChainItem
 
     EVENTS = [
-        EventListener("collapsible_expand",
+        EventListener("expand",
                       callback=lambda block: block._internal.update(
-                          bind_collapsible_expand_event=True),
+                          bind_expand_event=True),
                       doc="Callback function when expanded keys change."),
     ]
 
@@ -27,14 +29,15 @@ class AntdXThoughtChain(ModelScopeLayoutComponent):
 
     def __init__(
             self,
-            additional_props: dict | None = None,
             *,
-            collapsible: bool | dict | None = None,
+            expanded_keys: list[str] | None = None,
+            default_expanded_keys: list[str] | None = None,
             items: list[dict] | None = None,
-            size: Literal['small', 'middle', 'large'] | None = None,
+            line: bool | Literal['solid', 'dashed', "dotted"] | None = None,
             prefix_cls: str | None = None,
-            styles: dict | str | None= None,
-            class_names: dict | str | None= None,
+            additional_props: dict | None = None,
+            styles: dict | str | None = None,
+            class_names: dict | str | None = None,
             root_class_name: str | None = None,
             as_item: str | None = None,
             _internal: None = None,
@@ -53,9 +56,10 @@ def __init__(
                          elem_style=elem_style,
                          **kwargs)
         self.additional_props = additional_props
-        self.collapsible = collapsible
         self.items = items
-        self.size = size
+        self.expanded_keys = expanded_keys
+        self.default_expanded_keys = default_expanded_keys
+        self.line = line
         self.prefix_cls = prefix_cls
         self.styles = styles
         self.class_names = class_names