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

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

@@ -1,6 +1,6 @@
 waldiez/__init__.py,sha256=AuVJ8ZOPmJoTeKQ_ryezU14tG-DFRb7djvr72a5G63Y,251
 waldiez/__main__.py,sha256=9xR-F2ohZcRPDG6KrM7cJpXciKX-u6WdL221ckyJ04k,112
-waldiez/_version.py,sha256=A8aXjNec6D58qEEhm4QLBRHEUMFlCl0k8_SIGteYnJ8,62
+waldiez/_version.py,sha256=H1WLgLEQD6NHtFBb8DOYUVOZxL9vh2tPJJn9fu19O2k,62
 waldiez/cli.py,sha256=wStRkt056Y-F2CcuD6Zlov-ooUpWndyNesV9s_MjyHU,4798
 waldiez/exporter.py,sha256=iKe-l_Me8NRWsXHIdBcrOrnLT9XIyp4iYi4HLuuj2jA,9342
 waldiez/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -15,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
@@ -24,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
@@ -33,18 +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=YczhIw0PKEsxCSY6tAqElCqOUh8b-aEkSIGC1JjXxAs,5877
-waldiez/io/stream/__init__.py,sha256=6qst5j2iXmV-wDTnhgCna3llqUfJ6tkR0HBq7Vx9x-Q,197
-waldiez/io/stream/consumer.py,sha256=VQDJEomYpGjmAtOoFBJrpCyoYzwykJUIOrdiLn08Uj4,4049
-waldiez/io/stream/provider.py,sha256=xcjtlZeeq7q-9fh0pSKCvAXv8vzPFrqbSGzfZCQoSFw,9607
-waldiez/io/stream/server.py,sha256=G3mUmfyswp6jQZAeqY3eemsuhRns1ls7nsMr27mdT6o,12614
+waldiez/io/__init__.py,sha256=-w6ywIjG8ByxGtpSeRt1bjhnP4CZzpHf8WRWckjs4ng,5172
 waldiez/models/__init__.py,sha256=IMq8vzuAgmv92kHSSuZQLF38vVd31ojgOHonuHqWYj0,2888
-waldiez/models/waldiez.py,sha256=_odlEjyXPcvHXZVDq4wV0I3_bv27Syk3smBfMOUXIoU,9180
+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
@@ -52,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
@@ -88,8 +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-0.1.7.dist-info/METADATA,sha256=dGnSmaGRqnsTogxJ9sp9SNg5rxpd6iBrkJEaGtC3kJM,7726
-waldiez-0.1.7.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-waldiez-0.1.7.dist-info/entry_points.txt,sha256=5Po4yQXPa_QEdtTevpEBgr3rGoIvDMeQuJR2zqwBLBo,45
-waldiez-0.1.7.dist-info/licenses/LICENSE,sha256=VQEHM6WMQLRu1qaGl3GWsoOknDwro-69eGo4NLIJPIM,1064
-waldiez-0.1.7.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.7.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/__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/io/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

waldiez/io/stream/provider.py

