waldiez 0.5.1__py3-none-any.whl → 0.5.3__py3-none-any.whl

waldiez/running/base_runner.py

@@ -20,7 +20,6 @@ from typing_extensions import Self
 from waldiez.exporter import WaldiezExporter
 from waldiez.logger import WaldiezLogger, get_logger
 from waldiez.models import Waldiez
-from waldiez.utils import get_waldiez_version
 
 from .environment import refresh_environment, reset_env_vars, set_env_vars
 from .post_run import after_run
@@ -149,6 +148,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> Union[
         "ChatResult",
         list["ChatResult"],
@@ -165,6 +165,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> Union[
         "ChatResult",
         list["ChatResult"],
@@ -181,6 +182,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> None:
         """Start running the Waldiez flow in a non-blocking way."""
         raise NotImplementedError(
@@ -193,6 +195,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> None:
         """Start running the Waldiez flow in a non-blocking way asynchronously.
 
@@ -224,6 +227,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         uploads_root: Path | None,
         temp_dir: Path,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> None:
         """Run after the flow execution."""
         # Save results
@@ -237,6 +241,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             flow_name=self.waldiez.name,
             uploads_root=uploads_root,
             skip_mmd=skip_mmd,
+            skip_timeline=skip_timeline,
         )
         self.log.info("Cleanup completed")
 
@@ -251,6 +256,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         uploads_root: Path | None,
         temp_dir: Path,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> None:
         """Run after the flow execution asynchronously."""
         self._after_run(
@@ -259,6 +265,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             uploads_root=uploads_root,
             temp_dir=temp_dir,
             skip_mmd=skip_mmd,
+            skip_timeline=skip_timeline,
         )
 
     def _stop(self) -> None:
@@ -309,9 +316,6 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             for req in self.waldiez.requirements
             if req not in sys.modules and "waldiez" not in req
         }
-        waldiez_version = get_waldiez_version()
-        if "waldiez" not in sys.modules:
-            extra_requirements.add(f"waldiez=={waldiez_version}")
         return extra_requirements
 
     def install_requirements(self) -> None:
@@ -390,6 +394,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         threaded: bool | None = None,
         skip_patch_io: bool | None = None,
         skip_mmd: bool = False,
+        skip_timeline: bool = False,
     ) -> Union[
         "ChatResult",
         list["ChatResult"],
@@ -408,10 +413,12 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             by default False.
         threaded : bool | None
             Whether to run the flow in a threaded environment, by default None.
-        skip_mmd : bool
-            Whether to skip generating the mermaid diagram, by default False.
         skip_patch_io : bool | None
             Whether to skip patching the IO streams, by default None.
+        skip_mmd : bool
+            Whether to skip generating the mermaid diagram, by default False.
+        skip_timeline : bool
+            Whether to skip generating the timeline JSON.
 
         Returns
         -------
@@ -468,6 +475,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
                     output_file=output_file,
                     uploads_root=uploads_root_path,
                     skip_mmd=skip_mmd,
+                    skip_timeline=skip_timeline,
                 )
         finally:
             WaldiezBaseRunner._running = False
@@ -478,6 +486,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             uploads_root=uploads_root_path,
             temp_dir=temp_dir,
             skip_mmd=skip_mmd,
+            skip_timeline=skip_timeline,
         )
         if sys.path[0] == str(temp_dir):
             sys.path.pop(0)
@@ -490,6 +499,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         structured_io: bool | None = None,
         skip_patch_io: bool | None = None,
         skip_mmd: bool = False,
+        skip_timeline: bool = False,
     ) -> Union[
         "ChatResult",
         list["ChatResult"],
@@ -510,6 +520,8 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             Whether to skip patching I/O, by default None.
         skip_mmd : bool
             Whether to skip generating the mermaid diagram, by default False.
+        skip_timeline : bool
+            Whether to skip generating the timeline JSON, by default False.
 
         Returns
         -------
@@ -553,6 +565,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
                     output_file=output_file,
                     uploads_root=uploads_root_path,
                     skip_mmd=skip_mmd,
+                    skip_timeline=skip_timeline,
                 )
         finally:
             WaldiezBaseRunner._running = False
@@ -562,6 +575,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             uploads_root=uploads_root_path,
             temp_dir=temp_dir,
             skip_mmd=skip_mmd,
+            skip_timeline=skip_timeline,
         )
         if sys.path[0] == str(temp_dir):
             sys.path.pop(0)
@@ -574,6 +588,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         structured_io: bool | None = None,
         skip_patch_io: bool | None = None,
         skip_mmd: bool = False,
+        skip_timeline: bool = False,
     ) -> None:
         """Start running the Waldiez flow in a non-blocking way.
 
@@ -589,6 +604,8 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             Whether to skip patching I/O, by default None.
         skip_mmd : bool
             Whether to skip generating the mermaid diagram, by default False.
+        skip_timeline : bool
+            Whether to skip generating the timeline JSON, by default False.
 
         Raises
         ------
@@ -617,6 +634,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             output_file=output_file,
             uploads_root=uploads_root_path,
             skip_mmd=skip_mmd,
+            skip_timeline=skip_timeline,
         )
 
     async def a_start(
@@ -626,6 +644,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         structured_io: bool | None = None,
         skip_patch_io: bool | None = None,
         skip_mmd: bool = False,
+        skip_timeline: bool = False,
     ) -> None:
         """Asynchronously start running the Waldiez flow in a non-blocking way.
 
@@ -639,8 +658,10 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             Whether to use structured IO instead of the default 'input/print'.
         skip_patch_io : bool | None = None
             Whether to skip patching I/O, by default None.
-        skip_mmd : bool | None = None
-            Whether to skip generating the mermaid diagram, by default None.
+        skip_mmd : bool = False
+            Whether to skip generating the mermaid diagram, by default False.
+        skip_timeline : bool = False
+            Whether to skip generating the timeline JSON, by default False.
 
         Raises
         ------
@@ -669,6 +690,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             output_file=output_file,
             uploads_root=uploads_root_path,
             skip_mmd=skip_mmd,
+            skip_timeline=skip_timeline,
         )
 
     def after_run(
@@ -682,6 +704,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         uploads_root: Path | None,
         temp_dir: Path,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> None:
         """Actions to perform after running the flow.
 
@@ -697,6 +720,8 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             The path to the temporary directory used during the run.
         skip_mmd : bool
             Whether to skip generating the mermaid diagram.
+        skip_timeline : bool
+            Whether to skip generating the timeline JSON.
         """
         self._after_run(
             results=results,
@@ -704,6 +729,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             uploads_root=uploads_root,
             temp_dir=temp_dir,
             skip_mmd=skip_mmd,
+            skip_timeline=skip_timeline,
         )
 
     async def a_after_run(
@@ -717,6 +743,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         uploads_root: Path | None,
         temp_dir: Path,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> None:
         """Asynchronously perform actions after running the flow.
 
@@ -732,6 +759,8 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             The path to the temporary directory used during the run.
         skip_mmd : bool
             Whether to skip generating the mermaid diagram.
+        skip_timeline : bool
+
         """
         await self._a_after_run(
             results=results,
@@ -739,6 +768,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             uploads_root=uploads_root,
             temp_dir=temp_dir,
             skip_mmd=skip_mmd,
+            skip_timeline=skip_timeline,
         )
 
     def stop(self) -> None:

waldiez/running/environment.py

@@ -3,6 +3,7 @@
 # pylint: disable=import-outside-toplevel,reimported
 """Environment related utilities."""
 
+import importlib
 import os
 import site
 import sys
@@ -86,19 +87,19 @@ def reload_autogen() -> None:  # noqa: C901
         default_io_stream = IOStream.get_default()
     except (ImportError, AttributeError):
         pass
-
+    autogen_modules = sorted(
+        [
+            name
+            for name in sys.modules
+            if name.startswith("autogen.")
+            and not name.startswith("autogen.io")
+            and not name.startswith("autogen.tools")
+        ],
+        key=len,
+        reverse=True,  # Longer names (deeper modules) first
+    )
     try:
         # Remove autogen modules in reverse dependency order
-        autogen_modules = sorted(
-            [
-                name
-                for name in sys.modules
-                if name.startswith("autogen.")
-                and not name.startswith("autogen.io")
-            ],
-            key=len,
-            reverse=True,  # Longer names (deeper modules) first
-        )
         for mod_name in autogen_modules:
             if mod_name in sys.modules:
                 del sys.modules[mod_name]
@@ -110,6 +111,14 @@ def reload_autogen() -> None:  # noqa: C901
         # pylint: disable=unused-import
         import autogen  # pyright: ignore
 
+        for mod_name in sorted(autogen_modules, key=len, reverse=False):
+            # Re-import each module
+            try:
+                importlib.import_module(mod_name)
+            except ImportError as e:
+                # If a module fails to import, we can log it or handle it
+                print(f"Failed to re-import {mod_name}: {e}", file=sys.stderr)
+
         # Restore IOStream state if we had it
         if default_io_stream is not None:
             try:
@@ -124,7 +133,18 @@ def reload_autogen() -> None:  # noqa: C901
         # If reload fails, at least try to re-import autogen
         try:
             import autogen  # type: ignore  # noqa: F401
-        except ImportError:
+
+            for mod_name in sorted(autogen_modules, key=len, reverse=False):
+                # Re-import each module
+                try:
+                    importlib.import_module(mod_name)
+                except ImportError as err:
+                    # If a module fails to import, we can log it or handle it
+                    print(
+                        f"Failed to re-import {mod_name}: {err}",
+                        file=sys.stderr,
+                    )
+        except Exception:  # pylint: disable=broad-exception-caught
             pass
         raise e
 

waldiez/running/import_runner.py

@@ -63,6 +63,7 @@ class WaldiezImportRunner(WaldiezBaseRunner):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> Union["ChatResult", list["ChatResult"], dict[int, "ChatResult"]]:
         """Run the Waldiez workflow."""
         if self.threaded:
@@ -71,6 +72,7 @@ class WaldiezImportRunner(WaldiezBaseRunner):
                 output_file=output_file,
                 uploads_root=uploads_root,
                 skip_mmd=skip_mmd,
+                skip_timeline=skip_timeline,
             )
 
         return self._run_not_threaded(
@@ -78,14 +80,17 @@ class WaldiezImportRunner(WaldiezBaseRunner):
             output_file=output_file,
             uploads_root=uploads_root,
             skip_mmd=skip_mmd,
+            skip_timeline=skip_timeline,
         )
 
+    # pylint: disable=too-many-locals
     def _run_not_threaded(
         self,
         temp_dir: Path,
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> Union[
         "ChatResult",
         list["ChatResult"],
@@ -149,6 +154,7 @@ class WaldiezImportRunner(WaldiezBaseRunner):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> Union[
         "ChatResult",
         list["ChatResult"],
@@ -246,6 +252,7 @@ class WaldiezImportRunner(WaldiezBaseRunner):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> Union[
         "ChatResult",
         list["ChatResult"],
@@ -336,6 +343,7 @@ class WaldiezImportRunner(WaldiezBaseRunner):
         uploads_root: Path | None,
         temp_dir: Path,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> None:
         super()._after_run(
             results=results,
@@ -343,6 +351,7 @@ class WaldiezImportRunner(WaldiezBaseRunner):
             uploads_root=uploads_root,
             temp_dir=temp_dir,
             skip_mmd=skip_mmd,
+            skip_timeline=skip_timeline,
         )
 
         # Clean up module reference
@@ -354,6 +363,7 @@ class WaldiezImportRunner(WaldiezBaseRunner):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool = False,
+        skip_timeline: bool = False,
     ) -> None:
         """Start the Waldiez workflow."""
 
@@ -366,6 +376,7 @@ class WaldiezImportRunner(WaldiezBaseRunner):
                     output_file=output_file,
                     uploads_root=uploads_root,
                     skip_mmd=skip_mmd,
+                    skip_timeline=skip_timeline,
                 )
             except Exception as e:  # pylint: disable=broad-exception-caught
                 self._last_exception = e
@@ -383,6 +394,7 @@ class WaldiezImportRunner(WaldiezBaseRunner):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool = False,
+        skip_timeline: bool = False,
     ) -> None:
         """Start the Waldiez workflow asynchronously."""
 
@@ -394,6 +406,7 @@ class WaldiezImportRunner(WaldiezBaseRunner):
                     output_file=output_file,
                     uploads_root=uploads_root,
                     skip_mmd=skip_mmd,
+                    skip_timeline=skip_timeline,
                 )
             except Exception as e:  # pylint: disable=broad-exception-caught
                 self._last_exception = e

waldiez/running/post_run.py

@@ -5,11 +5,14 @@
 """Utilities for running code."""
 
 import datetime
+import json
 import shutil
 from pathlib import Path
 from typing import Optional, Union
 
 from .gen_seq_diagram import generate_sequence_diagram
+from .patch_io_stream import get_printer
+from .timeline_processor import TimelineProcessor
 
 
 def after_run(
@@ -18,6 +21,7 @@ def after_run(
     flow_name: str,
     uploads_root: Optional[Path] = None,
     skip_mmd: bool = False,
+    skip_timeline: bool = False,
 ) -> None:
     """Actions to perform after running the flow.
 
@@ -34,25 +38,21 @@ def after_run(
     skip_mmd : bool, optional
         Whether to skip the mermaid sequence diagram generation,
         by default, False
+    skip_timeline : bool, optional
+        Whether to skip the timeline processing, by default False
     """
     if isinstance(output_file, str):
         output_file = Path(output_file)
     mmd_dir = output_file.parent if output_file 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_file
-                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
+        _make_mermaid_diagram(
+            temp_dir=temp_dir,
+            output_file=output_file,
+            flow_name=flow_name,
+            mmd_dir=mmd_dir,
+        )
+    if skip_timeline is False:
+        _make_timeline_json(temp_dir=temp_dir)
     if output_file:
         destination_dir = output_file.parent
         destination_dir = (
@@ -71,6 +71,63 @@ def after_run(
     shutil.rmtree(temp_dir)
 
 
+def _make_mermaid_diagram(
+    temp_dir: Path,
+    output_file: Optional[Union[str, Path]],
+    flow_name: str,
+    mmd_dir: Path,
+) -> None:
+    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_file
+            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
+
+
+def _make_timeline_json(
+    temp_dir: Path,
+) -> None:
+    """Make the timeline JSON file."""
+    events_csv_path = temp_dir / "logs" / "events.csv"
+    if events_csv_path.exists():
+        print("Processing timeline...")
+        log_files = TimelineProcessor.get_files(temp_dir / "logs")
+        if any(log_files.values()):
+            output_file = temp_dir / "timeline.json"
+            # pylint: disable=too-many-try-statements
+            try:
+                processor = TimelineProcessor()
+                processor.load_csv_files(
+                    agents_file=log_files["agents"],
+                    chat_file=log_files["chat"],
+                    events_file=log_files["events"],
+                    functions_file=log_files["functions"],
+                )
+                results = processor.process_timeline()
+                with open(output_file, "w", encoding="utf-8") as f:
+                    json.dump(results, f, indent=2, default=str)
+                short_results = TimelineProcessor.get_short_results(results)
+                printer = get_printer()
+                printer(
+                    json.dumps(
+                        {"type": "timeline", "content": short_results},
+                        default=str,
+                    ),
+                    flush=True,
+                )
+            except BaseException:  # pylint: disable=broad-exception-caught
+                pass
+
+
 def copy_results(
     temp_dir: Path,
     output_file: Path,

waldiez/running/pre_run.py

@@ -15,6 +15,7 @@ from .utils import strip_ansi
 
 def install_requirements(
     extra_requirements: set[str],
+    upgrade: bool = False,
     printer: Callable[..., None] = print,
 ) -> None:
     """Install the requirements.
@@ -23,6 +24,8 @@ def install_requirements(
     ----------
     extra_requirements : set[str]
         The extra requirements.
+    upgrade : bool, optional
+        Whether to upgrade the requirements, by default False.
     printer : Callable[..., None]
         The printer function to use, defaults to print.
     """
@@ -38,6 +41,8 @@ def install_requirements(
         os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = "1"
         if not is_root():
             pip_install.append("--user")
+    if upgrade:
+        pip_install.append("--upgrade")
     pip_install.extend(extra_requirements)
     # pylint: disable=too-many-try-statements
     try:
@@ -63,6 +68,7 @@ def install_requirements(
 
 async def a_install_requirements(
     extra_requirements: set[str],
+    upgrade: bool = False,
     printer: Callable[..., None] = print,
 ) -> None:
     """Install the requirements asynchronously.
@@ -71,6 +77,8 @@ async def a_install_requirements(
     ----------
     extra_requirements : set[str]
         The extra requirements.
+    upgrade : bool, optional
+        Whether to upgrade the requirements, by default False.
     printer : Callable[..., None]
         The printer function to use, defaults to print.
     """
@@ -83,6 +91,8 @@ async def a_install_requirements(
         os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = "1"
         if not is_root():
             pip_install.extend(["--user"])
+    if upgrade:
+        pip_install.append("--upgrade")
     pip_install.extend(extra_requirements)
     # pylint: disable=too-many-try-statements
     try:
@@ -103,3 +113,35 @@ async def a_install_requirements(
                 os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = break_system_packages
             else:
                 del os.environ["PIP_BREAK_SYSTEM_PACKAGES"]
+
+
+def install_waldiez(
+    upgrade: bool = True,
+    printer: Callable[..., None] = print,
+) -> None:
+    """Install Waldiez.
+
+    Parameters
+    ----------
+    upgrade : bool, optional
+        Whether to upgrade Waldiez, by default True.
+    printer : Callable[..., None]
+        The printer function to use, defaults to print.
+    """
+    install_requirements({"waldiez"}, upgrade, printer)
+
+
+async def a_install_waldiez(
+    upgrade: bool = True,
+    printer: Callable[..., None] = print,
+) -> None:
+    """Install Waldiez asynchronously.
+
+    Parameters
+    ----------
+    upgrade : bool, optional
+        Whether to upgrade Waldiez, by default True.
+    printer : Callable[..., None]
+        The printer function to use, defaults to print.
+    """
+    await a_install_requirements({"waldiez"}, upgrade, printer)

waldiez/running/protocol.py

@@ -204,6 +204,7 @@ class WaldiezRunnerProtocol(Protocol):
         uploads_root: Path | None,
         temp_dir: Path,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> None:
         """Actions to perform after running the flow.
 
@@ -221,6 +222,8 @@ class WaldiezRunnerProtocol(Protocol):
             The path to the temporary directory.
         skip_mmd : bool
             Whether to skip generating the mermaid diagram.
+        skip_timeline : bool
+            Whether to skip generating the timeline JSON.
         """
 
     async def a_after_run(
@@ -234,6 +237,7 @@ class WaldiezRunnerProtocol(Protocol):
         uploads_root: Path | None,
         temp_dir: Path,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> None:
         """Asynchronously perform actions after running the flow.
 
@@ -251,6 +255,8 @@ class WaldiezRunnerProtocol(Protocol):
             The path to the temporary directory.
         skip_mmd : bool
             Whether to skip generating the mermaid diagram.
+        skip_timeline : bool
+            Whether to skip generating the timeline JSON.
         """
 
     def is_running(self) -> bool:  # pyright: ignore

waldiez/running/subprocess_runner.py

@@ -49,6 +49,7 @@ class WaldiezSubprocessRunner(WaldiezBaseRunner):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> Union[
         "ChatResult",
         list["ChatResult"],
@@ -63,6 +64,7 @@ class WaldiezSubprocessRunner(WaldiezBaseRunner):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> Union[
         "ChatResult",
         list["ChatResult"],
@@ -77,6 +79,7 @@ class WaldiezSubprocessRunner(WaldiezBaseRunner):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> None:
         """Start the Waldiez workflow."""
         # This method should be implemented to start the workflow
@@ -88,6 +91,7 @@ class WaldiezSubprocessRunner(WaldiezBaseRunner):
         output_file: Path,
         uploads_root: Path | None,
         skip_mmd: bool,
+        skip_timeline: bool,
     ) -> None:
         """Start the Waldiez workflow asynchronously."""
         # This method should be implemented to start the workflow asynchronously