waldiez 0.1.7__tar.gz → 0.1.8__tar.gz

{waldiez-0.1.7 → waldiez-0.1.8}/PKG-INFO

@@ -1,11 +1,10 @@
 Metadata-Version: 2.3
 Name: waldiez
-Version: 0.1.7
+Version: 0.1.8
 Summary: waldiez
-Project-URL: homepage, https://waldiez.github.io/py/
-Project-URL: repository, https://github.com/waldiez/py.git
+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>
-License-File: LICENSE
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -17,28 +16,26 @@ 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: autogen-agentchat==0.2.37
+Requires-Dist: ag2==0.3.2
 Requires-Dist: jupytext
-Requires-Dist: twisted==24.10.0
-Provides-Extra: autogen-extras
-Requires-Dist: autogen-agentchat[anthropic]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[bedrock]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[gemini]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[groq]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[mistral]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[retrievechat-couchbase]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[retrievechat-mongodb]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[retrievechat-pgvector]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[retrievechat-qdrant]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[retrievechat]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[together]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[websurfer]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: chromadb==0.5.17; extra == 'autogen-extras'
-Requires-Dist: fastembed==0.4.1; extra == 'autogen-extras'
-Requires-Dist: pgvector==0.3.6; extra == 'autogen-extras'
-Requires-Dist: psycopg[binary]>=3.2.3; extra == 'autogen-extras'
-Requires-Dist: pymongo==4.10.1; extra == 'autogen-extras'
-Requires-Dist: qdrant-client==1.12.1; extra == 'autogen-extras'
+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'
@@ -50,7 +47,7 @@ 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.2; 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
@@ -58,11 +55,11 @@ 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.43; 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.26.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'
@@ -74,20 +71,20 @@ Description-Content-Type: text/markdown
 
 # Waldiez
 
-![CI Build](https://github.com/waldiez/py/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/py/badge.svg)](https://coveralls.io/github/waldiez/py) [![PyPI version](https://badge.fury.io/py/waldiez.svg)](https://badge.fury.io/py/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/py/refs/heads/main/docs/flow.png)
+![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 [autogen](https://github.com/microsoft/autogen/) agents and chats.
+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://autogen-ai.github.io/autogen/docs/reference/io/base#iostream) to handle input and output.
+- Provide a custom [IOSStream](https://ag2ai.github.io/ag2/docs/reference/io/base#iostream) to handle input and output.
 
 ## Installation
 
@@ -100,7 +97,7 @@ python -m pip install waldiez
 From the repository:
 
 ```bash
-python -m pip install git+https://github.com/waldiez/py.git
+python -m pip install git+https://github.com/waldiez/waldiez.git
 ```
 
 ## Usage
@@ -119,13 +116,13 @@ waldiez /path/to/a/flow.waldiez --output /path/to/an/output/flow[.py] [--force]
 ```shell
 CONTAINER_COMMAND=docker # or podman
 # pull the image
-$CONTAINER_COMMAND pull waldiez/py
+$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/py --export /flow.waldiez --output /output/flow[.py|.ipynb]
+  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 \
@@ -134,12 +131,12 @@ $CONTAINER_COMMAND run \
   -v /path/to/an/output:/output \
   --userns=keep-id \
   --security-opt label=disable \
-  waldiez/py --export /flow.waldiez --output /output/flow[.py|.ipynb]
+  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/py /flow.waldiez --output /output/output[.py]
+$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
@@ -170,40 +167,75 @@ runner.run(output_path=output_path)
 
 ```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 print_function(*values, **args) -> None:
-    """A custom print function."""
-    print(values)
 
-def on_prompt_input(prompt: str) -> str:
-    """A custom input function."""
-    return input(prompt)
+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!")
 
-io_stream = WaldiezIOStream(
-    print_function=print_function,
-    on_prompt_input=on_prompt_input,
+        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)
 
-io_stream.close()
-
 ```
 
 ### Tools
 
-- [autogen](https://github.com/microsoft/autogen/)
+- [ag2 (formerly AutoGen)](https://github.com/ag2ai/ag2)
 - [juptytext](https://github.com/mwouts/jupytext)
-- [twisted](https://github.com/twisted/twisted)
 - [pydantic](https://github.com/pydantic/pydantic)
 
 ## License
 
-This project is licensed under the MIT License - see the [LICENSE](https://github.com/waldiez/py/blob/main/LICENSE) file for details.
+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.7 → 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,27 +49,27 @@ dev = [
     'pydocstyle==6.3.0',
     'pylint==3.3.1',
     'python-dotenv==1.0.1',
-    'ruff==0.7.2',
+    'ruff==0.7.4',
     'types-PyYAML==6.0.12',
     'yamllint==1.35.1',
 ]
-autogen_extras =[
-    'chromadb==0.5.17',
-    'fastembed==0.4.1',
-    'pgvector==0.3.6',
-    '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.1',
 ]
@@ -88,9 +87,9 @@ docs = [
   'mkdocs==1.6.1',
   'mkdocs-jupyter==0.25.1',
   'mkdocs-macros-plugin==1.3.7',
-  'mkdocs-material==9.5.43',
+  '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'
 ]
 
@@ -278,6 +277,9 @@ filterwarnings = [
   #
   # 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 = [
@@ -291,7 +293,7 @@ addopts = """
     --exitfirst \
     --durations=10 \
     --color=yes \
-    --timeout=60
+    --timeout=120
 """
 
 

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

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

{waldiez-0.1.7 → 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.7 → 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.7 → 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)