waldiez 0.2.2__py3-none-any.whl → 0.3.1__py3-none-any.whl

waldiez/models/agents/agents.py

@@ -1,3 +1,5 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Waldiez agents model."""
 
 from typing import Iterator, List
@@ -10,6 +12,7 @@ from .agent import WaldiezAgent
 from .assistant import WaldiezAssistant
 from .group_manager.group_manager import WaldiezGroupManager
 from .rag_user import WaldiezRagUser
+from .swarm_agent import WaldiezSwarmAgent
 from .user_proxy import WaldiezUserProxy
 
 
@@ -23,7 +26,7 @@ class WaldiezAgents(WaldiezBase):
     assistants : List[WaldiezAssistant]
         Assistant agents.
     managers : List[WaldiezGroupManager]
-        Group chat mangers.
+        Group chat managers.
     rag_users : List[WaldiezRagUser]
         RAG user agents.
     """
@@ -48,7 +51,7 @@ class WaldiezAgents(WaldiezBase):
         List[WaldiezGroupManager],
         Field(
             title="Managers.",
-            description="Group chat mangers",
+            description="Group chat managers",
             default_factory=list,
         ),
     ]
@@ -60,6 +63,14 @@ class WaldiezAgents(WaldiezBase):
             default_factory=list,
         ),
     ]
+    swarm_agents: Annotated[
+        List[WaldiezSwarmAgent],
+        Field(
+            title="Swarm Agents.",
+            description="Swarm agents",
+            default_factory=list,
+        ),
+    ]
 
     @property
     def members(self) -> Iterator[WaldiezAgent]:
@@ -72,8 +83,9 @@ class WaldiezAgents(WaldiezBase):
         """
         yield from self.users
         yield from self.assistants
-        yield from self.managers
         yield from self.rag_users
+        yield from self.swarm_agents
+        yield from self.managers
 
     @model_validator(mode="after")
     def validate_agents(self) -> Self:

waldiez/models/agents/assistant/__init__.py

@@ -1,3 +1,5 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Assistant agent model."""
 
 from .assistant import WaldiezAssistant

waldiez/models/agents/assistant/assistant.py

@@ -1,3 +1,5 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Assistant agent model."""
 
 from pydantic import Field

waldiez/models/agents/assistant/assistant_data.py

@@ -1,3 +1,5 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Assistant agent data model."""
 
 from pydantic import Field

waldiez/models/agents/group_manager/__init__.py