@@ -1,343 +0,0 @@
-"""TCP socket input provider.
-
-It connects to a TCP server,
-listens for `PROVIDE:` messages to receive prompts,
-and sends `USE:` messages to pass the response.
-"""
-
-import logging
-import socket
-import threading
-import time
-from types import TracebackType
-from typing import Optional, Type
-
-END_OF_MESSAGE = b"\r\n"
-LOGGER = logging.getLogger("tcp::provider")
-
-
-class InputProviderThread(threading.Thread):
-    """Input provider thread."""
-
-    def __init__(
-        self,
-        host: str,
-        port: int,
-        response: Optional[str],
-        timeout: Optional[float] = None,
-    ) -> None:
-        """Create a new input provider thread.
-
-        Parameters
-        ----------
-        host : str
-            The host to connect to.
-        port : int
-            The port to connect to.
-        response : str, optional
-            The response to send.
-        timeout : float, optional
-            The timeout for the provider, by default None (no timeout).
-        """
-        super().__init__(
-            name="InputProviderThread",
-            daemon=True,
-            target=self.run,
-        )
-        self.host = host
-        self.port = port
-        self.timeout = timeout
-        self._response = response
-        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0)
-        self.running = False
-
-    @property
-    def response(self) -> Optional[str]:
-        """Get the response."""
-        return self._response
-
-    @response.setter
-    def response(self, response: str) -> None:
-        """Set the response.
-
-        Parameters
-        ----------
-        response : str
-            The response to set.
-        """
-        self._response = response
-
-    def _run_no_timeout(self) -> None:
-        """Start the provider."""
-        self.socket.connect((self.host, self.port))
-        self.socket.sendall("PROVIDER\r\n".encode("utf-8"))
-        self.running = True
-        while self.running:
-            data = b""
-            while not data.endswith(END_OF_MESSAGE):
-                try:
-                    data += self.socket.recv(1024)
-                except BaseException:  # pylint: disable=broad-except
-                    break
-            if not data:
-                break
-            if data.startswith(b"PROVIDE:"):
-                prompt = data[len(b"PROVIDE:") : -len(END_OF_MESSAGE)]
-                LOGGER.debug("Got a prompt: %s", prompt)
-                if self.response:
-                    message = f"USE:{self.response}" + "\r\n"
-                    self.socket.sendall(message.encode("utf-8"))
-        self.socket.close()
-        self.running = False
-
-    def _run_with_timeout(self, timeout: float) -> None:
-        """Start the provider with a timeout."""
-        self.socket.connect((self.host, self.port))
-        self.socket.sendall("PROVIDER\r\n".encode("utf-8"))
-        self.running = True
-        start_time = time.monotonic()
-        while self.running:
-            data = b""
-            while time.monotonic() - start_time < timeout:
-                try:
-                    data += self.socket.recv(1024)
-                except BaseException:  # pylint: disable=broad-except
-                    break
-                if data.endswith(END_OF_MESSAGE):
-                    break
-            if not data:
-                break
-            if data.startswith(b"PROVIDE:"):
-                prompt = data[len(b"PROVIDE:") : -len(END_OF_MESSAGE)]
-                LOGGER.debug("Got a prompt: %s", prompt)
-                if self.response:
-                    message = f"USE:{self.response}" + "\r\n"
-                    self.socket.sendall(message.encode("utf-8"))
-        self.socket.close()
-        self.running = False
-
-    def run(self) -> None:
-        """Run the provider."""
-        if self.timeout is None or self.timeout < 1:
-            self._run_no_timeout()
-        else:
-            self._run_with_timeout(self.timeout)
-
-
-class InputProviderWrapper:
-    """Input provider wrapper."""
-
-    thread: Optional[InputProviderThread] = None
-
-    def __init__(
-        self,
-        host: str,
-        port: int,
-        response: Optional[str] = None,
-        timeout: Optional[float] = None,
-    ) -> None:
-        """Create a new input provider wrapper.
-
-        Parameters
-        ----------
-        host : str
-            The host to connect to.
-        port : int
-            The port to connect to.
-        response : str, optional
-            The response to send.
-        timeout : float, optional
-            The timeout for the provider, by default None (no timeout).
-        """
-        self._host = host
-        self._port = port
-        self._timeout = timeout
-        self._response = response
-
-    @property
-    def response(self) -> Optional[str]:
-        """Get the response."""
-        if self.thread:
-            return self.thread.response
-        return self._response
-
-    def start(self) -> None:
-        """Start the provider."""
-        self.thread = InputProviderThread(
-            self._host, self._port, self._response, self._timeout
-        )
-        self.thread.daemon = True
-        self.thread.start()
-
-    def set_response(self, response: str) -> None:
-        """Set the response.
-
-        Parameters
-        ----------
-        response : str
-            The response to set.
-        """
-        self._response = response
-        if self.thread:
-            self.thread.response = response
-
-    def stop(self) -> None:
-        """Stop the provider."""
-        if self.thread:
-            self.thread.running = False
-            try:
-                self.thread.socket.close()
-            except OSError:  # pragma: no cover
-                pass
-            self.thread.join(timeout=0)
-            del self.thread
-            self.thread = None
-
-
-class TCPProvider:
-    """Input provider."""
-
-    _wrapper: Optional[InputProviderWrapper] = None
-
-    def __init__(
-        self,
-        host: str,
-        port: int,
-        response: Optional[str] = None,
-        timeout: Optional[float] = 30,
-    ) -> None:
-        """Create a new input provider.
-
-        Parameters
-        ----------
-        host : str
-            The host to connect to.
-        port : int
-            The port to connect to.
-        response : str, optional
-            The response to send.
-        timeout : float, optional
-            The timeout for the provider, by default 30.
-        """
-        self._host = host
-        self._port = port
-        self._response = response
-        self._timeout = timeout
-        self._start_called = False
-        self._init_wrapper()
-
-    def __enter__(self) -> "TCPProvider":
-        """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()
-
-    @property
-    def thread(self) -> Optional[InputProviderThread]:
-        """Get the thread."""
-        if not self._wrapper:
-            return None
-        return self._wrapper.thread
-
-    @property
-    def response(self) -> Optional[str]:
-        """Get the response."""
-        if self._wrapper:
-            return self._wrapper.response
-        return self._response
-
-    def _init_wrapper(self) -> None:
-        """Initialize the wrapper."""
-        self._wrapper = InputProviderWrapper(
-            host=self._host,
-            port=self._port,
-            response=self._response,
-            timeout=self._timeout,
-        )
-
-    def is_running(self) -> bool:
-        """Check if the provider is running.
-
-        Returns
-        -------
-        bool
-            True if the provider is running, False otherwise.
-        """
-        if not self._wrapper:
-            return False
-        if not self.thread:
-            return False
-        return self.thread.running is True
-
-    def wait(self, timeout: float) -> None:
-        """Wait until the provider is running.
-
-        Parameters
-        ----------
-        timeout : float
-            The timeout to wait.
-        """
-        start_time = time.time()
-        while not self.is_running():
-            if time.time() - start_time > timeout:
-                break
-            time.sleep(1)
-
-    def start(self) -> None:
-        """Start the provider.
-
-        Raises
-        ------
-        RuntimeError
-            If the wrapper is not initialized.
-        """
-        # avoid starting the provider multiple times
-        # (if the wrapped thread has not yet started)
-        if self._start_called is True:
-            return
-        self._start_called = True
-        if self.is_running():
-            return
-        if not self._wrapper:
-            raise RuntimeError("Wrapper not initialized")
-        self._wrapper.start()
-
-    def set_response(self, response: str) -> None:
-        """Set the response.
-
-        Parameters
-        ----------
-        response : str
-            The response to set.
-
-        Raises
-        ------
-        RuntimeError
-            If the wrapper is not initialized.
-        """
-        if not self._wrapper:
-            raise RuntimeError("Wrapper not initialized")
-        self._wrapper.set_response(response or "\n")
-
-    def stop(self) -> None:
-        """Stop the provider."""
-        self._start_called = False
-        if not self.is_running():
-            return
-        if self._wrapper:
-            self._wrapper.stop()
-        del self._wrapper
-        self._init_wrapper()
-
-    def restart(self) -> None:
-        """Restart the provider."""
-        self.stop()
-        self.start()