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

waldiez/running/environment.py

@@ -8,7 +8,7 @@ import os
 import site
 import sys
 import warnings
-from typing import Dict, List, Tuple
+from typing import Dict, Generator, List, Tuple
 
 
 def in_virtualenv() -> bool:
@@ -50,6 +50,35 @@ def refresh_environment() -> None:
         for mod in modules_to_reload:
             if mod not in sys.modules:
                 importlib.import_module(mod)
+        # a swarm chat without a user agent
+        # creates a new user (this has a default code execution with docker)
+        # captain also generates new agents that also have
+        # default code execution with docker
+        # temp (until we handle/detect docker setup)
+        os.environ["AUTOGEN_USE_DOCKER"] = "0"
+        # we might get:
+        # module 'numpy' has no attribute '_no_nep50_warning'
+        # in autogen/agentchat/contrib/captainagent/tool_retriever.py
+        os.environ["NEP50_DEPRECATION_WARNING"] = "0"
+        os.environ["NEP50_DISABLE_WARNING"] = "1"
+        os.environ["NPY_PROMOTION_STATE"] = "weak"
+        import numpy as np
+
+        if not hasattr(np, "_no_pep50_warning"):
+            import contextlib
+
+            @contextlib.contextmanager
+            def _np_no_nep50_warning() -> Generator[None, None, None]:
+                """Dummy function to avoid the warning.
+
+                Yields
+                ------
+                None
+                    Dummy value.
+                """
+                yield
+
+            setattr(np, "_no_pep50_warning", _np_no_nep50_warning)  # noqa
 
 
 def set_env_vars(flow_env_vars: List[Tuple[str, str]]) -> Dict[str, str]:

waldiez/running/running.py

