waldiez 0.5.10__py3-none-any.whl → 0.6.0__py3-none-any.whl

waldiez/ws/server.py

@@ -1,5 +1,11 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
+# pyright: reportMissingImports=false,reportUnknownVariableType=false
+# pyright: reportPossiblyUnboundVariable=false,reportUnknownMemberType=false
+# pyright: reportUnknownParameterType=false,reportUnknownArgumentType=false
+# pyright: reportAttributeAccessIssue=false
+# pylint: disable=import-error,line-too-long
+# flake8: noqa: E501
 """WebSocket server implementation for Waldiez."""
 
 import asyncio
@@ -12,16 +18,42 @@ import uuid
 from pathlib import Path
 from typing import Any, Sequence
 
-import websockets
-from websockets.exceptions import ConnectionClosed, WebSocketException
-
 from .client_manager import ClientManager
 from .errors import ErrorHandler, MessageParsingError, ServerOverloadError
 from .models import ConnectionNotification
-from .reloader import create_file_watcher
 from .session_manager import SessionManager
 from .utils import get_available_port, is_port_available
 
+HAS_WATCHDOG = False
+try:
+    from .reloader import create_file_watcher
+
+    HAS_WATCHDOG = True  # pyright: ignore
+except ImportError:
+    # pylint: disable=unused-argument,missing-param-doc,missing-return-doc
+    # noinspection PyUnusedLocal
+    def create_file_watcher(*args: Any, **kwargs: Any) -> Any:  # type: ignore
+        """No file watcher available."""
+
+
+HAS_WEBSOCKETS = False
+try:
+    import websockets  # type: ignore[unused-ignore, unused-import, import-not-found, import-untyped]
+    from websockets.exceptions import (  # type: ignore[unused-ignore, unused-import, import-not-found, import-untyped]
+        ConnectionClosed,
+        WebSocketException,
+    )
+
+    HAS_WEBSOCKETS = True  # pyright: ignore
+except ImportError:
+    from ._mock import (  # type: ignore[no-redef, unused-ignore, unused-import, import-not-found, import-untyped]
+        websockets,
+    )
+
+    ConnectionClosed = websockets.ConnectionClosed  # type: ignore[no-redef,unused-ignore,unused-import,import-not-found,import-untyped,misc]
+    WebSocketException = websockets.WebSocketException  # type: ignore[no-redef,unused-ignore,unused-import,import-not-found,import-untyped,misc]
+
+
 logger = logging.getLogger(__name__)
 
 CWD = Path.cwd()
@@ -35,6 +67,7 @@ class WaldiezWsServer:
         self,
         host: str = "localhost",
         port: int = 8765,
+        auto_reload: bool = False,
         workspace_dir: Path = CWD,
         max_clients: int = 1,
         allowed_origins: Sequence[re.Pattern[str]] | None = None,
@@ -48,11 +81,13 @@ class WaldiezWsServer:
             Server host address
         port : int
             Server port
+        auto_reload : bool
+            Enable automatic reloading of the server on code changes
         workspace_dir : Path
             Path to the workspace directory
         max_clients : int
             Maximum number of concurrent clients (default: 1)
-        allowed_origins : list[str] | None
+        allowed_origins : Sequence[re.Pattern[str]] | None
             List of allowed origins for CORS (default: None)
         ping_interval : float | None
             Ping interval in seconds
