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

waldiez/models/waldiez.py

@@ -12,15 +12,12 @@ from dataclasses import dataclass
 from pathlib import Path
 from typing import Any, Dict, Iterator, List, Optional, Tuple, Union
 
-from .agents import (
-    WaldiezAgent,
-    get_retrievechat_extra_requirements,
-)
+from .agents import WaldiezAgent, get_retrievechat_extra_requirements
 from .chat import WaldiezChat
 from .common import get_autogen_version
 from .flow import WaldiezFlow, get_flow_data
 from .model import WaldiezModel, get_models_extra_requirements
-from .skill import WaldiezSkill
+from .skill import WaldiezSkill, get_skills_extra_requirements
 
 
 @dataclass(frozen=True, slots=True)
@@ -155,6 +152,11 @@ class Waldiez:
         """Check if the flow has multimodal agents."""
         return any(agent.data.is_multimodal for agent in self.agents)
 
+    @property
+    def has_captain_agents(self) -> bool:
+        """Check if the flow has captain agents."""
+        return any(agent.agent_type == "captain" for agent in self.agents)
+
     @property
     def chats(self) -> List[Tuple[WaldiezChat, WaldiezAgent, WaldiezAgent]]:
         """Get the chats."""
@@ -243,8 +245,36 @@ class Waldiez:
             requirements.update(rag_extras)
         if self.has_multimodal_agents:
             requirements.add(f"pyautogen[lmm]=={autogen_version}")
+        if self.has_captain_agents:
+            # pysqlite3-binary might not get installed on windows
+            captain_extras = [
+                "chromadb",
+                "sentence-transformers",
+                "huggingface-hub",
+                # tools:
+                "pillow",
+                "markdownify",
+                "arxiv",
+                "pymupdf",
+                "wikipedia-api",
+                "easyocr",
+                "python-pptx",
+                "openai-whisper",
+                "pandas",
+                "scipy",
+            ]
+            requirements.update(captain_extras)
+        requirements.update(
+            get_models_extra_requirements(
+                self.models,
+                autogen_version=autogen_version,
+            )
+        )
         requirements.update(
-            get_models_extra_requirements(self.models, autogen_version)
+            get_skills_extra_requirements(
+                self.skills,
+                autogen_version=autogen_version,
+            )
         )
         return sorted(list(requirements))
 

waldiez/runner.py

@@ -14,7 +14,7 @@ import sys
 import tempfile
 from pathlib import Path
 from types import TracebackType
-from typing import TYPE_CHECKING, Dict, List, Optional, Type, Union
+from typing import TYPE_CHECKING, Dict, List, Optional, Set, Type, Union
 
 from .exporter import WaldiezExporter
 from .models.waldiez import Waldiez
@@ -30,6 +30,7 @@ from .running import (
     reset_env_vars,
     set_env_vars,
 )
+from .utils import check_pysqlite3
 
 if TYPE_CHECKING:
     from autogen import ChatResult  # type: ignore
@@ -135,13 +136,26 @@ class WaldiezRunner:
         """Get the running status."""
         return self._running
 
+    def gather_requirements(self) -> Set[str]:
+        """Gather extra requirements to install before running the flow.
+
+        Returns
+        -------
+        Set[str]
+            The extra requirements.
+        """
+        extra_requirements = set(
+            req for req in self.waldiez.requirements if req not in sys.modules
+        )
+        if self.waldiez.has_captain_agents:
+            check_pysqlite3()
+        return extra_requirements
+
     def install_requirements(self) -> None:
         """Install the requirements for the flow."""
         self._called_install_requirements = True
         printer = get_printer()
-        extra_requirements = set(
-            req for req in self.waldiez.requirements if req not in sys.modules
-        )
+        extra_requirements = self.gather_requirements()
         if extra_requirements:
             install_requirements(extra_requirements, printer)
             refresh_environment()
@@ -150,9 +164,7 @@ class WaldiezRunner:
         """Install the requirements for the flow asynchronously."""
         self._called_install_requirements = True
         printer = get_printer()
