waldiez 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

{waldiez-0.1.6.dist-info → waldiez-0.1.8.dist-info}/RECORD

@@ -1,11 +1,10 @@
-waldiez/__init__.py,sha256=pX6lHj8fNlfqCpHEfF6MNQBCR6qRjFYjwBSOCoxrlwk,313
+waldiez/__init__.py,sha256=AuVJ8ZOPmJoTeKQ_ryezU14tG-DFRb7djvr72a5G63Y,251
 waldiez/__main__.py,sha256=9xR-F2ohZcRPDG6KrM7cJpXciKX-u6WdL221ckyJ04k,112
-waldiez/_version.py,sha256=g8Nna0tpgQ0zozBL1MTb08rmIQ8H6TPV1dTCiRbfuxk,62
-waldiez/cli.py,sha256=vIQDqs9XBOlDpshq7rNgvoTcnnkeZrsXa_51gMWs6H4,4690
+waldiez/_version.py,sha256=H1WLgLEQD6NHtFBb8DOYUVOZxL9vh2tPJJn9fu19O2k,62
+waldiez/cli.py,sha256=wStRkt056Y-F2CcuD6Zlov-ooUpWndyNesV9s_MjyHU,4798
 waldiez/exporter.py,sha256=iKe-l_Me8NRWsXHIdBcrOrnLT9XIyp4iYi4HLuuj2jA,9342
-waldiez/io_stream.py,sha256=YczhIw0PKEsxCSY6tAqElCqOUh8b-aEkSIGC1JjXxAs,5877
 waldiez/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-waldiez/runner.py,sha256=BvDiczgTip-cj5gWxBWuXca03aR63sBl3AP_6-5eZ9k,10003
+waldiez/runner.py,sha256=YDjEyu9zZF2JX22c6C1g1iYwxRDEdP2pPox6m0kJhfw,10194
 waldiez/exporting/__init__.py,sha256=GMY7qTRpNmc7tpaCFNGLT5wX9eu26NvoNPuYX8MzP50,344
 waldiez/exporting/agents/__init__.py,sha256=v5KA112W_EFYwXE2TSBKYyO8rRKUOUAOpFS5CMSnfRs,110
 waldiez/exporting/agents/agent.py,sha256=otNYclKfi57CUNsM0lVLAbD7YPboe_2TnrdllBwgKzE,7228
@@ -16,7 +15,7 @@ waldiez/exporting/agents/llm_config.py,sha256=A88e-RKp09r8n9MG11hArpITzxK8nVrTZ6
 waldiez/exporting/agents/teachability.py,sha256=ame4hHJCZRBp7hAQGZzv2Cjs6QtcV9vlQ1zheMEMac0,1103
 waldiez/exporting/agents/termination_message.py,sha256=tzI4-tcveYKBVx5PsznQZwAoghSX3mbn_vPu4rX8tuU,1276
 waldiez/exporting/agents/rag_user/__init__.py,sha256=01F4gwgUwtSpZbGXcfieqIuLNT64u9KiqMIB2f0mplI,196
-waldiez/exporting/agents/rag_user/chroma_utils.py,sha256=jBncfPsepTzgSo9wxyeLFEVvmoWDpIC-dJCpyqRQvx4,4632
+waldiez/exporting/agents/rag_user/chroma_utils.py,sha256=4kZor8wgTaA7Mw9lUEHu594QtbTmiD2iZafhmOo7_9s,4579
 waldiez/exporting/agents/rag_user/mongo_utils.py,sha256=y5IL-Anfktg9cYo2o-ED1A7lwHQWdVMWD_W1AT5_RmE,2664
 waldiez/exporting/agents/rag_user/pgvector_utils.py,sha256=EyGDwvo1Pe8bqoJl3NFpqK6zizN81lPPaSMBMQF79Dc,2722
 waldiez/exporting/agents/rag_user/qdrant_utils.py,sha256=tn1A7pjQjJfGS33ALgnPYTz7xu8rVBidcc5-WHLq8e8,3487
@@ -25,7 +24,7 @@ waldiez/exporting/agents/rag_user/vector_db.py,sha256=64Gr_y1VTZLXi1pC4KjChsZ6DT
 waldiez/exporting/chats/__init__.py,sha256=v5aR1gWqSN5xeDDMIFo-ceC_Z9UgL8qJZofC2sU8HqQ,296
 waldiez/exporting/chats/chats.py,sha256=xI5ZzWpcqYz8Kuu7B9pU6iHN16wUwHxOvYFhH5vxWuA,1259
 waldiez/exporting/chats/helpers.py,sha256=K-IwBAG4-t7NLBP8Qs8_a2AEoKyoLw_B6eHzgXpOshA,13352
