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

waldiez/exporting/core/extras/serializer.py

@@ -10,7 +10,7 @@ from typing import Any, Optional
 from ..protocols import Serializer
 
 
-# pylint: disable=too-few-public-methods
+# pylint: disable=too-few-public-methods,no-self-use
 class DefaultSerializer(Serializer):
     """Default serializer for Waldiez items."""
 
@@ -37,7 +37,7 @@ class DefaultSerializer(Serializer):
 def serialize_item(
     item: Any,
     tabs: int = 1,
-    _visited: Optional[set[int]] = None,
+    visited: Optional[set[int]] = None,
 ) -> str:
     """Convert an item to a formatted string with given indentation.
 
@@ -47,6 +47,8 @@ def serialize_item(
         The item to convert.
     tabs : int, optional
         The number of tabs, by default 1.
+    visited : set[int], optional
+        A set of visited IDs, by default None
 
     Returns
     -------
@@ -81,8 +83,8 @@ def serialize_item(
     }
     ```
     """
-    if _visited is None:
-        _visited = set()
+    if visited is None:
+        visited = set()
 
     if callable(item):
         return item.__name__
@@ -92,37 +94,37 @@ def serialize_item(
         return _format_primitive(item)
 
     # Handle circular references in containers
-    if isinstance(item, (dict, list, tuple, set)) and id(item) in _visited:
+    if isinstance(item, (dict, list, tuple, set)) and id(item) in visited:
         return '"<circular reference>"'
 
     next_indent = " " * 4 * (tabs + 1)
-    _visited.add(id(item))
+    visited.add(id(item))
 
     if isinstance(item, dict):
         items: list[str] = []
         for key, value in item.items():
             key_str = f'{next_indent}"{key}"'
-            value_str = serialize_item(value, tabs + 1, _visited)
+            value_str = serialize_item(value, tabs + 1, visited)
             items.append(f"{key_str}: {value_str}")
         return _format_container(items, "{", "}", tabs)
 
     if isinstance(item, list):
         items = [
-            f"{next_indent}{serialize_item(sub_item, tabs + 1, _visited)}"
+            f"{next_indent}{serialize_item(sub_item, tabs + 1, visited)}"
             for sub_item in item
         ]
         return _format_container(items, "[", "]", tabs)
 
     if isinstance(item, tuple):
         items = [
-            f"{next_indent}{serialize_item(sub_item, tabs + 1, _visited)}"
+            f"{next_indent}{serialize_item(sub_item, tabs + 1, visited)}"
             for sub_item in item
         ]
         return _format_container(items, "(", ")", tabs)
 
     if isinstance(item, set):
         items = [
-            f"{next_indent}{serialize_item(sub_item, tabs + 1, _visited)}"
+            f"{next_indent}{serialize_item(sub_item, tabs + 1, visited)}"
             for sub_item in item
         ]
         return _format_container(items, "{", "}", tabs)

waldiez/exporting/core/protocols.py

@@ -139,144 +139,3 @@ class ContentGenerator(Protocol):
         ExporterContentError
             If exporting fails.
         """
-
-
-@runtime_checkable
-class ExportingLogger(Protocol):
-    """Protocol for logging during exporting."""
-
-    def log(
-        self, message: Any, *args: Any, level: str = "info", **kwargs: Any
-    ) -> None:
-        """Log a message with the specified level.
-
-        Parameters
-        ----------
-        message : Any
-            The message to log or message template for formatting.
-        level : str, optional
-            The logging level to use (e.g., "debug", "info", "warning", "error",
-            "critical"). Defaults to "info".
-        *args : Any
-            Arguments to format into the message using % formatting.
-        **kwargs : Any
-            Keyword arguments to format into the message using str.format().
-        """
-
-    def error(self, message: Any, *args: Any, **kwargs: Any) -> None:
-        """Log an error message.
-
-        Parameters
-        ----------
-        message : Any
-            The error message to log or message template.
-        *args : Any
-            Arguments to format into the message.
-        **kwargs : Any
-            Keyword arguments to format into the message using str.format().
-        """
-
-    def warning(self, message: Any, *args: Any, **kwargs: Any) -> None:
-        """Log a warning message.
-
-        Parameters
-        ----------
-        message : Any
-            The warning message to log or message template.
-        *args : Any
-            Arguments to format into the message.
-        **kwargs : Any
-            Keyword arguments to format into the message using str.format().
-        """
-
-    def info(self, message: Any, *args: Any, **kwargs: Any) -> None:
-        """Log an informational message.
-
-        Parameters
-        ----------
-        message : Any
-            The informational message to log or message template.
-        *args : Any
-            Arguments to format into the message.
-        **kwargs : Any
-            Keyword arguments to format into the message using str.format().
-        """
-
-    def success(self, message: Any, *args: Any, **kwargs: Any) -> None:
-        """Log a success message.
-
-        Parameters
-        ----------
-        message : Any
-            The success message to log or message template.
-        *args : Any
-            Arguments to format into the message.
-        **kwargs : Any
-            Keyword arguments to format into the message using str.format().
-        """
-
-    def debug(self, message: Any, *args: Any, **kwargs: Any) -> None:
-        """Log a debug message.
-
-        Parameters
-        ----------
-        message : Any
-            The debug message to log or message template.
-        *args : Any
-            Arguments to format into the message.
-        **kwargs : Any
-            Keyword arguments to format into the message using str.format().
-        """
-        self.log(message, *args, level="debug", **kwargs)
-
-    def critical(self, message: Any, *args: Any, **kwargs: Any) -> None:
-        """Log a critical error message.
-
-        Parameters
-        ----------
-        message : Any
-            The critical error message to log or message template.
-        *args : Any
-            Arguments to format into the message.
-        **kwargs : Any
-            Keyword arguments to format into the message using str.format().
-        """
-        self.log(message, *args, level="critical", **kwargs)
-
-    def exception(self, message: Any, *args: Any, **kwargs: Any) -> None:
-        """Log an exception message.
-
-        Parameters
-        ----------
-        message : Any
-            The exception message to log or message template.
-        *args : Any
-            Arguments to format into the message.
-        **kwargs : Any
-            Keyword arguments to format into the message using str.format().
-        """
-        self.log(message, *args, level="exception", **kwargs)
-
-    def set_level(self, level: str) -> None:
-        """Set the logging level.
-
-        Parameters
-        ----------
-        level : str
-            The logging level to set
-            (e.g., "debug", "info", "warning", "error", "critical").
-
-        Raises
-        ------
-        ValueError
-            If the provided level is invalid.
-        """
-
-    def get_level(self) -> str:  # pyright: ignore
-        """Get the current logging level.
-
-        Returns
-        -------
-        str
-            The current logging level.
-        """

waldiez/exporting/core/result.py

@@ -59,7 +59,7 @@ class ExportResult:
         position : ImportPosition, optional
             The position of the import, by default THIRD_PARTY
         """
-        if statement and statement.strip():
+        if statement and statement.strip():  # pragma: no branch
             self.imports.add(
                 ImportStatement(
                     statement=statement.strip(),
@@ -123,7 +123,7 @@ class ExportResult:
         comment : Optional[str], optional
             Optional comment for the argument, by default None
         """
-        if name and value is not None:
+        if name and value is not None:  # pragma: no branch
             arg = InstanceArgument(
                 instance_id=instance_id,
                 name=name,
@@ -173,7 +173,7 @@ class ExportResult:
         position : ExportPosition, optional
             The position for the merged content, by default AGENTS
         """
-        if other.main_content:
+        if other.main_content:  # pragma: no branch
             self.main_content = (
                 (self.main_content or "") + "\n" + other.main_content
             ).strip()
@@ -241,7 +241,7 @@ class ExportResult:
             Additional metadata for the content, by default None
         """
         order_value = order.value if isinstance(order, ContentOrder) else order
-        if content and content.strip():
+        if content and content.strip():  # pragma: no branch
             positioned = PositionedContent(
                 content=content.strip() if not skip_strip else content,
                 position=position,
@@ -273,7 +273,7 @@ class ExportResult:
         required : bool, optional
             Whether the variable is required, by default True
         """
-        if name and value:
+        if name and value:  # pragma: no branch
             env_var = EnvironmentVariable(
                 name=name,
                 value=value,

waldiez/exporting/core/types.py

@@ -314,6 +314,7 @@ class ExportConfig:
         if for_notebook:
             output_extension = "ipynb"
         cache_seed = kwargs.pop("cache_seed", None)
+        # noinspection PyUnreachableCode
         if cache_seed is not None and not isinstance(cache_seed, int):
             cache_seed = None
         return cls(

waldiez/exporting/core/utils/llm_config.py

@@ -97,7 +97,7 @@ def _get_agent_llm_config_arg_as_dict(
 ) -> str:
     tab = " " * tab_leng * tabs if tabs > 0 else ""
     if not agent.data.model_ids:
-        return f'{tab}"llm_config": False' + "\n"
+        return f'{tab}"llm_config": False,' + "\n"
     content = f'{tab}"llm_config": autogen.LLMConfig(' + "\n"
     content += f"{tab}    config_list=["
     got_at_least_one_model = False
@@ -110,7 +110,7 @@ def _get_agent_llm_config_arg_as_dict(
             content += "\n" + f"{tab}        {model_name}_llm_config,"
             got_at_least_one_model = True
     if not got_at_least_one_model:  # pragma: no cover
-        return f'{tab}"llm_config": False' + "\n"
+        return f'{tab}"llm_config": False,' + "\n"
     content += "\n" + f"{tab}    ]," + "\n"
     content += f"{tab}    cache_seed={cache_seed}," + "\n"
     if temperature is not None:

waldiez/exporting/flow/execution_generator.py

@@ -125,6 +125,7 @@ class ExecutionGenerator:
         flow_content += "    result_dicts: list[dict[str, Any]] = []\n"
         space = "    "
         if cache_seed is not None:
+            # noinspection SqlDialectInspection
             flow_content += (
                 f"    with Cache.disk(cache_seed={cache_seed}"
                 ") as cache:  # pyright: ignore\n"

waldiez/exporting/flow/merger.py

@@ -167,7 +167,7 @@ class ContentMerger:
             # Log conflicts but continue
             for conflict in conflicts:
                 self.context.get_logger().warning(
-                    f"Import conflict: {conflict}"
+                    "Import conflict: %s", conflict
                 )
 
         return set(grouped.values())
@@ -309,7 +309,7 @@ class ContentMerger:
             # Log conflicts but continue (non-fatal)
             for conflict in conflicts:
                 self.context.get_logger().warning(
-                    f"Environment variable conflict: {conflict}"
+                    "Environment variable conflict: %s", conflict
                 )
 
         return list(env_vars.values())

waldiez/exporting/flow/orchestrator.py

@@ -249,6 +249,7 @@ class ExportOrchestrator:
                 agent_names=self.agent_names,
                 tools=self.tools,
                 tool_names=self.tool_names,
+                is_async=self.waldiez.is_async,
                 output_dir=self.config.output_directory,
                 context=self.context,
             )

waldiez/exporting/flow/utils/common.py

@@ -2,7 +2,7 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 # pylint: disable=line-too-long
 # flake8: noqa: E501
-"""Common utils for the final generatio."""
+"""Common utils for the final flow generation."""
 
 from waldiez.models import Waldiez
 
@@ -120,8 +120,8 @@ def main_doc_string() -> str:
 
     Raises
     ------
-    RuntimeError
-        If the chat session fails.
+    SystemExit
+        If the user interrupts the chat session.
     """'''
 
 

waldiez/exporting/flow/utils/importing.py

@@ -7,6 +7,7 @@ from typing import Optional
 from waldiez.exporting.core import ImportPosition
 
 BUILTIN_IMPORTS = [
+    "import asyncio",
     "import csv",
     "import importlib",
     "import json",

waldiez/exporting/flow/utils/logging.py

@@ -82,47 +82,13 @@ def get_start_logging(is_async: bool, for_notebook: bool) -> str:
 '''
 
 
-# pylint: disable=differing-param-doc,differing-type-doc
-# noinspection PyUnresolvedReferences
 def get_sync_sqlite_out() -> str:
-    r"""Get the sqlite to csv and json conversion code string.
+    """Get the sqlite to csv and json conversion code string.
 
     Returns
     -------
     str
         The sqlite to csv and json conversion code string.
-
-    Example
-    -------
-    ```python
-    >>> get_sqlite_outputs()
-    def get_sqlite_out(dbname: str, table: str, csv_file: str) -> None:
-        \"\"\"Convert a sqlite table to csv and json files.
-
-    Parameters
-    ----------
-        dbname : str
-            The sqlite database name.
-        table : str
-            The table name.
-        csv_file : str
-            The csv file name.
-        \"\"\"
-        conn = sqlite3.connect(dbname)
-        query = f"SELECT * FROM {table}"  # nosec
-        cursor = conn.execute(query)
-        rows = cursor.fetchall()
-        column_names = [description[0] for description in cursor.description]
-        data = [dict(zip(column_names, row)) for row in rows]
-        conn.close()
-        with open(csv_file, "w", newline="", encoding="utf-8") as file:
-            csv_writer = csv.DictWriter(file, fieldnames=column_names)
-            csv_writer.writeheader()
-            csv_writer.writerows(data)
-        json_file = csv_file.replace(".csv", ".json")
-        with open(json_file, "w", encoding="utf-8") as file:
-            json.dump(data, file, indent=4, ensure_ascii=False)
-    ```
     """
     content = "\n\n"
     content += (
@@ -148,7 +114,8 @@ def get_sync_sqlite_out() -> str:
     content += "    rows = cursor.fetchall()\n"
     content += "    column_names = [description[0] for description "
     content += "in cursor.description]\n"
-    content += "    data = [dict(zip(column_names, row)) for row in rows]\n"
+    # pylint: disable=line-too-long
+    content += "    data = [dict(zip(column_names, row, strict=True)) for row in rows]\n"
     content += "    conn.close()\n"
     content += (
         '    with open(csv_file, "w", newline="", encoding="utf-8") as file:\n'
@@ -165,52 +132,14 @@ def get_sync_sqlite_out() -> str:
     return content
 
 
-# pylint: disable=differing-param-doc,differing-type-doc,line-too-long
-# noinspection PyUnresolvedReferences
+# pylint: disable=line-too-long
 def get_async_sqlite_out() -> str:
-    r"""Get the sqlite to csv and json conversion code string.
+    """Get the sqlite to csv and json conversion code string.
 
     Returns
     -------
     str
         The sqlite to csv and json conversion code string.
-
-    Example
-    -------
-    ```python
-    >>> get_sqlite_outputs()
-    async def get_sqlite_out(dbname: str, table: str, csv_file: str) -> None:
-        \"\"\"Convert a sqlite table to csv and json files.
-
-    Parameters
-    ----------
-        dbname : str
-            The sqlite database name.
-        table : str
-            The table name.
-        csv_file : str
-            The csv file name.
-        \"\"\"
-        conn = await aiosqlite.connect(dbname)
-        query = f"SELECT * FROM {table}"  # nosec
-        try:
-            cursor = await conn.execute(query)
-        except BaseException:  # pylint: disable=broad-exception-caught
-            await conn.close()
-            return
-        rows = await cursor.fetchall()
-        column_names = [description[0] for description in cursor.description]
-        data = [dict(zip(column_names, row)) for row in rows]
-        await cursor.close()
-        await conn.close()
-        async with aiofiles.open(csv_file, "w", newline="", encoding="utf-8") as file:
-            csv_writer = csv.DictWriter(file, fieldnames=column_names)
-            csv_writer.writeheader()
-            csv_writer.writerows(data)
-        json_file = csv_file.replace(".csv", ".json")
-        async with aiofiles.open(json_file, "w", encoding="utf-8") as file:
-            await file.write(json.dumps(data, indent=4, ensure_ascii=False)
-    ```
     """
     # fmt: off
     content = "\n\n"
@@ -235,7 +164,7 @@ def get_async_sqlite_out() -> str:
     content += "    rows = await cursor.fetchall()\n"
     content += "    column_names = [description[0] for description "
     content += "in cursor.description]\n"
-    content += "    data = [dict(zip(column_names, row)) for row in rows]\n"
+    content += "    data = [dict(zip(column_names, row, strict=True)) for row in rows]\n"
     content += "    await cursor.close()\n"
     content += "    await conn.close()\n"
     content += '    async with aiofiles.open(csv_file, "w", newline="", encoding="utf-8") as file:\n'
@@ -369,9 +298,7 @@ def get_stop_logging(is_async: bool, tabs: int = 0) -> str:
     content += "def stop_logging() -> None:\n"
     content += '    """Stop logging."""\n'
     if is_async:
-        content += f"{tab}    # pylint: disable=import-outside-toplevel\n"
-        content += f"{tab}    from asyncer import asyncify\n\n"
-        content += f"{tab}    await asyncify(runtime_logging.stop)()\n"
+        content += f"{tab}    await asyncio.to_thread(runtime_logging.stop)\n"
     else:
         content += f"{tab}    runtime_logging.stop()\n"
     content += get_sqlite_out_call(tabs + 1, is_async)

waldiez/exporting/tools/exporter.py

@@ -32,6 +32,7 @@ class ToolsExporter(Exporter[ToolExtras]):
         agent_names: dict[str, str],
         tools: list[WaldiezTool],
         tool_names: dict[str, str],
+        is_async: bool,
         output_dir: str | Path | None = None,
         context: ExporterContext | None = None,
         **kwargs: Any,
@@ -50,6 +51,8 @@ class ToolsExporter(Exporter[ToolExtras]):
             The tools to export.
         tool_names : dict[str, str]
             Mapping of tool IDs to names.
+        is_async : bool
+            Whether the flow is asynchronous.
         output_dir : str | Path | None
             Output directory for generated files, by default None
         context : ExporterContext | None
@@ -64,6 +67,7 @@ class ToolsExporter(Exporter[ToolExtras]):
         self.agent_names = agent_names
         self.tools = tools
         self.tool_names = tool_names
+        self.is_async = is_async
         self.output_dir = Path(output_dir) if output_dir else None
 
         # Initialize extras with processed tool content
@@ -83,6 +87,7 @@ class ToolsExporter(Exporter[ToolExtras]):
             flow_name=self.flow_name,
             tools=self.tools,
             tool_names=self.tool_names,
+            is_async=self.is_async,
             output_dir=self.output_dir,
         )
 

waldiez/exporting/tools/factory.py

@@ -16,6 +16,7 @@ def create_tools_exporter(
     agent_names: dict[str, str],
     tools: list[WaldiezTool],
     tool_names: dict[str, str],
+    is_async: bool,
     output_dir: str | Path | None = None,
     context: ExporterContext | None = None,
 ) -> ToolsExporter:
@@ -33,6 +34,8 @@ def create_tools_exporter(
         The tools to export.
     tool_names : dict[str, str]
         Mapping of tool IDs to names.
+    is_async : bool
+        Whether the flow is asynchronous.
     output_dir : str | Path | None, optional
         Output directory for generated files, by default None
     context : ExporterContext | None, optional
@@ -52,5 +55,6 @@ def create_tools_exporter(
         tools=tools,
         tool_names=tool_names,
         output_dir=output_dir,
+        is_async=is_async,
         context=context,
     )

waldiez/exporting/tools/processor.py

@@ -33,6 +33,7 @@ class ToolProcessor:
         flow_name: str,
         tools: list[WaldiezTool],
         tool_names: dict[str, str],
+        is_async: bool,
         output_dir: Path | None = None,
     ):
         """Initialize the tool processor.
@@ -52,6 +53,7 @@ class ToolProcessor:
         self.tools = tools
         self.tool_names = tool_names
         self.output_dir = output_dir
+        self.is_async = is_async
 
     def process(self) -> ToolProcessingResult:
         """Process all tools and return consolidated result.
@@ -94,7 +96,9 @@ class ToolProcessor:
             The result to add processed content to.
         """
         # Get tool content
-        tool_content = tool.get_content()
+        tool_content = tool.get_content(
+            runtime_kwargs={"is_async": self.is_async}
+        )
         if tool_content:  # pragma: no branch
             # Add interop conversion if needed
             if tool.is_interop:

waldiez/io/__init__.py

@@ -30,7 +30,7 @@ from .models import (
     VideoMediaContent,
 )
 from .structured import StructuredIOStream
-from .utils import MediaType, MessageType
+from .utils import DEBUG_INPUT_PROMPT, START_CHAT_PROMPT, MediaType, MessageType
 
 try:
     from .redis import RedisIOStream  # type: ignore[no-redef,unused-ignore]
@@ -125,4 +125,6 @@ __all__ = [
     "AudioContent",
     "VideoMediaContent",
     "VideoContent",
+    "DEBUG_INPUT_PROMPT",
+    "START_CHAT_PROMPT",
 ]

waldiez/io/_ws.py

@@ -2,9 +2,12 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 # flake8: noqa: E501
 # pylint: disable=line-too-long
+# pyright: reportUnknownMemberType=false,reportUnknownParameterType=false
+# pyright: reportUnknownVariableType=false,reportUnknownArgumentType=false
 """WebSocket IOStream implementation for AsyncIO."""
 
 import asyncio
+import logging
 from typing import Any, Protocol
 
 HAS_WS_LIB = False
@@ -13,7 +16,7 @@ try:
     from starlette.websockets import WebSocket  # type: ignore[unused-ignore, unused-import, import-not-found, import-untyped]  # noqa
 
     HAS_WS_LIB = True  # pyright: ignore
-except ImportError:
+except ImportError:  # pragma: no cover
     pass
 
 try:
@@ -68,6 +71,7 @@ class WebSocketsAdapter:
             The websockets library connection.
         """
         self.websocket = websocket
+        self.log = logging.getLogger(__name__)
 
     async def send_message(self, message: str) -> None:
         """Send a message using websockets library.
@@ -100,11 +104,14 @@ class WebSocketsAdapter:
             )
             if isinstance(response, bytes):
                 return response.decode("utf-8")
+            # noinspection PyUnreachableCode
             return response if isinstance(response, str) else str(response)
         except asyncio.TimeoutError:
             return ""
-        except BaseException:  # pylint: disable=broad-exception-caught
-            # Handle other exceptions that may occur during receive
+        except BaseException as exc:  # pylint: disable=broad-exception-caught
+            self.log.error(
+                "WebsocketsAdapter: Error receiving message: %s", exc
+            )  # pragma: no cover
             return ""
 
 
@@ -120,6 +127,7 @@ class StarletteAdapter:
             The Starlette/FastAPI WebSocket connection.
         """
         self.websocket = websocket
+        self.log = logging.getLogger(__name__)
 
     async def send_message(self, message: str) -> None:
         """Send a message using Starlette WebSocket.
@@ -151,8 +159,10 @@ class StarletteAdapter:
             )
         except asyncio.TimeoutError:
             return ""
-        except BaseException:  # pylint: disable=broad-exception-caught
-            # Handle other exceptions that may occur during receive
+        except BaseException as exc:  # pylint: disable=broad-exception-caught
+            self.log.error(
+                "StarletteAdapter: Error receiving message: %s", exc
+            )  # pragma: no cover
             return ""
 
 

waldiez/io/models/content/image.py

@@ -61,6 +61,7 @@ class ImageUrlMediaContent(BaseModel):
     type: Literal["image_url"] = "image_url"
     image_url: ImageContent
 
+    # noinspection DuplicatedCode
     def to_string(
         self,
         uploads_root: Path | None,