-        extra_requirements = set(
-            req for req in self.waldiez.requirements if req not in sys.modules
-        )
+        extra_requirements = self.gather_requirements()
         if extra_requirements:
             await a_install_requirements(extra_requirements, printer)
             refresh_environment()

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.12.dist-info → waldiez-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.3.12
+Version: 0.4.0
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/waldiez/python
 Project-URL: repository, https://github.com/waldiez/python.git
@@ -25,9 +25,10 @@ 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
@@ -44,36 +45,38 @@ 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: psycopg[binary]>=3.2.4; 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[cohere]==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[ollama]==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: pydantic-ai==0.0.21; 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'
+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'
@@ -83,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.2; 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.14.1; 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.12; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.4.0; extra == 'jupyter'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.3.12; 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.12.dist-info → waldiez-0.4.0.dist-info}/RECORD

@@ -1,19 +1,19 @@
 waldiez/__init__.py,sha256=cXWdy5P9i_x6fHwlL5DRaIhmpfnhabo5GjASStlcLWw,819
 waldiez/__main__.py,sha256=0dYzNrQbovRwQQvmZC6_1FDR1m71SUIOkTleO5tBnBw,203
-waldiez/_version.py,sha256=8g1vIaGTCQVgdq8Z3RCbEIHIp4T5eWWWaTpPmb13Ui0,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=mLRocSnhXMTK0KPXN_F_tKTrSXBkg9uyAGDkhPPsPhs,10329
+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=jA-UjyK0AMr4yTjlXRIv9CQPytzNOGyR44iZ6iZ84Vo,7355
+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
@@ -30,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
@@ -41,31 +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=Kvaei1uhMfBBdzOTsnxEW-f-Q41OKj9kpB4e5rCjiso,18117
-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=GxEBfX6cUGFCMdxiu_hp5TLIhIC4ohczxA1A01mHnOo,2746
-waldiez/exporting/flow/utils/flow_content.py,sha256=RIMtpBFZCJf2G67ipOa3Co5OQinaobYvthqisgir5sQ,4505
+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/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=V6rrdD9P9mD8bVYWldha53gfZocaWqaadQTUI6IoQT0,8928
+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=us1qPNCwKdUao9UUVXDHbcjuUUA5yexWEQkozvWLUDY,1044
-waldiez/models/agents/agent/agent.py,sha256=Qfyl1VibDyg7Zz22gjQ6vot702TtnEHjdlKmEX9jpRQ,8269
+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
@@ -75,70 +76,72 @@ waldiez/models/agents/assistant/__init__.py,sha256=y2s2F2e7njMcmwye1UOUULHSvU3Vs
 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=ZjB0JYQ2SbWeP4QeR5OflKGvkoBxAezzMwfTtaWE_-A,1208
+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=a61rn0Nl0V-BFvJJxx9CDKc2CTxSsErnFPQ4QE3IYwQ,565
+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=CyTwRppegHviN-J_oj5XsPvX0edMSmEbLVeTkrG2HVo,7010
+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=X_jelP5lHuQt1OZE6sxLlhEKbkkt8jZgbkF79e9rQig,7140
 waldiez/models/model/__init__.py,sha256=0klSsHb8poHgtnVdCi-rDiSOb_yOofk9RPehjLQnGFc,482
-waldiez/models/model/extra_requirements.py,sha256=XAjZs-P2g_FjU0Jb27NYrwLi8kyVn-XIvLCGr1UrKAU,1595
+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=8GZerOBu5eGvpIfZm4aypFiiCUxz-SQtIfVLr3E59rg,3753
-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/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.12.dist-info/METADATA,sha256=ggItUOmudOaRVTAAGSn0ysxGF_i0TUe3ZmRbk4d8aKE,9661
-waldiez-0.3.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-waldiez-0.3.12.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
-waldiez-0.3.12.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
-waldiez-0.3.12.dist-info/licenses/NOTICE.md,sha256=lrKsUNrpE18LPSLPJY_kt2ZJFZJEwH55wPv5Axcheyc,133
-waldiez-0.3.12.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,,