waldiez 0.4.3__tar.gz → 0.4.4__tar.gz

{waldiez-0.4.3 → waldiez-0.4.4}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.4.3
-Summary: waldiez
+Version: 0.4.4
+Summary: Make AI Agents Collaborate: Drag, Drop, and Orchestrate with Waldiez.
 Project-URL: homepage, https://waldiez.io
 Project-URL: repository, https://github.com/waldiez/python.git
 Project-URL: documentation, https://waldiez.github.io/python/
@@ -28,20 +28,27 @@ Requires-Dist: asyncer==0.0.8
 Requires-Dist: graphviz==0.20.3
 Requires-Dist: httpx<1
 Requires-Dist: jupytext
-Requires-Dist: numpy<=2.2.4
+Requires-Dist: nest-asyncio==1.6.0
+Requires-Dist: numpy<=2.2.5
 Requires-Dist: pandas>=2
 Requires-Dist: parso==0.8.4
-Requires-Dist: pyautogen[openai]==0.8.6
+Requires-Dist: pyautogen[openai]==0.8.7
 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.10; extra == 'ag2-extras'
-Requires-Dist: embedchain; extra == 'ag2-extras'
+Requires-Dist: chromadb>=0.5.10; (sys_platform != 'win32') and extra == 'ag2-extras'
+Requires-Dist: chromadb>=0.5.10; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
+Requires-Dist: embedchain; (sys_platform != 'win32') and extra == 'ag2-extras'
+Requires-Dist: embedchain; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
+Requires-Dist: google-api-python-client<3.0,>=2.163.0; extra == 'ag2-extras'
+Requires-Dist: google-auth-httplib2<0.3,>=0.2.0; extra == 'ag2-extras'
+Requires-Dist: google-auth-oauthlib<2.0,>=1.2.1; 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: mcp<1.6,>=1.4.0; extra == 'ag2-extras'
 Requires-Dist: pgvector>=0.4.0; extra == 'ag2-extras'
 Requires-Dist: protobuf>=4.25.3; extra == 'ag2-extras'
 Requires-Dist: psycopg==3.2.6; (sys_platform == 'linux') and extra == 'ag2-extras'