@@ -91,12 +91,6 @@ def before_run(
     str
         The file name.
     """
-    printer = get_printer()
-    printer(
-        "Requirements installed.\n"
-        "NOTE: If new packages were added and you are using Jupyter, "
-        "you might need to restart the kernel."
-    )
     if not uploads_root:
         uploads_root = Path(tempfile.mkdtemp())
     else:

waldiez/utils/pysqlite3_checker.py

@@ -1,5 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
+# pylint: disable=import-error
 # flake8: noqa: E501
 """Try to install pysqlite3-binary.
 
@@ -205,18 +206,30 @@ def check_pysqlite3() -> None:
         import pysqlite3  # type: ignore[unused-ignore, import-untyped, import-not-found]  # noqa
     except ImportError:
         print("pysqlite3 not found or cannot be imported.")
-        # Uninstall pysqlite3-binary if it is already installed
-        pip_uninstall("pysqlite3", "pysqlite3-binary")
-        source_path = download_sqlite_amalgamation()
-        install_pysqlite3(source_path)
+        if sys.platform.lower() == "linux":
+            pip_install("pysqlite3-binary")
+        else:
+            # Uninstall pysqlite3-binary if it is already installed
+            pip_uninstall("pysqlite3", "pysqlite3-binary")
+            cwd = os.getcwd()
+            tmpdir = tempfile.mkdtemp()
+            os.chdir(tmpdir)
+            source_path = download_sqlite_amalgamation()
+            install_pysqlite3(source_path)
+            os.chdir(cwd)
+            shutil.rmtree(tmpdir)
         site.main()
         # Re-import pysqlite3 as sqlite3
         import pysqlite3  # type: ignore[unused-ignore, import-untyped, import-not-found]  # noqa
 
+        sys.modules["sqlite3"] = sys.modules["pysqlite3"]
+
 
 def test_sqlite_usage() -> None:
     """Test the usage of the sqlite3 module."""
-    # pylint: disable=import-outside-toplevel
+    # pylint: disable=import-outside-toplevel, unused-import, line-too-long
+    import pysqlite3  # type: ignore[unused-ignore, import-untyped, import-not-found]  # noqa
+
     sys.modules["sqlite3"] = sys.modules.pop("pysqlite3")
     import sqlite3  # noqa
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.3.11
+Version: 0.4.0
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/waldiez/python
 Project-URL: repository, https://github.com/waldiez/python.git
@@ -20,51 +20,63 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: <3.13,>=3.10
 Requires-Dist: aiocsv==1.3.2
 Requires-Dist: aiofiles==24.1.0
-Requires-Dist: aiosqlite==0.20.0
+Requires-Dist: aiosqlite==0.21.0
 Requires-Dist: asyncer==0.0.8
 Requires-Dist: graphviz==0.20.3
 Requires-Dist: httpx<1
 Requires-Dist: jupytext
+Requires-Dist: numpy<2.2.0
 Requires-Dist: pandas>=2
 Requires-Dist: parso==0.8.4
-Requires-Dist: pyautogen==0.7.3
+Requires-Dist: pyautogen==0.7.4
 Requires-Dist: pydantic<3,>=2.6.1
 Requires-Dist: typer<0.16,>=0.9
 Provides-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; (platform_system != 'Linux') and 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: 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: protobuf>=4.25.3; extra == 'ag2-extras'
 Requires-Dist: psycopg[binary]>=3.2.3; 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; (platform_system != 'Linux') and 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: 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: 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: mypy==1.14.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: 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.4; extra == 'dev'
+Requires-Dist: ruff==0.9.6; 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,16 +86,16 @@ 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.1; extra == 'docs'
-Requires-Dist: mkdocs-minify-html-plugin==0.2.3; 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==1.6.1; extra == 'docs'
-Requires-Dist: mkdocstrings-python==1.13.0; extra == 'docs'
-Requires-Dist: mkdocstrings[crystal,python]==0.27.0; extra == 'docs'
+Requires-Dist: mkdocstrings-python==1.15.0; extra == 'docs'
+Requires-Dist: mkdocstrings[crystal,python]==0.28.0; extra == 'docs'
 Provides-Extra: jupyter
 Requires-Dist: jupyterlab>=4.3.0; extra == 'jupyter'
-Requires-Dist: waldiez-jupyter==0.3.11; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.4.0; extra == 'jupyter'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.3.11; extra == 'studio'
+Requires-Dist: waldiez-studio==0.4.0; extra == 'studio'
 Provides-Extra: test
 Requires-Dist: pytest-asyncio==0.25.3; extra == 'test'
 Requires-Dist: pytest-cov==6.0.0; extra == 'test'

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

@@ -1,20 +1,19 @@
 waldiez/__init__.py,sha256=cXWdy5P9i_x6fHwlL5DRaIhmpfnhabo5GjASStlcLWw,819
 waldiez/__main__.py,sha256=0dYzNrQbovRwQQvmZC6_1FDR1m71SUIOkTleO5tBnBw,203
-waldiez/_version.py,sha256=xakNljN5CllPqeJEkZbxbd1Lwzqsba0yrHbJfxNpzSU,156
-waldiez/cli.py,sha256=ZhcXOH5WtGVncQpoOix9yHXctQKBwckgiIdYLaRj40c,7108
+waldiez/_version.py,sha256=kKOFfvxa6VJYqFJTYlWnFWHOdlAHJD4A0wscEAebzAU,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=rFuCPFSmdOxujEWdEtCrGv8FpZ0cZhPagIslPs-2DZU,11317
+waldiez/runner.py,sha256=iukz3kQRGOBQ5aatJsFwHCw4_lqHRMO9QFZSDDSjaSw,11673
 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=ZvEcBG1kNlaAitTZbqIfxF7wqnxUtobDvAC1gPcLME8,10060
-waldiez/exporting/agent/utils/__init__.py,sha256=Pg8Gwj9lA8MAb1SkkFDn6PRa2wtVJh4pABOmEC_ClRI,909
-waldiez/exporting/agent/utils/agent_class_name.py,sha256=vzeAafjj1Sb5-u-LziTgbMnnY2s7D6Z9yu0GUBqiAao,999
-waldiez/exporting/agent/utils/agent_imports.py,sha256=UgP3hWvKngMWUk2S6n2J2mhIsnfFCkEKHVVdXLoPu3M,1775
+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/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
-waldiez/exporting/agent/utils/swarm_agent.py,sha256=rAg_d9WDtWRQKMsI85xSrr5MqR8yYjZZTeQveyCmRFU,15775
+waldiez/exporting/agent/utils/swarm_agent.py,sha256=VjmRAF8Vc08-xUkBPqQ60okw0Hz0KNh4J4PpEFjZUyw,15967
 waldiez/exporting/agent/utils/teachability.py,sha256=4RXj-NfNZwdSTKBqw583EIXUzTuWMXYtMp2kRsCixsk,1268
 waldiez/exporting/agent/utils/termination_message.py,sha256=tQ4rCOLumn-xbQzTPK397BQI_fOV6QrjbylA0F4TMrw,1280
 waldiez/exporting/agent/utils/rag_user/__init__.py,sha256=B7vM06Cqe1rfeB_srQxzl1wOHb4wzOwdsTXQy4BFEQc,289
@@ -31,7 +30,7 @@ waldiez/exporting/base/export_position.py,sha256=bV50acpqHv6BQ6q6UNXg0e17Tv_koya
 waldiez/exporting/base/import_position.py,sha256=EzgHF8GZZ8UsyFdNo_da9hTrQEmbVtM2ofiv9W2-C_I,447
 waldiez/exporting/base/mixin.py,sha256=_QkBitZ3T6jdDx9yZTDFKToa0f9WsLKCW4FpZGjCztc,3376
 waldiez/exporting/base/utils/__init__.py,sha256=2dp5wO0TYWywgXXptU9VLDjNQYCf8x7g4e4vPlLKW98,508
-waldiez/exporting/base/utils/comments.py,sha256=M1W0lWMTyOjRZv5f58rktIPeYA9XF95lZlfcvB05tYU,2257
+waldiez/exporting/base/utils/comments.py,sha256=08L5W7SuXU79UsePpG-dAuDZtkO4ltdfDEPuezdIIHM,2291
 waldiez/exporting/base/utils/naming.py,sha256=G4e9MPnbg9W8QLVblQj31sV46GKmAOqEA8iChu6mo58,3670
 waldiez/exporting/base/utils/path_check.py,sha256=5XGKmHA3oi9kf2gI4ZHB3K9gnjDP0NEfpn0tP8cme-8,1425
 waldiez/exporting/base/utils/to_string.py,sha256=fo8i6QaLZJsym3biN2IhVooTWlXiXVCzuU8FNFJVXME,2183
@@ -42,30 +41,32 @@ waldiez/exporting/chats/utils/common.py,sha256=kZ9xO26Oo3tKSxZescFfB2laE9I8F5Hnq
 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=KJTtgQFt8c_jR800vC8R14ArYVm6ZgpsUyGbiOykmMI,6929
+waldiez/exporting/chats/utils/swarm.py,sha256=i8n-QNSgAOGHGMcV1pwbEVqb_LJZtNxtJ4yR42yLl6I,6928
 waldiez/exporting/flow/__init__.py,sha256=TAJ8lq9saP1IX0U-A6yGKCfvbQnEocmRAX4Hc60f3yY,183
-waldiez/exporting/flow/flow_exporter.py,sha256=DIiplGZaUretIpp6z2inEiiKIxNVhPw8-DF6tC23gjg,18280
-waldiez/exporting/flow/utils/__init__.py,sha256=Fe0sg5i1K2Vu0ruWsIuPj7oTRHoJTW4KdKmktKRxyrc,1389
+waldiez/exporting/flow/flow_exporter.py,sha256=xXLf5hx5xYwnPgv1iTAa6x1RtdMBno3jGQACxgtYfVM,18196
+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=3sINpH9q5h4v14tsV5G1abUgm6-v5HdVwUyfsN8CRN4,2662
-waldiez/exporting/flow/utils/flow_content.py,sha256=RIMtpBFZCJf2G67ipOa3Co5OQinaobYvthqisgir5sQ,4505
+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_names.py,sha256=GtfVRF8WYUP8Xx47AiyZ5FuuCUb22jhJn3diblbbBfM,3463
-waldiez/exporting/flow/utils/importing_utils.py,sha256=h8FqyAU0ApSFaie2lipxk69rNWNGwp_xAuhPpT991-M,5552
+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/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=o_16s0VjMoaDwe-JJdbqDP2BzQJV9SAM57k_4KxAix0,5290
-waldiez/exporting/skills/utils.py,sha256=SAJ55fpNlU9DmmAcSHafq1ddgg4A-bjBxbL3aaWps50,8120
-waldiez/models/__init__.py,sha256=R_1yLo1mSJHMyrxokXuO-24PVxE3sA0ep2Tj5EG-mTM,3816
-waldiez/models/waldiez.py,sha256=Hc54e_aTzckmi8jsfgNP-an2ocx2HRQhTVhWaexNzBI,10737
-waldiez/models/agents/__init__.py,sha256=HOY3IG1WWNqepRkETjFZihHRZHgipxWO5cxeVWm1Rv0,4553
-waldiez/models/agents/agents.py,sha256=pSB3UjeUcYpmWqKoxKpT_CAp2YMzN6l2-ZEYh8Pdddc,4322
-waldiez/models/agents/agent/__init__.py,sha256=us1qPNCwKdUao9UUVXDHbcjuUUA5yexWEQkozvWLUDY,1044
-waldiez/models/agents/agent/agent.py,sha256=qZLvi3kAW13DiOzL6IqsjrgkA4q65gTJY9ymC0Pf044,5803
+waldiez/exporting/skills/skills_exporter.py,sha256=l7a662slVl8h7M2I4lrI1oWwolP7ewUCZxlt_ius54A,5606
+waldiez/exporting/skills/utils.py,sha256=MbhK2TtJZyE8ysEs70JRMCY6jVlT67L-xebpAcPwqS4,10952
+waldiez/models/__init__.py,sha256=OsLOhFRoKYgB-7zNEdTG8BmHPiSRIlZv8NKZ6c2bVJs,3996
+waldiez/models/waldiez.py,sha256=y1KFXOV0xM3yjlmrf5C_R1QTT7PO0SYMbo4mnxbGi50,9976
+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_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
 waldiez/models/agents/agent/linked_skill.py,sha256=VpVDtI13bUPRUDk8usoIJVHs7ZLPWS3pp5GvuzIiZec,760
 waldiez/models/agents/agent/nested_chat.py,sha256=wPwc_6qUMz8JgQL9rjx2ry4HPJKvGlLzDq5SmDXmku0,1850
@@ -74,65 +75,73 @@ waldiez/models/agents/agent/termination_message.py,sha256=eKd_vvKx0Eegq-gU1o7BJx
 waldiez/models/agents/assistant/__init__.py,sha256=y2s2F2e7njMcmwye1UOUULHSvU3VsDLreTLvRU1I4M0,268
 waldiez/models/agents/assistant/assistant.py,sha256=POBoEv5iOCKF3s7a5Ww7OojM16PREye6r9jO7e6epd8,1188
 waldiez/models/agents/assistant/assistant_data.py,sha256=KFxgCXdcmr6v6kG13nLQYUrktIVV0DMNcpV-7kt-5rk,916
+waldiez/models/agents/captain_agent/__init__.py,sha256=xM2coa1L09hEgm36xh9GQJk_RPa-Ev8Jot_KuCYUaKQ,406
+waldiez/models/agents/captain_agent/captain_agent.py,sha256=Dl8Nsf2UfRKit9k6lGxBHep3bHeHcdXNXbt67Ql2lnY,1213
+waldiez/models/agents/captain_agent/captain_agent_data.py,sha256=o_MsCyg6tTRbRdofAGP6fJdS4F7yE6Q7aPgCOWLNk50,1723
+waldiez/models/agents/captain_agent/captain_agent_lib_entry.py,sha256=oxuKB6w8LuCUPqNcc78NsdVeiAMM_BnvuFtUOlZfeG0,868
 waldiez/models/agents/group_manager/__init__.py,sha256=MlMzrAfiSRprVJvAVNLUuulckNsBuWsiHXa0-tcUCWM,894
 waldiez/models/agents/group_manager/group_manager.py,sha256=LG5-PEi6Hf69fZSkwOBjHnhC_SikRUewqBU37x4DxXk,2780
 waldiez/models/agents/group_manager/group_manager_data.py,sha256=LZYG2RF6El041rIE0X_9tSfe7bElgixsSja04v4KiiA,2878
-waldiez/models/agents/group_manager/speakers.py,sha256=0QAvlPswcBGFx3heCrX9AfWaVSHJxj0xV52IyYCHGjk,7848
+waldiez/models/agents/group_manager/speakers.py,sha256=Dzar1Ztul6cKiQfKYUcpj75Qb-k5PjsIVcFbxMKcoto,8005
 waldiez/models/agents/rag_user/__init__.py,sha256=sGMVeg7J0QYclnS_iiM5XP6Lz_M9_Vvdz8_mpf0Ft3o,1382
 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=asCScPLIlw_Itt4WArzkzTVzD55Mbpd49uuUYRX3DYw,29964
+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/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
-waldiez/models/agents/reasoning/reasoning_agent_reason_config.py,sha256=QEEI74IrgmvwS3T3NXWJWczqn6T1lILN45tX1rFRWOQ,2840
+waldiez/models/agents/reasoning/reasoning_agent_reason_config.py,sha256=ip5feO_B6xPcmOspZl8gpBvgdU1XPHg2xWYdS6x8GUM,2874
 waldiez/models/agents/swarm_agent/__init__.py,sha256=nPTmjGJS4LCaEqRHX-TlK-AJHzrDN1hR8gfPzZfx1cU,1609
-waldiez/models/agents/swarm_agent/after_work.py,sha256=mKqk1P3AFrV4tveEkXLbMtdXJPTobUiFCkA3i12blcQ,6047
-waldiez/models/agents/swarm_agent/on_condition.py,sha256=VZ9efN_r3KF_s0-DME_SmHwlh0F1QnGpW7XA6IziFnc,3158
-waldiez/models/agents/swarm_agent/on_condition_available.py,sha256=jphgubLaNOn05XHEAi7BQ0J7asBwb_c9lum8SIc6XwM,4502
+waldiez/models/agents/swarm_agent/after_work.py,sha256=_mXYzw6_hrPqOqEIeNxulEYYynNKHmTK_Wdeb0OtfB0,6166
+waldiez/models/agents/swarm_agent/on_condition.py,sha256=WHOeQGEMCvp1R5NrpbmG5bIdMlI1N0SP15_Mccl40Lo,3212
+waldiez/models/agents/swarm_agent/on_condition_available.py,sha256=wZFknfoxcVvvDBvYrGFs39aYvIl3Zdv8EEW4prDQ1-s,4550
 waldiez/models/agents/swarm_agent/on_condition_target.py,sha256=g41yZ_EnNBJyfZIOyC51MIYNT6KCjF75UXBvfDWLOCM,981
 waldiez/models/agents/swarm_agent/swarm_agent.py,sha256=Uz2E8u0oolZCgXCxWlGv03lLjY36t2UWe_3YL03Sel8,2756
-waldiez/models/agents/swarm_agent/swarm_agent_data.py,sha256=rQ4Krebir7YRWRgF5Fdx56Kj2uXa2b1YJP0pL82_yi0,4053
-waldiez/models/agents/swarm_agent/update_system_message.py,sha256=aIHXqvpQnh8EFeYmpmFyvmsIUdVADbaU-tmUG2DVw44,4812
+waldiez/models/agents/swarm_agent/swarm_agent_data.py,sha256=WQkm68A8Bl7_q-_2_X3QCpM1aqhDZaIPkIUtPN3A9yo,4046
+waldiez/models/agents/swarm_agent/update_system_message.py,sha256=tOAEAFvPI85egVxexf-AonOtZqG-1FxnZmbYSFnQjkc,4858
 waldiez/models/agents/user_proxy/__init__.py,sha256=jBojxeNmG7B5P7mFg7Z10ETH6AHjkwe7-XIuxACpXSs,271
 waldiez/models/agents/user_proxy/user_proxy.py,sha256=QT18qFeQUmnzClqIs15jwILfG60dhva58eGrlcGamjI,1164
 waldiez/models/agents/user_proxy/user_proxy_data.py,sha256=ni9RTUmHHWPfBk_L7QTUkUkU6AopsHfGw3uYEOTtUzQ,926
 waldiez/models/chat/__init__.py,sha256=uivo-wYhIiu0OyqhiXeNdES4dRToPxQTNmSWGjkfH6A,1049
 waldiez/models/chat/chat.py,sha256=XL50OtIYSUFvkmOxkecMsz_sWqlh3crguv6mb8b1nr0,10602
 waldiez/models/chat/chat_data.py,sha256=a0ScDNtU_VgAMGzrR2Lj7mPZpY-yQKazYiiwDTfhe3A,12482
-waldiez/models/chat/chat_message.py,sha256=FyQCIVFl6Z1nlyCpDYHZ5YySN6pHSNbMSp4e56hodJQ,8306
+waldiez/models/chat/chat_message.py,sha256=2vnauAWUEUByqZFSlf-mrIPZGdik5W-_AZZSVWdlQnw,8344
 waldiez/models/chat/chat_nested.py,sha256=C1ygS7TeB5emRIIIs_xFabKgXzze0HTEO4CIexedGKw,4132
-waldiez/models/chat/chat_summary.py,sha256=RDeRw7vTV4Niw5_v_UZTOWQkIZUI0q9vdQ01Ja1isM4,2916
-waldiez/models/common/__init__.py,sha256=I_j63y68FCpSVDz9YXW7aKQsA-X3MOWDvNokyrVpv4w,493
-waldiez/models/common/base.py,sha256=WTqElKnOIoDUN9tfaVR4q3-U-DyyrkyV8x7296eZ8jE,1941
-waldiez/models/common/date_utils.py,sha256=hdnxYU8GKQTnucOnAfs1DAqER0oe59A5G3UVDktl-ro,344
-waldiez/models/common/dict_utils.py,sha256=ydUP9Am9QxQveqnDB3Rdt62FvTHeFfD4kTmCLXyutGY,1050
-waldiez/models/common/method_utils.py,sha256=CyTwRppegHviN-J_oj5XsPvX0edMSmEbLVeTkrG2HVo,7010
-waldiez/models/flow/__init__.py,sha256=3UdwoUZdkLr6G78tNTyaijoiwvmHZ-N0tY3G7S_xbRs,255
+waldiez/models/chat/chat_summary.py,sha256=O7mHkIVpoB1Mqzf585qqemjmB1ojj8-jlK6x8TGWQcA,2960
+waldiez/models/common/__init__.py,sha256=dOocZwZn8OHMNHK1WyiKXboi5BkB3zIoGkJErA6orWg,617
+waldiez/models/common/ag2_version.py,sha256=p9EF8N2zsb14YXrbZtMIWYLUGFB_Jnjbm4o6tODCw-Q,744
+waldiez/models/common/base.py,sha256=caFhCnUH41OYyx19Zl2gHXzhlZbl8Ewmpc5t7LpYhjY,1961
+waldiez/models/common/date_utils.py,sha256=NdVmShZjvhDVT-Y5rT8FgwQTHcDoljBJQauTsE7gwbE,437
+waldiez/models/common/dict_utils.py,sha256=xfZ6fFJXGyW2yiPiBcGO7WKqhAlCH1sMRp_Z4__xyr0,1143
+waldiez/models/common/method_utils.py,sha256=dRgkPVwIIYROXLou2qtFyDNlr5KSDX7XUo0YnsVKyvI,10284
+waldiez/models/flow/__init__.py,sha256=3UFbMMn_5c5AXs6ppkvhNeWb9jelru2lNHUjpEq0N68,309
 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=Bd8dWYlCenuU4VeusAL_qxkXzYSuxl9e3a7FhHs6UCY,5435
-waldiez/models/model/__init__.py,sha256=1cOyDThhZV8o7pZN1DaIhh12jWty_YMegoQ4Jg7uKyI,383
-waldiez/models/model/model.py,sha256=k31Tr5mjW4zmahBI52e0pBPnOSxXY0KgOgiNKvNVLQw,7139
-waldiez/models/model/model_data.py,sha256=EwObYtzPyxZqADdTeWjULNmb3dLgF91mzK3r1nxxJJk,3819
-waldiez/models/skill/__init__.py,sha256=kAyp9-LwBacI3ndQpU-dCGWjsjojRQy73zXsTddGAmg,306
-waldiez/models/skill/skill.py,sha256=6r8heHUgRYwfcBJbdsV5DywvTXcEeWPuJN4jzCh3-O4,4255
-waldiez/models/skill/skill_data.py,sha256=Spk3YxK0IgkXpJgk05DJwvZwUPrh-2l95o9PnS1WnoQ,908
+waldiez/models/flow/utils.py,sha256=X_jelP5lHuQt1OZE6sxLlhEKbkkt8jZgbkF79e9rQig,7140
+waldiez/models/model/__init__.py,sha256=0klSsHb8poHgtnVdCi-rDiSOb_yOofk9RPehjLQnGFc,482
+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/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
+waldiez/models/skill/skill_data.py,sha256=22ZrXHmjOy-7WAr8_QmmfW9RNIQiMUEBI8smucVC0hQ,1347
+waldiez/models/skill/skill_type.py,sha256=TUL5MADIwZRxODoViDWC65AmRLpi7ofqW-X90hm2XhA,271
 waldiez/running/__init__.py,sha256=hptO_sDFv0Alg-YJmLN-A49SeQ4YyWUMP8nBH21qO-o,633
-waldiez/running/environment.py,sha256=q4h3GWBSGR79F9lqLTqgGYh3xyt2xL-kPYDc65orryY,2540
+waldiez/running/environment.py,sha256=Qqg0n7cRie16vJNDCE6d23uBSn4UP8eiMqxnDeBDU5Q,3680
 waldiez/running/gen_seq_diagram.py,sha256=wIsbAddASKNb-pRQhCIhouWGLuhX2Xja5x5xHCCSqHk,5716
-waldiez/running/running.py,sha256=Uqe4CCt_Rple4OGpOqLDeSGNXbqssXtdIezm0_Sm-PA,10091
+waldiez/running/running.py,sha256=7rDA_-t__N_s5Ahb5qj6wHEM9GC6Wj5qzKm_qKeZMF4,9889
 waldiez/utils/__init__.py,sha256=M7ZqIInj2371ojAO73OuEPZ3fF17WZVbGE-37oyJwl0,415
 waldiez/utils/conflict_checker.py,sha256=v7BNDenJSHfPBfqb4PnxaVZwjPqzWlh5w31WC9uOJL4,1369
 waldiez/utils/flaml_warnings.py,sha256=jgdS8zwzuWPodTgm9nnx3wO5MrOUbR7P7IRj9-aq2qk,601
-waldiez/utils/pysqlite3_checker.py,sha256=LypcHlCMbzdC0QUfrr68gzmrEbwzvKLUn-riYiCKu70,7636
+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.3.11.dist-info/METADATA,sha256=olpy7txsUlYNHBlFo5-6fU9F1rRRk8_sgeNiw7hq0Hc,9422
-waldiez-0.3.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-waldiez-0.3.11.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
-waldiez-0.3.11.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
-waldiez-0.3.11.dist-info/licenses/NOTICE.md,sha256=lrKsUNrpE18LPSLPJY_kt2ZJFZJEwH55wPv5Axcheyc,133
-waldiez-0.3.11.dist-info/RECORD,,
+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/exporting/agent/utils/agent_class_name.py

@@ -1,36 +0,0 @@
-# SPDX-License-Identifier: Apache-2.0.
-# Copyright (c) 2024 - 2025 Waldiez and contributors.
-"""Get the agent class name."""
-
-from waldiez.models import WaldiezAgent
-
-
-# pylint: disable=too-many-return-statements
-def get_agent_class_name(agent: WaldiezAgent) -> str:
-    """Get the agent class name.
-
-    Parameters
-    ----------
-    agent : WaldiezAgent
-        The agent.
-
-    Returns
-    -------
-    str
-        The agent class name.
-    """
-    if agent.data.is_multimodal:
-        return "MultimodalConversableAgent"
-    if agent.agent_type == "assistant":
-        return "AssistantAgent"
-    if agent.agent_type == "user":
-        return "UserProxyAgent"
-    if agent.agent_type == "manager":
-        return "GroupChatManager"
-    if agent.agent_type == "rag_user":
-        return "RetrieveUserProxyAgent"
-    if agent.agent_type == "swarm":
-        return "SwarmAgent"
-    if agent.agent_type == "reasoning":
-        return "ReasoningAgent"
-    return "ConversableAgent"  # pragma: no cover

waldiez/exporting/agent/utils/agent_imports.py

@@ -1,55 +0,0 @@
-# SPDX-License-Identifier: Apache-2.0.
-# Copyright (c) 2024 - 2025 Waldiez and contributors.
-"""Get the imports needed for the agent."""
-
-from typing import Set
-
-
-def get_agent_imports(agent_class: str) -> Set[str]:
-    """Get the imports needed for the agent.
-
-    Parameters
-    ----------
-    agent_class : str
-        The agent class name.
-
-    Returns
-    -------
-    Set[str]
-        The imports needed for the agent.
-    """
-    imports = set(["import autogen"])
-    if agent_class == "AssistantAgent":
-        imports.add("from autogen import AssistantAgent")
-    elif agent_class == "UserProxyAgent":
-        imports.add("from autogen import UserProxyAgent")
-    elif agent_class == "GroupChatManager":
-        imports.add("from autogen import GroupChatManager")
-    elif agent_class == "RetrieveUserProxyAgent":
-        imports.add(
-            "from autogen.agentchat.contrib.retrieve_user_proxy_agent "
-            "import RetrieveUserProxyAgent"
-        )
-    elif agent_class == "MultimodalConversableAgent":
-        imports.add(
-            "from autogen.agentchat.contrib.multimodal_conversable_agent "
-            "import MultimodalConversableAgent"
-        )
-    elif agent_class == "SwarmAgent":
-        imports.add(
-            "from autogen import "
-            "AFTER_WORK, "
-            "ON_CONDITION, "
-            "UPDATE_SYSTEM_MESSAGE, "
-            "AfterWorkOption, "
-            "SwarmAgent, "
-            "SwarmResult"
-        )
-    elif agent_class == "ReasoningAgent":
-        # pylint: disable=line-too-long
-        imports.add(
-            "from autogen.agentchat.contrib.reasoning_agent import ReasoningAgent, visualize_tree"  # noqa: E501
-        )
-    else:
-        imports.add("from autogen import ConversableAgent")
-    return imports