@@ -69,6 +104,7 @@ class WaldiezWsServer:
         """
         self.host = host
         self.port = port
+        self.auto_reload = auto_reload and HAS_WATCHDOG
         self.workspace_dir = workspace_dir
         self.max_clients = max_clients
         self.allowed_origins = allowed_origins
@@ -148,7 +184,8 @@ class WaldiezWsServer:
             )
 
             # Message handling loop
-            async for raw_message in websocket:
+            # noinspection PyTypeChecker
+            async for raw_message in websocket:  # pyright: ignore
                 try:
                     # Parse message
                     if isinstance(raw_message, bytes):
@@ -230,7 +267,7 @@ class WaldiezWsServer:
 
         await self.session_manager.start()
         # Check port availability
-        if not is_port_available(self.port):
+        if not self.auto_reload and not is_port_available(self.port):
             logger.warning("Port %d is not available", self.port)
             self.port = get_available_port()
             logger.info("Using port %d", self.port)
@@ -239,6 +276,7 @@ class WaldiezWsServer:
         # pylint: disable=too-many-try-statements,broad-exception-caught
         try:
             # Create server
+            # noinspection PyTypeChecker
             self.server = await websockets.serve(
                 self._handle_client,
                 self.host,
@@ -401,6 +439,7 @@ async def run_server(
     server = WaldiezWsServer(
         host=host,
         port=port,
+        auto_reload=auto_reload,
         workspace_dir=workspace_dir,
         **server_kwargs,
     )
@@ -420,7 +459,7 @@ async def run_server(
 
     # Set up auto-reload if requested
     file_watcher = None
-    if auto_reload:
+    if auto_reload and HAS_WATCHDOG:
         # pylint: disable=import-outside-toplevel,too-many-try-statements
         try:
             # Determine watch directories

waldiez/ws/utils.py

@@ -1,5 +1,9 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
+# pylint: disable=import-error,line-too-long
+# pyright: reportUnknownMemberType=false,reportUnknownVariableType=false
+# pyright: reportUnknownArgumentType=false,reportAttributeAccessIssue=false
+# flake8: noqa: E501
 """Utilities for WebSocket server management."""
 
 import asyncio
@@ -11,7 +15,11 @@ from contextlib import closing
 from dataclasses import asdict, dataclass
 from typing import TYPE_CHECKING, Any
 
-import websockets
+try:
+    import websockets  # type: ignore[unused-ignore, unused-import, import-not-found, import-untyped] # noqa
+except ImportError:  # pragma: no cover
+    from ._mock import websockets  # type: ignore[no-redef,unused-ignore]
+
 
 if TYPE_CHECKING:
     from .server import WaldiezWsServer
@@ -329,7 +337,10 @@ async def test_server_connection(
     try:
         uri = f"ws://{host}:{port}"
 
-        async with websockets.connect(uri, ping_interval=None) as websocket:
+        async with websockets.connect(
+            uri,
+            ping_interval=None,
+        ) as websocket:
             # Send ping message
             ping_msg = {"action": "ping"}
             await websocket.send(json.dumps(ping_msg))
@@ -350,6 +361,7 @@ async def test_server_connection(
     return result
 
 
+# noinspection PyBroadException
 def is_port_available(port: int) -> bool:
     """Check if the port is available.
 
@@ -363,13 +375,26 @@ def is_port_available(port: int) -> bool:
     bool
         True if port is available
     """
+    # Check IPv4
     try:
         with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as sock:
             sock.bind(("", port))
-            return True
-    except OSError:
+    except BaseException:  # pylint: disable=broad-exception-caught
         return False
 
+    # Check IPv6
+    try:  # pragma: no cover
+        with closing(
+            socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
+        ) as sock:
+            # Disable dual-stack to only check IPv6
+            sock.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_V6ONLY, 1)
+            sock.bind(("", port))
+    except BaseException:  # pylint: disable=broad-exception-caught
+        return False
+
+    return True
+
 
 def get_available_port() -> int:
     """Get an available port.