-waldiez/exporting/chats/nested.py,sha256=cDB22ekFrzBZj5eb-nM1oGfPc4BEDGrXTZT2AlNXRsQ,8356
+waldiez/exporting/chats/nested.py,sha256=nYzxOTHSBpyFdqvX_NtZUvUgTdmoTZcCRaGTFzdnmII,7916
 waldiez/exporting/flow/__init__.py,sha256=WhdPrjXQAcihrS1KUtPNgbx0y1tqD5HtGykzpEAcsBM,98
 waldiez/exporting/flow/def_main.py,sha256=YpdhqO4iFng3r7if69ZPMJAibPVNDqnrOrWvGw7CJq8,1052
 waldiez/exporting/flow/flow.py,sha256=x2AlXr7aJJfrPMeuXQYfT9sG5GycMwf0kRkovZWKGag,6206
@@ -34,13 +33,14 @@ waldiez/exporting/skills/__init__.py,sha256=oIA9f5ABtYSbS0kMY_4ovU3-m6meVk5blEu_
 waldiez/exporting/utils/__init__.py,sha256=tP1V4g9-MyARlfOEL_1YWMJNW7UivUrrukq7DIwdq6k,1018
 waldiez/exporting/utils/comments.py,sha256=X9j8w48rh3DfFDjiMverU9DBSuE9yuMMbbStxBbN1sE,3190
 waldiez/exporting/utils/importing.py,sha256=dA4HCQ-OxmodUjovgXyLI9IwNvLwbY67P41969XoZ7g,8649
-waldiez/exporting/utils/logging_utils.py,sha256=uoV6O23lfB5ztOYEZiYu8Mn-2xEUwp_Qx404Mr62i7M,5822
+waldiez/exporting/utils/logging_utils.py,sha256=DfnC13PqnadG0nRxz_gHA4IIZGnMawpTPlkKYdGT94w,5864
 waldiez/exporting/utils/method_utils.py,sha256=7-RUMTylNM2W0iM1bPX2_Gn3553XZSl2s2VGEijxNp4,891
 waldiez/exporting/utils/naming.py,sha256=VdoVODQduhXIs9hQFWUVEVqTaSyNDt7rkECsuIgXYwI,3196
 waldiez/exporting/utils/object_string.py,sha256=2kdIu4in3iUV92a2KbLWwp9POhvY-fzF_r2AGVnCKls,2166
 waldiez/exporting/utils/path_check.py,sha256=laGSLLiiwIIKxpAp5JIbtEvurbvXGkT4Hx0lemJcA9s,976
+waldiez/io/__init__.py,sha256=-w6ywIjG8ByxGtpSeRt1bjhnP4CZzpHf8WRWckjs4ng,5172
 waldiez/models/__init__.py,sha256=IMq8vzuAgmv92kHSSuZQLF38vVd31ojgOHonuHqWYj0,2888
-waldiez/models/waldiez.py,sha256=B_ujtMVnuGFAB_d9gMDydOdbe2XtP80EH7FVefwpHeg,8909
+waldiez/models/waldiez.py,sha256=Oxav9QCYlkfxrNLlcJC89gT0UC4Ls5fTn4kL0PypyGw,9066
 waldiez/models/agents/__init__.py,sha256=3ZyVYBHMFzZjRMIdPrBF6HLg82LPAlEubL6utL6KhfU,1856
 waldiez/models/agents/agents.py,sha256=zIqihnoBjzaQLL_P6FcVoHddcusRNYsWPIFLZD091bE,3641
 waldiez/models/agents/agent/__init__.py,sha256=inA0zV3dnwmcQlcObH_FLaZSURjFG31E_XUampJAnJU,746
@@ -48,7 +48,7 @@ waldiez/models/agents/agent/agent.py,sha256=DAwreQtIdoM2x_vVccIkALl5whyS07GvfKRU
 waldiez/models/agents/agent/agent_data.py,sha256=A3qIvOKyUKBIqqPbsBKVE5JTNQ8JcoDEGMqazNkN_rA,5009
 waldiez/models/agents/agent/code_execution.py,sha256=kgL3pbEEyuMvJid0sIbfe4os7SWKpzL1Bv4O525Biyk,1942
 waldiez/models/agents/agent/linked_skill.py,sha256=8RHWHkHXqFuv7lEe1PuQoK1hTO3kBQ7ILKm9kCEWqNs,667
