waldiez 0.1.16__tar.gz → 0.1.18__tar.gz

{waldiez-0.1.16 → waldiez-0.1.18}/PKG-INFO

@@ -1,10 +1,11 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: waldiez
-Version: 0.1.16
+Version: 0.1.18
 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>, Stella Ioannidou <stella@humancentered.gr>
+License-File: LICENSE
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -17,25 +18,26 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Python: <3.13,>=3.10
 Requires-Dist: jupytext
-Requires-Dist: pyautogen==0.5.0
+Requires-Dist: pyautogen==0.5.3
 Requires-Dist: pydantic>=2.0
 Requires-Dist: typer<0.13,>=0.9
 Provides-Extra: ag2-extras
 Requires-Dist: pgvector>=0.3.6; extra == 'ag2-extras'
 Requires-Dist: psycopg[binary]>=3.2.3; extra == 'ag2-extras'
-Requires-Dist: pyautogen[anthropic]==0.5.0; extra == 'ag2-extras'
-Requires-Dist: pyautogen[bedrock]==0.5.0; extra == 'ag2-extras'
-Requires-Dist: pyautogen[captainagent]==0.5.0; extra == 'ag2-extras'
-Requires-Dist: pyautogen[gemini]==0.5.0; extra == 'ag2-extras'
-Requires-Dist: pyautogen[groq]==0.5.0; extra == 'ag2-extras'
-Requires-Dist: pyautogen[lmm]==0.5.0; extra == 'ag2-extras'
-Requires-Dist: pyautogen[mistral]==0.5.0; extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat-mongodb]==0.5.0; extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat-pgvector]==0.5.0; extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat-qdrant]==0.5.0; (python_version < '3.13') and extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat]==0.5.0; extra == 'ag2-extras'
-Requires-Dist: pyautogen[together]==0.5.0; extra == 'ag2-extras'
-Requires-Dist: pyautogen[websurfer]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[anthropic]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[bedrock]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[captainagent]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[gemini]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[groq]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[lmm]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[mistral]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[neo4j]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat-mongodb]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat-pgvector]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat-qdrant]==0.5.3; (python_version < '3.13') and extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[together]==0.5.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[websurfer]==0.5.3; extra == 'ag2-extras'
 Requires-Dist: pymongo==4.10.1; extra == 'ag2-extras'
 Requires-Dist: qdrant-client[fastembed]; (python_version >= '3.13') and extra == 'ag2-extras'
 Provides-Extra: dev
@@ -49,7 +51,7 @@ Requires-Dist: pre-commit==4.0.1; extra == 'dev'
 Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
 Requires-Dist: pylint==3.3.2; extra == 'dev'
 Requires-Dist: python-dotenv==1.0.1; extra == 'dev'
-Requires-Dist: ruff==0.8.2; extra == 'dev'
+Requires-Dist: ruff==0.8.3; extra == 'dev'
 Requires-Dist: toml; (python_version <= '3.10') and extra == 'dev'
 Requires-Dist: types-pyyaml==6.0.12.20240917; extra == 'dev'
 Requires-Dist: types-toml==0.10.8.20240310; extra == 'dev'
