waldiez 0.3.9__tar.gz → 0.3.11__tar.gz

{waldiez-0.3.9 → waldiez-0.3.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.3.9
+Version: 0.3.11
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/waldiez/python
 Project-URL: repository, https://github.com/waldiez/python.git
@@ -23,48 +23,48 @@ Requires-Dist: aiofiles==24.1.0
 Requires-Dist: aiosqlite==0.20.0
 Requires-Dist: asyncer==0.0.8
 Requires-Dist: graphviz==0.20.3
-Requires-Dist: httpx<0.28.0,>=0.27.0
+Requires-Dist: httpx<1
 Requires-Dist: jupytext
 Requires-Dist: pandas>=2
 Requires-Dist: parso==0.8.4
-Requires-Dist: pyautogen==0.7.2
+Requires-Dist: pyautogen==0.7.3
 Requires-Dist: pydantic<3,>=2.6.1
-Requires-Dist: typer<0.13,>=0.9
+Requires-Dist: typer<0.16,>=0.9
 Provides-Extra: ag2-extras
-Requires-Dist: autogen[captainagent]==0.7.2; (platform_system == 'Linux') and extra == 'ag2-extras'
+Requires-Dist: autogen[captainagent]==0.7.3; (platform_system == 'Linux') and extra == 'ag2-extras'
 Requires-Dist: chromadb; (platform_system != 'Linux') and extra == 'ag2-extras'
-Requires-Dist: huggingface-hub>=0.0.17; (platform_system != 'Linux') and extra == 'ag2-extras'
+Requires-Dist: huggingface-hub; (platform_system != 'Linux') and 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.7.2; extra == 'ag2-extras'
-Requires-Dist: pyautogen[bedrock]==0.7.2; extra == 'ag2-extras'
-Requires-Dist: pyautogen[gemini]==0.7.2; extra == 'ag2-extras'
-Requires-Dist: pyautogen[groq]==0.7.2; extra == 'ag2-extras'
-Requires-Dist: pyautogen[lmm]==0.7.2; extra == 'ag2-extras'
-Requires-Dist: pyautogen[mistral]==0.7.2; extra == 'ag2-extras'
-Requires-Dist: pyautogen[neo4j]==0.7.2; extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat-mongodb]==0.7.2; extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat-pgvector]==0.7.2; extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat-qdrant]==0.7.2; (python_version < '3.13') and extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat]==0.7.2; extra == 'ag2-extras'
-Requires-Dist: pyautogen[together]==0.7.2; extra == 'ag2-extras'
-Requires-Dist: pyautogen[websurfer]==0.7.2; extra == 'ag2-extras'
-Requires-Dist: pymongo==4.10.1; extra == 'ag2-extras'
+Requires-Dist: pyautogen[anthropic]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[bedrock]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[gemini]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[groq]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[lmm]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[mistral]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[neo4j]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat-mongodb]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat-pgvector]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat-qdrant]==0.7.3; (python_version < '3.13') and extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[together]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[websurfer]==0.7.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'
-Requires-Dist: sentence-transformers>=2.0.0; (platform_system != 'Linux') and extra == 'ag2-extras'
+Requires-Dist: sentence-transformers; (platform_system != 'Linux') and extra == 'ag2-extras'
 Provides-Extra: dev
 Requires-Dist: autoflake==2.3.1; extra == 'dev'
 Requires-Dist: bandit==1.8.2; extra == 'dev'
-Requires-Dist: black[jupyter]==24.10.0; extra == 'dev'
+Requires-Dist: black[jupyter]==25.1.0; extra == 'dev'
 Requires-Dist: flake8==7.1.1; extra == 'dev'
-Requires-Dist: isort==5.13.2; extra == 'dev'
+Requires-Dist: isort==6.0.0; extra == 'dev'
 Requires-Dist: mypy==1.14.1; extra == 'dev'
 Requires-Dist: pandas-stubs; extra == 'dev'
 Requires-Dist: pre-commit==4.1.0; extra == 'dev'
 Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