-waldiez/models/agents/agent/nested_chat.py,sha256=DAF3TCbTwyDvg6PGbeETtBLCQ4Xz5Saw5a4Xi-jr6jA,1929
+waldiez/models/agents/agent/nested_chat.py,sha256=VKzHI38CUCnOlB6oBdO7CJdCk-wmG0aGXIkWpJPa4-Q,1757
 waldiez/models/agents/agent/teachability.py,sha256=IIR4LY9dwx3T7Ok17RYN2g6zGiga2gGizGteaeI3eGs,1703
 waldiez/models/agents/agent/termination_message.py,sha256=aGsIdwGs42BxDktkxoqHGEWM1wzhXr-zh_MlETsbycA,5674
 waldiez/models/agents/assistant/__init__.py,sha256=Zlf-4EI9HXl-LrqGosL7UucoyqIl74GZzohZlRLx2QI,175
@@ -84,12 +84,8 @@ waldiez/models/model/model_data.py,sha256=pDPKUbltaXWjCuDArgwTOEHw_igfk_DkxzFzdn
 waldiez/models/skill/__init__.py,sha256=rU88bajKOGMYoHFcE8MP0jW9H0MswbQmvz5wxS35BYE,169
 waldiez/models/skill/skill.py,sha256=fhsAI413an2_d4DBIkf7dzEuWk6rGs2t4sl97a4dj20,3473
 waldiez/models/skill/skill_data.py,sha256=RTWn8Od6w7g-nRIpsS29sqZ8sPm5dCPiK7-qXmU-KD4,815
-waldiez/stream/__init__.py,sha256=6qst5j2iXmV-wDTnhgCna3llqUfJ6tkR0HBq7Vx9x-Q,197
-waldiez/stream/consumer.py,sha256=VQDJEomYpGjmAtOoFBJrpCyoYzwykJUIOrdiLn08Uj4,4049
-waldiez/stream/provider.py,sha256=JqnnR1yAimfK4HxW4gr4HIoJ15Ly71kXn75mv7FZBmw,9478
-waldiez/stream/server.py,sha256=Ya4AKP5Wr_xAsbG3Q-qShqHmfIyamj9LGXizJsL-N4Y,11815
-waldiez-0.1.6.dist-info/METADATA,sha256=1kZfdzqSXE08jzxkgf4Obp2NXC3nigKyorE7le8jTHo,6786
-waldiez-0.1.6.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-waldiez-0.1.6.dist-info/entry_points.txt,sha256=5Po4yQXPa_QEdtTevpEBgr3rGoIvDMeQuJR2zqwBLBo,45
-waldiez-0.1.6.dist-info/licenses/LICENSE,sha256=VQEHM6WMQLRu1qaGl3GWsoOknDwro-69eGo4NLIJPIM,1064
-waldiez-0.1.6.dist-info/RECORD,,
+waldiez-0.1.8.dist-info/METADATA,sha256=M3jPlHQhUMCvzgvm8LHzaBIjKl2GX8ARC-mtrsz90hE,8679
+waldiez-0.1.8.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+waldiez-0.1.8.dist-info/entry_points.txt,sha256=5Po4yQXPa_QEdtTevpEBgr3rGoIvDMeQuJR2zqwBLBo,45
+waldiez-0.1.8.dist-info/licenses/LICENSE,sha256=VQEHM6WMQLRu1qaGl3GWsoOknDwro-69eGo4NLIJPIM,1064
+waldiez-0.1.8.dist-info/RECORD,,

