waldiez 0.4.0__py3-none-any.whl → 0.4.2__py3-none-any.whl

waldiez/_version.py

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

waldiez/exporting/agent/utils/captain_agent.py

@@ -142,10 +142,11 @@ def get_llm_config(
     max_tokens: Optional[int] = 2048
     if agent.data.model_ids:
         waldiez_model = get_waldiez_model(agent.data.model_ids[0], all_models)
-        model_name = waldiez_model.name
-        temperature = waldiez_model.data.temperature
-        top_p = waldiez_model.data.top_p
-        max_tokens = waldiez_model.data.max_tokens
+        llm_config = waldiez_model.get_llm_config(skip_price=True)
+        for key in ["temperature", "top_p", "max_tokens"]:
+            if key not in llm_config:
+                llm_config[key] = None
+        return llm_config
     config_dict = {
         "model": model_name,
         "temperature": temperature,

waldiez/exporting/agent/utils/rag_user/vector_db.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 """Vector DB exporting utils for RAG user agents."""
 

waldiez/exporting/base/base_exporter.py

@@ -10,7 +10,7 @@ from .export_position import ExportPosition
 from .import_position import ImportPosition
 
 
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 class ExporterReturnType(TypedDict):
     """Exporter Return Type.

waldiez/exporting/chats/chats_exporter.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 """Chats exporter."""
 

waldiez/exporting/chats/utils/single_chat.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 """Utilities for exporting a single chat in a flow."""
 

waldiez/exporting/chats/utils/swarm.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 """Swarm chat exporting functions."""
 
 from typing import Callable, Dict, List, Optional, Tuple

waldiez/exporting/flow/flow_exporter.py

@@ -19,7 +19,7 @@
 - If the flow is async, the `run` function is async.
 """
 
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 
 from functools import partial
@@ -213,7 +213,7 @@ class FlowExporter(BaseExporter, ExporterMixin):
         content += imports[0] + "\n"
         content += get_np_no_nep50_handle() + "\n"
         content += self.get_comment("logging", self.for_notebook) + "\n"
-        content += get_start_logging(tabs=0) + "\n"
+        content += get_start_logging(is_async=is_async, tabs=0) + "\n"
         content += "start_logging()\n\n"
         if models_output:
             content += self.get_comment("models", self.for_notebook) + "\n"

waldiez/exporting/flow/utils/flow_content.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 """Utils to generate the content of a flow."""
 
@@ -22,7 +22,7 @@ def get_py_content_start(waldiez: Waldiez) -> str:
         The first part of the python script.
     """
     content = "#!/usr/bin/env python\n"
-    content += "# flake8: noqa E501\n"
+    content += "# flake8: noqa: E501\n"
     content += get_pylint_ignore_comment(False)
     content += "# cspell: disable\n"
     content += f'"""{waldiez.name}.' + "\n\n"
@@ -60,7 +60,7 @@ def get_ipynb_content_start(
         # fmt: off
         content += "# " + f"!{{sys.executable}} -m pip install -q {requirements}" + "\n"
         # fmt: on
-    content += "# flake8: noqa E501"
+    content += "# flake8: noqa: E501"
     content += get_pylint_ignore_comment(True)
     content += "# cspell: disable\n"
     return content
@@ -135,8 +135,8 @@ def get_after_run_content(
     str
         The content to add after the flow is run.
     """
-    # if th eflow has reasoning agents, we add
-    # visualize_tree(agent._root) for each agent
+    # if the flow has a reasoning agents, we add
+    # agent.visualize_tree() for each agent
     content = ""
     tab = "    "
     space = tab * tabs
@@ -146,7 +146,7 @@ def get_after_run_content(
             content += f"""
 {space}# pylint: disable=broad-except,too-many-try-statements
 {space}try:
-{space}{tab}visualize_tree({agent_name}._root)  # pylint: disable=protected-access
+{space}{tab}{agent_name}.visualize_tree()
 {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)

waldiez/exporting/flow/utils/importing_utils.py

@@ -127,7 +127,6 @@ def get_the_imports_string(
             "\nimport anyio"
             "\nimport nest_asyncio"
             "\nfrom aiocsv import AsyncDictWriter"
-            "\nfrom asyncer import asyncify"
         )
     if got_import_autogen:
         final_string += "\nimport autogen  # type: ignore\n"

waldiez/exporting/flow/utils/logging_utils.py

@@ -16,11 +16,13 @@ get_sqlite_to_csv_call_string
 """
 
 
-def get_start_logging(tabs: int = 0) -> str:
+def get_start_logging(is_async: bool, tabs: int = 0) -> str:
     """Get the logging start call string.
 
     Parameters
     ----------
+    is_async : bool
+        Whether to use async mode.
     tabs : int, optional
         The number of tabs to use for indentation, by default 0
 
@@ -41,7 +43,8 @@ def get_start_logging(tabs: int = 0) -> str:
         )
     """
     tab = "    " * tabs
-    content = f'''
+    if is_async is False:
+        return f'''
 {tab}def start_logging() -> None:
 {tab}    """Start logging."""
 {tab}    runtime_logging.start(
@@ -49,7 +52,20 @@ def get_start_logging(tabs: int = 0) -> str:
 {tab}        config={{"dbname": "flow.db"}},
 {tab}    )
 '''
-    return content
+    return f'''
+{tab}def start_logging() -> None:
+{tab}    """Start logging."""
+{tab}    # pylint: disable=import-outside-toplevel
+{tab}    from anyio.from_thread import start_blocking_portal
+
+{tab}    with start_blocking_portal(backend="asyncio") as portal:
+{tab}        portal.call(
+{tab}            runtime_logging.start,
+{tab}            None,
+{tab}            "sqlite",
+{tab}            {{"dbname": "flow.db"}},
+{tab}        )
+'''
 
 
 # pylint: disable=differing-param-doc,differing-type-doc
@@ -336,6 +352,11 @@ def get_stop_logging(tabs: int, is_async: bool) -> str:
         content += "async "
     content += "def stop_logging() -> None:\n"
     content += '    """Stop logging."""\n'
-    content += f"{tab}    runtime_logging.stop()\n"
+    if is_async:
+        content += f"{tab}    # pylint: disable=import-outside-toplevel\n"
+        content += f"{tab}    from asyncer import asyncify\n\n"
+        content += f"{tab}    await asyncify(runtime_logging.stop)()\n"
+    else:
+        content += f"{tab}    runtime_logging.stop()\n"
     content += get_sqlite_out_call(tabs + 1, is_async)
     return content

waldiez/exporting/skills/utils.py

@@ -263,8 +263,8 @@ def get_skill_secrets_import(flow_name: str, skill: WaldiezSkill) -> str:
         return ""
         # fmt: on
     module_name = f"{flow_name}_{skill.name}"
-    ignore_noqa = "  # type: ignore # noqa"
-    return f"import {module_name}_secrets{ignore_noqa}" + "\n"
+    type_ignore_noqa = "  # type: ignore # noqa"
+    return f"import {module_name}_secrets{type_ignore_noqa}" + "\n"
 
 
 def get_agent_skill_registrations(

waldiez/models/agents/agent/agent.py

@@ -177,8 +177,7 @@ class WaldiezAgent(WaldiezBase):
             )
         elif agent_class == "ReasoningAgent":
             imports.add(
-                "from autogen.agentchat.contrib.reasoning_agent "
-                "import ReasoningAgent, visualize_tree"
+                "from autogen.agents.experimental import ReasoningAgent"
             )
         elif agent_class == "CaptainAgent":
             imports.add(

waldiez/models/agents/rag_user/vector_db_config.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 
 """The vector db config for the RAG user agent."""
 

waldiez/models/model/__init__.py

@@ -3,12 +3,13 @@
 """Waldiez model."""
 
 from .extra_requirements import get_models_extra_requirements
-from .model import DEFAULT_BASE_URLS, WaldiezModel
+from .model import DEFAULT_BASE_URLS, MODEL_NEEDS_BASE_URL, WaldiezModel
 from .model_data import WaldiezModelAPIType, WaldiezModelData, WaldiezModelPrice
 
 __all__ = [
     "get_models_extra_requirements",
     "DEFAULT_BASE_URLS",
+    "MODEL_NEEDS_BASE_URL",
     "WaldiezModel",
     "WaldiezModelData",
     "WaldiezModelPrice",

waldiez/models/model/model.py

@@ -23,6 +23,22 @@ DEFAULT_BASE_URLS: Dict[WaldiezModelAPIType, str] = {
 }
 
 
+# we can omit the base_url for these models
+MODEL_NEEDS_BASE_URL: Dict[WaldiezModelAPIType, bool] = {
+    "openai": False,
+    "azure": False,
+    "google": False,
+    "anthropic": False,
+    "cohere": False,
+    "other": False,  # falls back to openai
+    "deepseek": True,
+    "mistral": True,
+    "groq": True,
+    "together": True,
+    "nim": True,
+}
+
+
 class WaldiezModel(WaldiezBase):
     """Waldiez Model class.
 
@@ -225,10 +241,13 @@ def set_default_base_url(
     Dict[str, Any]
         The llm config dictionary with the default base url set.
     """
-    if api_type in ("openai", "other", "azure", "cohere"):
-        return llm_config
+    dict_copy = llm_config.copy()
     if "base_url" not in llm_config or not llm_config["base_url"]:
-        dict_copy = llm_config.copy()
-        dict_copy["base_url"] = DEFAULT_BASE_URLS.get(api_type, "")
-        return dict_copy
-    return llm_config
+        if MODEL_NEEDS_BASE_URL.get(api_type, True):
+            dict_copy["base_url"] = DEFAULT_BASE_URLS.get(api_type, "")
+    if (
+        not llm_config.get("base_url", "")
+        and MODEL_NEEDS_BASE_URL.get(api_type, True) is False
+    ):
+        dict_copy.pop("base_url", None)
+    return dict_copy

waldiez/models/model/model_data.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 """Waldiez Model Data."""
 
 from typing import Dict, Optional

waldiez/models/waldiez.py

@@ -239,7 +239,7 @@ class Waldiez:
             self.flow.requirements,
         )
         requirements = set(requirements_list)
-        requirements.add(f"pyautogen=={autogen_version}")
+        requirements.add(f"pyautogen[openai]=={autogen_version}")
         if self.has_rag_agents:
             rag_extras = get_retrievechat_extra_requirements(self.agents)
             requirements.update(rag_extras)

waldiez/runner.py

@@ -217,6 +217,7 @@ class WaldiezRunner:
             spec.loader.exec_module(module)
             printer("<Waldiez> - Starting workflow...")
             results = module.main()
+            printer("<Waldiez> - Workflow finished")
             sys.path.pop(0)
             reset_env_vars(old_vars)
         after_run(

waldiez/utils/cli_extras/studio.py

@@ -2,6 +2,7 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 # pylint: skip-file
 # isort: skip_file
+# flake8: noqa: E501,E402
 """Waldiez-studio extra typer commands for CLI."""
 
 from typing import Any, Callable
@@ -13,7 +14,7 @@ HAVE_STUDIO = False
 studio_app: Callable[..., Any] | None = None
 
 try:
-    from waldiez_studio.cli import run  # type: ignore[unused-ignore, import-not-found, import-untyped]  # noqa  # flake8: noqa
+    from waldiez_studio.cli import run  # type: ignore[unused-ignore, import-not-found, import-untyped]
 
     studio_app = run
 

{waldiez-0.4.0.dist-info → waldiez-0.4.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.4.0
+Version: 0.4.2
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/waldiez/python
 Project-URL: repository, https://github.com/waldiez/python.git
@@ -17,7 +17,8 @@ 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
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: <3.14,>=3.10
 Requires-Dist: aiocsv==1.3.2
 Requires-Dist: aiofiles==24.1.0
 Requires-Dist: aiosqlite==0.21.0
@@ -28,82 +29,78 @@ Requires-Dist: jupytext
 Requires-Dist: numpy<2.2.0
 Requires-Dist: pandas>=2
 Requires-Dist: parso==0.8.4
-Requires-Dist: pyautogen==0.7.4
-Requires-Dist: pydantic<3,>=2.6.1
+Requires-Dist: pyautogen[openai]==0.8.4
+Requires-Dist: pydantic<3,>=2.10.2
 Requires-Dist: typer<0.16,>=0.9
 Provides-Extra: ag2-extras
 Requires-Dist: beautifulsoup4; extra == 'ag2-extras'
-Requires-Dist: chromadb>=0.5.23; extra == 'ag2-extras'
-Requires-Dist: crewai-tools==0.33.0; extra == 'ag2-extras'
-Requires-Dist: crewai>0.98.0; extra == 'ag2-extras'
-Requires-Dist: embedchain==0.1.126; extra == 'ag2-extras'
+Requires-Dist: chromadb>=0.5.10; extra == 'ag2-extras'
+Requires-Dist: embedchain; extra == 'ag2-extras'
 Requires-Dist: huggingface-hub; extra == 'ag2-extras'
 Requires-Dist: ipython; extra == 'ag2-extras'
 Requires-Dist: langchain-community<1,>=0.3.12; extra == 'ag2-extras'
 Requires-Dist: markdownify; extra == 'ag2-extras'
-Requires-Dist: pgvector>=0.3.6; extra == 'ag2-extras'
+Requires-Dist: pgvector>=0.4.0; extra == 'ag2-extras'
 Requires-Dist: protobuf>=4.25.3; extra == 'ag2-extras'
-Requires-Dist: psycopg[binary]>=3.2.3; extra == 'ag2-extras'
-Requires-Dist: psycopg[binary]>=3.2.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[anthropic]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[bedrock]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[cohere]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[gemini]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[groq]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[interop-crewai]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[interop-langchain]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[lmm]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[mistral]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[neo4j]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[ollama]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[together]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[websurfer]==0.7.4; extra == 'ag2-extras'
+Requires-Dist: psycopg[binary]>=3.2.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[anthropic]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[bedrock]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[cohere]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[gemini]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[groq]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[interop-crewai]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[interop-langchain]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[lmm]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[mistral]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[neo4j]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[ollama]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[together]==0.8.4; extra == 'ag2-extras'
+Requires-Dist: pyautogen[websurfer]==0.8.4; extra == 'ag2-extras'
 Requires-Dist: pydantic-ai>=0.0.21; extra == 'ag2-extras'
 Requires-Dist: pymongo>=4.11; extra == 'ag2-extras'
 Requires-Dist: pypdf; extra == 'ag2-extras'
 Requires-Dist: qdrant-client[fastembed]; extra == 'ag2-extras'
 Requires-Dist: sentence-transformers; extra == 'ag2-extras'
-Requires-Dist: weaviate-client>=4.10.2; extra == 'ag2-extras'
 Provides-Extra: dev
 Requires-Dist: autoflake==2.3.1; extra == 'dev'
-Requires-Dist: bandit==1.8.2; extra == 'dev'
+Requires-Dist: bandit==1.8.3; extra == 'dev'
 Requires-Dist: black[jupyter]==25.1.0; extra == 'dev'
-Requires-Dist: flake8==7.1.1; extra == 'dev'
-Requires-Dist: isort==6.0.0; extra == 'dev'
+Requires-Dist: flake8==7.2.0; extra == 'dev'
+Requires-Dist: isort==6.0.1; extra == 'dev'
 Requires-Dist: mypy==1.15.0; extra == 'dev'
 Requires-Dist: pandas-stubs; extra == 'dev'
-Requires-Dist: pre-commit==4.1.0; extra == 'dev'
+Requires-Dist: pre-commit==4.2.0; extra == 'dev'
 Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
-Requires-Dist: pylint==3.3.4; extra == 'dev'
-Requires-Dist: python-dotenv==1.0.1; extra == 'dev'
-Requires-Dist: ruff==0.9.6; extra == 'dev'
+Requires-Dist: pylint==3.3.6; extra == 'dev'
+Requires-Dist: python-dotenv==1.1.0; extra == 'dev'
+Requires-Dist: ruff==0.11.2; 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-pyyaml==6.0.12.20250326; extra == 'dev'
 Requires-Dist: types-toml==0.10.8.20240310; extra == 'dev'
-Requires-Dist: yamllint==1.35.1; extra == 'dev'
+Requires-Dist: yamllint==1.37.0; 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.6.4; extra == 'docs'
-Requires-Dist: mkdocs-minify-html-plugin==0.2.4; extra == 'docs'
+Requires-Dist: mkdocs-material==9.6.9; extra == 'docs'
+Requires-Dist: mkdocs-minify-html-plugin==0.3.1; extra == 'docs'
 Requires-Dist: mkdocs==1.6.1; extra == 'docs'
-Requires-Dist: mkdocstrings-python==1.15.0; extra == 'docs'
-Requires-Dist: mkdocstrings[crystal,python]==0.28.0; extra == 'docs'
+Requires-Dist: mkdocstrings-python==1.16.8; extra == 'docs'
+Requires-Dist: mkdocstrings[crystal,python]==0.29.0; extra == 'docs'
 Provides-Extra: jupyter
 Requires-Dist: jupyterlab>=4.3.0; extra == 'jupyter'
-Requires-Dist: waldiez-jupyter==0.4.0; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.4.2; extra == 'jupyter'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.4.0; extra == 'studio'
+Requires-Dist: waldiez-studio==0.4.2; extra == 'studio'
 Provides-Extra: test
-Requires-Dist: pytest-asyncio==0.25.3; extra == 'test'
+Requires-Dist: pytest-asyncio==0.26.0; 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.4; extra == 'test'
+Requires-Dist: pytest==8.3.5; extra == 'test'
 Description-Content-Type: text/markdown
 
 # Waldiez

{waldiez-0.4.0.dist-info → waldiez-0.4.2.dist-info}/RECORD

@@ -1,15 +1,15 @@
 waldiez/__init__.py,sha256=cXWdy5P9i_x6fHwlL5DRaIhmpfnhabo5GjASStlcLWw,819
 waldiez/__main__.py,sha256=0dYzNrQbovRwQQvmZC6_1FDR1m71SUIOkTleO5tBnBw,203
-waldiez/_version.py,sha256=kKOFfvxa6VJYqFJTYlWnFWHOdlAHJD4A0wscEAebzAU,155
+waldiez/_version.py,sha256=UtKLzWSct51WNdME7_hVxvsAM1wToEe77efK_17lcFM,155
 waldiez/cli.py,sha256=FaPWh9IHBHoC9rtyt2bKBrk7S018BX__mwtTArfv4AA,6992
 waldiez/exporter.py,sha256=ZuMF8cra0Shw4nkDiFuPqaQKFYrF0vPj_I3EhPyesPw,4788
 waldiez/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-waldiez/runner.py,sha256=iukz3kQRGOBQ5aatJsFwHCw4_lqHRMO9QFZSDDSjaSw,11673
+waldiez/runner.py,sha256=ea09I3c7-evD3Yh7KpTMsZ5gf5OOfXujCcgo2Ypews8,11726
 waldiez/exporting/__init__.py,sha256=nlMWBnn_MViGBJmDhrTCYpPv19fEP_WixvmmY9jt5c8,399
 waldiez/exporting/agent/__init__.py,sha256=GOgri16QNICpx3YNZPzVn_KYVxuQACYKQyazhJNtYK0,188
 waldiez/exporting/agent/agent_exporter.py,sha256=Rv-Fcsf3r_nEib1HVk3SglKkLU6fN7cMLgUwJ4Vi4sw,10482
 waldiez/exporting/agent/utils/__init__.py,sha256=UlyOZCm3jLz5PHOJ9fhZuCqbNBBlomzvRd7K_tKR-No,844
-waldiez/exporting/agent/utils/captain_agent.py,sha256=31KfNOQpaZIcS97-G9yzC7du7bYrU7SWXkNor9twlX4,7233
+waldiez/exporting/agent/utils/captain_agent.py,sha256=TJZbjf7TtIdS00wTWciqneoNrGC4aJiYL2Rn6WpxMLs,7277
 waldiez/exporting/agent/utils/code_execution.py,sha256=sRHczOmm12U9E-Q1v9UQRwao1LUEve4fnKvJwsokNNk,2352
 waldiez/exporting/agent/utils/group_manager.py,sha256=VCqd-lsCertwVTWVez9Z1HS_H-UmBGyNGizvC736VnI,7360
 waldiez/exporting/agent/utils/reasoning.py,sha256=EBE42Dar9n6fBrhg9D9wch0KptsP_m3Dm2WNEDuKb7g,999
@@ -22,10 +22,10 @@ waldiez/exporting/agent/utils/rag_user/mongo_utils.py,sha256=XWo6ixphBtcpmFIe9fE
 waldiez/exporting/agent/utils/rag_user/pgvector_utils.py,sha256=KScLbdMdiZRWfBXhMTDQUIYYoC0gaH6rJWhJ4TZsVq8,2922
 waldiez/exporting/agent/utils/rag_user/qdrant_utils.py,sha256=eBOGRUoTnb7QMXpScWIlMnhOv1jS4qlRJkwze0SB0Bo,3605
 waldiez/exporting/agent/utils/rag_user/rag_user.py,sha256=gg9vH98Pf5dV42IW9HHB1Ajs6kPFVuzblFhGc_il6CU,7024
-waldiez/exporting/agent/utils/rag_user/vector_db.py,sha256=cbZWrzd3PmIvQ9U95kZmc1mND0ZIh3Y4UiK8EnWR-sE,3807
+waldiez/exporting/agent/utils/rag_user/vector_db.py,sha256=EUv7mgFHAKEamkPjdFQHJAw6mTRPjZBCP2LjFLsn0Wo,3808
 waldiez/exporting/base/__init__.py,sha256=CTfIOJt8Kgljv-jrktP3_AXnKhtn5ijaMP3QCk4b6NE,816
 waldiez/exporting/base/agent_position.py,sha256=1qKE4MxNfJXwgsAjd674eXYub5X82WInBUc6Lh1Tw98,1642
-waldiez/exporting/base/base_exporter.py,sha256=P4_TcF4jpsRszNUJ1LZvWCgL3IaG-xYHnUNj6Yv9HKo,3522
+waldiez/exporting/base/base_exporter.py,sha256=mfBxbnTll1NJeDmARYwxp7Dx2bgSsE09QZG0lu5Wsdc,3523
 waldiez/exporting/base/export_position.py,sha256=bV50acpqHv6BQ6q6UNXg0e17Tv_koyardSA_RZDgFP4,1106
 waldiez/exporting/base/import_position.py,sha256=EzgHF8GZZ8UsyFdNo_da9hTrQEmbVtM2ofiv9W2-C_I,447
 waldiez/exporting/base/mixin.py,sha256=_QkBitZ3T6jdDx9yZTDFKToa0f9WsLKCW4FpZGjCztc,3376
@@ -35,36 +35,36 @@ waldiez/exporting/base/utils/naming.py,sha256=G4e9MPnbg9W8QLVblQj31sV46GKmAOqEA8
 waldiez/exporting/base/utils/path_check.py,sha256=5XGKmHA3oi9kf2gI4ZHB3K9gnjDP0NEfpn0tP8cme-8,1425
 waldiez/exporting/base/utils/to_string.py,sha256=fo8i6QaLZJsym3biN2IhVooTWlXiXVCzuU8FNFJVXME,2183
 waldiez/exporting/chats/__init__.py,sha256=BS8F-hgrzsKPtWEyEMPE-e0JL6q5478i76vR1bKjXXY,195
-waldiez/exporting/chats/chats_exporter.py,sha256=xbVSa9HJ5X8tU197XqAeWfORdtSxcZf9V9scbO_6cNg,8515
+waldiez/exporting/chats/chats_exporter.py,sha256=mLvT2Rrfec82L0_nPZq_7YB5cJuDK6GsJbVZxhafOA0,8516
 waldiez/exporting/chats/utils/__init__.py,sha256=FMS94ObfHIlH_Pg_5s5PR-QaXYzcpZErP1Vut2PYaTY,442
 waldiez/exporting/chats/utils/common.py,sha256=kZ9xO26Oo3tKSxZescFfB2laE9I8F5Hnq3c8oojEVGo,2410
 waldiez/exporting/chats/utils/nested.py,sha256=evjvJeWHATumNNffAPZUInlyQuzXLmBWnlHfYQaub0I,9227
 waldiez/exporting/chats/utils/sequential.py,sha256=TYAte2hwpGkEENjCMer-6PcL2r8PE-o95hgoi_AX2eg,7841
-waldiez/exporting/chats/utils/single_chat.py,sha256=FYz0jvtO0yFYPk5CBuXb7rOW8VxB0NJWvWY1PEHPIAQ,9535
-waldiez/exporting/chats/utils/swarm.py,sha256=i8n-QNSgAOGHGMcV1pwbEVqb_LJZtNxtJ4yR42yLl6I,6928
+waldiez/exporting/chats/utils/single_chat.py,sha256=8pgqPuug4K6DcMc7WANAkHPw5EuONhdahytgZ37scR8,9536
+waldiez/exporting/chats/utils/swarm.py,sha256=KOJnUVcYql3tQgMgkzBo1q8rFtovw7o88x3L6ZUWw-c,6929
 waldiez/exporting/flow/__init__.py,sha256=TAJ8lq9saP1IX0U-A6yGKCfvbQnEocmRAX4Hc60f3yY,183
-waldiez/exporting/flow/flow_exporter.py,sha256=xXLf5hx5xYwnPgv1iTAa6x1RtdMBno3jGQACxgtYfVM,18196
+waldiez/exporting/flow/flow_exporter.py,sha256=oRXJEG2zkEWACvRTZd0VhTt10Rdz4oW7cuyPDxnP6h4,18216
 waldiez/exporting/flow/utils/__init__.py,sha256=Alk9M4_BLvnZ5yO3y0T-Xie__OgtjCPtjzfZx_tH0Ig,1380
 waldiez/exporting/flow/utils/agent_utils.py,sha256=QL1nzlgYeh6Y-l9R3hmCsrxSzluvBVciuuEIFdBXXKM,6109
 waldiez/exporting/flow/utils/chat_utils.py,sha256=q6HS4JTNQJe15eh78H1BlN-wId9XzvVQziGhxsUIEFs,1882
 waldiez/exporting/flow/utils/def_main.py,sha256=GxEBfX6cUGFCMdxiu_hp5TLIhIC4ohczxA1A01mHnOo,2746
-waldiez/exporting/flow/utils/flow_content.py,sha256=w2JUqx1EmYRnptGtsYrU-gfETHi6bkimV9vd2Ao-YBI,5548
+waldiez/exporting/flow/utils/flow_content.py,sha256=RtkCPs9tE99beivh7gG-NihdvnashxZ5BuCM-qbHpPE,5507
 waldiez/exporting/flow/utils/flow_names.py,sha256=GtfVRF8WYUP8Xx47AiyZ5FuuCUb22jhJn3diblbbBfM,3463
-waldiez/exporting/flow/utils/importing_utils.py,sha256=juKDzk06A4IuHrjFjR-TUnc8AcH0-6b4S_2_rt5MuFs,7151
-waldiez/exporting/flow/utils/logging_utils.py,sha256=m3Ja00Qb_07rLVgwSOFSBaJk48SfCNsyPuQbRy5VMNU,10963
+waldiez/exporting/flow/utils/importing_utils.py,sha256=ysryaFF872wrtqRDAlZB2mFvXwjmkX_PG-j5vLPViWQ,7106
+waldiez/exporting/flow/utils/logging_utils.py,sha256=DsBuv7LLWb4i18HQkXrJV2YpjG_Jzk0ZDsM-8HJ5bMg,11720
 waldiez/exporting/models/__init__.py,sha256=QbyYuS-Px-M9vMrdL-UKqnDzhYWRkNmRGBbc1IBuwpM,212
 waldiez/exporting/models/models_exporter.py,sha256=i9aSnL7JNU8iWJotfSxWbm1CRjUP6yqd9CtfXWiRbV4,6289
 waldiez/exporting/models/utils.py,sha256=g15N_JA21Ly5nzCfburEbHf6fTWX7hG5rxyTvAwyXl0,4997
 waldiez/exporting/skills/__init__.py,sha256=sJm4ZnN3haldxRBgNlLK0qMK5YtydF2XnSaqnxu2hQo,195
 waldiez/exporting/skills/skills_exporter.py,sha256=l7a662slVl8h7M2I4lrI1oWwolP7ewUCZxlt_ius54A,5606
-waldiez/exporting/skills/utils.py,sha256=MbhK2TtJZyE8ysEs70JRMCY6jVlT67L-xebpAcPwqS4,10952
+waldiez/exporting/skills/utils.py,sha256=GK6UDdtz4dMMTiqvkeK6ROJNGcX52OFKDYLwDxq6hFI,10962
 waldiez/models/__init__.py,sha256=OsLOhFRoKYgB-7zNEdTG8BmHPiSRIlZv8NKZ6c2bVJs,3996
-waldiez/models/waldiez.py,sha256=y1KFXOV0xM3yjlmrf5C_R1QTT7PO0SYMbo4mnxbGi50,9976
+waldiez/models/waldiez.py,sha256=G0lmD-Vm3_QO2QB9t-zHWFVwgEnHjccsnB0fxLrwWBU,9984
 waldiez/models/agents/__init__.py,sha256=bB1TKT8rqb_-SfCIIdjNi3Xkof4IVBtuhCubg2UEmXI,4970
 waldiez/models/agents/agents.py,sha256=kwUOlV4VJIgqhYBQHbAZhWzx3iENOvYilN3nDonZTIU,4605
 waldiez/models/agents/extra_requirements.py,sha256=6v61EsAUCKAiUi52_p4kanB9EgwGN9nRy7qV-f13tyc,2681
 waldiez/models/agents/agent/__init__.py,sha256=Uu3t5OLBElOWlM1QqsZv0ZujZaBjCrlVmBvQCiS_Mrg,1067
-waldiez/models/agents/agent/agent.py,sha256=7Ku3gPJg2uvXwx1mmqJg9u1r21vz3KqepdwviylZP6s,8202
+waldiez/models/agents/agent/agent.py,sha256=GfyDC5NXhcTo4gcVHuHFYCaGX0w4aFFfcPSWYVG37oU,8153
 waldiez/models/agents/agent/agent_data.py,sha256=rqRRMdqOzUF5jjdVoYika-tqe0YUWHQ4at5fNL9GAYM,5437
 waldiez/models/agents/agent/agent_type.py,sha256=3IHXHt3TrMmYYKaeN64q266GP08ACRArnbHt4aIB1Mw,449
 waldiez/models/agents/agent/code_execution.py,sha256=lBfJMRvzl1QfNIHwQXxY2ote2Kjh9iT3twz-Tj8JrZA,2035
@@ -87,7 +87,7 @@ waldiez/models/agents/rag_user/__init__.py,sha256=sGMVeg7J0QYclnS_iiM5XP6Lz_M9_V
 waldiez/models/agents/rag_user/rag_user.py,sha256=eWCW0TxqS8eJ1ph-9UAYfD-MHwgAOhi8derMYDY4PkU,1620
 waldiez/models/agents/rag_user/rag_user_data.py,sha256=Aboyy6JJcDzh2DQIttt1l-Uzqebn46HKlw11WObbUrA,963
 waldiez/models/agents/rag_user/retrieve_config.py,sha256=nZZ6r3yEl74J12kPsKJA28FpdjF79h-J72Fp4BDaiuI,30107
-waldiez/models/agents/rag_user/vector_db_config.py,sha256=4aaYBTqtliUzUHmxVKBfyLs6gGtmx-f9hZdiwdlab9c,5015
+waldiez/models/agents/rag_user/vector_db_config.py,sha256=USf39Pwuj4p8qR8lEdgmeuJ6cAIwuYwxfEZkPuQqlTs,5016
 waldiez/models/agents/reasoning/__init__.py,sha256=ALDeBqJBVoOz5z8ftpfr0Edbku54uQrkkImPuoM98LQ,429
 waldiez/models/agents/reasoning/reasoning_agent.py,sha256=ti4gtBkKphOOlBqaWF1onEfO6rikszEbzd2UwWez9sk,1134
 waldiez/models/agents/reasoning/reasoning_agent_data.py,sha256=1ozpf8-wpmb0keJYCujOaOmBmy23dnQ4T5bA1gqG5pI,4098
@@ -119,10 +119,10 @@ waldiez/models/flow/__init__.py,sha256=3UFbMMn_5c5AXs6ppkvhNeWb9jelru2lNHUjpEq0N
 waldiez/models/flow/flow.py,sha256=Shyg-XwzloSzqLLFPwY9g_MjZHxAGy2h_qfPmK58UZs,16876
 waldiez/models/flow/flow_data.py,sha256=afD97Ff9Yd7QSXE4DphaENY7wqrq4QIl4EVgI8O4ZFk,5464
 waldiez/models/flow/utils.py,sha256=X_jelP5lHuQt1OZE6sxLlhEKbkkt8jZgbkF79e9rQig,7140
-waldiez/models/model/__init__.py,sha256=0klSsHb8poHgtnVdCi-rDiSOb_yOofk9RPehjLQnGFc,482
+waldiez/models/model/__init__.py,sha256=jyJHJWzqNxnmDHAhpXWMnMK_4VF2olPIE3k7KV7_xVg,532
 waldiez/models/model/extra_requirements.py,sha256=KgqzixbsvzM6Ec8rdQEVoWtyNusqW33ukdRoBHS5LAg,1690
-waldiez/models/model/model.py,sha256=wYT_ajxIYY-wxJPRiZctAs78nazdUYkYSlCVw3ZE4Go,7283
-waldiez/models/model/model_data.py,sha256=5bdUz0gi3RNowVh4SCKiTds6sZMMBvf69pNO6mx4LnA,3793
+waldiez/models/model/model.py,sha256=A1N90QZsbd3vEddeffAMdKIja9RjspZl46Ff4q2CP9w,7743
+waldiez/models/model/model_data.py,sha256=xB6ovuB5Rvz9Jc3A_K8CEI55E4EPLofGuVcN4ehxkvE,3794
 waldiez/models/skill/__init__.py,sha256=EMkjfsMUU-2H5cZ4NhzHJWuyaSp605whT-zf7Gxt4aA,470
 waldiez/models/skill/extra_requirements.py,sha256=q3ZY5v6mWQ_9C0GuAmFm8DK2j8TDr9CQafTmFY0w5bc,1082
 waldiez/models/skill/skill.py,sha256=Dw7sqxj78i430eTIOrkH8OFBJVLefgyJkoBn11foOx0,9083
@@ -138,10 +138,10 @@ waldiez/utils/flaml_warnings.py,sha256=jgdS8zwzuWPodTgm9nnx3wO5MrOUbR7P7IRj9-aq2
 waldiez/utils/pysqlite3_checker.py,sha256=8m0yBAbg3tMyXOBktL2q1WXMgjym_Ttn2NbgpL1vyys,8128
 waldiez/utils/cli_extras/__init__.py,sha256=FQ6biuz6Ig-U3xkmgac3WMUynH_7XWjeGI-DqWFTC8c,625
 waldiez/utils/cli_extras/jupyter.py,sha256=C4fOiS_PbU15X-6HUZBPHvfgEjOrY39e07mClKvswPI,2971
-waldiez/utils/cli_extras/studio.py,sha256=te6VmwYDeEEK6Jps-Ou84QPZmCYMsF-oE1hNu6ewFHg,887
-waldiez-0.4.0.dist-info/METADATA,sha256=MikezMp_G1SZ7l8ObfJ5OqbzsoKyZNRffqR4s3ZQN7o,9830
-waldiez-0.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-waldiez-0.4.0.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
-waldiez-0.4.0.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
-waldiez-0.4.0.dist-info/licenses/NOTICE.md,sha256=lrKsUNrpE18LPSLPJY_kt2ZJFZJEwH55wPv5Axcheyc,133
-waldiez-0.4.0.dist-info/RECORD,,
+waldiez/utils/cli_extras/studio.py,sha256=JTlkLuXgqDC0z79hT-LNiSqniXcql7jyz1nQ517-xKI,889
+waldiez-0.4.2.dist-info/METADATA,sha256=otXaCZcP7Izruf6uJKmfAWayYIFGge5JGgeHNmuIopw,9648
+waldiez-0.4.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+waldiez-0.4.2.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
+waldiez-0.4.2.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
+waldiez-0.4.2.dist-info/licenses/NOTICE.md,sha256=lrKsUNrpE18LPSLPJY_kt2ZJFZJEwH55wPv5Axcheyc,133
+waldiez-0.4.2.dist-info/RECORD,,