waldiez 0.1.6__tar.gz → 0.1.8__tar.gz

waldiez-0.1.8/PKG-INFO

@@ -0,0 +1,241 @@
+Metadata-Version: 2.3
+Name: waldiez
+Version: 0.1.8
+Summary: waldiez
+Project-URL: homepage, https://waldiez.github.io/waldiez/
+Project-URL: repository, https://github.com/waldiez/waldiez.git
+Author-email: Panagiotis Kasnesis <pkasnesis@thingenious.io>, Lazaros Toumanidis <laztoum@protonmail.com>
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: <3.13,>=3.10
+Requires-Dist: ag2==0.3.2
+Requires-Dist: jupytext
+Provides-Extra: ag2-extras
+Requires-Dist: ag2[anthropic]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[bedrock]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[gemini]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[groq]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[mistral]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[retrievechat-mongodb]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[retrievechat-pgvector]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[retrievechat-qdrant]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[retrievechat]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[together]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[websurfer]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: chromadb==0.5.3; extra == 'ag2-extras'
+Requires-Dist: fastembed==0.4.2; extra == 'ag2-extras'
+Requires-Dist: pgvector>=0.2.5; extra == 'ag2-extras'
+Requires-Dist: psycopg[binary]>=3.1.18; extra == 'ag2-extras'
+Requires-Dist: pymongo==4.10.1; extra == 'ag2-extras'
+Requires-Dist: qdrant-client==1.12.1; extra == 'ag2-extras'
+Provides-Extra: dev
+Requires-Dist: autoflake==2.3.1; extra == 'dev'
+Requires-Dist: bandit==1.7.10; extra == 'dev'
+Requires-Dist: black[jupyter]==24.10.0; extra == 'dev'
+Requires-Dist: flake8==7.1.1; extra == 'dev'
+Requires-Dist: isort==5.13.2; extra == 'dev'
+Requires-Dist: mypy==1.13.0; extra == 'dev'
+Requires-Dist: pre-commit==4.0.1; extra == 'dev'
+Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
+Requires-Dist: pylint==3.3.1; extra == 'dev'
+Requires-Dist: python-dotenv==1.0.1; extra == 'dev'
+Requires-Dist: ruff==0.7.4; extra == 'dev'
+Requires-Dist: types-pyyaml==6.0.12; extra == 'dev'
+Requires-Dist: yamllint==1.35.1; extra == 'dev'
+Provides-Extra: docs
+Requires-Dist: mdx-include==1.4.2; extra == 'docs'
+Requires-Dist: mdx-truly-sane-lists==1.3; extra == 'docs'
+Requires-Dist: mkdocs-jupyter==0.25.1; extra == 'docs'
+Requires-Dist: mkdocs-macros-plugin==1.3.7; extra == 'docs'
+Requires-Dist: mkdocs-material==9.5.44; extra == 'docs'
+Requires-Dist: mkdocs-minify-html-plugin==0.2.3; extra == 'docs'
+Requires-Dist: mkdocs==1.6.1; extra == 'docs'
+Requires-Dist: mkdocstrings-python==1.12.2; extra == 'docs'
+Requires-Dist: mkdocstrings[crystal,python]==0.27.0; extra == 'docs'
+Provides-Extra: test
+Requires-Dist: pytest-cov==6.0.0; extra == 'test'
+Requires-Dist: pytest-html==4.1.1; extra == 'test'
+Requires-Dist: pytest-sugar==1.0.0; extra == 'test'
+Requires-Dist: pytest-timeout==2.3.1; extra == 'test'
+Requires-Dist: pytest-xdist==3.6.1; extra == 'test'
+Requires-Dist: pytest==8.3.3; extra == 'test'
+Description-Content-Type: text/markdown
+
+# Waldiez
+
+![CI Build](https://github.com/waldiez/waldiez/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/waldiez/badge.svg)](https://coveralls.io/github/waldiez/waldiez) [![PyPI version](https://badge.fury.io/waldiez/waldiez.svg)](https://badge.fury.io/py/waldiez)
+
+Translate a Waldiez flow:
+
+![Flow](https://raw.githubusercontent.com/waldiez/waldiez/refs/heads/main/docs/static/images/overview.webp)
+
+To a python script or a jupyter notebook with the corresponding [ag2](https://github.com/ag2ai/ag2/) agents and chats.
+
+## Features
+
+- Export .waldiez flows to .py or .ipynb
+- Run a .waldiez flow
+- Include a `logs` folder with the logs of the flow in csv format
+- Provide a custom [IOSStream](https://ag2ai.github.io/ag2/docs/reference/io/base#iostream) to handle input and output.
+
+## Installation
+
+On PyPI:
+
+```bash
+python -m pip install waldiez
+```
+
+From the repository:
+
+```bash
+python -m pip install git+https://github.com/waldiez/waldiez.git
+```
+
+## Usage
+
+### CLI
+
+```bash
+# Export a Waldiez flow to a python script or a jupyter notebook
+waldiez --export /path/to/a/flow.waldiez --output /path/to/an/output/flow[.py|.ipynb]
+# Export and run the script, optionally force generation if the output file already exists
+waldiez /path/to/a/flow.waldiez --output /path/to/an/output/flow[.py] [--force]
+```
+
+### Using docker/podman
+
+```shell
+CONTAINER_COMMAND=docker # or podman
+# pull the image
+$CONTAINER_COMMAND pull waldiez/waldiez
+# Export a Waldiez flow to a python script or a jupyter notebook
+$CONTAINER_COMMAND run \
+  --rm \
+  -v /path/to/a/flow.waldiez:/flow.waldiez \
+  -v /path/to/an/output:/output \
+  waldiez/waldiez --export /flow.waldiez --output /output/flow[.py|.ipynb]
+
+# with selinux and/or podman, you might get permission (or file not found) errors, so you can try:
+$CONTAINER_COMMAND run \
+  --rm \
+  -v /path/to/a/flow.waldiez:/flow.waldiez \
+  -v /path/to/an/output:/output \
+  --userns=keep-id \
+  --security-opt label=disable \
+  waldiez/waldiez --export /flow.waldiez --output /output/flow[.py|.ipynb]
+```
+
+```shell
+# Export and run the script
+$CONTAINER_COMMAND run --rm -v /path/to/a/flow.waldiez:/flow.waldiez -v /path/to/an/output:/output waldiez/waldiez /flow.waldiez --output /output/output[.py]
+```
+
+### As a library
+
+#### Export a flow
+
+```python
+# Export a Waldiez flow to a python script or a jupyter notebook
+from waldiez import WaldiezExporter
+flow_path = "/path/to/a/flow.waldiez"
+output_path = "/path/to/an/output.py"  # or .ipynb
+exporter = WaldiezExporter.load(flow_path)
+exporter.export(output_path)
+```
+  
+#### Run a flow
+
+```python
+# Run a flow
+from waldiez import WaldiezRunner
+flow_path = "/path/to/a/flow.waldiez"
+output_path = "/path/to/an/output.py"
+runner = WaldiezRunner.load(flow_path)
+runner.run(output_path=output_path)
+```
+
+#### Run a flow with a custom IOStream
+
+```python
+# Run the flow with a custom IOStream
+# In case the standard 'input' and 'print' functions cannot be used
+import time
+import threading
+
+from typing import Any
+
+from waldiez import WaldiezRunner
+from waldiez.io import WaldiezIOStream
+
+flow_path = "/path/to/a/flow.waldiez"
+output_path = "/path/to/an/output.py"
+
+
+def custom_print_function(*args: Any, sep: str = " ", **kwargs: Any) -> None:
+    """Custom print function."""
+    print(*args, sep=sep, **kwargs)
+
+
+# Custom input handler
+class InputProcessorWrapper:
+    """Wrapper input processor.
+    
+    To manage the interaction between the custom input processor and IOStream.
+    """
+
+    def __init__(self):
+        self.stream = None  # Placeholder for the WaldiezIOStream instance
+        self.lock = threading.Lock()  # Ensure thread-safe operations
+
+    def custom_input_processor(self, prompt: str) -> None:
+        """Simulate external input and send it back to the IOStream."""
+        def external_input_simulation():
+            with self.lock:  # Ensure thread-safe access
+                time.sleep(2)  # Simulate delay for network input
+                if self.stream:
+                    self.stream.set_input("Simulated external input")
+                else:
+                    raise RuntimeError("Stream reference not set!")
+
+        threading.Thread(target=external_input_simulation, daemon=True).start()
+
+    def set_stream(self, stream: "WaldiezIOStream"):
+        """Set the WaldiezIOStream instance."""
+        with self.lock:  # Ensure thread-safe setting of the stream
+            self.stream = stream
+
+processor_wrapper = InputProcessorWrapper()
+
+stream = WaldiezIOStream(
+    input_timeout=30,
+    print_function=
+    on_prompt_input=processor_wrapper.custom_input_processor,
+)
+
+# Link the processor wrapper to the WaldiezIOStream instance
+processor_wrapper.set_stream(custom_stream)
+
+with WaldiezIOStream.set_default(io_stream):
+    runner = WaldiezRunner.load(flow_path)
+    runner.run(stream=io_stream, output_path=output_path)
+
+```
+
+### Tools
+
+- [ag2 (formerly AutoGen)](https://github.com/ag2ai/ag2)
+- [juptytext](https://github.com/mwouts/jupytext)
+- [pydantic](https://github.com/pydantic/pydantic)
+
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](https://github.com/waldiez/waldiez/blob/main/LICENSE) file for details.

waldiez-0.1.8/README.md

@@ -0,0 +1,170 @@
+# Waldiez
+
+![CI Build](https://github.com/waldiez/waldiez/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/waldiez/badge.svg)](https://coveralls.io/github/waldiez/waldiez) [![PyPI version](https://badge.fury.io/waldiez/waldiez.svg)](https://badge.fury.io/py/waldiez)
+
+Translate a Waldiez flow:
+
+![Flow](https://raw.githubusercontent.com/waldiez/waldiez/refs/heads/main/docs/static/images/overview.webp)
+
+To a python script or a jupyter notebook with the corresponding [ag2](https://github.com/ag2ai/ag2/) agents and chats.
+
+## Features
+
+- Export .waldiez flows to .py or .ipynb
+- Run a .waldiez flow
+- Include a `logs` folder with the logs of the flow in csv format
+- Provide a custom [IOSStream](https://ag2ai.github.io/ag2/docs/reference/io/base#iostream) to handle input and output.
+
+## Installation
+
+On PyPI:
+
+```bash
+python -m pip install waldiez
+```
+
+From the repository:
+
+```bash
+python -m pip install git+https://github.com/waldiez/waldiez.git
+```
+
+## Usage
+
+### CLI
+
+```bash
+# Export a Waldiez flow to a python script or a jupyter notebook
+waldiez --export /path/to/a/flow.waldiez --output /path/to/an/output/flow[.py|.ipynb]
+# Export and run the script, optionally force generation if the output file already exists
+waldiez /path/to/a/flow.waldiez --output /path/to/an/output/flow[.py] [--force]
+```
+
+### Using docker/podman
+
+```shell
+CONTAINER_COMMAND=docker # or podman
+# pull the image
+$CONTAINER_COMMAND pull waldiez/waldiez
+# Export a Waldiez flow to a python script or a jupyter notebook
+$CONTAINER_COMMAND run \
+  --rm \
+  -v /path/to/a/flow.waldiez:/flow.waldiez \
+  -v /path/to/an/output:/output \
+  waldiez/waldiez --export /flow.waldiez --output /output/flow[.py|.ipynb]
+
+# with selinux and/or podman, you might get permission (or file not found) errors, so you can try:
+$CONTAINER_COMMAND run \
+  --rm \
+  -v /path/to/a/flow.waldiez:/flow.waldiez \
+  -v /path/to/an/output:/output \
+  --userns=keep-id \
+  --security-opt label=disable \
+  waldiez/waldiez --export /flow.waldiez --output /output/flow[.py|.ipynb]
+```
+
+```shell
+# Export and run the script
+$CONTAINER_COMMAND run --rm -v /path/to/a/flow.waldiez:/flow.waldiez -v /path/to/an/output:/output waldiez/waldiez /flow.waldiez --output /output/output[.py]
+```
+
+### As a library
+
+#### Export a flow
+
+```python
+# Export a Waldiez flow to a python script or a jupyter notebook
+from waldiez import WaldiezExporter
+flow_path = "/path/to/a/flow.waldiez"
+output_path = "/path/to/an/output.py"  # or .ipynb
+exporter = WaldiezExporter.load(flow_path)
+exporter.export(output_path)
+```
+  
+#### Run a flow
+
+```python
+# Run a flow
+from waldiez import WaldiezRunner
+flow_path = "/path/to/a/flow.waldiez"
+output_path = "/path/to/an/output.py"
+runner = WaldiezRunner.load(flow_path)
+runner.run(output_path=output_path)
+```
+
+#### Run a flow with a custom IOStream
+
+```python
+# Run the flow with a custom IOStream
+# In case the standard 'input' and 'print' functions cannot be used
+import time
+import threading
+
+from typing import Any
+
+from waldiez import WaldiezRunner
+from waldiez.io import WaldiezIOStream
+
+flow_path = "/path/to/a/flow.waldiez"
+output_path = "/path/to/an/output.py"
+
+
+def custom_print_function(*args: Any, sep: str = " ", **kwargs: Any) -> None:
+    """Custom print function."""
+    print(*args, sep=sep, **kwargs)
+
+
+# Custom input handler
+class InputProcessorWrapper:
+    """Wrapper input processor.
+    
+    To manage the interaction between the custom input processor and IOStream.
+    """
+
+    def __init__(self):
+        self.stream = None  # Placeholder for the WaldiezIOStream instance
+        self.lock = threading.Lock()  # Ensure thread-safe operations
+
+    def custom_input_processor(self, prompt: str) -> None:
+        """Simulate external input and send it back to the IOStream."""
+        def external_input_simulation():
+            with self.lock:  # Ensure thread-safe access
+                time.sleep(2)  # Simulate delay for network input
+                if self.stream:
+                    self.stream.set_input("Simulated external input")
+                else:
+                    raise RuntimeError("Stream reference not set!")
+
+        threading.Thread(target=external_input_simulation, daemon=True).start()
+
+    def set_stream(self, stream: "WaldiezIOStream"):
+        """Set the WaldiezIOStream instance."""
+        with self.lock:  # Ensure thread-safe setting of the stream
+            self.stream = stream
+
+processor_wrapper = InputProcessorWrapper()
+
+stream = WaldiezIOStream(
+    input_timeout=30,
+    print_function=
+    on_prompt_input=processor_wrapper.custom_input_processor,
+)
+
+# Link the processor wrapper to the WaldiezIOStream instance
+processor_wrapper.set_stream(custom_stream)
+
+with WaldiezIOStream.set_default(io_stream):
+    runner = WaldiezRunner.load(flow_path)
+    runner.run(stream=io_stream, output_path=output_path)
+
+```
+
+### Tools
+
+- [ag2 (formerly AutoGen)](https://github.com/ag2ai/ag2)
+- [juptytext](https://github.com/mwouts/jupytext)
+- [pydantic](https://github.com/pydantic/pydantic)
+
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](https://github.com/waldiez/waldiez/blob/main/LICENSE) file for details.

{waldiez-0.1.6 → waldiez-0.1.8}/pyproject.toml

@@ -22,13 +22,12 @@ classifiers = [
 ]
 dependencies = [
     "jupytext",
-    "autogen-agentchat==0.2.37",
-    "twisted==24.10.0",
+    "ag2==0.3.2"
 ]
 
 [project.urls]
-homepage = 'https://waldiez.github.io/py/'
-repository = 'https://github.com/waldiez/py.git'
+homepage = 'https://waldiez.github.io/waldiez/'
+repository = 'https://github.com/waldiez/waldiez.git'
 
 [tool.hatch.build.targets.sdist]
 packages = ["waldiez", "waldiez.*"]
@@ -50,33 +49,33 @@ dev = [
     'pydocstyle==6.3.0',
     'pylint==3.3.1',
     'python-dotenv==1.0.1',
-    'ruff==0.7.1',
+    'ruff==0.7.4',
     'types-PyYAML==6.0.12',
     'yamllint==1.35.1',
 ]
-autogen_extras =[
-    'chromadb==0.5.15',
-    'fastembed==0.4.1',
-    'pgvector==0.3.5',
-    'psycopg[binary]>=3.2.3',
-    'autogen-agentchat[retrievechat]==0.2.37',
-    'autogen-agentchat[retrievechat-pgvector]==0.2.37',
-    'autogen-agentchat[retrievechat-mongodb]==0.2.37',
-    'autogen-agentchat[retrievechat-qdrant]==0.2.37',
-    'autogen-agentchat[retrievechat-couchbase]==0.2.37',
-    'autogen-agentchat[gemini]==0.2.37',
-    'autogen-agentchat[together]==0.2.37',
-    'autogen-agentchat[anthropic]==0.2.37',
-    'autogen-agentchat[mistral]==0.2.37',
-    'autogen-agentchat[groq]==0.2.37',
-    'autogen-agentchat[bedrock]==0.2.37',
-    'autogen-agentchat[websurfer]==0.2.37',
+ag2_extras =[
+    'chromadb==0.5.3',
+    'fastembed==0.4.2',
+    'pgvector>=0.2.5',
+    'psycopg[binary]>=3.1.18',
+    'ag2[retrievechat]==0.3.2',
+    'ag2[retrievechat-pgvector]==0.3.2',
+    'ag2[retrievechat-mongodb]==0.3.2',
+    'ag2[retrievechat-qdrant]==0.3.2',
+    'ag2[gemini]==0.3.2',
+    'ag2[together]==0.3.2',
+    'ag2[anthropic]==0.3.2',
+    'ag2[mistral]==0.3.2',
+    'ag2[groq]==0.3.2',
+    'ag2[bedrock]==0.3.2',
+    'ag2[websurfer]==0.3.2',
+    'ag2[together]==0.3.2',
     'pymongo==4.10.1',
-    'qdrant-client==1.12.0',
+    'qdrant-client==1.12.1',
 ]
 test = [
     'pytest==8.3.3',
-    'pytest-cov==5.0.0',
+    'pytest-cov==6.0.0',
     'pytest-html==4.1.1',
     'pytest-sugar==1.0.0',
     'pytest-timeout==2.3.1',
@@ -87,10 +86,10 @@ docs = [
   'mdx_truly_sane_lists==1.3',
   'mkdocs==1.6.1',
   'mkdocs-jupyter==0.25.1',
-  'mkdocs-macros-plugin==1.3.6',
-  'mkdocs-material==9.5.42',
+  'mkdocs-macros-plugin==1.3.7',
+  'mkdocs-material==9.5.44',
   'mkdocs-minify-html-plugin==0.2.3',
-  'mkdocstrings[crystal,python]==0.26.2',
+  'mkdocstrings[crystal,python]==0.27.0',
   'mkdocstrings-python==1.12.2'
 ]
 
@@ -271,7 +270,18 @@ line-ending = "lf"
 
 # pytest
 [tool.pytest.ini_options]
-# filterwarnings = []
+filterwarnings = [
+  # DeprecationWarning:
+  #   Type google._upb._message.MessageMapContainer uses PyType_Spec with a metaclass that has custom tp_new.
+  #   This is deprecated and will no longer be allowed in Python 3.14.
+  #
+  # autogen/logger/logger_utils.py:9: DeprecationWarning:
+  # datetime.datetime.utcnow() is deprecated and scheduled for removal in a future version
+  #
+  # DeprecationWarning:
+  # reactor.stop cannot be used inside unit tests
+  "ignore::DeprecationWarning",
+]
 python_files = [
     "tests.py",
     "test_*.py",
@@ -283,7 +293,7 @@ addopts = """
     --exitfirst \
     --durations=10 \
     --color=yes \
-    --timeout=30
+    --timeout=120
 """
 
 

{waldiez-0.1.6 → waldiez-0.1.8}/waldiez/__init__.py

@@ -2,14 +2,12 @@
 
 from ._version import __version__
 from .exporter import WaldiezExporter
-from .io_stream import WaldiezIOStream
 from .models import Waldiez
 from .runner import WaldiezRunner
 
 __all__ = [
     "Waldiez",
     "WaldiezExporter",
-    "WaldiezIOStream",
     "WaldiezRunner",
     "__version__",
 ]

{waldiez-0.1.6 → waldiez-0.1.8}/waldiez/_version.py

@@ -1,3 +1,3 @@
 """Version information for Waldiez."""
 
-__version__ = "0.1.6"
+__version__ = "0.1.8"

{waldiez-0.1.6 → waldiez-0.1.8}/waldiez/cli.py

@@ -6,14 +6,15 @@ import logging
 import os
 import sys
 from pathlib import Path
-from typing import Any, Dict, Optional
-
-from autogen import ChatResult  # type: ignore[import-untyped]
+from typing import TYPE_CHECKING, Any, Dict, Optional
 
 from . import Waldiez, __version__
 from .exporter import WaldiezExporter
 from .runner import WaldiezRunner
 
+if TYPE_CHECKING:
+    from autogen import ChatResult  # type: ignore[import-untyped]
+
 
 def get_parser() -> argparse.ArgumentParser:
     """Get the argument parser for the Waldiez package.
@@ -28,7 +29,7 @@ def get_parser() -> argparse.ArgumentParser:
         prog="waldiez",
     )
     parser.add_argument(
-        "waldiez",
+        "file",
         type=str,
         help="Path to the Waldiez flow (*.waldiez) file.",
     )
@@ -66,7 +67,7 @@ def get_parser() -> argparse.ArgumentParser:
     return parser
 
 
-def _log_result(result: ChatResult) -> None:
+def _log_result(result: "ChatResult") -> None:
     """Log the result of the Waldiez flow."""
     logger = logging.getLogger("waldiez::cli")
     logger.info("Chat History:\n")
@@ -94,9 +95,12 @@ def _run(data: Dict[str, Any], output_path: Optional[str]) -> None:
 def main() -> None:
     """Parse the command line arguments and run the Waldiez flow."""
     parser = get_parser()
+    if len(sys.argv) == 1:
+        parser.print_help()
+        sys.exit(0)
     args = parser.parse_args()
     logger = _get_logger()
-    waldiez_file: str = args.waldiez
+    waldiez_file: str = args.file
     if not os.path.exists(waldiez_file):
         logger.error("File not found: %s", waldiez_file)
         sys.exit(1)

{waldiez-0.1.6 → waldiez-0.1.8}/waldiez/exporting/agents/rag_user/chroma_utils.py

@@ -109,8 +109,7 @@ def get_chroma_db_args(
         f"            embedding_function={embedding_function_arg},\n"
     )
     # The RAG example:
-    # https://microsoft.github.io/autogen/docs/\
-    #                                       notebooks/agentchat_groupchat_RAG
+    # https://ag2ai.github.io/ag2/docs/notebooks/agentchat_groupchat_RAG/
     # raises `InvalidCollectionException`: Collection groupchat does not exist.
     # https://github.com/chroma-core/chroma/issues/861
     # https://github.com/microsoft/autogen/issues/3551#issuecomment-2366930994

{waldiez-0.1.6 → waldiez-0.1.8}/waldiez/exporting/chats/nested.py

@@ -14,7 +14,6 @@ from .helpers import escape_summary_args_quotes
 
 
 def get_nested_chat_trigger_agent_names(
-    all_chats: List[WaldiezChat],
     nested_chat: WaldiezAgentNestedChat,
     agent_names: Dict[str, str],
 ) -> str:
@@ -22,8 +21,6 @@ def get_nested_chat_trigger_agent_names(
 
     Parameters
     ----------
-    all_chats : List[WaldiezChat]
-        All the chats in the flow.
     nested_chat : WaldiezAgentNestedChat
         The nested chat.
     agent_names : Dict[str, str]
@@ -34,14 +31,7 @@ def get_nested_chat_trigger_agent_names(
     str
         The trigger agent names.
     """
-    trigger_agent_ids: List[str] = []
-    for message in nested_chat.triggered_by:
-        waldiez_chat = next(chat for chat in all_chats if chat.id == message.id)
-        if message.is_reply:
-            trigger_agent_ids.append(waldiez_chat.target)
-        else:
-            trigger_agent_ids.append(waldiez_chat.source)
-    agents = [agent_names[agent_id] for agent_id in trigger_agent_ids]
+    agents = [agent_names[agent_id] for agent_id in nested_chat.triggered_by]
     trigger_string = f'{[", ".join(agents)]}'
     return trigger_string.replace("'", '"')
 
@@ -228,7 +218,7 @@ def export_nested_chat(
     use_suffix = len(agent.data.nested_chats) > 1
     for index, entry in enumerate(agent.data.nested_chats):
         trigger_names = get_nested_chat_trigger_agent_names(
-            all_chats=all_chats, nested_chat=entry, agent_names=agent_names
+            nested_chat=entry, agent_names=agent_names
         )
         chat_queue, extra_methods = get_nested_chat_queue(
             nested_chat=entry,

{waldiez-0.1.6 → waldiez-0.1.8}/waldiez/exporting/utils/logging_utils.py

@@ -14,6 +14,7 @@ get_sqlite_to_csv_call_string
 
 
 # Check issue:
+# Also check if in ag2 this still applies
 # https://github.com/microsoft/autogen/issues/2286
 # we cannot log new agents if they have code_execution enabled
 # we get `Path` is not JSON serializable (on code_executor)