@@ -49,40 +56,48 @@ Requires-Dist: psycopg==3.2.6; (sys_platform == 'win32' and platform_machine ==
 Requires-Dist: psycopg==3.2.6; (sys_platform == 'win32' and platform_machine == 'ARM64') and extra == 'ag2-extras'
 Requires-Dist: psycopg==3.2.6; (sys_platform == 'win32' and platform_machine == 'aarch64') and extra == 'ag2-extras'
 Requires-Dist: psycopg==3.2.6; (sys_platform == 'win32' and platform_machine == 'arm64') and extra == 'ag2-extras'
+Requires-Dist: psycopg>=3.2.6; (sys_platform == 'win32') and extra == 'ag2-extras'
 Requires-Dist: psycopg[binary]==3.2.6; (sys_platform != 'linux' and platform_machine != 'arm64' and platform_machine != 'ARM64' and platform_machine != 'aarch64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
-Requires-Dist: psycopg[binary]>=3.2.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[anthropic]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[bedrock]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[cohere]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[gemini]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[groq]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[interop-crewai]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[interop-langchain]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[lmm]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[mistral]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[neo4j]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[ollama]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[together]==0.8.6; extra == 'ag2-extras'
-Requires-Dist: pyautogen[websurfer]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: psycopg[binary]>=3.2.6; (sys_platform != 'win32') and extra == 'ag2-extras'
+Requires-Dist: pyautogen[anthropic]==0.8.7; extra == 'ag2-extras'
+Requires-Dist: pyautogen[bedrock]==0.8.7; extra == 'ag2-extras'
+Requires-Dist: pyautogen[cohere]==0.8.7; extra == 'ag2-extras'
+Requires-Dist: pyautogen[gemini]==0.8.7; (sys_platform != 'win32') and extra == 'ag2-extras'
+Requires-Dist: pyautogen[gemini]==0.8.7; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
+Requires-Dist: pyautogen[groq]==0.8.7; extra == 'ag2-extras'
+Requires-Dist: pyautogen[interop-crewai]==0.8.7; extra == 'ag2-extras'
+Requires-Dist: pyautogen[interop-langchain]==0.8.7; extra == 'ag2-extras'
+Requires-Dist: pyautogen[lmm]==0.8.7; extra == 'ag2-extras'
+Requires-Dist: pyautogen[mistral]==0.8.7; extra == 'ag2-extras'
+Requires-Dist: pyautogen[neo4j]==0.8.7; (sys_platform != 'win32') and extra == 'ag2-extras'
+Requires-Dist: pyautogen[neo4j]==0.8.7; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
+Requires-Dist: pyautogen[ollama]==0.8.7; extra == 'ag2-extras'
+Requires-Dist: pyautogen[together]==0.8.7; (sys_platform != 'win32') and extra == 'ag2-extras'
+Requires-Dist: pyautogen[together]==0.8.7; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
+Requires-Dist: pyautogen[websurfer]==0.8.7; 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: pysqlite3-binary==0.5.4; (sys_platform == 'linux' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
-Requires-Dist: qdrant-client[fastembed]; extra == 'ag2-extras'
-Requires-Dist: sentence-transformers; extra == 'ag2-extras'
+Requires-Dist: qdrant-client[fastembed]; (sys_platform != 'win32') and extra == 'ag2-extras'
+Requires-Dist: qdrant-client[fastembed]; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
+Requires-Dist: sentence-transformers; (sys_platform == 'linux') and extra == 'ag2-extras'
+Requires-Dist: wikipedia-api<1.0,>=0.8.1; extra == 'ag2-extras'
 Provides-Extra: dev
 Requires-Dist: autoflake==2.3.1; extra == 'dev'
 Requires-Dist: bandit==1.8.3; extra == 'dev'
 Requires-Dist: black[jupyter]==25.1.0; extra == 'dev'
 Requires-Dist: flake8==7.2.0; extra == 'dev'
 Requires-Dist: isort==6.0.1; extra == 'dev'
+Requires-Dist: jupyter-server==2.15.0; extra == 'dev'
+Requires-Dist: jupyterlab>=4.3.0; extra == 'dev'
 Requires-Dist: mypy==1.15.0; extra == 'dev'
 Requires-Dist: pandas-stubs==2.2.3.250308; extra == 'dev'
 Requires-Dist: pre-commit==4.2.0; extra == 'dev'
 Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
 Requires-Dist: pylint==3.3.6; extra == 'dev'
 Requires-Dist: python-dotenv==1.1.0; extra == 'dev'
-Requires-Dist: ruff==0.11.5; extra == 'dev'
+Requires-Dist: ruff==0.11.7; extra == 'dev'
 Requires-Dist: toml==0.10.2; (python_version <= '3.10') and extra == 'dev'
 Requires-Dist: types-pyyaml==6.0.12.20250402; extra == 'dev'
 Requires-Dist: types-toml==0.10.8.20240310; extra == 'dev'
@@ -92,16 +107,19 @@ 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.11; extra == 'docs'
+Requires-Dist: mkdocs-material==9.6.12; 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.16.10; extra == 'docs'
 Requires-Dist: mkdocstrings[crystal,python]==0.29.1; extra == 'docs'
 Provides-Extra: jupyter
+Requires-Dist: jupyter-server==2.15.0; extra == 'jupyter'
 Requires-Dist: jupyterlab>=4.3.0; extra == 'jupyter'
-Requires-Dist: waldiez-jupyter==0.4.3; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.4.4; extra == 'jupyter'
+Provides-Extra: runner
+Requires-Dist: waldiez-runner==0.4.4; extra == 'runner'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.4.3; extra == 'studio'
+Requires-Dist: waldiez-studio==0.4.4; extra == 'studio'
 Provides-Extra: test
 Requires-Dist: pytest-asyncio==0.26.0; extra == 'test'
 Requires-Dist: pytest-cov==6.1.1; extra == 'test'

{waldiez-0.4.3 → waldiez-0.4.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "waldiez"
-description = 'waldiez'
+description = 'Make AI Agents Collaborate: Drag, Drop, and Orchestrate with Waldiez.'
 authors = [
   { name = 'Panagiotis Kasnesis', email = 'pkasnesis@waldiez.io' },
   { name= 'Lazaros Toumanidis', email= 'laztoum@waldiez.io' },
@@ -27,11 +27,16 @@ dependencies =[
     "aiofiles==24.1.0",
     "aiosqlite==0.21.0",
     "asyncer==0.0.8",
-    "numpy<=2.2.4",
+    "numpy<=2.2.5",
+    "nest_asyncio==1.6.0",
     "graphviz==0.20.3",
     "pandas>=2",
-    "pyautogen[openai]==0.8.6",
+    "pyautogen[openai]==0.8.7",
     "jupytext",
+    # "rpds-py @ git+https://github.com/crate-py/rpds.git@v0.24.0 ;sys_platform == 'win32' and platform_machine == 'arm64'",
+    # "rpds-py @ git+https://github.com/crate-py/rpds.git@v0.24.0 ;sys_platform == 'win32' and platform_machine == 'ARM64'",
+    # "rpds-py @ git+https://github.com/crate-py/rpds.git@v0.24.0 ;sys_platform == 'win32' and platform_machine == 'aarch64'",
+    # "rpds-py @ git+https://github.com/crate-py/rpds.git@v0.24.0 ;sys_platform == 'win32' and platform_machine == 'AARCH64'",
     "parso==0.8.4",
     "pydantic>=2.10.2,<3",
     "typer>=0.9,<0.16",
@@ -44,6 +49,9 @@ repository = 'https://github.com/waldiez/python.git'
 documentation = "https://waldiez.github.io/python/"
 issues = "https://github.com/waldiez/python/issues"
 
+[tool.hatch.metadata]
+allow-direct-references = true
+
 [tool.hatch.build.targets.sdist]
 packages = ["waldiez", "waldiez.*"]
 exclude = [ "**/example.py", ".gitignore" ]
@@ -54,17 +62,23 @@ exclude = [ "**/example.py" ]
 
 [project.optional-dependencies]
 studio = [
-    "waldiez_studio==0.4.3",
+    "waldiez_studio==0.4.4",
+]
+runner = [
+    "waldiez_runner==0.4.4",
 ]
 jupyter = [
-    "waldiez_jupyter==0.4.3",
+    "waldiez_jupyter==0.4.4",
     'jupyterlab>=4.3.0',
+    "jupyter_server==2.15.0",
 ]
 dev = [
     'autoflake==2.3.1',
     'bandit==1.8.3',
     'black[jupyter]==25.1.0',
     'flake8==7.2.0',
+    'jupyterlab>=4.3.0',
+    "jupyter_server==2.15.0",
     'isort==6.0.1',
     'mypy==1.15.0',
     'pandas-stubs==2.2.3.250308',
@@ -72,7 +86,7 @@ dev = [
     'pydocstyle==6.3.0',
     'pylint==3.3.6',
     'python-dotenv==1.1.0',
-    'ruff==0.11.5',
+    'ruff==0.11.7',
     'types-pyyaml==6.0.12.20250402',
     'yamllint==1.37.0',
     'toml==0.10.2; python_version <= "3.10"',
@@ -80,43 +94,55 @@ dev = [
 ]
 ag2_extras =[
     # models
-    'pyautogen[anthropic]==0.8.6',
-    'pyautogen[bedrock]==0.8.6',
-    'pyautogen[cohere]==0.8.6',
-    'pyautogen[gemini]==0.8.6',
-    'pyautogen[groq]==0.8.6',
-    'pyautogen[lmm]==0.8.6',
-    'pyautogen[mistral]==0.8.6',
-    'pyautogen[ollama]==0.8.6',
-    'pyautogen[together]==0.8.6',
+    'pyautogen[anthropic]==0.8.7',
+    'pyautogen[bedrock]==0.8.7',
+    'pyautogen[cohere]==0.8.7',
+    'pyautogen[gemini]==0.8.7 ;sys_platform != "win32"',
+    'pyautogen[gemini]==0.8.7 ;sys_platform == "win32" and platform_machine != "arm64" and platform_machine != "aarch64" and platform_machine != "ARM64" and platform_machine != "AARCH64"',
+    'pyautogen[groq]==0.8.7',
+    'pyautogen[lmm]==0.8.7',
+    'pyautogen[mistral]==0.8.7',
+    'pyautogen[ollama]==0.8.7',
+    'pyautogen[together]==0.8.7 ;sys_platform != "win32"',
+    'pyautogen[together]==0.8.7 ;sys_platform == "win32" and platform_machine != "arm64" and platform_machine != "aarch64" and platform_machine != "ARM64" and platform_machine != "AARCH64"',
     # utils
-    'pyautogen[interop-crewai]==0.8.6',
-    'pyautogen[interop-langchain]==0.8.6',
-    'pyautogen[neo4j]==0.8.6',
-    'pyautogen[websurfer]==0.8.6',
-    'chromadb>=0.5.10',
+    'pyautogen[interop-crewai]==0.8.7',
+    'pyautogen[interop-langchain]==0.8.7',
+    "pyautogen[neo4j]==0.8.7 ;sys_platform != 'win32'",
+    "pyautogen[neo4j]==0.8.7 ;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
+    'pyautogen[websurfer]==0.8.7',
+    "chromadb>=0.5.10 ;sys_platform != 'win32'",
+    "chromadb>=0.5.10 ;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
     'pgvector>=0.4.0',
-    "psycopg==3.2.6; sys_platform == 'linux'",
-    "psycopg==3.2.6; sys_platform == 'win32' and platform_machine == 'arm64'",
-    "psycopg==3.2.6; sys_platform == 'win32' and platform_machine == 'ARM64'",
-    "psycopg==3.2.6; sys_platform == 'win32' and platform_machine == 'aarch64'",
-    "psycopg==3.2.6; sys_platform == 'win32' and platform_machine == 'AARCH64'",
-    "psycopg[binary]==3.2.6; sys_platform != 'linux' and platform_machine != 'arm64' and platform_machine != 'ARM64' and platform_machine != 'aarch64' and platform_machine != 'AARCH64'",
+    "psycopg==3.2.6 ;sys_platform == 'linux'",
+    "psycopg==3.2.6 ;sys_platform == 'win32' and platform_machine == 'arm64'",
+    "psycopg==3.2.6 ;sys_platform == 'win32' and platform_machine == 'ARM64'",
+    "psycopg==3.2.6 ;sys_platform == 'win32' and platform_machine == 'aarch64'",
+    "psycopg==3.2.6 ;sys_platform == 'win32' and platform_machine == 'AARCH64'",
+    "psycopg[binary]==3.2.6 ;sys_platform != 'linux' and platform_machine != 'arm64' and platform_machine != 'ARM64' and platform_machine != 'aarch64' and platform_machine != 'AARCH64'",
     'protobuf>=4.25.3',
     'pypdf',
-    "pysqlite3-binary==0.5.4; sys_platform == 'linux' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
+    "pysqlite3-binary==0.5.4 ;sys_platform == 'linux' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
     'ipython',
     'markdownify',
     'beautifulsoup4',
-    'sentence-transformers',
+    "sentence-transformers ;sys_platform == 'linux'",
     'huggingface-hub',
-    'qdrant-client[fastembed]',
+    "qdrant-client[fastembed] ;sys_platform != 'win32'",
+    "qdrant-client[fastembed] ;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
     'pgvector>=0.4.0',
-    'psycopg[binary]>=3.2.6',
-    'embedchain',
+    "psycopg[binary]>=3.2.6 ;sys_platform != 'win32'",
+    "psycopg>=3.2.6 ;sys_platform == 'win32'",
+    "embedchain ;sys_platform != 'win32'",
+    "embedchain ;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
     'langchain-community>=0.3.12,<1',
     'pydantic-ai>=0.0.21',
     'pymongo>=4.11',
+    'wikipedia-api>=0.8.1, <1.0',
+    'google-api-python-client>=2.163.0,<3.0',
+    'google-auth-httplib2>=0.2.0,<0.3',
+    'google-auth-oauthlib>=1.2.1,<2.0',
+    'mcp>=1.4.0,<1.6',
 ]
 test = [
     'pytest==8.3.5',
@@ -133,7 +159,7 @@ docs = [
   'mkdocs==1.6.1',
   'mkdocs-jupyter==0.25.1',
   'mkdocs-macros-plugin==1.3.7',
-  'mkdocs-material==9.6.11',
+  'mkdocs-material==9.6.12',
   'mkdocs-minify-html-plugin==0.3.1',
   'mkdocstrings[crystal,python]==0.29.1',
   'mkdocstrings-python==1.16.10'

{waldiez-0.4.3 → waldiez-0.4.4}/waldiez/_version.py

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

{waldiez-0.4.3 → waldiez-0.4.4}/waldiez/running/environment.py

@@ -3,7 +3,7 @@
 # pylint: disable=import-outside-toplevel,reimported
 """Environment related utilities."""
 
-import importlib.util
+import importlib
 import os
 import site
 import sys
@@ -20,7 +20,8 @@ def in_virtualenv() -> bool:
         True if inside a virtualenv, False otherwise.
     """
     return hasattr(sys, "real_prefix") or (
-        hasattr(sys, "base_prefix") and sys.base_prefix != sys.prefix
+        hasattr(sys, "base_prefix")
+        and os.path.realpath(sys.base_prefix) != os.path.realpath(sys.prefix)
     )
 
 

{waldiez-0.4.3 → waldiez-0.4.4}/waldiez/running/running.py

@@ -120,20 +120,34 @@ def install_requirements(
     requirements_string = ", ".join(extra_requirements)
     printer(f"Installing requirements: {requirements_string}")
     pip_install = [sys.executable, "-m", "pip", "install"]
-    if not in_virtualenv():
+    if not in_virtualenv():  # it should
+        # if not, let's try to install as user
+        # not sure if --break-system-packages is safe
+        # but it might fail if we don't
+        break_system_packages = os.environ.get("PIP_BREAK_SYSTEM_PACKAGES", "")
+        os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = "1"
         pip_install.append("--user")
     pip_install.extend(extra_requirements)
-    with subprocess.Popen(
-        pip_install,
-        stdout=subprocess.PIPE,
-        stderr=subprocess.PIPE,
-    ) as proc:
-        if proc.stdout:
-            for line in io.TextIOWrapper(proc.stdout, encoding="utf-8"):
-                printer(line.strip())
-        if proc.stderr:
-            for line in io.TextIOWrapper(proc.stderr, encoding="utf-8"):
-                printer(line.strip())
+    # pylint: disable=too-many-try-statements
+    try:
+        with subprocess.Popen(
+            pip_install,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+        ) as proc:
+            if proc.stdout:
+                for line in io.TextIOWrapper(proc.stdout, encoding="utf-8"):
+                    printer(line.strip())
+            if proc.stderr:
+                for line in io.TextIOWrapper(proc.stderr, encoding="utf-8"):
+                    printer(line.strip())
+    finally:
+        if not in_virtualenv():
+            # restore the old env var
+            if break_system_packages:
+                os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = break_system_packages
+            else:
+                del os.environ["PIP_BREAK_SYSTEM_PACKAGES"]
 
 
 async def a_install_requirements(
@@ -152,19 +166,29 @@ async def a_install_requirements(
     printer(f"Installing requirements: {requirements_string}")
     pip_install = [sys.executable, "-m", "pip", "install"]
     if not in_virtualenv():
-        pip_install.extend(["--user", "--break-system-packages"])
+        break_system_packages = os.environ.get("PIP_BREAK_SYSTEM_PACKAGES", "")
+        os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = "1"
+        pip_install.extend(["--user"])
     pip_install.extend(extra_requirements)
-    proc = await asyncio.create_subprocess_exec(
-        *pip_install,
-        stdout=asyncio.subprocess.PIPE,
-        stderr=asyncio.subprocess.PIPE,
-    )
-    if proc.stdout:
-        async for line in proc.stdout:
-            printer(line.decode().strip())
-    if proc.stderr:
-        async for line in proc.stderr:
-            printer(line.decode().strip())
+    # pylint: disable=too-many-try-statements
+    try:
+        proc = await asyncio.create_subprocess_exec(
+            *pip_install,
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
+        )
+        if proc.stdout:
+            async for line in proc.stdout:
+                printer(line.decode().strip())
+        if proc.stderr:
+            async for line in proc.stderr:
+                printer(line.decode().strip())
+    finally:
+        if not in_virtualenv():
+            if break_system_packages:
+                os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = break_system_packages
+            else:
+                del os.environ["PIP_BREAK_SYSTEM_PACKAGES"]
 
 
 def after_run(

{waldiez-0.4.3 → waldiez-0.4.4}/waldiez/utils/cli_extras/__init__.py

@@ -7,6 +7,7 @@
 import typer
 
 from .jupyter import add_jupyter_cli
+from .runner import add_runner_cli
 from .studio import add_studio_cli
 
 
@@ -24,6 +25,7 @@ def add_cli_extras(app: typer.Typer) -> None:
         The app with the extra commands added
     """
     add_jupyter_cli(app)
+    add_runner_cli(app)
     add_studio_cli(app)
 
 

waldiez-0.4.4/waldiez/utils/cli_extras/runner.py

@@ -0,0 +1,37 @@
+# SPDX-License-Identifier: Apache-2.0.
+# Copyright (c) 2024 - 2025 Waldiez and contributors.
+# pylint: skip-file
+# isort: skip_file
+# flake8: noqa: E501,E402
+"""Waldiez-runner extra typer commands for CLI."""
+
+from typing import Any, Callable
+
+import typer
+from typer.models import CommandInfo
+
+HAVE_RUNNER = False
+runner_app: Callable[..., Any] | None = None
+
+try:
+    from waldiez_runner.cli import run  # type: ignore[unused-ignore, import-not-found, import-untyped]
+
+    runner_app = run
+
+    HAVE_RUNNER = True
+except BaseException:
+    pass
+
+
+def add_runner_cli(app: typer.Typer) -> None:
+    """Add runner command to the app if available.
+
+    Parameters
+    ----------
+    app : typer.Typer
+        The Typer app to add the runner command to.
+    """
+    if HAVE_RUNNER:
+        app.registered_commands.append(
+            CommandInfo(name="serve", callback=runner_app)
+        )