waldiez 0.5.7__py3-none-any.whl → 0.5.8__py3-none-any.whl

waldiez/running/base_runner.py

@@ -7,6 +7,7 @@
 
 """Base runner for Waldiez workflows."""
 
+import shutil
 import sys
 import tempfile
 import threading
@@ -19,9 +20,9 @@ from typing import (
     Coroutine,
     Optional,
     Type,
-    Union,
 )
 
+from aiofiles.os import wrap
 from anyio.from_thread import start_blocking_portal
 from typing_extensions import Self
 
@@ -43,15 +44,19 @@ from .utils import (
 
 if TYPE_CHECKING:
     from autogen.events import BaseEvent  # type: ignore[import-untyped]
-    from autogen.io.run_response import (  # type: ignore[import-untyped]
-        AsyncRunResponseProtocol,
-        RunResponseProtocol,
-    )
 
 
 class WaldiezBaseRunner(WaldiezRunnerProtocol):
     """Base runner for Waldiez.
 
+    Initialization parameters:
+        - waldiez: The Waldiez flow to run.
+        - output_path: Path to save the output file.
+        - uploads_root: Root directory for uploads.
+        - structured_io: Whether to use structured I/O.
+        - skip_patch_io: Whether to skip patching I/O functions.
+        - dot_env: Path to a .env file for environment variables.
+
     Methods to override:
         - _before_run: Actions to perform before running the flow.
         - _a_before_run: Async actions to perform before running the flow.
@@ -68,6 +73,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
     _structured_io: bool
     _output_path: str | Path | None
     _uploads_root: str | Path | None
+    _dot_env_path: str | Path | None
     _skip_patch_io: bool
     _running: bool
 
@@ -78,6 +84,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         uploads_root: str | Path | None,
         structured_io: bool,
         skip_patch_io: bool = False,
+        dot_env: str | Path | None = None,
     ) -> None:
         """Initialize the Waldiez manager."""
         self._waldiez = waldiez
@@ -86,6 +93,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         WaldiezBaseRunner._output_path = output_path
         WaldiezBaseRunner._uploads_root = uploads_root
         WaldiezBaseRunner._skip_patch_io = skip_patch_io
+        WaldiezBaseRunner._dot_env_path = dot_env
         self._called_install_requirements = False
         self._exporter = WaldiezExporter(waldiez)
         self._stop_requested = threading.Event()
@@ -95,9 +103,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         self._input: (
             Callable[..., str] | Callable[..., Coroutine[Any, Any, str]]
         ) = input
-        self._last_results: Union[
-            list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]
-        ] = []
+        self._last_results: list[dict[str, Any]] = []
         self._last_exception: Exception | None = None
         self._execution_complete_event = threading.Event()
         self._execution_thread: Optional[threading.Thread] = None
@@ -186,6 +192,11 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
                 uploads_root=uploads_root,
                 structured_io=WaldiezBaseRunner._structured_io,
             )
+            if self._dot_env_path:
+                shutil.copyfile(
+                    str(self._dot_env_path),
+                    str(temp_dir / ".env"),
+                )
         return temp_dir
 
     async def _a_before_run(
@@ -203,6 +214,12 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
                 structured_io=self.structured_io,
                 force=True,
             )
+            if self._dot_env_path:
+                wrapped = wrap(shutil.copyfile)
+                await wrapped(
+                    str(self._dot_env_path),
+                    str(temp_dir / ".env"),
+                )
         return temp_dir
 
     def _run(
@@ -213,7 +230,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         skip_mmd: bool,
         skip_timeline: bool,
         **kwargs: Any,
-    ) -> Union[list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]]:
+    ) -> list[dict[str, Any]]:
         """Run the Waldiez flow."""
         raise NotImplementedError(
             "The _run method must be implemented in the subclass."
@@ -227,7 +244,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         skip_mmd: bool,
         skip_timeline: bool,
         **kwargs: Any,
-    ) -> Union[list["AsyncRunResponseProtocol"], list["RunResponseProtocol"]]:
+    ) -> list[dict[str, Any]]:
         """Run the Waldiez flow asynchronously."""
         raise NotImplementedError(
             "The _a_run method must be implemented in the subclass."
@@ -273,9 +290,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
 
     def _after_run(
         self,
-        results: Union[
-            list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]
-        ],
+        results: list[dict[str, Any]],
         output_file: Path,
         uploads_root: Path | None,
         temp_dir: Path,
@@ -300,9 +315,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
 
     async def _a_after_run(
         self,
-        results: Union[
-            list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]
-        ],
+        results: list[dict[str, Any]],
         output_file: Path,
         uploads_root: Path | None,
         temp_dir: Path,
@@ -367,6 +380,8 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             for req in self.waldiez.requirements
             if req not in sys.modules and "waldiez" not in req
         }
+        if "python-dotenv" not in extra_requirements:
+            extra_requirements.add("python-dotenv")
         return extra_requirements
 
     def install_requirements(self) -> None:
@@ -444,8 +459,9 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         structured_io: bool | None = None,
         skip_mmd: bool = False,
         skip_timeline: bool = False,
+        dot_env: str | Path | None = None,
         **kwargs: Any,
-    ) -> Union[list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]]:
+    ) -> list[dict[str, Any]]:
         """Run the Waldiez flow in blocking mode.
 
         Parameters
@@ -461,20 +477,25 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             Whether to skip generating the mermaid diagram, by default False.
         skip_timeline : bool
             Whether to skip generating the timeline JSON.
+        dot_env : str | Path | None
+            The path to the .env file, if any.
         **kwargs : Any
             Additional keyword arguments for the run method.
 
         Returns
         -------
-        Union[list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]]
-            The result of the run, which can be a list of RunResponseProtocol
-            or a list of AsyncRunResponseProtocol.
+        list[dict[str, Any]]
+            The result of the run.
 
         Raises
         ------
         RuntimeError
             If the runner is already running.
         """
+        if dot_env is not None:
+            resolved = Path(dot_env).resolve()
+            if resolved.is_file():
+                WaldiezBaseRunner._dot_env_path = resolved
         if structured_io is not None:
             WaldiezBaseRunner._structured_io = structured_io
         if self.is_running():
@@ -499,9 +520,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         self.install_requirements()
         refresh_environment()
         WaldiezBaseRunner._running = True
-        results: Union[
-            list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]
-        ]
+        results: list[dict[str, Any]] = []
         old_env_vars = set_env_vars(self.waldiez.get_flow_env_vars())
         try:
             with chdir(to=temp_dir):
@@ -537,7 +556,9 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         structured_io: bool | None = None,
         skip_mmd: bool = False,
         skip_timeline: bool = False,
-    ) -> Union[list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]]:
+        dot_env: str | Path | None = None,
+        **kwargs: Any,
+    ) -> list[dict[str, Any]]:
         """Run the Waldiez flow asynchronously.
 
         Parameters
