waldiez 0.4.8__py3-none-any.whl → 0.4.9__py3-none-any.whl

waldiez/_version.py

@@ -5,4 +5,4 @@
 This file is automatically generated by Hatchling.
 Do not edit this file directly.
 """
-__version__ = VERSION = "0.4.8"
+__version__ = VERSION = "0.4.9"

waldiez/cli.py

@@ -2,43 +2,25 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 # flake8: noqa: E501
 # pylint: disable=missing-function-docstring,missing-param-doc,missing-raises-doc
+# pylint: disable=line-too-long
 """Command line interface to convert or run a waldiez file."""
 
 import json
 import os
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Optional, Union
+from typing import Any, Optional
 
 import anyio
 import typer
 from dotenv import load_dotenv
 from typing_extensions import Annotated
 
-# pylint: disable=import-error,line-too-long
-# pyright: reportMissingImports=false
-try:  # pragma: no cover
-    # fmt: off
-    from ._version import __version__ # type: ignore[unused-ignore, unused-import, import-not-found, import-untyped]  # noqa
-    # fmt: on
-except ImportError:  # pragma: no cover
-    import warnings
-
-    warnings.warn(
-        "Importing __version__ failed. Using 'dev' as version.", stacklevel=2
-    )
-    __version__ = "dev"
-
-
 from .exporter import WaldiezExporter
 from .io import StructuredIOStream
 from .logger import get_logger
 from .models import Waldiez
 from .runner import WaldiezRunner
-from .utils import add_cli_extras
-
-if TYPE_CHECKING:
-    from autogen import ChatResult  # type: ignore[import-untyped]
-
+from .utils import add_cli_extras, get_waldiez_version
 
 load_dotenv()
 LOG = get_logger()
@@ -71,7 +53,7 @@ def show_version(
 ) -> None:
     """Show the version of the Waldiez package and exit."""
     if version:
-        typer.echo(f"waldiez version: {__version__}")
+        typer.echo(f"waldiez version: {get_waldiez_version()}")
         raise typer.Exit()
 
 
@@ -130,27 +112,12 @@ def run(
         except json.decoder.JSONDecodeError as error:
             typer.echo("Invalid .waldiez file. Not a valid json?")
             raise typer.Exit(code=1) from error
-    results = _do_run(
+    _do_run(
         data,
         structured=structured,
         uploads_root=uploads_root,
         output_path=output_path,
     )
-    if isinstance(results, list):
-        LOG.info("Results:")
-        for result in results:
-            _log_result(result)
-            sep = "-" * 80
-            print("\n" + f"{sep}" + "\n")
-    elif isinstance(results, dict):
-        LOG.info("Results:")
-        for key, result in results.items():
-            LOG.info("Order: %s", key)
-            _log_result(result)
-            sep = "-" * 80
-            print("\n" + f"{sep}" + "\n")
-    else:
-        _log_result(results)
 
 
 @app.command()
@@ -239,7 +206,7 @@ def _do_run(
     structured: bool,
     uploads_root: Optional[Path],
     output_path: Optional[Path],
-) -> Union["ChatResult", list["ChatResult"], dict[int, "ChatResult"]]:
+) -> None:
     """Run the Waldiez flow and get the results."""
     waldiez = Waldiez.from_dict(data)
     if structured:
@@ -247,30 +214,29 @@ def _do_run(
         with StructuredIOStream.set_default(stream):
             runner = WaldiezRunner(waldiez)
             if waldiez.is_async:
-                results = anyio.run(
+                anyio.run(
                     runner.a_run,
                     output_path,
                     uploads_root,
                 )
             else:
-                results = runner.run(
+                runner.run(
                     output_path=output_path,
                     uploads_root=uploads_root,
                 )
     else:
         runner = WaldiezRunner(waldiez)
         if waldiez.is_async:
-            results = anyio.run(
+            anyio.run(
                 runner.a_run,
                 output_path,
                 uploads_root,
             )
         else:
-            results = runner.run(
+            runner.run(
                 output_path=output_path,
                 uploads_root=uploads_root,
             )
-    return results
 
 
 def _get_output_path(output: Optional[Path], force: bool) -> Optional[Path]:
@@ -286,16 +252,6 @@ def _get_output_path(output: Optional[Path], force: bool) -> Optional[Path]:
     return output
 
 
-def _log_result(result: "ChatResult") -> None:
-    """Log the result of the Waldiez flow."""
-    LOG.info("Chat History:\n")
-    LOG.info(result.chat_history)
-    LOG.info("Summary:\n")
-    LOG.info(result.summary)
-    LOG.info("Cost:\n")
-    LOG.info(result.cost)
-
-
 add_cli_extras(app)
 
 if __name__ == "__main__":

waldiez/exporting/agent/extras/rag/chroma_extras.py

@@ -65,15 +65,13 @@ def _get_chroma_embedding_function_string(
     to_import = ""
     embedding_function_arg = ""
     embedding_function_content = ""
-    vector_db_model = agent.retrieve_config.db_config.model
     if not agent.retrieve_config.use_custom_embedding:
         to_import = (
             "from chromadb.utils.embedding_functions."
             "sentence_transformer_embedding_function import "
             "SentenceTransformerEmbeddingFunction"
         )
-        embedding_function_arg = "SentenceTransformerEmbeddingFunction("
-        embedding_function_arg += f'model_name="{vector_db_model}")'
+        embedding_function_arg = f"{agent_name}_embedding_function"
     else:
         embedding_function_content, embedding_function_arg = (
             agent.retrieve_config.get_custom_embedding_function(
@@ -124,21 +122,35 @@ def get_chroma_db_args(
     # manually initializing the collection before running the flow,
     # might be a workaround.
     content_before = f"{agent_name}_client = {client_str}" + "\n"
+    vector_db_model = agent.retrieve_config.db_config.model
+    if not embedding_function_body:
+        content_before += (
+            f"{agent_name}_embedding_function = "
+            "SentenceTransformerEmbeddingFunction(\n"
+            f'    model_name="{vector_db_model}",' + "\n"
+            ")\n"
+        )
     collection_name = agent.retrieve_config.collection_name
     get_or_create = agent.retrieve_config.get_or_create
     if collection_name:
         if get_or_create:
             content_before += (
-                f"{agent_name}_client.get_or_create_collection("
-                f'"{collection_name}")' + "\n"
+                f"{agent_name}_client.get_or_create_collection" + "(\n"
+                f'    "{collection_name}",' + "\n"
+                f"    embedding_function={embedding_function_arg}," + "\n"
+                ")" + "\n"
             )
         else:
             content_before += (
                 "try:\n"
-                f'    {agent_name}_client.get_collection("{collection_name}")'
-                + "\n"
+                f"    {agent_name}_client.get_collection(" + "\n"
+                f'        "{collection_name}",' + "\n"
+                f"        embedding_function={embedding_function_arg}," + "\n"
+                "    )\n"
                 "except ValueError:\n"
-                f"    {agent_name}_client.create_collection("
-                f'"{collection_name}")' + "\n"
+                f"    {agent_name}_client.create_collection(" + "\n"
+                f'        "{collection_name}",' + "\n"
+                f"        embedding_function={embedding_function_arg}," + "\n"
+                "    )\n"
             )
     return kwarg_string, to_import, embedding_function_body, content_before

waldiez/io/structured.py

@@ -36,9 +36,13 @@ class StructuredIOStream(IOStream):
     uploads_root: Path | None = None
 
     def __init__(
-        self, timeout: float = 120, uploads_root: Path | str | None = None
+        self,
+        timeout: float = 120,
+        uploads_root: Path | str | None = None,
+        is_async: bool = False,
     ) -> None:
         self.timeout = timeout
+        self.is_async = is_async
         if uploads_root is not None:
             self.uploads_root = Path(uploads_root).resolve()
             if not self.uploads_root.exists():
@@ -106,6 +110,22 @@ class StructuredIOStream(IOStream):
             uploads_root=self.uploads_root,
             base_name=request_id,
         )
+        # let's keep an eye here:
+        # https://github.com/ag2ai/ag2/blob/main/autogen/agentchat/conversable_agent.py#L2973
+        # reply = await iostream.input(prompt) ???? (await???)
+        if self.is_async:
+            # let's make a coroutine to just return the user response
+            async def async_input() -> str:
+                """Asynchronous input method.
+
+                Returns
+                -------
+                str
+                    The line read from the input stream.
+                """
+                return user_response
+
+            return async_input()  # type: ignore
         return user_response
 
     # noinspection PyMethodMayBeStatic

waldiez/runner.py

@@ -18,7 +18,6 @@ from pathlib import Path
 from types import ModuleType, TracebackType
 from typing import (
     TYPE_CHECKING,
-    Callable,
     Optional,
     Set,
     Type,
@@ -34,7 +33,6 @@ from .running import (
     after_run,
     before_run,
     chdir,
-    get_printer,
     install_requirements,
     refresh_environment,
     reset_env_vars,
@@ -166,25 +164,22 @@ class WaldiezRunner:
     def install_requirements(self) -> None:
         """Install the requirements for the flow."""
         self._called_install_requirements = True
-        printer = get_printer()
         extra_requirements = self.gather_requirements()
         if extra_requirements:
-            install_requirements(extra_requirements, printer)
+            install_requirements(extra_requirements)
 
     async def a_install_requirements(self) -> None:
         """Install the requirements for the flow asynchronously."""
         self._called_install_requirements = True
-        printer = get_printer()
         extra_requirements = self.gather_requirements()
         if extra_requirements:
-            await a_install_requirements(extra_requirements, printer)
+            await a_install_requirements(extra_requirements)
 
     def _before_run(
         self,
         temp_dir: Path,
         file_name: str,
         module_name: str,
-        printer: Callable[..., None],
     ) -> tuple[ModuleType, dict[str, str]]:
         self._exporter.export(Path(file_name))
         # unique_names = self._exporter.context.get_unique_names()
@@ -197,8 +192,8 @@ class WaldiezRunner:
         old_vars = set_env_vars(self.waldiez.get_flow_env_vars())
         module = importlib.util.module_from_spec(spec)
         spec.loader.exec_module(module)
-        printer("<Waldiez> - Starting workflow...")
-        printer(self.waldiez.info.model_dump_json())
+        print("<Waldiez> - Starting workflow...")
+        print(self.waldiez.info.model_dump_json())
         return module, old_vars
 
     def _run(
@@ -232,8 +227,7 @@ class WaldiezRunner:
         if not self._called_install_requirements:
             self.install_requirements()
         refresh_environment()
-        printer = get_printer()
-        printer(
+        print(
             "Requirements installed.\n"
             "NOTE: If new packages were added and you are using Jupyter, "
             "you might need to restart the kernel."
@@ -246,7 +240,6 @@ class WaldiezRunner:
                 temp_dir=temp_dir,
                 file_name=file_name,
                 module_name=module_name,
-                printer=printer,
             )
             if use_structured_io:
                 stream = StructuredIOStream(
@@ -257,13 +250,12 @@ class WaldiezRunner:
                     results = module.main()
             else:
                 results = module.main()
-            printer("<Waldiez> - Workflow finished")
+            print("<Waldiez> - Workflow finished")
             sys.path.pop(0)
             reset_env_vars(old_vars)
         after_run(
             temp_dir=temp_dir,
             output_path=output_path,
-            printer=printer,
             flow_name=self.waldiez.name,
             skip_mmd=skip_mmd,
         )
@@ -285,8 +277,7 @@ class WaldiezRunner:
         if not self._called_install_requirements:
             await self.a_install_requirements()
         refresh_environment()
-        printer = get_printer()
-        printer(
+        print(
             "Requirements installed.\n"
             "NOTE: If new packages were added and you are using Jupyter, "
             "you might need to restart the kernel."
@@ -299,7 +290,6 @@ class WaldiezRunner:
                 temp_dir=temp_dir,
                 file_name=file_name,
                 module_name=module_name,
-                printer=printer,
             )
             if use_structured_io:
                 stream = StructuredIOStream(
@@ -315,7 +305,6 @@ class WaldiezRunner:
         after_run(
             temp_dir=temp_dir,
             output_path=output_path,
-            printer=printer,
             flow_name=self.waldiez.name,
             skip_mmd=skip_mmd,
         )

waldiez/running/__init__.py

@@ -9,23 +9,29 @@ from .environment import (
     reset_env_vars,
     set_env_vars,
 )
-from .running import (
-    a_chdir,
+from .post_run import after_run
+from .pre_run import (
     a_install_requirements,
-    after_run,
     before_run,
-    chdir,
-    get_printer,
     install_requirements,
 )
+from .util import (
+    a_chdir,
+    chdir,
+    create_async_subprocess,
+    create_sync_subprocess,
+    strip_ansi,
+)
 
 __all__ = [
     "a_chdir",
     "a_install_requirements",
     "after_run",
     "before_run",
+    "create_async_subprocess",
+    "create_sync_subprocess",
+    "strip_ansi",
     "chdir",
-    "get_printer",
     "in_virtualenv",
     "is_root",
     "install_requirements",

waldiez/running/post_run.py

@@ -0,0 +1,119 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
+"""Utilities for running code."""
+
+import datetime
+import shutil
+from pathlib import Path
+from typing import Optional, Union
+
+from .gen_seq_diagram import generate_sequence_diagram
+
+
+def after_run(
+    temp_dir: Path,
+    output_path: Optional[Union[str, Path]],
+    flow_name: str,
+    skip_mmd: bool = False,
+) -> None:
+    """Actions to perform after running the flow.
+
+    Parameters
+    ----------
+    temp_dir : Path
+        The temporary directory.
+    output_path : Optional[Union[str, Path]]
+        The output path.
+    flow_name : str
+        The flow name.
+    skip_mmd : bool, optional
+        Whether to skip the mermaid sequence diagram generation,
+        by default, False
+    """
+    if isinstance(output_path, str):
+        output_path = Path(output_path)
+    mmd_dir = output_path.parent if output_path else Path.cwd()
+    if skip_mmd is False:
+        events_csv_path = temp_dir / "logs" / "events.csv"
+        if events_csv_path.exists():
+            print("Generating mermaid sequence diagram...")
+            mmd_path = temp_dir / f"{flow_name}.mmd"
+            generate_sequence_diagram(events_csv_path, mmd_path)
+            if (
+                not output_path
+                and mmd_path.exists()
+                and mmd_path != mmd_dir / f"{flow_name}.mmd"
+            ):
+                try:
+                    shutil.copyfile(mmd_path, mmd_dir / f"{flow_name}.mmd")
+                except BaseException:  # pylint: disable=broad-exception-caught
+                    pass
+    if output_path:
+        destination_dir = output_path.parent
+        destination_dir = (
+            destination_dir
+            / "waldiez_out"
+            / datetime.datetime.now().strftime("%Y%m%d%H%M%S")
+        )
+        destination_dir.mkdir(parents=True, exist_ok=True)
+        # copy the contents of the temp dir to the destination dir
+        print(f"Copying the results to {destination_dir}")
+        copy_results(
+            temp_dir=temp_dir,
+            output_path=output_path,
+            output_dir=output_path.parent,
+            destination_dir=destination_dir,
+        )
+    shutil.rmtree(temp_dir)
+
+
+def copy_results(
+    temp_dir: Path,
+    output_path: Path,
+    output_dir: Path,
+    destination_dir: Path,
+) -> None:
+    """Copy the results to the output directory.
+
+    Parameters
+    ----------
+    temp_dir : Path
+        The temporary directory.
+    output_path : Path
+        The output path.
+    output_dir : Path
+        The output directory.
+    destination_dir : Path
+        The destination directory.
+    """
+    temp_dir.mkdir(parents=True, exist_ok=True)
+    for item in temp_dir.iterdir():
+        # skip cache files
+        if (
+            item.name.startswith("__pycache__")
+            or item.name.endswith(".pyc")
+            or item.name.endswith(".pyo")
+            or item.name.endswith(".pyd")
+            or item.name == ".cache"
+        ):
+            continue
+        if item.is_file():
+            # let's also copy the "tree of thoughts" image
+            # to the output directory
+            if item.name.endswith("tree_of_thoughts.png") or item.name.endswith(
+                "reasoning_tree.json"
+            ):
+                shutil.copy(item, output_dir / item.name)
+            shutil.copy(item, destination_dir)
+        else:
+            shutil.copytree(item, destination_dir / item.name)
+    if output_path.is_file():
+        if output_path.suffix == ".waldiez":
+            output_path = output_path.with_suffix(".py")
+        if output_path.suffix == ".py":
+            src = temp_dir / output_path.name
+            if src.exists():
+                dst = destination_dir / output_path.name
+                if dst.exists():
+                    dst.unlink()
+                shutil.copyfile(src, output_dir / output_path.name)

waldiez/running/pre_run.py

@@ -0,0 +1,149 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
+"""Actions to perform before running the flow."""
+
+import asyncio
+import io
+import os
+import subprocess
+import sys
+import tempfile
+from pathlib import Path
+from typing import Callable, Optional, Union
+
+from .environment import in_virtualenv, is_root
+from .util import strip_ansi
+
+
+def before_run(
+    output_path: Optional[Union[str, Path]],
+    uploads_root: Optional[Union[str, Path]],
+) -> str:
+    """Actions to perform before running the flow.
+
+    Parameters
+    ----------
+    output_path : Optional[Union[str, Path]]
+        The output path.
+    uploads_root : Optional[Union[str, Path]]
+        The runtime uploads root.
+
+    Returns
+    -------
+    str
+        The file name.
+    """
+    if not uploads_root:
+        uploads_root = Path(tempfile.mkdtemp())
+    else:
+        uploads_root = Path(uploads_root)
+    if not uploads_root.exists():
+        uploads_root.mkdir(parents=True)
+    output_dir = Path.cwd()
+    if output_path and isinstance(output_path, str):
+        output_path = Path(output_path)
+    if output_path:
+        if output_path.is_dir():
+            output_dir = output_path
+        else:
+            output_dir = output_path.parent if output_path else Path.cwd()
+    if not output_dir.exists():
+        output_dir.mkdir(parents=True, exist_ok=True)
+    file_name = Path(output_path).name if output_path else "waldiez_flow.py"
+    if file_name.endswith((".json", ".waldiez")):
+        file_name = file_name.replace(".json", ".py").replace(".waldiez", ".py")
+    if not file_name.endswith(".py"):
+        file_name += ".py"
+    return file_name
+
+
+def install_requirements(
+    extra_requirements: set[str],
+    printer: Callable[..., None] = print,
+) -> None:
+    """Install the requirements.
+
+    Parameters
+    ----------
+    extra_requirements : set[str]
+        The extra requirements.
+    printer : Callable[..., None]
+        The printer function to use, defaults to print.
+    """
+    requirements_string = ", ".join(extra_requirements)
+    printer(f"Installing requirements: {requirements_string}")
+    pip_install = [sys.executable, "-m", "pip", "install"]
+    break_system_packages = ""
+    if not in_virtualenv():  # it should
+        # if not, let's try to install as user
+        # not sure if --break-system-packages is safe,
+        # but it might fail if we don't
+        break_system_packages = os.environ.get("PIP_BREAK_SYSTEM_PACKAGES", "")
+        os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = "1"
+        if not is_root():
+            pip_install.append("--user")
+    pip_install.extend(extra_requirements)
+    # pylint: disable=too-many-try-statements
+    try:
+        with subprocess.Popen(
+            pip_install,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+        ) as proc:
+            if proc.stdout:
+                for line in io.TextIOWrapper(proc.stdout, encoding="utf-8"):
+                    printer(strip_ansi(line.strip()))
+            if proc.stderr:
+                for line in io.TextIOWrapper(proc.stderr, encoding="utf-8"):
+                    printer(strip_ansi(line.strip()))
+    finally:
+        if not in_virtualenv():
+            # restore the old env var
+            if break_system_packages:
+                os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = break_system_packages
+            else:
+                del os.environ["PIP_BREAK_SYSTEM_PACKAGES"]
+
+
+async def a_install_requirements(
+    extra_requirements: set[str],
+    printer: Callable[..., None] = print,
+) -> None:
+    """Install the requirements asynchronously.
+
+    Parameters
+    ----------
+    extra_requirements : set[str]
+        The extra requirements.
+    printer : Callable[..., None]
+        The printer function to use, defaults to print.
+    """
+    requirements_string = ", ".join(extra_requirements)
+    printer(f"Installing requirements: {requirements_string}")
+    pip_install = [sys.executable, "-m", "pip", "install"]
+    break_system_packages = ""
+    if not in_virtualenv():
+        break_system_packages = os.environ.get("PIP_BREAK_SYSTEM_PACKAGES", "")
+        os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = "1"
+        if not is_root():
+            pip_install.extend(["--user"])
+    pip_install.extend(extra_requirements)
+    # pylint: disable=too-many-try-statements
+    try:
+        proc = await asyncio.create_subprocess_exec(
+            *pip_install,
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
+        )
+        if proc.stdout:
+            async for line in proc.stdout:
+                printer(strip_ansi(line.decode().strip()))
+        if proc.stderr:
+            async for line in proc.stderr:
+                printer(strip_ansi(line.decode().strip()))
+    finally:
+        if not in_virtualenv():
+            if break_system_packages:
+                os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = break_system_packages
+            else:
+                del os.environ["PIP_BREAK_SYSTEM_PACKAGES"]

waldiez/running/util.py

@@ -0,0 +1,134 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
+"""Common utilities for the waldiez runner."""
+
+import asyncio
+import os
+import re
+import subprocess
+import sys
+from asyncio.subprocess import Process
+from contextlib import asynccontextmanager, contextmanager
+from dataclasses import dataclass
+from pathlib import Path
+from typing import AsyncIterator, Iterator, Union
+
+
+@dataclass
+class ProcessSetup:
+    """Container for subprocess setup data."""
+
+    temp_dir: Path
+    file_path: Path
+    old_vars: dict[str, str]
+    skip_mmd: bool
+
+
+@contextmanager
+def chdir(to: Union[str, Path]) -> Iterator[None]:
+    """Change the current working directory in a context.
+
+    Parameters
+    ----------
+    to : Union[str, Path]
+        The directory to change to.
+
+    Yields
+    ------
+    Iterator[None]
+        The context manager.
+    """
+    old_cwd = str(os.getcwd())
+    os.chdir(to)
+    try:
+        yield
+    finally:
+        os.chdir(old_cwd)
+
+
+@asynccontextmanager
+async def a_chdir(to: Union[str, Path]) -> AsyncIterator[None]:
+    """Asynchronously change the current working directory in a context.
+
+    Parameters
+    ----------
+    to : Union[str, Path]
+        The directory to change to.
+
+    Yields
+    ------
+    AsyncIterator[None]
+        The async context manager.
+    """
+    old_cwd = str(os.getcwd())
+    os.chdir(to)
+    try:
+        yield
+    finally:
+        os.chdir(old_cwd)
+
+
+def strip_ansi(text: str) -> str:
+    """Remove ANSI escape sequences from text.
+
+    Parameters
+    ----------
+    text : str
+        The text to strip.
+
+    Returns
+    -------
+    str
+        The text without ANSI escape sequences.
+    """
+    ansi_pattern = re.compile(r"\x1b\[[0-9;]*m|\x1b\[.*?[@-~]")
+    return ansi_pattern.sub("", text)
+
+
+def create_sync_subprocess(setup: ProcessSetup) -> subprocess.Popen[bytes]:
+    """Create a synchronous subprocess.
+
+    Parameters
+    ----------
+    setup : ProcessSetup
+        The setup data for the subprocess.
+
+    Returns
+    -------
+    subprocess.Popen[bytes]
+        The created subprocess.
+    """
+    return subprocess.Popen(
+        [sys.executable, "-u", str(setup.file_path)],
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE,
+        stdin=subprocess.PIPE,
+        # text=True,
+        # bufsize=1,  # Line buffered for real-time output
+        # universal_newlines=True,
+        env={**os.environ},
+    )
+
+
+async def create_async_subprocess(setup: ProcessSetup) -> Process:
+    """Create an asynchronous subprocess.
+
+    Parameters
+    ----------
+    setup : ProcessSetup
+        The setup data for the subprocess.
+
+    Returns
+    -------
+    Process
+        The created asynchronous subprocess.
+    """
+    return await asyncio.create_subprocess_exec(
+        sys.executable,
+        "-u",
+        str(setup.file_path),
+        # stdout=asyncio.subprocess.PIPE,
+        # stderr=asyncio.subprocess.PIPE,
+        # stdin=asyncio.subprocess.PIPE,
+        env={**os.environ},
+    )

waldiez/utils/__init__.py

@@ -5,9 +5,11 @@
 from .cli_extras import add_cli_extras
 from .conflict_checker import check_conflicts
 from .flaml_warnings import check_flaml_warnings
+from .version import get_waldiez_version
 
 __all__ = [
     "check_conflicts",
     "check_flaml_warnings",
     "add_cli_extras",
+    "get_waldiez_version",
 ]

waldiez/utils/version.py

@@ -0,0 +1,47 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
+"""Try to get the Waldiez version."""
+
+import json
+from pathlib import Path
+
+
+def _get_waldiez_version_from_package_json() -> str | None:
+    """Get the Waldiez version from package.json."""
+    package_json_path = Path(__file__).parent.parent.parent / "package.json"
+    if package_json_path.exists():
+        with open(package_json_path, "r", encoding="utf-8") as f:
+            package_data = json.load(f)
+            return package_data.get("version", None)
+    return None
+
+
+def _get_waldiez_version_from_version_py() -> str | None:
+    """Get the Waldiez version from _version.py."""
+    version_py_path = Path(__file__).parent.parent / "_version.py"
+    if version_py_path.exists():
+        with open(version_py_path, "r", encoding="utf-8") as f:
+            for line in f:
+                if line.startswith("__version__"):
+                    return line.split('"')[1]
+    return None
+
+
+def get_waldiez_version() -> str:
+    """Get the Waldiez version.
+
+    Returns
+    -------
+    str
+        The Waldiez version, or "dev" if not found.
+    """
+    w_version = _get_waldiez_version_from_package_json()
+    if not w_version:
+        w_version = _get_waldiez_version_from_version_py()
+    if not w_version:
+        w_version = "dev"
+    return w_version
+
+
+if __name__ == "__main__":
+    print(get_waldiez_version())

{waldiez-0.4.8.dist-info → waldiez-0.4.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.4.8
+Version: 0.4.9
 Dynamic: Keywords
 Summary: Make AG2 Agents Collaborate: Drag, Drop, and Orchestrate with Waldiez.
 Project-URL: Homepage, https://waldiez.io
@@ -157,15 +157,15 @@ Requires-Dist: natsort==8.4.0; extra == 'docs'
 Provides-Extra: jupyter
 Requires-Dist: jupyter-server==2.16.0; extra == 'jupyter'
 Requires-Dist: jupyterlab<5.0,>=4.3.0; extra == 'jupyter'
-Requires-Dist: waldiez-jupyter==0.4.8; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.4.9; extra == 'jupyter'
 Provides-Extra: mqtt
 Requires-Dist: paho-mqtt<3.0,>=2.1.0; extra == 'mqtt'
 Provides-Extra: redis
 Requires-Dist: ag2[redis]==0.9.2; extra == 'redis'
 Provides-Extra: runner
-Requires-Dist: waldiez-runner==0.4.8; (python_version >= '3.11') and extra == 'runner'
+Requires-Dist: waldiez-runner==0.4.9; (python_version >= '3.11') and extra == 'runner'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.4.8; extra == 'studio'
+Requires-Dist: waldiez-studio==0.4.9; extra == 'studio'
 Provides-Extra: test
 Requires-Dist: ag2[redis]==0.9.2; extra == 'test'
 Requires-Dist: ag2[websockets]==0.9.2; extra == 'test'

{waldiez-0.4.8.dist-info → waldiez-0.4.9.dist-info}/RECORD

@@ -1,11 +1,11 @@
 waldiez/__init__.py,sha256=JVVh6d5Zw3ch4GYtIwXZEBpMEyJZAc5t0mtu08tLbZs,992
 waldiez/__main__.py,sha256=0dYzNrQbovRwQQvmZC6_1FDR1m71SUIOkTleO5tBnBw,203
-waldiez/_version.py,sha256=Nlo6ovErTlkRrYOk191gkWS6guip7lym_CNGuVCS1nI,249
-waldiez/cli.py,sha256=LAxEdPMBP7Un0hUGUmW7U9KaSgzJxtjQejS9CZ8LPKk,8804
+waldiez/_version.py,sha256=De8aF7saKAU1lagF5W1Ha6s0ndjgEGVM1xVTicmFN_8,249
+waldiez/cli.py,sha256=FuWHXi1NWNhuXyUHVEvSnS0450Hh0LPekfsZurML2us,7427
 waldiez/exporter.py,sha256=Gl23KC-pF2986gnHLqGf1-7RAPfvbf2sz4VX236gXO8,6134
 waldiez/logger.py,sha256=UFdPwS2AOl2LkFQVonDSET9Dguq5jsn2mV817iTZFJc,16375
 waldiez/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-waldiez/runner.py,sha256=mzMsk-YKhgsKpIPW5DW1nUv-wXo_P56k8QkKpntlw5A,13875
+waldiez/runner.py,sha256=E-e5_QPFYWvWgcY7FwOz5lso2toV1enDD2pnP_Ej1fE,13526
 waldiez/exporting/__init__.py,sha256=xVOqb3gADkrTmp_BVrQ-dMPiySPUDU4xZmqYqhzL8BI,1026
 waldiez/exporting/agent/__init__.py,sha256=GpNfVQLAvGoJP_U4WkMMIGpEB8k7vEV-FVR446s2zhY,260
 waldiez/exporting/agent/code_execution.py,sha256=sxjulr2OmeSg0SrYz7rgpBgF_AfqpHfV86kmtGU49Yk,3795
@@ -28,7 +28,7 @@ waldiez/exporting/agent/extras/handoffs/condition.py,sha256=kURtSLS5B0WbXPARff3t
 waldiez/exporting/agent/extras/handoffs/handoff.py,sha256=Xp0sYpXIoXQ6GbLrLPxylDpyNPiE2w1oXmQ-UqfnK7U,5918
 waldiez/exporting/agent/extras/handoffs/target.py,sha256=R7KerfkL1kXkGMnoy1GXaOTH6DM6LSKXl7ohpTcInCE,6527
 waldiez/exporting/agent/extras/rag/__init__.py,sha256=jkKSaHLfwA8Tj7QJH0MSdAphaP4HowuhQXDj7V122Ek,259
-waldiez/exporting/agent/extras/rag/chroma_extras.py,sha256=m6PMx4opQ9VJPHdEW-kOmOZeqibLBeVDkkLxiZFkUTY,5018
+waldiez/exporting/agent/extras/rag/chroma_extras.py,sha256=vf4EuikCZPtqFEOCoB01YcD85Hi_SA37b1JO869DqSA,5555
 waldiez/exporting/agent/extras/rag/mongo_extras.py,sha256=bn_5bfmyfdW5eQCOjcCRYPKLIJ7vZXKEiUsa4zPx7Sk,2822
 waldiez/exporting/agent/extras/rag/pgvector_extras.py,sha256=Y08rU8Z3-oHUbh7huO1Jo_PZLlOW-NRa1Gz3-d9WXso,2961
 waldiez/exporting/agent/extras/rag/qdrant_extras.py,sha256=hTzY7GUL8RBMFNEGefLDAxKFMSlD5Qmkzm3Ducgp9vU,3633
@@ -97,7 +97,7 @@ waldiez/io/__init__.py,sha256=x4-LxBgHd0ekLXyccgcgV8ERwKRENaJQxaegULkRW3g,3361
 waldiez/io/_ws.py,sha256=uyuIfCrBK7LDXowTQ2fsbP4jvjUcIQZqPQefx2UXmp8,5652
 waldiez/io/mqtt.py,sha256=T01XCGyoWYdIEhzxS5M9igbYLgZqhF-K63j3JQDmj5E,22418
 waldiez/io/redis.py,sha256=diz0UqgcpIejBGFb9TOoT2iVzGtHWFnhg0cfEOox25Q,25523
-waldiez/io/structured.py,sha256=n-YeW-BFgNzFJU7QXbG3rG6v6R58ZltttouYZm2rZ6E,13717
+waldiez/io/structured.py,sha256=ZFmiaX-OOIX86SDMg3h0Gw1OwmrNgi7PgIKu9V1gUrk,14409
 waldiez/io/utils.py,sha256=J2I0cuBMMjJadM8OjP7kiqFjoZp8hc9rXF9n5fdltEE,4488
 waldiez/io/ws.py,sha256=aiORAYjvSA57ZqI1mXINkkHnZzqkibaICTj2bZ9Yze8,9107
 waldiez/io/models/__init__.py,sha256=N21ET1_QEkCsJNrax8y5cDDiTrGj2rv9BfEsVvUY1m0,1240
@@ -181,20 +181,23 @@ waldiez/models/tool/extra_requirements.py,sha256=kNrmWxS96FhO93NV3B5Hr5wAJrjuUWb
 waldiez/models/tool/tool.py,sha256=Mzst4x9tS-1exryTYZSAlr6B3E7ba7Q5yF9WLxXp5G8,8987
 waldiez/models/tool/tool_data.py,sha256=lOtIetRM35tUvDCtjiHjddVSoqlwV4NFkz_496GLunA,1305
 waldiez/models/tool/tool_type.py,sha256=VYcgnAlj1n6ZqXA8D0VdbRfZ-_U89CPSDuNgk822_UA,268
-waldiez/running/__init__.py,sha256=EqH4knSyNxYjN31YBifoLfXS5nEW4ItWMA6-CNpZszk,662
+waldiez/running/__init__.py,sha256=Q_lhpEdJ2HoXOAISQYdCesUYMctGqq59AdYsa3Gr1yI,817
 waldiez/running/environment.py,sha256=DxiI9GqFrFiF6p5icrpCFkbsvq6zgACoqXYyHZniffQ,3472
 waldiez/running/gen_seq_diagram.py,sha256=wsF9b9uG0NA5iVAlW-wOtYn0w_9-bySBO9nWkUIBNUc,5943
-waldiez/running/running.py,sha256=0nZ33hwvUAVGcemT2C2Rec8fqHkB5xU4TRiCzs1fAKE,11750
-waldiez/utils/__init__.py,sha256=ITQcsqe2jhO376VSeZGfcf0n0kUhMw-4rKxUWtFB8SY,345
+waldiez/running/post_run.py,sha256=rxVmRndpKnoJs2uZrk0JDNrrFNihoU52rExyRxgJe0k,3891
+waldiez/running/pre_run.py,sha256=lM5l6lLLgqlKpWS3o2Gbe_VhDvVJ6xgJZsbL8rYKJpY,5019
+waldiez/running/util.py,sha256=7DnuL-FNkraDhu3QUV3GEZC_dNgo3IMlptBzG9tMUKA,3002
+waldiez/utils/__init__.py,sha256=lamPdfOEeOZ9iYtDXKXEyOL2RhjZ7wpyK2W5SmVxPNc,413
 waldiez/utils/conflict_checker.py,sha256=bskxC2KmPrAOjjYfgQhGXgNxdU5Z10ZZmx-_c0W8I7I,1493
 waldiez/utils/flaml_warnings.py,sha256=gN1yzlAAHor04w0-_xXUK0wID6aXCizKYElaNHflkg8,630
+waldiez/utils/version.py,sha256=pYu3CwBW5Rwv9LqWG8W5DxrvdIU1nc2MOY1RDJR8oc4,1417
 waldiez/utils/cli_extras/__init__.py,sha256=ZvuLaN3IGuffWMh7mladTGkJxx3kn5IepUF3jk4ZAWY,684
 waldiez/utils/cli_extras/jupyter.py,sha256=nOQraO7Xg1G8SCP93OpIVheXieXW5ireSC9Oir-dXzA,3047
 waldiez/utils/cli_extras/runner.py,sha256=V_aoTKZsvtPgjszvWlPamKjnOG93L2aY6sBcHnIh9XI,983
 waldiez/utils/cli_extras/studio.py,sha256=p4CMDCYW6lxrPvHvWX2kpsE1Ix7NnKZg1Jf15weWirE,984
-waldiez-0.4.8.dist-info/METADATA,sha256=3Ta0RN91r4JKHXSf8hZ7P4KekETUeJeLgquleK0pS6w,22620
-waldiez-0.4.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-waldiez-0.4.8.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
-waldiez-0.4.8.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-waldiez-0.4.8.dist-info/licenses/NOTICE.md,sha256=L7xtckFRYvYJjhjQNtFpURWCiAvEuq4ePvxJsC-XAdk,785
-waldiez-0.4.8.dist-info/RECORD,,
+waldiez-0.4.9.dist-info/METADATA,sha256=FefkfD_VJwlzOf2tuWvcPpjYvBsQ8AYlw9sOgPmzh9Y,22620
+waldiez-0.4.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+waldiez-0.4.9.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
+waldiez-0.4.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+waldiez-0.4.9.dist-info/licenses/NOTICE.md,sha256=L7xtckFRYvYJjhjQNtFpURWCiAvEuq4ePvxJsC-XAdk,785
+waldiez-0.4.9.dist-info/RECORD,,

waldiez/running/running.py

@@ -1,388 +0,0 @@
-# SPDX-License-Identifier: Apache-2.0.
-# Copyright (c) 2024 - 2025 Waldiez and contributors.
-"""Utilities for running code."""
-
-import asyncio
-import datetime
-import io
-import os
-import re
-import shutil
-import subprocess
-import sys
-import tempfile
-from contextlib import asynccontextmanager, contextmanager
-from pathlib import Path
-from typing import (
-    Any,
-    AsyncIterator,
-    Callable,
-    Iterator,
-    Optional,
-    Set,
-    Union,
-)
-
-from .environment import in_virtualenv, is_root
-from .gen_seq_diagram import generate_sequence_diagram
-
-# pylint: disable=import-outside-toplevel
-
-
-@contextmanager
-def chdir(to: Union[str, Path]) -> Iterator[None]:
-    """Change the current working directory in a context.
-
-    Parameters
-    ----------
-    to : Union[str, Path]
-        The directory to change to.
-
-    Yields
-    ------
-    Iterator[None]
-        The context manager.
-    """
-    old_cwd = str(os.getcwd())
-    os.chdir(to)
-    try:
-        yield
-    finally:
-        os.chdir(old_cwd)
-
-
-@asynccontextmanager
-async def a_chdir(to: Union[str, Path]) -> AsyncIterator[None]:
-    """Asynchronously change the current working directory in a context.
-
-    Parameters
-    ----------
-    to : Union[str, Path]
-        The directory to change to.
-
-    Yields
-    ------
-    AsyncIterator[None]
-        The async context manager.
-    """
-    old_cwd = str(os.getcwd())
-    os.chdir(to)
-    try:
-        yield
-    finally:
-        os.chdir(old_cwd)
-
-
-def strip_ansi(text: str) -> str:
-    """Remove ANSI escape sequences from text.
-
-    Parameters
-    ----------
-    text : str
-        The text to strip.
-
-    Returns
-    -------
-    str
-        The text without ANSI escape sequences.
-    """
-    ansi_pattern = re.compile(r"\x1b\[[0-9;]*m|\x1b\[.*?[@-~]")
-    return ansi_pattern.sub("", text)
-
-
-def before_run(
-    output_path: Optional[Union[str, Path]],
-    uploads_root: Optional[Union[str, Path]],
-) -> str:
-    """Actions to perform before running the flow.
-
-    Parameters
-    ----------
-    output_path : Optional[Union[str, Path]]
-        The output path.
-    uploads_root : Optional[Union[str, Path]]
-        The runtime uploads root.
-
-    Returns
-    -------
-    str
-        The file name.
-    """
-    if not uploads_root:
-        uploads_root = Path(tempfile.mkdtemp())
-    else:
-        uploads_root = Path(uploads_root)
-    if not uploads_root.exists():
-        uploads_root.mkdir(parents=True)
-    file_name = "waldiez_flow.py" if not output_path else Path(output_path).name
-    if file_name.endswith((".json", ".waldiez")):
-        file_name = file_name.replace(".json", ".py").replace(".waldiez", ".py")
-    if not file_name.endswith(".py"):
-        file_name += ".py"
-    return file_name
-
-
-def install_requirements(
-    extra_requirements: Set[str], printer: Callable[..., None]
-) -> None:
-    """Install the requirements.
-
-    Parameters
-    ----------
-    extra_requirements : Set[str]
-        The extra requirements.
-    printer : Callable[..., None]
-        The printer function.
-    """
-    requirements_string = ", ".join(extra_requirements)
-    printer(f"Installing requirements: {requirements_string}")
-    pip_install = [sys.executable, "-m", "pip", "install"]
-    break_system_packages = ""
-    if not in_virtualenv():  # it should
-        # if not, let's try to install as user
-        # not sure if --break-system-packages is safe,
-        # but it might fail if we don't
-        break_system_packages = os.environ.get("PIP_BREAK_SYSTEM_PACKAGES", "")
-        os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = "1"
-        if not is_root():
-            pip_install.append("--user")
-    pip_install.extend(extra_requirements)
-    # pylint: disable=too-many-try-statements
-    try:
-        with subprocess.Popen(
-            pip_install,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE,
-        ) as proc:
-            if proc.stdout:
-                for line in io.TextIOWrapper(proc.stdout, encoding="utf-8"):
-                    printer(strip_ansi(line.strip()))
-            if proc.stderr:
-                for line in io.TextIOWrapper(proc.stderr, encoding="utf-8"):
-                    printer(strip_ansi(line.strip()))
-    finally:
-        if not in_virtualenv():
-            # restore the old env var
-            if break_system_packages:
-                os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = break_system_packages
-            else:
-                del os.environ["PIP_BREAK_SYSTEM_PACKAGES"]
-
-
-async def a_install_requirements(
-    extra_requirements: Set[str], printer: Callable[..., None]
-) -> None:
-    """Install the requirements asynchronously.
-
-    Parameters
-    ----------
-    extra_requirements : Set[str]
-        The extra requirements.
-    printer : Callable[..., None]
-        The printer function.
-    """
-    requirements_string = ", ".join(extra_requirements)
-    printer(f"Installing requirements: {requirements_string}")
-    pip_install = [sys.executable, "-m", "pip", "install"]
-    break_system_packages = ""
-    if not in_virtualenv():
-        break_system_packages = os.environ.get("PIP_BREAK_SYSTEM_PACKAGES", "")
-        os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = "1"
-        if not is_root():
-            pip_install.extend(["--user"])
-    pip_install.extend(extra_requirements)
-    # pylint: disable=too-many-try-statements
-    try:
-        proc = await asyncio.create_subprocess_exec(
-            *pip_install,
-            stdout=asyncio.subprocess.PIPE,
-            stderr=asyncio.subprocess.PIPE,
-        )
-        if proc.stdout:
-            async for line in proc.stdout:
-                printer(strip_ansi(line.decode().strip()))
-        if proc.stderr:
-            async for line in proc.stderr:
-                printer(strip_ansi(line.decode().strip()))
-    finally:
-        if not in_virtualenv():
-            if break_system_packages:
-                os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = break_system_packages
-            else:
-                del os.environ["PIP_BREAK_SYSTEM_PACKAGES"]
-
-
-def after_run(
-    temp_dir: Path,
-    output_path: Optional[Union[str, Path]],
-    printer: Callable[..., None],
-    flow_name: str,
-    skip_mmd: bool = False,
-) -> None:
-    """Actions to perform after running the flow.
-
-    Parameters
-    ----------
-    temp_dir : Path
-        The temporary directory.
-    output_path : Optional[Union[str, Path]]
-        The output path.
-    printer : Callable[..., None]
-        The printer function.
-    flow_name : str
-        The flow name.
-    skip_mmd : bool, optional
-        Whether to skip the mermaid sequence diagram generation,
-        by default, False
-    """
-    if isinstance(output_path, str):
-        output_path = Path(output_path)
-    output_dir = output_path.parent if output_path else Path.cwd()
-    if skip_mmd is False:
-        events_csv_path = temp_dir / "logs" / "events.csv"
-        if events_csv_path.exists():
-            printer("Generating mermaid sequence diagram...")
-            mmd_path = temp_dir / f"{flow_name}.mmd"
-            generate_sequence_diagram(events_csv_path, mmd_path)
-            if mmd_path.exists():
-                shutil.copyfile(mmd_path, output_dir / f"{flow_name}.mmd")
-    if output_path:
-        destination_dir = output_path.parent
-        destination_dir = (
-            destination_dir
-            / "waldiez_out"
-            / datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-        )
-        destination_dir.mkdir(parents=True, exist_ok=True)
-        # copy the contents of the temp dir to the destination dir
-        printer(f"Copying the results to {destination_dir}")
-        copy_results(
-            temp_dir=temp_dir,
-            output_path=output_path,
-            output_dir=output_dir,
-            destination_dir=destination_dir,
-        )
-    shutil.rmtree(temp_dir)
-
-
-def copy_results(
-    temp_dir: Path,
-    output_path: Path,
-    output_dir: Path,
-    destination_dir: Path,
-) -> None:
-    """Copy the results to the output directory.
-
-    Parameters
-    ----------
-    temp_dir : Path
-        The temporary directory.
-    output_path : Path
-        The output path.
-    output_dir : Path
-        The output directory.
-    destination_dir : Path
-        The destination directory.
-    """
-    temp_dir.mkdir(parents=True, exist_ok=True)
-    for item in temp_dir.iterdir():
-        # skip cache files
-        if (
-            item.name.startswith("__pycache__")
-            or item.name.endswith(".pyc")
-            or item.name.endswith(".pyo")
-            or item.name.endswith(".pyd")
-            or item.name == ".cache"
-        ):
-            continue
-        if item.is_file():
-            # let's also copy the "tree of thoughts" image
-            # to the output directory
-            if item.name.endswith("tree_of_thoughts.png") or item.name.endswith(
-                "reasoning_tree.json"
-            ):
-                shutil.copy(item, output_dir / item.name)
-            shutil.copy(item, destination_dir)
-        else:
-            shutil.copytree(item, destination_dir / item.name)
-    if output_path.is_file():
-        if output_path.suffix == ".waldiez":
-            output_path = output_path.with_suffix(".py")
-        if output_path.suffix == ".py":
-            src = temp_dir / output_path.name
-            if src.exists():
-                dst = destination_dir / output_path.name
-                if dst.exists():
-                    dst.unlink()
-                shutil.copyfile(src, output_dir / output_path.name)
-
-
-def get_printer() -> Callable[..., None]:
-    """Get the printer function.
-
-    Returns
-    -------
-    Callable[..., None]
-        The printer function.
-    """
-    from autogen.io import IOStream  # type: ignore
-
-    # noinspection PyUnboundLocalVariable
-    printer = IOStream.get_default().print
-
-    def safe_printer(*args: Any, **kwargs: Any) -> None:
-        try:
-            printer(*args, **kwargs)
-        except UnicodeEncodeError:
-            try:
-                # Use the helper to get safely encoded message
-                safe_msg, flush = get_what_to_print(*args, **kwargs)
-                # Try printing the safe message
-                # (without end since it's already included)
-                printer(safe_msg, end="", flush=flush)
-            except UnicodeEncodeError:
-                # pylint: disable=too-many-try-statements
-                try:
-                    # If that still fails, write directly to buffer
-                    msg, flush = get_what_to_print(*args, **kwargs)
-                    safe_msg_bytes = msg.encode("utf-8", errors="replace")
-                    sys.stdout.buffer.write(safe_msg_bytes)
-                    if flush:
-                        sys.stdout.buffer.flush()
-                except Exception:  # pylint: disable=broad-exception-caught
-                    error_msg = (
-                        b"Could not print message due to encoding issues.\n"
-                    )
-                    sys.stderr.buffer.write(error_msg)
-                    sys.stderr.buffer.flush()
-
-    return safe_printer
-
-
-def get_what_to_print(*args: object, **kwargs: object) -> tuple[str, bool]:
-    """Get what to print.
-
-    Parameters
-    ----------
-    args : object
-        The print arguments.
-    kwargs : object
-        The keyword arguments.
-
-    Returns
-    -------
-    tuple[str, bool]
-        The message and whether to flush.
-    """
-    sep = kwargs.get("sep", " ")
-    if not isinstance(sep, str):
-        sep = " "
-    end = kwargs.get("end", "\n")
-    if not isinstance(end, str):
-        end = "\n"
-    flush = kwargs.get("flush", False)
-    if not isinstance(flush, bool):
-        flush = False
-    msg = sep.join(str(arg) for arg in args) + end
-    utf8_msg = msg.encode("utf-8", errors="replace").decode("utf-8")
-    return utf8_msg, flush