{waldiez-0.5.10.dist-info → waldiez-0.6.0.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.5.10
+Version: 0.6.0
 Dynamic: Keywords
 Summary: Make AG2 Agents Collaborate: Drag, Drop, and Orchestrate with Waldiez.
 Project-URL: Homepage, https://waldiez.io
 Project-URL: Bug Tracker, https://github.com/waldiez/waldiez/issues
 Project-URL: Repository, https://github.com/waldiez/waldiez.git
-Author-email: Panagiotis Kasnesis <pkasnesis@waldiez.io>, Lazaros Toumanidis <laztoum@waldiez.io>, Stella Ioannidou <stella@waldiez.io>
+Author-email: Panagiotis Kasnesis <pkasnesis@waldiez.io>, Lazaros Toumanidis <laztoum@waldiez.io>, Christos Chatzigeorgiou <chrihatz@waldiez.io>
 License-File: LICENSE
 License-File: NOTICE.md
 Classifier: Development Status :: 4 - Beta
@@ -36,18 +36,16 @@ Requires-Dist: graphviz<=0.21
 Requires-Dist: httpx<1
 Requires-Dist: jupytext
 Requires-Dist: nest-asyncio==1.6.0
-Requires-Dist: numpy<=2.3.2
+Requires-Dist: numpy<=2.3.3
 Requires-Dist: pandas>=2
-Requires-Dist: parso==0.8.4
+Requires-Dist: parso==0.8.5
 Requires-Dist: pillow
 Requires-Dist: pip>=25.1.1
-Requires-Dist: platformdirs==4.3.8
+Requires-Dist: platformdirs==4.4.0
 Requires-Dist: psutil==7.0.0
 Requires-Dist: pydantic<3,>=2.10.2
-Requires-Dist: rpds-py==0.27.0
+Requires-Dist: rpds-py==0.27.1
 Requires-Dist: typer<1,>=0.9.0
-Requires-Dist: watchdog==6.0.0
-Requires-Dist: websockets==15.0.1
 Provides-Extra: ag2-extras
 Requires-Dist: ag2[anthropic]==0.9.9; extra == 'ag2-extras'
 Requires-Dist: ag2[autobuild]==0.9.9; extra == 'ag2-extras'
@@ -92,12 +90,13 @@ Requires-Dist: llama-index-vector-stores-chroma; extra == 'ag2-extras'
 Requires-Dist: llama-index-vector-stores-mongodb; extra == 'ag2-extras'
 Requires-Dist: llama-index<1,>=0.12; extra == 'ag2-extras'
 Requires-Dist: pgvector>=0.4.0; extra == 'ag2-extras'
-Requires-Dist: psycopg>=3.2.6; (sys_platform == 'linux') 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 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[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>=3.2.9; (sys_platform == 'linux') and extra == 'ag2-extras'
+Requires-Dist: psycopg>=3.2.9; (sys_platform == 'win32' and platform_machine == 'AARCH64') and extra == 'ag2-extras'
+Requires-Dist: psycopg>=3.2.9; (sys_platform == 'win32' and platform_machine == 'ARM64') and extra == 'ag2-extras'
+Requires-Dist: psycopg>=3.2.9; (sys_platform == 'win32' and platform_machine == 'aarch64') and extra == 'ag2-extras'
+Requires-Dist: psycopg>=3.2.9; (sys_platform == 'win32' and platform_machine == 'arm64') and extra == 'ag2-extras'
+Requires-Dist: psycopg[binary]>=3.2.9; (sys_platform == 'darwin') and extra == 'ag2-extras'
+Requires-Dist: psycopg[binary]>=3.2.9; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'ARM64' and platform_machine != 'aarch64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
 Requires-Dist: pydantic-ai>=0.0.21; extra == 'ag2-extras'
 Requires-Dist: pymongo>=4.11; extra == 'ag2-extras'
 Requires-Dist: qdrant-client[fastembed]; (sys_platform != 'win32') and extra == 'ag2-extras'
@@ -107,86 +106,96 @@ Requires-Dist: selenium<5,>=4.28.1; extra == 'ag2-extras'
 Requires-Dist: webdriver-manager==4.0.2; extra == 'ag2-extras'
 Provides-Extra: dev
 Requires-Dist: ag2[redis]==0.9.9; extra == 'dev'
-Requires-Dist: ag2[websockets]==0.9.9; extra == 'dev'
 Requires-Dist: autoflake==2.3.1; extra == 'dev'
 Requires-Dist: bandit==1.8.6; extra == 'dev'
 Requires-Dist: black[jupyter]==25.1.0; extra == 'dev'
 Requires-Dist: build==1.3.0; extra == 'dev'
-Requires-Dist: fakeredis==2.31.0; extra == 'dev'
+Requires-Dist: fakeredis==2.31.1; extra == 'dev'
 Requires-Dist: fastjsonschema>=2.21.2; extra == 'dev'
 Requires-Dist: flake8==7.3.0; extra == 'dev'
 Requires-Dist: hatchling==1.27.0; extra == 'dev'
 Requires-Dist: jsonschema==4.25.1; extra == 'dev'
-Requires-Dist: jupyter-server==2.16.0; extra == 'dev'
-Requires-Dist: jupyterlab==4.4.6; extra == 'dev'
+Requires-Dist: jupyter-server==2.17.0; extra == 'dev'
+Requires-Dist: jupyterlab==4.4.7; extra == 'dev'
 Requires-Dist: mypy-extensions>=1.1.0; extra == 'dev'
-Requires-Dist: mypy==1.17.1; extra == 'dev'
+Requires-Dist: mypy==1.18.1; extra == 'dev'
 Requires-Dist: nbclient>=0.10.2; extra == 'dev'
 Requires-Dist: nbconvert>=7.16.6; extra == 'dev'
 Requires-Dist: nbformat>=5.10.4; extra == 'dev'
 Requires-Dist: nodeenv>=1.9.1; extra == 'dev'
 Requires-Dist: notebook-shim>=0.2.4; extra == 'dev'
 Requires-Dist: paho-mqtt<3.0,>=2.1.0; extra == 'dev'
-Requires-Dist: pandas-stubs==2.3.0.250703; extra == 'dev'
+Requires-Dist: pandas-stubs==2.3.2.250827; extra == 'dev'
 Requires-Dist: pre-commit==4.3.0; extra == 'dev'
 Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
 Requires-Dist: pylint==3.3.8; extra == 'dev'
 Requires-Dist: python-dotenv>=1.1.1; extra == 'dev'
-Requires-Dist: ruff==0.12.9; extra == 'dev'
+Requires-Dist: ruff==0.13.0; extra == 'dev'
 Requires-Dist: toml==0.10.2; (python_version <= '3.10') and extra == 'dev'
-Requires-Dist: types-aiofiles==24.1.0.20250809; extra == 'dev'
-Requires-Dist: types-jsonschema==4.25.0.20250809; extra == 'dev'
-Requires-Dist: types-psutil==7.0.0.20250801; extra == 'dev'
-Requires-Dist: types-pyyaml==6.0.12.20250809; extra == 'dev'
+Requires-Dist: types-aiofiles==24.1.0.20250822; extra == 'dev'
+Requires-Dist: types-jsonschema==4.25.1.20250822; extra == 'dev'
+Requires-Dist: types-psutil==7.0.0.20250822; extra == 'dev'
+Requires-Dist: types-pyyaml==6.0.12.20250822; extra == 'dev'
 Requires-Dist: types-redis==4.6.0.20241004; extra == 'dev'
-Requires-Dist: types-requests==2.32.4.20250809; extra == 'dev'
+Requires-Dist: types-requests==2.32.4.20250913; extra == 'dev'
 Requires-Dist: types-toml==0.10.8.20240310; extra == 'dev'
-Requires-Dist: watchdog==6.0.0; extra == 'dev'
+Requires-Dist: watchdog>=5.0.0; extra == 'dev'
+Requires-Dist: websockets<16,>=14.0; extra == 'dev'
 Requires-Dist: yamllint==1.37.1; extra == 'dev'
 Provides-Extra: docs
 Requires-Dist: markdown-callouts==0.4.0; extra == 'docs'
 Requires-Dist: mdx-include==1.4.2; extra == 'docs'
 Requires-Dist: mdx-truly-sane-lists==1.3; extra == 'docs'
-Requires-Dist: mkdocs-autorefs==1.4.2; extra == 'docs'
-Requires-Dist: mkdocs-awesome-nav==3.1.2; extra == 'docs'
+Requires-Dist: mkdocs-autorefs==1.4.3; extra == 'docs'
+Requires-Dist: mkdocs-awesome-nav==3.2.0; extra == 'docs'
 Requires-Dist: mkdocs-jupyter==0.25.1; extra == 'docs'
 Requires-Dist: mkdocs-macros-plugin==1.3.9; extra == 'docs'
-Requires-Dist: mkdocs-material==9.6.17; extra == 'docs'
-Requires-Dist: mkdocs-minify-html-plugin==0.3.4; extra == 'docs'
+Requires-Dist: mkdocs-material==9.6.19; extra == 'docs'
+Requires-Dist: mkdocs-minify-html-plugin; (python_version >= '3.13' and sys_platform == 'win32' and platform_machine == 'AARCH64') and extra == 'docs'
+Requires-Dist: mkdocs-minify-html-plugin; (python_version >= '3.13' and sys_platform == 'win32' and platform_machine == 'ARM64') and extra == 'docs'
+Requires-Dist: mkdocs-minify-html-plugin; (python_version >= '3.13' and sys_platform == 'win32' and platform_machine == 'aarch64') and extra == 'docs'
+Requires-Dist: mkdocs-minify-html-plugin; (python_version >= '3.13' and sys_platform == 'win32' and platform_machine == 'arm64') and extra == 'docs'
+Requires-Dist: mkdocs-minify-html-plugin==0.3.4; (python_version < '3.13') and extra == 'docs'
+Requires-Dist: mkdocs-minify-html-plugin==0.3.4; (python_version >= '3.13' and sys_platform != 'win32') and extra == 'docs'
+Requires-Dist: mkdocs-minify-html-plugin==0.3.4; (python_version >= '3.13' and sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'docs'
 Requires-Dist: mkdocs-open-in-new-tab==1.0.8; extra == 'docs'
 Requires-Dist: mkdocs==1.6.1; extra == 'docs'
 Requires-Dist: mkdocstrings-crystal==0.3.7; extra == 'docs'
-Requires-Dist: mkdocstrings-python==1.17.0; extra == 'docs'
+Requires-Dist: mkdocstrings-python==1.18.2; extra == 'docs'
 Requires-Dist: mkdocstrings[crystal,python]==0.30.0; extra == 'docs'
 Requires-Dist: natsort==8.4.0; extra == 'docs'
 Provides-Extra: jupyter
 Requires-Dist: ipywidgets==8.1.7; extra == 'jupyter'
-Requires-Dist: jupyter-server==2.16.0; extra == 'jupyter'
-Requires-Dist: jupyterlab==4.4.6; extra == 'jupyter'
-Requires-Dist: waldiez-jupyter==0.5.10; extra == 'jupyter'
+Requires-Dist: jupyter-server==2.17.0; extra == 'jupyter'
+Requires-Dist: jupyterlab==4.4.7; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.6.0; extra == 'jupyter'
 Provides-Extra: mqtt
 Requires-Dist: paho-mqtt<3.0,>=2.1.0; extra == 'mqtt'
 Provides-Extra: redis
 Requires-Dist: ag2[redis]==0.9.9; extra == 'redis'
+Provides-Extra: reload
+Requires-Dist: watchdog>=5.0.0; extra == 'reload'
 Provides-Extra: runner
-Requires-Dist: waldiez-runner==0.5.10; (python_version >= '3.11') and extra == 'runner'
+Requires-Dist: waldiez-runner==0.6.0; (python_version >= '3.11') and extra == 'runner'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.5.10; extra == 'studio'
+Requires-Dist: waldiez-studio==0.6.0; extra == 'studio'
 Provides-Extra: test
 Requires-Dist: ag2[redis]==0.9.9; extra == 'test'
 Requires-Dist: ag2[websockets]==0.9.9; extra == 'test'
-Requires-Dist: fakeredis==2.31.0; extra == 'test'
+Requires-Dist: fakeredis==2.31.1; extra == 'test'
 Requires-Dist: paho-mqtt<3.0,>=2.1.0; extra == 'test'
-Requires-Dist: pytest-asyncio==1.1.0; extra == 'test'
-Requires-Dist: pytest-cov==6.2.1; extra == 'test'
+Requires-Dist: pytest-asyncio==1.2.0; extra == 'test'
+Requires-Dist: pytest-cov==7.0.0; extra == 'test'
 Requires-Dist: pytest-env==1.1.5; extra == 'test'
 Requires-Dist: pytest-html==4.1.1; extra == 'test'
-Requires-Dist: pytest-sugar==1.1.0; extra == 'test'
+Requires-Dist: pytest-sugar==1.1.1; extra == 'test'
 Requires-Dist: pytest-timeout==2.4.0; extra == 'test'
 Requires-Dist: pytest-xdist==3.8.0; extra == 'test'
-Requires-Dist: pytest==8.4.1; extra == 'test'
+Requires-Dist: pytest==8.4.2; extra == 'test'
+Requires-Dist: watchdog>=5.0.0; extra == 'test'
+Requires-Dist: websockets<16,>=14.0; extra == 'test'
 Provides-Extra: websockets
-Requires-Dist: ag2[websockets]==0.9.9; extra == 'websockets'
+Requires-Dist: websockets<16,>=14.0; extra == 'websockets'
 Description-Content-Type: text/markdown
 
 # Waldiez
@@ -300,7 +309,7 @@ docker run --rm `
 
 ```shell
 CONTAINER_COMMAND=docker # or podman
-# Asuming ./flow.waldiez exists
+# Assuming ./flow.waldiez exists
 HOST_INPUT="$(pwd)/flow.waldiez"
 CONTAINER_INPUT="/home/waldiez/workspace/flow.waldiez"
 HOST_OUTPUT_DIR="$(pwd)/output"