@@ -553,18 +574,25 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             Whether to skip generating the mermaid diagram, by default False.
         skip_timeline : bool
             Whether to skip generating the timeline JSON, by default False.
+        dot_env : str | Path | None
+            The path to the .env file, if any.
+        **kwargs : Any
+            Additional keyword arguments for the run method.
 
         Returns
         -------
-        Union[list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]]
-            The result of the run, which can be a list of RunResponseProtocol
-            or a list of AsyncRunResponseProtocol.
+        list[dict[str, Any]]
+            The result of the run.
 
         Raises
         ------
         RuntimeError
             If the runner is already running.
         """
+        if dot_env is not None:
+            resolved = Path(dot_env).resolve()
+            if resolved.is_file():
+                WaldiezBaseRunner._dot_env_path = resolved
         if structured_io is not None:
             WaldiezBaseRunner._structured_io = structured_io
         if self.is_running():
@@ -580,9 +608,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         await self.a_install_requirements()
         refresh_environment()
         WaldiezBaseRunner._running = True
-        results: Union[
-            list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]
-        ]
+        results: list[dict[str, Any]] = []
         old_env_vars = set_env_vars(self.waldiez.get_flow_env_vars())
         try:
             async with a_chdir(to=temp_dir):
@@ -616,6 +642,8 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         structured_io: bool | None = None,
         skip_mmd: bool = False,
         skip_timeline: bool = False,
+        dot_env: str | Path | None = None,
+        **kwargs: Any,
     ) -> None:
         """Start running the Waldiez flow in a non-blocking way.
 
@@ -631,12 +659,20 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             Whether to skip generating the mermaid diagram, by default False.
         skip_timeline : bool
             Whether to skip generating the timeline JSON, by default False.
+        dot_env : str | Path | None
+            The path to the .env file, if any.
+        **kwargs : Any
+            Additional keyword arguments for the start method.
 
         Raises
         ------
         RuntimeError
             If the runner is already running.
         """
+        if dot_env is not None:
+            resolved = Path(dot_env).resolve()
+            if resolved.is_file():
+                WaldiezBaseRunner._dot_env_path = resolved
         if structured_io is not None:
             WaldiezBaseRunner._structured_io = structured_io
         if self.is_running():
@@ -668,6 +704,8 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         structured_io: bool | None = None,
         skip_mmd: bool = False,
         skip_timeline: bool = False,
+        dot_env: str | Path | None = None,
+        **kwargs: Any,
     ) -> None:
         """Asynchronously start running the Waldiez flow in a non-blocking way.
 