{waldiez-0.1.6.dist-info → waldiez-0.1.8.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.25.0
+Generator: hatchling 1.26.3
 Root-Is-Purelib: true
 Tag: py3-none-any

waldiez/io_stream.py

@@ -1,181 +0,0 @@
-"""Custom IOStream class to use with autogen.
-
-It is meant to be used when we want to use custom
-`print` and `input`. For example, when a websocket
-is used to trigger a UI element that requires user input.
-and sends back the user's input to the websocket. In the same
-way, we can use it to forward what is meant to be printed.
-
-
-We use:
-
-- one tcp server to handle messaging between the clients
-- one tcp client (provider) to set and forward the user's input
-    that we got elsewhere (e.g. from a websocket connection)
-- one tcp client (consumer) to ask and get the input from the provider
-"""
-
-import socket
-from contextlib import closing
-from typing import Any, Callable, Optional
-
-from autogen.io import IOStream  # type: ignore[import-untyped]
-
-from .stream import TCPConsumer, TCPProvider, TCPServer
-
-
-class WaldiezIOStream(IOStream):
-    """Custom IOStream class to handle the `print` and `input` functions."""
-
-    def __init__(
-        self,
-        port: int = 0,
-        input_timeout: float = 60,
-        print_function: Optional[Callable[..., None]] = None,
-        on_prompt_input: Optional[Callable[[str], None]] = None,
-    ) -> None:
-        """Initialize the IOStream.
-
-        Parameters
-        ----------
-        port : int, optional
-            The port to use, by default 0 (auto-assign).
-        input_timeout : float, optional
-            The input timeout, by default 60.
-        print_function : Optional[Callable[..., None]], optional
-            The print function to use, by default None.
-        on_prompt_input : Optional[Callable[[str], None]], optional
-            The function to call for getting an input, by default None.
-        """
-        self._print_function = print_function
-        if port == 0:
-            port = get_available_port()
-        self._port = port
-        self._input_timeout = input_timeout
-        self._server = TCPServer(port)
-        self._server.start()
-        self._provider = TCPProvider("localhost", port, response=None)
-        self._on_prompt_input = on_prompt_input
-
-    @property
-    def print_function(self) -> Optional[Callable[..., None]]:
-        """Get the print function."""
-        return self._print_function
-
-    def open(self) -> None:
-        """Start the server."""
-        if not self._server.is_running():
-            self._server.start()
-
-    def close(self) -> None:
-        """Stop the server and the provider."""
-        # pylint: disable=broad-except
-        if self._server.is_running():
-            try:
-                self._server.stop()
-            except BaseException:  # pragma: no cover
-                pass
-        try:
-            self._provider.stop()
-        except BaseException:  # pragma: no cover
-            pass
-
-    def __del__(self) -> None:  # pragma: no cover
-        """Delete the instance."""
-        self.close()
-
-    def forward_input(self, input_data: str) -> None:
-        """Forward the user's input to the provider.
-
-        When we have the input data
-            e.g. from 'input(..)' or from a websocket connection,
-            we can forward it to the provider (the tcp client)
-            to make it available to the consumer (the other tcp client).
-        Parameters
-        ----------
-        input_data : str
-            The input data to forward.
-        """
-        if not self._provider.is_running():
-            self._provider.start()
-        self._provider.set_response(input_data)
-
-    def print(
-        self,
-        *objects: Any,
-        sep: str = " ",
-        end: str = "\n",
-        flush: bool = False,
-    ) -> None:
-        """Mock the print function.
-
-        Parameters
-        ----------
-        objects : Any
-            The objects to print.
-        sep : str, optional
-            The separator, by default " ".
-        end : str, optional
-            The end, by default a new line.
-        flush : bool, optional
-            Whether to flush, by default False.
-        """
-        print_function: Callable[..., None] = self.print_function or print
-        print_function(*objects, sep=sep, end=end, flush=flush)
-
-    def input(self, prompt: str = "", *, password: bool = False) -> str:
-        """Mock the input function.
-
-        Parameters
-        ----------
-        prompt : str, optional
-            The prompt to show, by default "".
-        password : bool, optional (not used)
-            Whether to show the input as password, by default False.
-
-        Returns
-        -------
-        str
-            The user's input.
-        """
-        _prompt = prompt or "Your input:"
-        if _prompt in (">", "> "):
-            _prompt = "Your input:"
-        if prompt:
-            if self._on_prompt_input:
-                self._on_prompt_input(_prompt)
-            self.print(_prompt, end="")
-        if not self._provider.is_running():
-            self._provider.start()
-        # wait for the provider to start
-        self._provider.wait(timeout=self._input_timeout)
-        if not self._provider.is_running():  # pragma: no cover
-            self.print(
-                "WARNING: Provider is not running. Was an input expected?"
-            )
-            return "\n"
-        consumer = TCPConsumer(
-            "localhost", self._port, timeout=self._input_timeout
-        )
-        consumer.start()
-        # send the prompt and wait for the response
-        consumer.send_prompt(_prompt)
-        response = consumer.get_response()
-        consumer.stop()
-        self._provider.stop()
-        # return the response or a line break (i.e. no input)
-        return response or "\n"
-
-
-def get_available_port() -> int:
-    """Get an available port.
-
-    Returns
-    -------
-    int
-        Available port.
-    """
-    with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as soc:
-        soc.bind(("", 0))
-        soc.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-        return soc.getsockname()[1]

waldiez/stream/__init__.py

@@ -1,7 +0,0 @@
-"""IO stream using twisted and sockets."""
-
-from .consumer import TCPConsumer
-from .provider import TCPProvider
-from .server import TCPServer
-
-__all__ = ["TCPServer", "TCPProvider", "TCPConsumer"]

waldiez/stream/consumer.py

@@ -1,139 +0,0 @@
-"""TCP socket input consumer.
-
-It connects to a TCP server,
-listens for `INPUT:` messages to get the user's input,
-and sends `REQUEST:` messages to prompt the user,
-"""
-
-import socket
-import time
-from types import TracebackType
-from typing import Optional, Type
-
-END_OF_MESSAGE = b"\r\n"
-
-
-class TCPConsumer:
-    """TCP socket input consumer."""
-
-    def __init__(
-        self, host: str, port: int, timeout: Optional[float] = None
-    ) -> None:
-        """Create a new input consumer.
-
-        Parameters
-        ----------
-        host : str
-            The host to connect to.
-        port : int
-            The port to connect to.
-        timeout : float, optional
-            The timeout for the consumer, by default None (no timeout).
-        """
-        self.host = host
-        self.port = port
-        self.timeout = timeout
-        self._running = False
-        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0)
-
-    def __enter__(self) -> "TCPConsumer":
-        """Enter the context."""
-        self.start()
-        return self
-
-    def __exit__(
-        self,
-        exc_type: Optional[Type[BaseException]],
-        exc_value: Optional[BaseException],
-        traceback: Optional[TracebackType],
-    ) -> None:
-        """Exit the context."""
-        self.stop()
-
-    def is_running(self) -> bool:
-        """Check if the consumer is running.
-
-        Returns
-        -------
-        bool
-            True is the consumer is running, else False
-        """
-        return self._running
-
-    def start(self) -> None:
-        """Start the consumer."""
-        if self._running:
-            return
-        self._running = True
-        self.socket.connect((self.host, self.port))
-        self.socket.sendall("CONSUMER\r\n".encode("utf-8"))
-
-    def _get_response_no_timeout(self) -> Optional[str]:
-        """Get the response."""
-        data = self.socket.recv(1024)
-        while not data.endswith(END_OF_MESSAGE):
-            data += self.socket.recv(1024)
-        if data.startswith(b"INPUT:"):
-            response = data[len(b"INPUT:") :]
-            if response.endswith(END_OF_MESSAGE):
-                response = response[: -len(END_OF_MESSAGE)]
-            return response.decode("utf-8")
-        return None
-
-    def _get_response_with_timeout(self, timeout: float) -> Optional[str]:
-        """Get the response using a timeout."""
-        start_time = time.monotonic()
-        data = b""
-        self.socket.settimeout(timeout)
-        while time.monotonic() - start_time < timeout:
-            try:
-                data += self.socket.recv(1024)
-            except TimeoutError:
-                return None
-            if data.endswith(END_OF_MESSAGE):
-                break
-        if not data:
-            return None
-        if data.startswith(b"INPUT:"):
-            response = data[len(b"INPUT:") :]
-            if response.endswith(END_OF_MESSAGE):
-                response = response[: -len(END_OF_MESSAGE)]
-            return response.decode("utf-8")
-        return None
-
-    def get_response(self) -> Optional[str]:
-        """Get the response.
-
-        Returns
-        -------
-        Optional[str]
-            The response if available, None otherwise.
-        """
-        if not self._running:
-            self.start()
-        if self.timeout is None or self.timeout < 1:
-            return self._get_response_no_timeout()
-        return self._get_response_with_timeout(self.timeout)
-
-    def send_prompt(self, prompt: str) -> None:
-        """Send a prompt.
-
-        Parameters
-        ----------
-        prompt : str
-            The prompt to send.
-        """
-        if not self._running:
-            self.start()
-        message = f"REQUEST:{prompt}" + "\r\n"
-        self.socket.sendall(message.encode("utf-8"))
-
-    def stop(self) -> None:
-        """Close the consumer."""
-        try:
-            self.socket.close()
-        except OSError:
-            pass
-        del self.socket
-        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0)
-        self._running = False