@@ -59,7 +61,7 @@ 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.47; extra == 'docs'
+Requires-Dist: mkdocs-material==9.5.49; 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'
@@ -87,7 +89,7 @@ To a python script or a jupyter notebook with the corresponding [ag2](https://gi
 
 - Convert .waldiez flows to .py or .ipynb
 - Run a .waldiez flow
-- Provide a custom [IOSStream](https://ag2ai.github.io/ag2/docs/reference/io/base#iostream) to handle input and output.
+- Store the runtime logs of a flow to csv for further analysis
 
 ## Installation
 
@@ -177,73 +179,6 @@ 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)

{waldiez-0.1.16 → waldiez-0.1.18}/README.md

@@ -1,195 +1,128 @@
-# 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/py/waldiez.svg?icon=si%3Apython)](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
-
-- Convert .waldiez flows to .py or .ipynb
-- Run a .waldiez flow
-- 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
-# Convert a Waldiez flow to a python script or a jupyter notebook
-waldiez convert --file /path/to/a/flow.waldiez --output /path/to/an/output/flow[.py|.ipynb]
-# Convert and run the script, optionally force generation if the output file already exists
-waldiez run --file /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
-# Convert 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 convert --file /flow.waldiez --output /output/flow[.py|.ipynb] [--force]
-
-# 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 convert --file /flow.waldiez --output /output/flow[.py|.ipynb] [--force]
-```
-
-```shell
-# Convert and run the script
-$CONTAINER_COMMAND run \
-  --rm \
-  -v /path/to/a/flow.waldiez:/flow.waldiez \
-  -v /path/to/an/output:/output \
-  waldiez/waldiez run --file /flow.waldiez --output /output/output[.py]
-```
-
-### UI
-
-For creating-only (no exporting or running) waldiez flows, you can use the playground at <https://waldiez.github.io>.  
-The repo for the js library is [here](https://github.com/waldiez/react).  
-We are currently working on waldiez-studio to provide a visual interface for creating and running Waldiez flows (you can find more [here](https://github.com/waldiez/studio)).  
-Until then, you can use our [Jupyter](https://github.com/waldiez/jupyter) or the [VSCode](https://github.com/waldiez/vscode) extension to create and run Waldiez flows.
-
-### 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)
-- [typer](https://github.com/fastapi/typer)
-
-## Known Conflicts
-
-- **autogen-agentchat**: This package conflicts with `ag2` / `pyautogen`. Ensure that `autogen-agentchat` is uninstalled before installing `waldiez`. If you have already installed `autogen-agentchat`, you can uninstall it with the following command:
-
-  ```shell
-  pip uninstall autogen-agentchat -y
-  ```
-
-  If already installed waldiez you might need to reinstall it after uninstalling `autogen-agentchat`:
-  
-    ```shell
-    pip install --force --no-cache waldiez pyautogen
-    ```
-
-## License
-
-This project is licensed under the MIT License - see the [LICENSE](https://github.com/waldiez/waldiez/blob/main/LICENSE) file for details.
+# 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/py/waldiez.svg?icon=si%3Apython)](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
+
+- Convert .waldiez flows to .py or .ipynb
+- Run a .waldiez flow
+- Store the runtime logs of a flow to csv for further analysis
+
+## 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
+# Convert a Waldiez flow to a python script or a jupyter notebook
+waldiez convert --file /path/to/a/flow.waldiez --output /path/to/an/output/flow[.py|.ipynb]
+# Convert and run the script, optionally force generation if the output file already exists
+waldiez run --file /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
+# Convert 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 convert --file /flow.waldiez --output /output/flow[.py|.ipynb] [--force]
+
+# 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 convert --file /flow.waldiez --output /output/flow[.py|.ipynb] [--force]
+```
+
+```shell
+# Convert and run the script
+$CONTAINER_COMMAND run \
+  --rm \
+  -v /path/to/a/flow.waldiez:/flow.waldiez \
+  -v /path/to/an/output:/output \
+  waldiez/waldiez run --file /flow.waldiez --output /output/output[.py]
+```
+
+### UI
+
+For creating-only (no exporting or running) waldiez flows, you can use the playground at <https://waldiez.github.io>.  
+The repo for the js library is [here](https://github.com/waldiez/react).  
+We are currently working on waldiez-studio to provide a visual interface for creating and running Waldiez flows (you can find more [here](https://github.com/waldiez/studio)).  
+Until then, you can use our [Jupyter](https://github.com/waldiez/jupyter) or the [VSCode](https://github.com/waldiez/vscode) extension to create and run Waldiez flows.
+
+### 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)
+```
+
+### Tools
+
+- [ag2 (formerly AutoGen)](https://github.com/ag2ai/ag2)
+- [juptytext](https://github.com/mwouts/jupytext)
+- [pydantic](https://github.com/pydantic/pydantic)
+- [typer](https://github.com/fastapi/typer)
+
+## Known Conflicts
+
+- **autogen-agentchat**: This package conflicts with `ag2` / `pyautogen`. Ensure that `autogen-agentchat` is uninstalled before installing `waldiez`. If you have already installed `autogen-agentchat`, you can uninstall it with the following command:
+
+  ```shell
+  pip uninstall autogen-agentchat -y
+  ```
+
+  If already installed waldiez you might need to reinstall it after uninstalling `autogen-agentchat`:
+  
+    ```shell
+    pip install --force --no-cache waldiez pyautogen
+    ```
+
+## 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.16 → waldiez-0.1.18}/pyproject.toml

@@ -22,7 +22,7 @@ classifiers = [
   "Intended Audience :: Developers",
 ]
 dependencies =[
-    "pyautogen==0.5.0",
+    "pyautogen==0.5.3",
     "jupytext",
     "pydantic>=2.0",
     # together(ag2 extra) 1.2.0 depends on typer<0.13 and >=0.9
@@ -53,7 +53,7 @@ dev = [
     'pydocstyle==6.3.0',
     'pylint==3.3.2',
     'python-dotenv==1.0.1',
-    'ruff==0.8.2',
+    'ruff==0.8.3',
     'types-PyYAML==6.0.12.20240917',
     'yamllint==1.35.1',
     'toml; python_version <= "3.10"',
@@ -82,21 +82,22 @@ dev = [
 ag2_extras =[
     'pgvector>=0.3.6',
     'psycopg[binary]>=3.2.3',
-    'pyautogen[captainagent]==0.5.0',
-    'pyautogen[retrievechat]==0.5.0',
-    'pyautogen[retrievechat-pgvector]==0.5.0',
-    'pyautogen[retrievechat-mongodb]==0.5.0',
-    'pyautogen[retrievechat-qdrant]==0.5.0; python_version < "3.13"',
+    'pyautogen[anthropic]==0.5.3',
+    'pyautogen[bedrock]==0.5.3',
+    'pyautogen[captainagent]==0.5.3',
+    'pyautogen[gemini]==0.5.3',
+    'pyautogen[groq]==0.5.3',
+    'pyautogen[lmm]==0.5.3',
+    'pyautogen[mistral]==0.5.3',
+    'pyautogen[neo4j]==0.5.3',
+    'pyautogen[retrievechat-mongodb]==0.5.3',
+    'pyautogen[retrievechat-pgvector]==0.5.3',
+    'pyautogen[retrievechat-qdrant]==0.5.3; python_version < "3.13"',
     'qdrant-client[fastembed]; python_version >= "3.13"',
-    'pyautogen[gemini]==0.5.0',
-    'pyautogen[together]==0.5.0',
-    'pyautogen[anthropic]==0.5.0',
-    'pyautogen[lmm]==0.5.0',
-    'pyautogen[mistral]==0.5.0',
-    'pyautogen[groq]==0.5.0',
-    'pyautogen[bedrock]==0.5.0',
-    'pyautogen[websurfer]==0.5.0',
-    'pyautogen[together]==0.5.0',
+    'pyautogen[retrievechat]==0.5.3',
+    'pyautogen[together]==0.5.3',
+    'pyautogen[together]==0.5.3',
+    'pyautogen[websurfer]==0.5.3',
     'pymongo==4.10.1',
 ]
 test = [
@@ -113,7 +114,7 @@ docs = [
   'mkdocs==1.6.1',
   'mkdocs-jupyter==0.25.1',
   'mkdocs-macros-plugin==1.3.7',
-  'mkdocs-material==9.5.47',
+  'mkdocs-material==9.5.49',
   'mkdocs-minify-html-plugin==0.2.3',
   'mkdocstrings[crystal,python]==0.27.0',
   'mkdocstrings-python==1.12.2'

waldiez-0.1.18/waldiez/__init__.py

@@ -0,0 +1,74 @@
+"""Waldiez package."""
+
+import logging
+
+from ._version import __version__
+from .conflict_checker import check_conflicts
+from .exporter import WaldiezExporter
+from .models import Waldiez
+from .runner import WaldiezRunner
+
+
+# pylint: disable=too-few-public-methods
+class FlamlFilter(logging.Filter):
+    """Filter out flaml.automl is not available message."""
+
+    def filter(self, record: logging.LogRecord) -> bool:
+        """Filter out flaml.automl is not available message.
+
+        this is just annoying:
+
+        ```
+        flaml.automl is not available.
+        Please install flaml[automl] to enable AutoML functionalities.
+        ```
+
+        Parameters
+        ----------
+        record : logging.LogRecord
+            Log record to filter.
+
+        Returns
+        -------
+        bool
+            Whether to filter out the log record.
+        """
+        return "flaml.automl is not available" not in record.getMessage()
+
+
+# flag to check if ag2 and autogen-agentchat
+# are installed at the same time
+__WALDIEZ_CHECKED_FOR_CONFLICTS = False
+# flag to handle flaml logging
+# suppress the annoying message about flaml.automl
+__WALDIEZ_HANDLED_FLAML_LOGGING = False
+
+
+def _check_conflicts() -> None:
+    """Check for conflicts once."""
+    # pylint: disable=global-statement
+    global __WALDIEZ_CHECKED_FOR_CONFLICTS
+    if __WALDIEZ_CHECKED_FOR_CONFLICTS is False:
+        check_conflicts()
+        __WALDIEZ_CHECKED_FOR_CONFLICTS = True
+
+
+def _handle_flaml_logging() -> None:
+    """Handle flaml logging once."""
+    # pylint: disable=global-statement
+    global __WALDIEZ_HANDLED_FLAML_LOGGING
+    if __WALDIEZ_HANDLED_FLAML_LOGGING is False:
+        __WALDIEZ_HANDLED_FLAML_LOGGING = True
+        flam_logger = logging.getLogger("flaml")
+        flam_logger.addFilter(FlamlFilter())
+
+
+_check_conflicts()
+_handle_flaml_logging()
+
+__all__ = [
+    "Waldiez",
+    "WaldiezExporter",
+    "WaldiezRunner",
+    "__version__",
+]

{waldiez-0.1.16 → waldiez-0.1.18}/waldiez/_version.py

@@ -1,3 +1,3 @@
 """Version information for Waldiez."""
 
-__version__ = "0.1.16"
+__version__ = "0.1.18"

{waldiez-0.1.16 → waldiez-0.1.18}/waldiez/cli.py

@@ -90,7 +90,7 @@ def run(
             raise typer.Exit(code=1) from error
     waldiez = Waldiez.from_dict(data)
     runner = WaldiezRunner(waldiez)
-    results = runner.run(stream=None, output_path=output_path)
+    results = runner.run(output_path=output_path)
     logger = _get_logger()
     if isinstance(results, list):
         logger.info("Results:")

{waldiez-0.1.16 → waldiez-0.1.18}/waldiez/exporting/flow/def_main.py

@@ -1,10 +1,6 @@
 """Get the main function (if exporting to python)."""
 
-from ..utils import (
-    get_logging_start_string,
-    get_logging_stop_string,
-    get_sqlite_to_csv_call_string,
-)
+from ..utils import get_logging_stop_string, get_sqlite_to_csv_call_string
 
 
 def get_def_main(waldiez_chats: str) -> str:
@@ -29,7 +25,6 @@ def get_def_main(waldiez_chats: str) -> str:
     # type: () -> Union[ChatResult, List[ChatResult]]
     \"\"\"Start chatting.\"\"\"
 """
-    content += get_logging_start_string(1)
     content += f"    results = {waldiez_chats}" + "\n"
     content += get_logging_stop_string(1) + "\n"
     content += get_sqlite_to_csv_call_string(1) + "\n"

{waldiez-0.1.16 → waldiez-0.1.18}/waldiez/exporting/flow/flow.py

@@ -159,8 +159,8 @@ def _combine_strings(
 ) -> str:
     content = get_pylint_ignore_comment(notebook)
     content += imports_string
-    # content += get_comment("logging", notebook) + "\n"
-    # content += get_logging_start_string(tabs=0) + "\n\n"
+    content += get_comment("logging", notebook) + "\n"
+    content += get_logging_start_string(tabs=0) + "\n\n"
     content += models_string
     content += get_comment("agents", notebook) + "\n"
     content += agents_string
@@ -182,7 +182,6 @@ def _combine_strings(
     if not notebook:
         content += get_def_main(chats_content)
     else:
-        content += get_logging_start_string(tabs=0)
         content += "\n" + chats_content + "\n"
         content += get_logging_stop_string(tabs=0) + "\n"
         content += get_sqlite_to_csv_call_string(tabs=0) + "\n"