@@ -683,12 +721,20 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             Whether to skip generating the mermaid diagram, by default False.
         skip_timeline : bool = False
             Whether to skip generating the timeline JSON, by default False.
+        dot_env : str | Path | None = None
+            The path to the .env file, if any.
+        **kwargs : Any
+            Additional keyword arguments for the start method.
 
         Raises
         ------
         RuntimeError
             If the runner is already running.
         """
+        if dot_env is not None:
+            resolved = Path(dot_env).resolve()
+            if resolved.is_file():
+                WaldiezBaseRunner._dot_env_path = resolved
         if structured_io is not None:
             WaldiezBaseRunner._structured_io = structured_io
         if self.is_running():
@@ -714,9 +760,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
 
     def after_run(
         self,
-        results: Union[
-            list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]
-        ],
+        results: list[dict[str, Any]],
         output_file: Path,
         uploads_root: Path | None,
         temp_dir: Path,
@@ -727,7 +771,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
 
         Parameters
         ----------
-        results : Union[ChatResult, list[ChatResult], dict[int, ChatResult]]
+        results : list[dict[str, Any]]
             The results of the flow run.
         output_file : Path
             The path to the output file.
@@ -751,9 +795,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
 
     async def a_after_run(
         self,
-        results: Union[
-            list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]
-        ],
+        results: list[dict[str, Any]],
         output_file: Path,
         uploads_root: Path | None,
         temp_dir: Path,
@@ -764,7 +806,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
 
         Parameters
         ----------
-        results : Union[ChatResult, list[ChatResult], dict[int, ChatResult]]
+        results : list[dict[str, Any]]
             The results of the flow run.
         output_file : Path
             The path to the output file.
@@ -833,6 +875,11 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         """Check if the runner is using structured IO."""
         return WaldiezBaseRunner._structured_io
 
+    @property
+    def dot_env_path(self) -> str | Path | None:
+        """Get the path to the .env file, if any."""
+        return WaldiezBaseRunner._dot_env_path
+
     @property
     def output_path(self) -> str | Path | None:
         """Get the output path for the runner."""
@@ -859,6 +906,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         output_path: str | Path | None = None,
         uploads_root: str | Path | None = None,
         structured_io: bool = False,
+        dot_env: str | Path | None = None,
     ) -> "WaldiezBaseRunner":
         """Load a waldiez flow from a file and create a runner.
 
@@ -881,6 +929,8 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
         structured_io : bool, optional
             Whether to use structured IO instead of the default 'input/print',
             by default False.
+        dot_env : str | Path | None, optional
+            The path to the .env file, if any, by default None.
 
         Returns
         -------
@@ -899,6 +949,7 @@ class WaldiezBaseRunner(WaldiezRunnerProtocol):
             output_path=output_path,
             uploads_root=uploads_root,
             structured_io=structured_io,
+            dot_env=dot_env,
         )
 
     def __enter__(self) -> Self:

waldiez/running/protocol.py

@@ -1,16 +1,10 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-
+# pyright: reportReturnType=false
 """Waldiez Runner protocol."""
 
 from pathlib import Path
-from typing import TYPE_CHECKING, Protocol, Union, runtime_checkable
-
-if TYPE_CHECKING:
-    from autogen.io.run_response import (  # type: ignore[import-untyped]
-        AsyncRunResponseProtocol,
-        RunResponseProtocol,
-    )
+from typing import Any, Protocol, runtime_checkable
 
 
 @runtime_checkable
@@ -63,6 +57,9 @@ class WaldiezRunnerProtocol(Protocol):
         uploads_root: str | Path | None,
         structured_io: bool | None = None,
         skip_mmd: bool = False,
+        skip_timeline: bool = False,
+        dot_env: str | Path | None = None,
+        **kwargs: Any,
     ) -> None:
         """Start running the Waldiez flow in a non-blocking way.
 
@@ -78,6 +75,12 @@ class WaldiezRunnerProtocol(Protocol):
             Whether to use structured IO instead of the default 'input/print'.
         skip_mmd : bool
             Whether to skip generating the mermaid diagram.
+        skip_timeline : bool
+            Whether to skip generating the timeline JSON.
+        dot_env : str | Path | None
+            The path to the .env file, if any.
+        **kwargs : Any
+            Additional keyword arguments for the start method.
 
         Raises
         ------
@@ -91,6 +94,9 @@ class WaldiezRunnerProtocol(Protocol):
         uploads_root: str | Path | None,
         structured_io: bool | None = None,
         skip_mmd: bool = False,
+        skip_timeline: bool = False,
+        dot_env: str | Path | None = None,
+        **kwargs: Any,
     ) -> None:
         """Asynchronously start running the Waldiez flow in a non-blocking way.
 