-Requires-Dist: pylint==3.3.3; extra == 'dev'
+Requires-Dist: pylint==3.3.4; extra == 'dev'
 Requires-Dist: python-dotenv==1.0.1; extra == 'dev'
-Requires-Dist: ruff==0.9.3; extra == 'dev'
+Requires-Dist: ruff==0.9.4; extra == 'dev'
 Requires-Dist: toml; (python_version <= '3.10') and extra == 'dev'
 Requires-Dist: types-pyyaml==6.0.12.20241230; extra == 'dev'
 Requires-Dist: types-toml==0.10.8.20240310; extra == 'dev'
@@ -74,18 +74,18 @@ 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.50; extra == 'docs'
+Requires-Dist: mkdocs-material==9.6.1; 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.13.0; extra == 'docs'
 Requires-Dist: mkdocstrings[crystal,python]==0.27.0; extra == 'docs'
 Provides-Extra: jupyter
 Requires-Dist: jupyterlab>=4.3.0; extra == 'jupyter'
-Requires-Dist: waldiez-jupyter==0.3.9; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.3.11; extra == 'jupyter'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.3.9; extra == 'studio'
+Requires-Dist: waldiez-studio==0.3.11; extra == 'studio'
 Provides-Extra: test
-Requires-Dist: pytest-asyncio==0.25.2; extra == 'test'
+Requires-Dist: pytest-asyncio==0.25.3; 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'

{waldiez-0.3.9 → waldiez-0.3.11}/pyproject.toml

@@ -28,15 +28,13 @@ dependencies =[
     "asyncer==0.0.8",
     "graphviz==0.20.3",
     "pandas>=2",
-    "pyautogen==0.7.2",
+    "pyautogen==0.7.3",
     "jupytext",
     "parso==0.8.4",
     "pydantic>=2.6.1,<3",
     # together(ag2 extra) 1.2.0 depends on typer<0.13 and >=0.9
-    "typer>=0.9,<0.13",
-    # mistralai and anthropic need httpx<0.28
-    # TypeError: Client.__init__() got an unexpected keyword argument 'proxies'
-    "httpx<0.28.0,>=0.27.0",
+    "typer>=0.9,<0.16",
+    "httpx<1",
 ]
 
 [project.urls]