@@ -1,8 +1,13 @@
-"""Group chat manger agent."""
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
+"""Group chat manager agent."""
 
 from .group_manager import WaldiezGroupManager
 from .group_manager_data import WaldiezGroupManagerData
 from .speakers import (
+    CUSTOM_SPEAKER_SELECTION,
+    CUSTOM_SPEAKER_SELECTION_ARGS,
+    CUSTOM_SPEAKER_SELECTION_TYPES,
     WaldiezGroupManagerSpeakers,
     WaldiezGroupManagerSpeakersSelectionMethod,
     WaldiezGroupManagerSpeakersSelectionMode,
@@ -10,6 +15,9 @@ from .speakers import (
 )
 
 __all__ = [
+    "CUSTOM_SPEAKER_SELECTION",
+    "CUSTOM_SPEAKER_SELECTION_ARGS",
+    "CUSTOM_SPEAKER_SELECTION_TYPES",
     "WaldiezGroupManager",
     "WaldiezGroupManagerData",
     "WaldiezGroupManagerSpeakers",

waldiez/models/agents/group_manager/group_manager.py

@@ -1,3 +1,5 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Group chat manager agent."""
 
 from typing import List, Literal

waldiez/models/agents/group_manager/group_manager_data.py

@@ -1,3 +1,5 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Group chat manager data."""
 
 from typing import Optional

waldiez/models/agents/group_manager/speakers.py

@@ -1,12 +1,13 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Group chat speakers."""
 
-from typing import Dict, List, Optional, Union
+from typing import Dict, List, Optional, Tuple, Union
 
-from pydantic import ConfigDict, Field, model_validator
-from pydantic.alias_generators import to_camel
+from pydantic import Field, model_validator
 from typing_extensions import Annotated, Literal, Self
 
-from ...common import WaldiezBase, WaldiezMethodName, check_function
+from ...common import WaldiezBase, check_function, generate_function
 
 WaldiezGroupManagerSpeakersSelectionMethod = Literal[
     "auto",
@@ -18,6 +19,13 @@ WaldiezGroupManagerSpeakersSelectionMethod = Literal[
 WaldiezGroupManagerSpeakersSelectionMode = Literal["repeat", "transition"]
 WaldiezGroupManagerSpeakersTransitionsType = Literal["allowed", "disallowed"]
 
+CUSTOM_SPEAKER_SELECTION = "custom_speaker_selection"
+CUSTOM_SPEAKER_SELECTION_ARGS = ["last_speaker", "groupchat"]
+CUSTOM_SPEAKER_SELECTION_TYPES = (
+    ["ConversableAgent", "GroupChat"],
+    "Optional[Union[Agent, str]]",
+)
+
 
 class WaldiezGroupManagerSpeakers(WaldiezBase):
     """Group chat speakers.
@@ -72,13 +80,6 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
         Validate the speakers config.
     """
 
-    model_config = ConfigDict(
-        extra="forbid",
-        alias_generator=to_camel,
-        populate_by_name=True,
-        frozen=False,
-    )
-
     selection_method: Annotated[
         WaldiezGroupManagerSpeakersSelectionMethod,
         Field(
@@ -181,6 +182,40 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
         """
         return self._custom_method_string
 
+    def get_custom_method_function(
+        self,
+        name_prefix: Optional[str] = None,
+        name_suffix: Optional[str] = None,
+    ) -> Tuple[str, str]:
+        """Get the custom method function.
+
+        Parameters
+        ----------
+        name_prefix : str
+            The function name prefix.
+        name_suffix : str
+            The function name suffix.
+
+        Returns
+        -------
+        Tuple[str, str]
+            The custom method function and the function name.
+        """
+        function_name = CUSTOM_SPEAKER_SELECTION
+        if name_prefix:
+            function_name = f"{name_prefix}_{function_name}"
+        if name_suffix:
+            function_name = f"{function_name}_{name_suffix}"
+        return (
+            generate_function(
+                function_name=function_name,
+                function_args=CUSTOM_SPEAKER_SELECTION_ARGS,
+                function_types=CUSTOM_SPEAKER_SELECTION_TYPES,
+                function_body=self.custom_method_string or "",
+            ),
+            function_name,
+        )
+
     @model_validator(mode="after")
     def validate_group_speakers_config(self) -> Self:
         """Validate the speakers config.
@@ -198,9 +233,10 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
         if self.selection_method == "custom":
             if not self.selection_custom_method:
                 raise ValueError("No custom method provided.")
-            function_name: WaldiezMethodName = "custom_speaker_selection"
             is_valid, error_or_body = check_function(
-                self.selection_custom_method, function_name=function_name
+                code_string=self.selection_custom_method,
+                function_name=CUSTOM_SPEAKER_SELECTION,
+                function_args=CUSTOM_SPEAKER_SELECTION_ARGS,
             )
             if not is_valid or not error_or_body:
                 # pylint: disable=inconsistent-quotes

waldiez/models/agents/rag_user/__init__.py

@@ -1,11 +1,19 @@
-"""RAG user agent.
-# pylint: disable=line-too-long
-It extends a user agent and has RAG related parameters.
-"""
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
+"""It extends a user agent and has RAG related parameters."""
 
 from .rag_user import WaldiezRagUser
 from .rag_user_data import WaldiezRagUserData
 from .retrieve_config import (
+    CUSTOM_EMBEDDING_FUNCTION,
+    CUSTOM_EMBEDDING_FUNCTION_ARGS,
+    CUSTOM_EMBEDDING_FUNCTION_TYPES,
+    CUSTOM_TEXT_SPLIT_FUNCTION,
+    CUSTOM_TEXT_SPLIT_FUNCTION_ARGS,
+    CUSTOM_TEXT_SPLIT_FUNCTION_TYPES,
+    CUSTOM_TOKEN_COUNT_FUNCTION,
+    CUSTOM_TOKEN_COUNT_FUNCTION_ARGS,
+    CUSTOM_TOKEN_COUNT_FUNCTION_TYPES,
     WaldiezRagUserChunkMode,
     WaldiezRagUserModels,
     WaldiezRagUserRetrieveConfig,
@@ -15,6 +23,15 @@ from .retrieve_config import (
 from .vector_db_config import WaldiezRagUserVectorDbConfig
 
 __all__ = [
+    "CUSTOM_EMBEDDING_FUNCTION",
+    "CUSTOM_EMBEDDING_FUNCTION_ARGS",
+    "CUSTOM_EMBEDDING_FUNCTION_TYPES",
+    "CUSTOM_TEXT_SPLIT_FUNCTION",
+    "CUSTOM_TEXT_SPLIT_FUNCTION_ARGS",
+    "CUSTOM_TEXT_SPLIT_FUNCTION_TYPES",
+    "CUSTOM_TOKEN_COUNT_FUNCTION",
+    "CUSTOM_TOKEN_COUNT_FUNCTION_ARGS",
+    "CUSTOM_TOKEN_COUNT_FUNCTION_TYPES",
     "WaldiezRagUser",
     "WaldiezRagUserData",
     "WaldiezRagUserModels",

waldiez/models/agents/rag_user/rag_user.py

@@ -1,5 +1,7 @@
-# pylint: disable=line-too-long
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
 """RAG user agent.
+
 It extends a user agent and has RAG related parameters (`retrieve_config`).
 """
 

waldiez/models/agents/rag_user/rag_user_data.py

@@ -1,3 +1,5 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Waldiez RAG user agent data."""
 
 from pydantic import Field

waldiez/models/agents/rag_user/retrieve_config.py

@@ -1,12 +1,15 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
 """RAG user agent retrieve config."""
 
-from typing import Dict, List, Optional, Union
+import os
+from pathlib import Path
+from typing import Dict, List, Optional, Tuple, Union
 
-from pydantic import ConfigDict, Field, model_validator
-from pydantic.alias_generators import to_camel
+from pydantic import Field, model_validator
 from typing_extensions import Annotated, Literal, Self
 
-from ...common import WaldiezBase, WaldiezMethodName, check_function
+from ...common import WaldiezBase, check_function, generate_function
 from .vector_db_config import WaldiezRagUserVectorDbConfig
 
 WaldiezRagUserTask = Literal["code", "qa", "default"]
@@ -19,6 +22,39 @@ WaldiezRagUserModels: Dict[WaldiezRagUserVectorDb, str] = {
     "qdrant": "BAAI/bge-small-en-v1.5",
 }
 
+CUSTOM_EMBEDDING_FUNCTION = "custom_embedding_function"
+CUSTOM_EMBEDDING_FUNCTION_ARGS: List[str] = []
+CUSTOM_EMBEDDING_FUNCTION_TYPES: Tuple[List[str], str] = (
+    [],
+    "Callable[..., Any]",
+)
+CUSTOM_TOKEN_COUNT_FUNCTION = "custom_token_count_function"  # nosec
+CUSTOM_TOKEN_COUNT_FUNCTION_ARGS = ["text", "model"]  # nosec
+CUSTOM_TOKEN_COUNT_FUNCTION_TYPES = (
+    ["str", "str"],
+    "int",
+)
+
+CUSTOM_TEXT_SPLIT_FUNCTION = "custom_text_split_function"
+CUSTOM_TEXT_SPLIT_FUNCTION_ARGS = [
+    "text",
+    "max_tokens",
+    "chunk_mode",
+    "must_break_at_empty_line",
+    "overlap",
+]
+CUSTOM_TEXT_SPLIT_FUNCTION_TYPES = (
+    ["str", "int", "str", "bool", "int"],
+    "List[str]",
+)
+NOT_LOCAL = (
+    "http://",
+    "https://",
+    "ftp://",
+    "ftps://",
+    "sftp://",
+)
+
 
 class WaldiezRagUserRetrieveConfig(WaldiezBase):
     """RAG user agent.
@@ -140,13 +176,6 @@ class WaldiezRagUserRetrieveConfig(WaldiezBase):
         Validate the RAG user data.
     """
 
-    model_config = ConfigDict(
-        extra="forbid",
-        alias_generator=to_camel,
-        populate_by_name=True,
-        frozen=False,
-    )
-
     task: Annotated[
         WaldiezRagUserTask,
         Field(
@@ -502,6 +531,108 @@ class WaldiezRagUserRetrieveConfig(WaldiezBase):
         """
         return self._text_split_function_string
 
+    def get_custom_embedding_function(
+        self,
+        name_prefix: Optional[str] = None,
+        name_suffix: Optional[str] = None,
+    ) -> Tuple[str, str]:
+        """Generate the custom embedding function.
+
+        Parameters
+        ----------
+        name_prefix : str
+            The function name prefix.
+        name_suffix : str
+            The function name suffix.
+
+        Returns
+        -------
+        Tuple[str, str]
+            The custom embedding function and the function name.
+        """
+        function_name = CUSTOM_EMBEDDING_FUNCTION
+        if name_prefix:
+            function_name = f"{name_prefix}_{function_name}"
+        if name_suffix:
+            function_name = f"{function_name}_{name_suffix}"
+        return (
+            generate_function(
+                function_name=function_name,
+                function_args=CUSTOM_EMBEDDING_FUNCTION_ARGS,
+                function_types=CUSTOM_EMBEDDING_FUNCTION_TYPES,
+                function_body=self.embedding_function_string or "",
+            ),
+            function_name,
+        )
+
+    def get_custom_token_count_function(
+        self,
+        name_prefix: Optional[str] = None,
+        name_suffix: Optional[str] = None,
+    ) -> Tuple[str, str]:
+        """Generate the custom token count function.
+
+        Parameters
+        ----------
+        name_prefix : str
+            The function name prefix.
+        name_suffix : str
+            The function name suffix.
+
+        Returns
+        -------
+        Tuple[str, str]
+            The custom token count function and the function name.
+        """
+        function_name = CUSTOM_TOKEN_COUNT_FUNCTION
+        if name_prefix:
+            function_name = f"{name_prefix}_{function_name}"
+        if name_suffix:
+            function_name = f"{function_name}_{name_suffix}"
+        return (
+            generate_function(
+                function_name=function_name,
+                function_args=CUSTOM_TOKEN_COUNT_FUNCTION_ARGS,
+                function_types=CUSTOM_TOKEN_COUNT_FUNCTION_TYPES,
+                function_body=self.token_count_function_string or "",
+            ),
+            function_name,
+        )
+
+    def get_custom_text_split_function(
+        self,
+        name_prefix: Optional[str] = None,
+        name_suffix: Optional[str] = None,
+    ) -> Tuple[str, str]:
+        """Generate the custom text split function.
+
+        Parameters
+        ----------
+        name_prefix : str
+            The function name prefix.
+        name_suffix : str
+            The function name suffix.
+
+        Returns
+        -------
+        Tuple[str, str]
+            The custom text split function and the function name.
+        """
+        function_name = CUSTOM_TEXT_SPLIT_FUNCTION
+        if name_prefix:
+            function_name = f"{name_prefix}_{function_name}"
+        if name_suffix:
+            function_name = f"{function_name}_{name_suffix}"
+        return (
+            generate_function(
+                function_name=function_name,
+                function_args=CUSTOM_TEXT_SPLIT_FUNCTION_ARGS,
+                function_types=CUSTOM_TEXT_SPLIT_FUNCTION_TYPES,
+                function_body=self.text_split_function_string or "",
+            ),
+            function_name,
+        )
+
     def validate_custom_embedding_function(self) -> None:
         """Validate the custom embedding function.
 
@@ -516,9 +647,10 @@ class WaldiezRagUserRetrieveConfig(WaldiezBase):
                     "The embedding_function is required "
                     "if use_custom_embedding is True."
                 )
-            function_name: WaldiezMethodName = "custom_embedding_function"
             valid, error_or_content = check_function(
-                self.embedding_function, function_name
+                code_string=self.embedding_function,
+                function_name=CUSTOM_EMBEDDING_FUNCTION,
+                function_args=CUSTOM_EMBEDDING_FUNCTION_ARGS,
             )
             if not valid:
                 raise ValueError(error_or_content)
@@ -538,9 +670,10 @@ class WaldiezRagUserRetrieveConfig(WaldiezBase):
                     "The custom_token_count_function is required "
                     "if use_custom_token_count is True."
                 )
-            function_name: WaldiezMethodName = "custom_token_count_function"
             valid, error_or_content = check_function(
-                self.custom_token_count_function, function_name
+                code_string=self.custom_token_count_function,
+                function_name=CUSTOM_TOKEN_COUNT_FUNCTION,
+                function_args=CUSTOM_TOKEN_COUNT_FUNCTION_ARGS,
             )
             if not valid:
                 raise ValueError(error_or_content)
@@ -560,14 +693,49 @@ class WaldiezRagUserRetrieveConfig(WaldiezBase):
                     "The custom_text_split_function is required "
                     "if use_custom_text_split is True."
                 )
-            function_name: WaldiezMethodName = "custom_text_split_function"
             valid, error_or_content = check_function(
-                self.custom_text_split_function, function_name
+                code_string=self.custom_text_split_function,
+                function_name=CUSTOM_TEXT_SPLIT_FUNCTION,
+                function_args=CUSTOM_TEXT_SPLIT_FUNCTION_ARGS,
             )
             if not valid:
                 raise ValueError(error_or_content)
             self._text_split_function_string = error_or_content
 
+    def validate_docs_path(self) -> None:
+        """Validate the docs path.
+
+        Raises
+        ------
+        ValueError
+            If the validation fails.
+        """
+        if not self.docs_path:
+            return
+        # if urls or directories ok, if files they should resolve
+        doc_paths = (
+            [self.docs_path]
+            if isinstance(self.docs_path, str)
+            else self.docs_path
+        )
+        for index, path in enumerate(doc_paths):
+            is_remote, is_raw = is_remote_path(path)
+            if is_remote:
+                if not is_raw:
+                    doc_paths[index] = f'r"{path}"'
+                continue
+            if path.startswith("file://"):
+                path = path[len("file://") :]
+            if path.startswith('r"file://"'):
+                path = path[len('r"file://"') :]
+            if string_represents_folder(path):
+                if not is_raw:
+                    doc_paths[index] = f'r"{path}"'
+                continue
+            resolved = resolve_path(path, is_raw)
+            doc_paths[index] = resolved
+        self.docs_path = doc_paths
+
     @model_validator(mode="after")
     def validate_rag_user_data(self) -> Self:
         """Validate the RAG user data.
@@ -585,8 +753,91 @@ class WaldiezRagUserRetrieveConfig(WaldiezBase):
         self.validate_custom_embedding_function()
         self.validate_custom_token_count_function()
         self.validate_custom_text_split_function()
+        self.validate_docs_path()
         if not self.db_config.model:
             self.db_config.model = WaldiezRagUserModels[self.vector_db]
         if isinstance(self.n_results, int) and self.n_results < 1:
             self.n_results = None
         return self
+
+
+def string_represents_folder(path: str) -> bool:
+    """Check if a string represents a folder.
+
+    Parameters
+    ----------
+    path : str
+        The string to check (does not need to exist).
+
+    Returns
+    -------
+    bool
+        True if the path is likely a folder, False if it's likely a file.
+    """
+    if path.endswith(os.path.sep):
+        return True
+    if os.path.isdir(path):
+        return True
+    return not os.path.splitext(path)[1]
+
+
+def is_remote_path(path: str) -> Tuple[bool, bool]:
+    """Check if a path is a remote path.
+
+    Parameters
+    ----------
+    path : str
+        The path to check.
+
+    Returns
+    -------
+    Tuple[bool, bool]
+        If the path is a remote path and if it's a raw string.
+    """
+    is_raw = path.startswith(("r'", 'r"'))
+    for not_local in NOT_LOCAL:
+        if path.startswith((not_local, f'r"{not_local}"', f"r'{not_local}'")):
+            return True, is_raw
+    return False, is_raw
+
+
+def resolve_path(path: str, is_raw: bool) -> str:
+    """Try to resolve a path.
+
+    Parameters
+    ----------
+    path : str
+        The path to resolve.
+    is_raw : bool
+        If the path is a raw string.
+    Returns
+    -------
+    Path
+        The resolved path.
+
+    Raises
+    ------
+    ValueError
+        If the path is not a valid local path.
+    """
+    # pylint: disable=broad-except
+    path_string = path
+    if is_raw:
+        path_string = path[2:-1]
+    if path_string.startswith(NOT_LOCAL):
+        return path
+    try:
+        resolved = Path(path_string).resolve()
+    except BaseException as error:  # pragma: no cover
+        # check if 'r'... is needed
+        raw_string = f'r"{path}"'
+        try:
+            Path(raw_string).resolve()
+        except BaseException:
+            raise ValueError(
+                f"Path {path} is not a valid local path."
+            ) from error
+        return raw_string
+    if not resolved.exists():
+        raise ValueError(f"Path {path} does not exist.")
+    return f'r"{resolved}"'

waldiez/models/agents/rag_user/vector_db_config.py

@@ -1,7 +1,9 @@
-"""The vector db config for the RAG user agent."""
-
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
 # flake8: noqa E501
 
+"""The vector db config for the RAG user agent."""
+
 from pathlib import Path
 from typing import Any, Dict, Optional
 
@@ -45,7 +47,7 @@ class WaldiezRagUserVectorDbConfig(WaldiezBase):
     """
 
     model_config = ConfigDict(
-        extra="forbid",
+        extra="ignore",
         alias_generator=to_camel,
         populate_by_name=True,
         frozen=False,

waldiez/models/agents/swarm_agent/__init__.py

@@ -0,0 +1,49 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
+"""Swarm Agent."""
+
+from .after_work import (
+    CUSTOM_AFTER_WORK,
+    CUSTOM_AFTER_WORK_ARGS,
+    CUSTOM_AFTER_WORK_TYPES,
+    WaldiezSwarmAfterWork,
+    WaldiezSwarmAfterWorkOption,
+    WaldiezSwarmAfterWorkRecipientType,
+)
+from .on_condition import WaldiezSwarmOnCondition
+from .on_condition_available import (
+    CUSTOM_ON_CONDITION_AVAILABLE,
+    CUSTOM_ON_CONDITION_AVAILABLE_ARGS,
+    CUSTOM_ON_CONDITION_AVAILABLE_TYPES,
+    WaldiezSwarmOnConditionAvailable,
+)
+from .on_condition_target import WaldiezSwarmOnConditionTarget
+from .swarm_agent import WaldiezSwarmAgent
+from .swarm_agent_data import WaldiezSwarmAgentData
+from .update_system_message import (
+    CUSTOM_UPDATE_SYSTEM_MESSAGE,
+    CUSTOM_UPDATE_SYSTEM_MESSAGE_ARGS,
+    CUSTOM_UPDATE_SYSTEM_MESSAGE_TYPES,
+    WaldiezSwarmUpdateSystemMessage,
+)
+
+__all__ = [
+    "CUSTOM_AFTER_WORK",
+    "CUSTOM_AFTER_WORK_ARGS",
+    "CUSTOM_AFTER_WORK_TYPES",
+    "CUSTOM_ON_CONDITION_AVAILABLE",
+    "CUSTOM_ON_CONDITION_AVAILABLE_ARGS",
+    "CUSTOM_ON_CONDITION_AVAILABLE_TYPES",
+    "CUSTOM_UPDATE_SYSTEM_MESSAGE",
+    "CUSTOM_UPDATE_SYSTEM_MESSAGE_ARGS",
+    "CUSTOM_UPDATE_SYSTEM_MESSAGE_TYPES",
+    "WaldiezSwarmAfterWork",
+    "WaldiezSwarmAfterWorkOption",
+    "WaldiezSwarmAgent",
+    "WaldiezSwarmAgentData",
+    "WaldiezSwarmAfterWorkRecipientType",
+    "WaldiezSwarmOnCondition",
+    "WaldiezSwarmOnConditionTarget",
+    "WaldiezSwarmOnConditionAvailable",
+    "WaldiezSwarmUpdateSystemMessage",
+]