@@ -106,6 +112,12 @@ class WaldiezRunnerProtocol(Protocol):
             Whether to use structured IO instead of the default 'input/print'.
         skip_mmd : bool
             Whether to skip generating the mermaid diagram.
+        skip_timeline : bool
+            Whether to skip generating the timeline JSON.
+        dot_env : str | Path | None
+            The path to the .env file, if any.
+        **kwargs : Any
+            Additional keyword arguments for the start method.
 
         Raises
         ------
@@ -119,10 +131,10 @@ class WaldiezRunnerProtocol(Protocol):
         uploads_root: str | Path | None,
         structured_io: bool | None = None,
         skip_mmd: bool = False,
-    ) -> Union[
-        list["RunResponseProtocol"],
-        list["AsyncRunResponseProtocol"],
-    ]:  # pyright: ignore
+        skip_timeline: bool = False,
+        dot_env: str | Path | None = None,
+        **kwargs: Any,
+    ) -> list[dict[str, Any]]:
         """Run the Waldiez flow in a blocking way.
 
         Parameters
@@ -135,15 +147,17 @@ class WaldiezRunnerProtocol(Protocol):
             Whether to use structured IO instead of the default 'input/print'.
         skip_mmd : bool
             Whether to skip generating the mermaid diagram.
+        skip_timeline : bool
+            Whether to skip generating the timeline JSON.
+        dot_env : str | Path | None
+            The path to the .env file, if any.
+        **kwargs : Any
+            Additional keyword arguments for the run method.
 
         Returns
         -------
-        Union[
-            list["RunResponseProtocol"],
-            list["AsyncRunResponseProtocol"],
-        ]
-            The result of the run, which can be a list of RunResponseProtocol
-            or a list of AsyncRunResponseProtocol.
+        list[dict[str, Any]]
+            The result of the run.
         """
 
     async def a_run(
@@ -152,10 +166,10 @@ class WaldiezRunnerProtocol(Protocol):
         uploads_root: str | Path | None,
         structured_io: bool | None = None,
         skip_mmd: bool = False,
-    ) -> Union[
-        list["RunResponseProtocol"],
-        list["AsyncRunResponseProtocol"],
-    ]:  # pyright: ignore
+        skip_timeline: bool = False,
+        dot_env: str | Path | None = None,
+        **kwargs: Any,
+    ) -> list[dict[str, Any]]:
         """Run the Waldiez flow.
 
         Parameters
@@ -168,23 +182,22 @@ class WaldiezRunnerProtocol(Protocol):
             Whether to use structured IO instead of the default 'input/print'.
         skip_mmd : bool
             Whether to skip generating the mermaid diagram.
+        skip_timeline : bool
+            Whether to skip generating the timeline JSON.
+        dot_env : str | Path | None
+            The path to the .env file, if any.
+        **kwargs : Any
+            Additional keyword arguments for the a_run method.
 
         Returns
         -------
-        Union[
-            list["RunResponseProtocol"],
-            list["AsyncRunResponseProtocol"],
-        ]
-            The result of the run, which can be a list of RunResponseProtocol
-            or a list of AsyncRunResponseProtocol.
+        list[dict[str, Any]]
+            The result of the run.
         """
 
     def after_run(
         self,
-        results: Union[
-            list["RunResponseProtocol"],
-            list["AsyncRunResponseProtocol"],
-        ],
+        results: list[dict[str, Any]],
         output_file: Path,
         uploads_root: Path | None,
         temp_dir: Path,
@@ -195,12 +208,8 @@ class WaldiezRunnerProtocol(Protocol):
 
         Parameters
         ----------
-        results : Union[
-            list["RunResponseProtocol"],
-            list["AsyncRunResponseProtocol"],
-        ]
-            The results of the run, which can be a list of RunResponseProtocol
-            or a list of AsyncRunResponseProtocol.
+        results : list[dict[str, Any]]
+            The results of the run.
         output_file : Path
             The path to the output file.
         uploads_root : Path | None
@@ -215,10 +224,7 @@ class WaldiezRunnerProtocol(Protocol):
 
     async def a_after_run(
         self,
-        results: Union[
-            list["RunResponseProtocol"],
-            list["AsyncRunResponseProtocol"],
-        ],
+        results: list[dict[str, Any]],
         output_file: Path,
         uploads_root: Path | None,
         temp_dir: Path,
@@ -229,12 +235,8 @@ class WaldiezRunnerProtocol(Protocol):
 
         Parameters
         ----------
-        results : Union[
-            list["RunResponseProtocol"],
-            list["AsyncRunResponseProtocol"]
-        ]
-            The results of the run, which can be a list of RunResponseProtocol
-            or a list of AsyncRunResponseProtocol.
+        results : list[dict[str, Any]]
+            The results of the run.
         output_file : Path
             The path to the output file.
         uploads_root : Path | None

waldiez/running/run_results.py

@@ -3,20 +3,12 @@
 
 """Waldiez run results module."""
 
-from typing import TYPE_CHECKING, TypedDict, Union
-
-if TYPE_CHECKING:
-    from autogen.io.run_response import (  # type: ignore[import-untyped]
-        AsyncRunResponseProtocol,
-        RunResponseProtocol,
-    )
+from typing import Any, TypedDict
 
 
 class WaldiezRunResults(TypedDict):
     """Results of the Waldiez run."""
 
-    results: Union[
-        list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]
-    ]
+    results: list[dict[str, Any]]
     exception: Exception | None
     completed: bool

waldiez/running/standard_runner.py

@@ -32,10 +32,6 @@ from .utils import chdir
 
 if TYPE_CHECKING:
     from autogen.events import BaseEvent  # type: ignore
-    from autogen.io.run_response import (  # type: ignore
-        AsyncRunResponseProtocol,
-        RunResponseProtocol,
-    )
     from autogen.messages import BaseMessage  # type: ignore
 
 
@@ -48,6 +44,8 @@ class WaldiezStandardRunner(WaldiezBaseRunner):
         output_path: str | Path | None = None,
         uploads_root: str | Path | None = None,
         structured_io: bool = False,
+        dot_env: str | Path | None = None,
+        **kwargs: Any,
     ) -> None:
         """Initialize the Waldiez manager."""
         super().__init__(
@@ -55,6 +53,8 @@ class WaldiezStandardRunner(WaldiezBaseRunner):
             output_path=output_path,
             uploads_root=uploads_root,
             structured_io=structured_io,
+            dot_env=dot_env,
+            **kwargs,
         )
         self._execution_thread: threading.Thread | None = None
         self._loaded_module: ModuleType | None = None