@@ -53,25 +51,25 @@ exclude = [ "**/example.py" ]
 
 [project.optional-dependencies]
 studio = [
-    "waldiez_studio==0.3.9",
+    "waldiez_studio==0.3.11",
 ]
 jupyter = [
-    "waldiez_jupyter==0.3.9",
+    "waldiez_jupyter==0.3.11",
     "jupyterlab>=4.3.0",
 ]
 dev = [
     'autoflake==2.3.1',
     'bandit==1.8.2',
-    'black[jupyter]==24.10.0',
+    'black[jupyter]==25.1.0',
     'flake8==7.1.1',
-    'isort==5.13.2',
+    'isort==6.0.0',
     'mypy==1.14.1',
     "pandas-stubs",
     'pre-commit==4.1.0',
     'pydocstyle==6.3.0',
-    'pylint==3.3.3',
+    'pylint==3.3.4',
     'python-dotenv==1.0.1',
-    'ruff==0.9.3',
+    'ruff==0.9.4',
     'types-PyYAML==6.0.12.20241230',
     'yamllint==1.35.1',
     'toml; python_version <= "3.10"',
@@ -103,30 +101,30 @@ dev = [
 ag2_extras =[
     'pgvector>=0.3.6',
     'psycopg[binary]>=3.2.3',
-    'pyautogen[anthropic]==0.7.2',
-    'pyautogen[bedrock]==0.7.2',
-    'autogen[captainagent]==0.7.2; platform_system == "Linux"',
+    'pyautogen[anthropic]==0.7.3',
+    'pyautogen[bedrock]==0.7.3',
+    'autogen[captainagent]==0.7.3; platform_system == "Linux"',
     'chromadb; platform_system != "Linux"',
-    'sentence-transformers>=2.0.0; platform_system != "Linux"',
-    'huggingface-hub>=0.0.17; platform_system != "Linux"',
-    'pyautogen[gemini]==0.7.2',
-    'pyautogen[groq]==0.7.2',
-    'pyautogen[lmm]==0.7.2',
-    'pyautogen[mistral]==0.7.2',
-    'pyautogen[neo4j]==0.7.2',
-    'pyautogen[retrievechat-mongodb]==0.7.2',
-    'pyautogen[retrievechat-pgvector]==0.7.2',
-    'pyautogen[retrievechat-qdrant]==0.7.2; python_version < "3.13"',
+    'sentence-transformers; platform_system != "Linux"',
+    'huggingface-hub; platform_system != "Linux"',
+    'pyautogen[gemini]==0.7.3',
+    'pyautogen[groq]==0.7.3',
+    'pyautogen[lmm]==0.7.3',
+    'pyautogen[mistral]==0.7.3',
+    'pyautogen[neo4j]==0.7.3',
+    'pyautogen[retrievechat-mongodb]==0.7.3',
+    'pyautogen[retrievechat-pgvector]==0.7.3',
+    'pyautogen[retrievechat-qdrant]==0.7.3; python_version < "3.13"',
     'qdrant-client[fastembed]; python_version >= "3.13"',
-    'pyautogen[retrievechat]==0.7.2',
-    'pyautogen[together]==0.7.2',
-    'pyautogen[together]==0.7.2',
-    'pyautogen[websurfer]==0.7.2',
-    'pymongo==4.10.1',
+    'pyautogen[retrievechat]==0.7.3',
+    'pyautogen[together]==0.7.3',
+    'pyautogen[together]==0.7.3',
+    'pyautogen[websurfer]==0.7.3',
+    'pymongo>=4.10.1',
 ]
 test = [
     'pytest==8.3.4',
-    'pytest-asyncio==0.25.2',
+    'pytest-asyncio==0.25.3',
     'pytest-cov==6.0.0',
     'pytest-html==4.1.1',
     'pytest-sugar==1.0.0',
@@ -139,7 +137,7 @@ docs = [
   'mkdocs==1.6.1',
   'mkdocs-jupyter==0.25.1',
   'mkdocs-macros-plugin==1.3.7',
-  'mkdocs-material==9.5.50',
+  'mkdocs-material==9.6.1',
   'mkdocs-minify-html-plugin==0.2.3',
   'mkdocstrings[crystal,python]==0.27.0',
   'mkdocstrings-python==1.13.0'

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/_version.py

@@ -2,4 +2,4 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Version information for Waldiez."""
 
-__version__ = "0.3.9"
+__version__ = "0.3.11"

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporter.py

@@ -6,21 +6,12 @@ to an autogen's flow with one or more chats.
 
 The resulting file(s): a `flow.py` file with one `main()` function
 to trigger the chat(s).
-If additional tools/skills are used,
-they are exported as their `skill_name` in the same directory with
-the `flow.py` file. So the `flow.py` could have entries like:
-`form {flow_name}_{skill1_name} import {skill1_name}`
-`form {flow_name}_{skill2_name} import {skill2_name}`
 """
 
-# pylint: disable=inconsistent-quotes
-
-import os
-import shutil
-import subprocess
-import sys
 from pathlib import Path
-from typing import List, Optional, Union
+from typing import Union
+
+import jupytext  # type: ignore[import-untyped]
 
 from .exporting import FlowExporter
 from .models import Waldiez
@@ -125,25 +116,10 @@ class WaldiezExporter:
         py_path = path.with_suffix(".tmp.py")
         with open(py_path, "w", encoding="utf-8", newline="\n") as f:
             f.write(content)
-        if not shutil.which("jupytext"):  # pragma: no cover
-            run_command(
-                [sys.executable, "-m", "pip", "install", "jupytext"],
-                allow_error=False,
-            )
-        run_command(
-            [
-                sys.executable,
-                "-m",
-                "jupytext",
-                "--to",
-                "notebook",
-                str(py_path),
-            ],
-            allow_error=False,
-        )
+        with open(py_path, "r", encoding="utf-8") as py_out:
+            content = jupytext.read(py_out, fmt="py:light")
         ipynb_path = str(py_path).replace(".tmp.py", ".tmp.ipynb")
-        if not os.path.exists(ipynb_path):  # pragma: no cover
-            raise RuntimeError("Could not generate notebook")
+        jupytext.write(content, ipynb_path, fmt="ipynb")
         Path(ipynb_path).rename(ipynb_path.replace(".tmp.ipynb", ".ipynb"))
         py_path.unlink(missing_ok=True)
 
@@ -182,42 +158,3 @@ class WaldiezExporter:
         """
         with open(file_path, "w", encoding="utf-8", newline="\n") as file:
             file.write(self.waldiez.model_dump_json())
-
-
-def run_command(
-    cmd: List[str],
-    cwd: Optional[Path] = None,
-    allow_error: bool = True,
-) -> None:
-    """Run a command.
-
-    Parameters
-    ----------
-    cmd : List[str]
-        The command to run.
-    cwd : Path, optional
-        The working directory, by default None (current working directory).
-    allow_error : bool, optional
-        Whether to allow errors, by default True.
-
-    Raises
-    ------
-    RuntimeError
-        If the command fails and allow_error is False.
-    """
-    if not cwd:
-        cwd = Path.cwd()
-    # pylint: disable=broad-except
-    try:
-        subprocess.run(
-            cmd,
-            check=True,
-            cwd=cwd,
-            env=os.environ,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE,
-        )  # nosemgrep # nosec
-    except BaseException as error:  # pragma: no cover
-        if allow_error:
-            return
-        raise RuntimeError(f"Error running command: {error}") from error

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporting/base/utils/naming.py

@@ -15,11 +15,13 @@ get_escaped_string
 import re
 from typing import Dict, Tuple
 
+MAX_VARIABLE_LENGTH = 46
+
 
 def get_valid_python_variable_name(
     possible: str,
     prefix: str = "w",
-    max_length: int = 64,
+    max_length: int = MAX_VARIABLE_LENGTH,
 ) -> str:
     """Get a valid Python variable name from a possible name.
 
@@ -78,7 +80,7 @@ def get_valid_instance_name(
     instance: Tuple[str, str],
     current_names: Dict[str, str],
     prefix: str = "w",
-    max_length: int = 64,
+    max_length: int = MAX_VARIABLE_LENGTH,
 ) -> Dict[str, str]:
     """Get a valid instance name.
 
@@ -109,7 +111,9 @@ def get_valid_instance_name(
         # already in the current names (it's id)
         return current_names
     new_names = current_names.copy()
-    name = get_valid_python_variable_name(possible_name, prefix=prefix)
+    name = get_valid_python_variable_name(
+        possible_name, prefix=prefix, max_length=max_length
+    )
     if name in current_names.values():
         name = f"{prefix}_{name}"
     if name in current_names.values():

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporting/chats/chats_exporter.py

@@ -97,7 +97,7 @@ class ChatsExporter(BaseExporter, ExporterMixin):
                     recipient=recipient,
                     serializer=self.serializer,
                     string_escape=self.string_escape,
-                    tabs=0 if self.for_notebook else 1,
+                    tabs=1 if self.for_notebook else 2,
                     is_async=self.is_async,
                 )
                 return
@@ -109,7 +109,7 @@ class ChatsExporter(BaseExporter, ExporterMixin):
                 chat_names=self.chat_names,
                 serializer=self.serializer,
                 string_escape=self.string_escape,
-                tabs=0 if self.for_notebook else 1,
+                tabs=1 if self.for_notebook else 2,
                 is_async=self.is_async,
             )
             return
@@ -119,7 +119,7 @@ class ChatsExporter(BaseExporter, ExporterMixin):
             chat_names=self.chat_names,
             serializer=self.serializer,
             string_escape=self.string_escape,
-            tabs=0 if self.for_notebook else 1,
+            tabs=1 if self.for_notebook else 2,
             is_async=self.is_async,
         )
 

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporting/chats/utils/sequential.py

@@ -181,6 +181,7 @@ def _get_chat_dict_string(
     chat_string = "{"
     chat_string += "\n" + f'{tab}    "sender": {agent_names[sender.id]},'
     chat_string += "\n" + f'{tab}    "recipient": {agent_names[recipient.id]},'
+    chat_string += "\n" + f'{tab}    "cache": cache,'
     additional_methods_string = ""
     for key, value in chat_args.items():
         if isinstance(value, str):

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporting/chats/utils/single_chat.py

@@ -86,6 +86,7 @@ def export_single_chat(
     agent1.initiate_chat(
         agent2,
         message="Hello, how are you?",
+        cache=cache,
     )
     ```
     """
@@ -167,6 +168,7 @@ def get_simple_chat_string(
     recipient_name = agent_names[recipient.id]
     chat_string = "\n" + f"{tab}results = {sender_name}.{initiate}(" + "\n"
     chat_string += f"{tab}    {recipient_name},"
+    chat_string += "\n" + f"{tab}    cache=cache,"
     for key, value in chat_args.items():
         if isinstance(value, str):
             chat_string += "\n" + f'{tab}    {key}="{value}",'
@@ -228,6 +230,7 @@ def get_empty_simple_chat_string(
     initiate = "a_initiate_chat" if is_async else "initiate_chat"
     content = "\n" + f"{tab}results = {sender_name}.{initiate}(" + "\n"
     content += f"{tab}    {recipient_name}," + "\n"
+    content += f"{tab}    cache=cache," + "\n"
     message_arg, _ = get_chat_message(
         tab=tab,
         chat=chat,

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporting/flow/flow_exporter.py

@@ -202,6 +202,7 @@ class FlowExporter(BaseExporter, ExporterMixin):
             The merged export contents.
         """
         is_async = self.waldiez.is_async
+        cache_seed = self.waldiez.cache_seed
         content = (
             get_py_content_start(self.waldiez)
             if not self.for_notebook
@@ -236,9 +237,18 @@ class FlowExporter(BaseExporter, ExporterMixin):
                 chats_content,
                 after_run=after_run,
                 is_async=self.waldiez.is_async,
+                cache_seed=cache_seed,
             )
         else:
-            content += "\n" + chats_content + "\n"
+            # content += f"    with Cache.disk(cache_seed={cache_seed}" + "):\n"
+            # content += f"{flow_chats}" + "\n"
+            # content += "\n" + chats_content + "\n"
+            if chats_content.startswith("\n"):
+                chats_content = chats_content[1:]
+            content += (
+                "\n" + f"with Cache.disk(cache_seed={cache_seed}):"
+                "\n" + chats_content + "\n"
+            )
             if is_async:
                 content += "await stop_logging()"
             else:
@@ -332,6 +342,7 @@ class FlowExporter(BaseExporter, ExporterMixin):
             model_names=self.model_names,
             for_notebook=self.for_notebook,
             output_dir=self.output_dir,
+            cache_seed=self.waldiez.cache_seed,
         )
         return exporter.export()
 

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporting/flow/utils/def_main.py

@@ -4,8 +4,12 @@
 # pylint: disable=inconsistent-quotes, line-too-long
 """Get the main function."""
 
+from typing import Optional
 
-def get_def_main(flow_chats: str, after_run: str, is_async: bool) -> str:
+
+def get_def_main(
+    flow_chats: str, after_run: str, is_async: bool, cache_seed: Optional[int]
+) -> str:
     """Get the main function.
 
     When exporting to python, waldiez_chats string will be the
@@ -22,17 +26,21 @@ def get_def_main(flow_chats: str, after_run: str, is_async: bool) -> str:
         The content after the run of the flow.
     is_async : bool
         Whether the main function is asynchronous.
+    cache_seed : Optional[int]
+        The seed for the cache. If None, cache should be disabled.
     Returns
     -------
     str
         The main function.
     """
+    if flow_chats.startswith("\n"):
+        flow_chats = flow_chats[1:]
     content = ""
     if is_async:
         content += "async "
-    content += "def main():\n"
-    content += "    # type: () -> Union[ChatResult, List[ChatResult], Dict[int, ChatResult]]\n"
+    content += "def main() -> Union[ChatResult, List[ChatResult], Dict[int, ChatResult]]:\n"
     content += '    """Start chatting."""\n'
+    content += f"    with Cache.disk(cache_seed={cache_seed}" + ") as cache:\n"
     content += f"{flow_chats}" + "\n"
     if is_async:
         content += "    await stop_logging()"
@@ -45,7 +53,7 @@ def get_def_main(flow_chats: str, after_run: str, is_async: bool) -> str:
     else:
         content += "def call_main() -> None:\n"
     content += '    """Run the main function and print the results."""\n'
-    content += "    results: Union[ChatResult, List[ChatResult], Dict[int, ChatResult]] = "
+    content += "    results = "
     if is_async:
         content += "await "
     content += "main()\n"
@@ -58,7 +66,7 @@ def get_def_main(flow_chats: str, after_run: str, is_async: bool) -> str:
     content += "        for result in results:\n"
     content += "            pprint(asdict(result))\n"
     content += "    else:\n"
-    content += "        pprint(asdict(results))\n"
+    content += "        pprint(asdict(results))\n\n\n"
     content += 'if __name__ == "__main__":\n'
     if is_async:
         content += "    anyio.run(call_main)\n"

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporting/flow/utils/flow_content.py

@@ -136,25 +136,26 @@ def get_after_run_content(
     # if th eflow has reasoning agents, we add
     # visualize_tree(agent._root) for each agent
     content = ""
-    space = "    " * tabs
+    tab = "    "
+    space = tab * tabs
     for agent in waldiez.agents:
         if agent.agent_type == "reasoning":
             agent_name = agent_names[agent.id]
             content += f"""
 {space}# pylint: disable=broad-except,too-many-try-statements
 {space}try:
-{space}{space}visualize_tree({agent_name}._root)  # pylint: disable=protected-access
-{space}{space}if os.path.exists("tree_of_thoughts.png"):
-{space}{space}{space}new_name = "{agent_name}_tree_of_thoughts.png"
-{space}{space}{space}os.rename("tree_of_thoughts.png", new_name)
+{space}{tab}visualize_tree({agent_name}._root)  # pylint: disable=protected-access
+{space}{tab}if os.path.exists("tree_of_thoughts.png"):
+{space}{tab}{tab}new_name = "{agent_name}_tree_of_thoughts.png"
+{space}{tab}{tab}os.rename("tree_of_thoughts.png", new_name)
 {space}except BaseException:
-{space}{space}pass
+{space}{tab}pass
 {space}# save the tree to json
 {space}try:
-{space}{space}data = {agent_name}._root.to_dict()  # pylint: disable=protected-access
-{space}{space}with open("{agent_name}_reasoning_tree.json", "w", encoding="utf-8") as f:
-{space}{space}{space}json.dump(data, f)
+{space}{tab}data = {agent_name}._root.to_dict()  # pylint: disable=protected-access
+{space}{tab}with open("{agent_name}_reasoning_tree.json", "w", encoding="utf-8") as f:
+{space}{tab}{tab}json.dump(data, f)
 {space}except BaseException:
-{space}{space}pass
+{space}{tab}pass
 """
     return content

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporting/flow/utils/flow_names.py

@@ -31,7 +31,7 @@ class ResultType(TypedDict):
 def ensure_unique_names(
     waldiez: Waldiez,
     get_valid_instance_name: Callable[..., Dict[str, str]],
-    max_length: int = 64,
+    max_length: int = 46,
     flow_name_max_length: int = 20,
 ) -> ResultType:
     """Ensure unique names for agents, models, skills, and chats and flow.

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporting/flow/utils/importing_utils.py

@@ -28,6 +28,7 @@ TYPING_IMPORTS = [
 ]
 COMMON_AUTOGEN_IMPORTS = [
     "from autogen import Agent",
+    "from autogen import Cache",
     "from autogen import ConversableAgent",
     "from autogen import ChatResult",
     "from autogen import GroupChat",
@@ -128,6 +129,7 @@ def get_the_imports_string(
             "\nimport aiofiles"
             "\nimport aiosqlite"
             "\nimport anyio"
+            "\nimport nest_asyncio"
             "\nfrom aiocsv import AsyncDictWriter"
             "\nfrom asyncer import asyncify"
         )
@@ -143,6 +145,8 @@ def get_the_imports_string(
         final_string += "\n".join(local_imports) + "\n"
     while not final_string.endswith("\n\n"):
         final_string += "\n"
+    if is_async:
+        final_string += "\nnest_asyncio.apply()\n"
     return final_string.replace("\n\n\n", "\n\n")  # avoid too many newlines
 
 

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporting/models/models_exporter.py

@@ -32,6 +32,7 @@ class ModelsExporter(BaseExporter, ExporterMixin):
         models: List[WaldiezModel],
         model_names: Dict[str, str],
         for_notebook: bool,
+        cache_seed: Optional[int],
         output_dir: Optional[Union[str, Path]] = None,
     ) -> None:
         """Initialize the models exporter.
@@ -46,6 +47,10 @@ class ModelsExporter(BaseExporter, ExporterMixin):
             The models.
         model_names : Dict[str, str]
             The model names.
+        for_notebook : bool
+            Whether the export is for a notebook or not.
+        cache_seed : Optional[int]
+            The cache seed if any, by default None
         output_dir : Optional[Union[str, Path]], optional
             The output directory if any, by default None
         """
@@ -57,6 +62,7 @@ class ModelsExporter(BaseExporter, ExporterMixin):
         self.model_names = model_names
         if output_dir is not None and not isinstance(output_dir, Path):
             output_dir = Path(output_dir)
+        self.cache_seed = cache_seed
         self.output_dir = output_dir
         self._exported_string = None
 
@@ -127,6 +133,7 @@ class ModelsExporter(BaseExporter, ExporterMixin):
                         agent,
                         all_models=self.models,
                         model_names=self.model_names,
+                        cache_seed=self.cache_seed,
                     ),
                     AgentPosition(
                         agent=agent, position=AgentPositions.AS_ARGUMENT

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/exporting/models/utils.py

@@ -63,6 +63,7 @@ def get_agent_llm_config_arg(
     agent: WaldiezAgent,
     all_models: List[WaldiezModel],
     model_names: Dict[str, str],
+    cache_seed: Optional[int],
     tabs: int = 1,
 ) -> str:
     """Get the string representation of the agent's llm config argument.
@@ -75,6 +76,8 @@ def get_agent_llm_config_arg(
         All the models in the flow.
     model_names : Dict[str, str]
         A mapping of model ids to model names.
+    cache_seed : Optional[int]
+        The cache seed.
     tabs : int, optional
         The number of tabs for indentation, by default 1.
 
@@ -98,6 +101,7 @@ def get_agent_llm_config_arg(
     if not got_at_least_one_model:  # pragma: no cover
         return f"{tab}llm_config=False," + "\n"
     content += "\n" + f"{tab}    ]," + "\n"
+    content += f'{tab}    "cache_seed": {cache_seed},' + "\n"
     content += tab + "},\n"
     return content
 

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/models/agents/agents.py

@@ -115,8 +115,8 @@ class WaldiezAgents(WaldiezBase):
             If the agents are invalid.
         """
         all_agent_ids = [agent.id for agent in self.members]
-        if len(all_agent_ids) < 2:
-            raise ValueError("At least two agents are required.")
+        if len(all_agent_ids) < 1:
+            raise ValueError("At least one agent is required.")
         if len(all_agent_ids) != len(set(all_agent_ids)):
             raise ValueError("Agent IDs must be unique.")
         return self

{waldiez-0.3.9 → waldiez-0.3.11}/waldiez/models/common/base.py

@@ -64,3 +64,7 @@ class WaldiezBase(BaseModel):
         if not isinstance(by_alias, bool):
             by_alias = True
         return super().model_dump_json(by_alias=by_alias, **kwargs)
+
+    def __hash__(self) -> int:
+        """Return the hash of the object."""
+        return id(self)