@@ -107,7 +107,7 @@ class WaldiezStandardRunner(WaldiezBaseRunner):
         skip_mmd: bool,
         skip_timeline: bool,
         **kwargs: Any,
-    ) -> Union[list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]]:
+    ) -> list[dict[str, Any]]:
         """Run the Waldiez workflow."""
         from autogen.io import IOStream  # type: ignore
 
@@ -185,7 +185,7 @@ class WaldiezStandardRunner(WaldiezBaseRunner):
                     )
                     event.content.respond(user_input)
                 else:
-                    self._send(event)
+                    self._send(event)  # pyright: ignore
             self._processed_events += 1
         except Exception as e:
             raise RuntimeError(
@@ -301,7 +301,7 @@ class WaldiezStandardRunner(WaldiezBaseRunner):
                         user_input = await user_input
                     await event.content.respond(user_input)
                 else:
-                    self._send(event)
+                    self._send(event)  # pyright: ignore
             self._processed_events += 1
         except Exception as e:
             raise RuntimeError(
@@ -317,18 +317,18 @@ class WaldiezStandardRunner(WaldiezBaseRunner):
         skip_mmd: bool = False,
         skip_timeline: bool = False,
         **kwargs: Any,
-    ) -> Union[list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]]:
+    ) -> list[dict[str, Any]]:
         """Run the Waldiez workflow asynchronously."""
 
         # fmt: off
-        async def _execute_workflow() -> Union[list["RunResponseProtocol"], list["AsyncRunResponseProtocol"]]:
+        async def _execute_workflow() -> list[dict[str, Any]]:
             # fmt: on
             """Execute the workflow in an async context."""
             from autogen.io import IOStream  # pyright: ignore
 
             from waldiez.io import StructuredIOStream
 
-            results: Union[list["AsyncRunResponseProtocol"], list["RunResponseProtocol"]]
+            results: list[dict[str, Any]] = []
             try:
                 self._loaded_module = self._load_module(output_file, temp_dir)
                 if self._stop_requested.is_set():