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

{waldiez-0.1.7.dist-info → waldiez-0.1.9.dist-info}/METADATA

@@ -1,11 +1,10 @@
 Metadata-Version: 2.3
 Name: waldiez
-Version: 0.1.7
+Version: 0.1.9
 Summary: waldiez
-Project-URL: homepage, https://waldiez.github.io/py/
-Project-URL: repository, https://github.com/waldiez/py.git
+Project-URL: homepage, https://waldiez.github.io/waldiez/
+Project-URL: repository, https://github.com/waldiez/waldiez.git
 Author-email: Panagiotis Kasnesis <pkasnesis@thingenious.io>, Lazaros Toumanidis <laztoum@protonmail.com>
-License-File: LICENSE
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -17,28 +16,26 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Python: <3.13,>=3.10
-Requires-Dist: autogen-agentchat==0.2.37
+Requires-Dist: ag2==0.3.2
 Requires-Dist: jupytext
-Requires-Dist: twisted==24.10.0
-Provides-Extra: autogen-extras
-Requires-Dist: autogen-agentchat[anthropic]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[bedrock]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[gemini]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[groq]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[mistral]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[retrievechat-couchbase]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[retrievechat-mongodb]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[retrievechat-pgvector]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[retrievechat-qdrant]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[retrievechat]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[together]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: autogen-agentchat[websurfer]==0.2.37; extra == 'autogen-extras'
-Requires-Dist: chromadb==0.5.17; extra == 'autogen-extras'
-Requires-Dist: fastembed==0.4.1; extra == 'autogen-extras'
-Requires-Dist: pgvector==0.3.6; extra == 'autogen-extras'
-Requires-Dist: psycopg[binary]>=3.2.3; extra == 'autogen-extras'
-Requires-Dist: pymongo==4.10.1; extra == 'autogen-extras'
-Requires-Dist: qdrant-client==1.12.1; extra == 'autogen-extras'
+Provides-Extra: ag2-extras
+Requires-Dist: ag2[anthropic]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[bedrock]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[gemini]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[groq]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[mistral]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[retrievechat-mongodb]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[retrievechat-pgvector]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[retrievechat-qdrant]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[retrievechat]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[together]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: ag2[websurfer]==0.3.2; extra == 'ag2-extras'
+Requires-Dist: chromadb==0.5.3; extra == 'ag2-extras'
+Requires-Dist: fastembed==0.4.2; extra == 'ag2-extras'
+Requires-Dist: pgvector>=0.2.5; extra == 'ag2-extras'
+Requires-Dist: psycopg[binary]>=3.1.18; extra == 'ag2-extras'
+Requires-Dist: pymongo==4.10.1; extra == 'ag2-extras'
+Requires-Dist: qdrant-client==1.12.1; extra == 'ag2-extras'
 Provides-Extra: dev
 Requires-Dist: autoflake==2.3.1; extra == 'dev'
 Requires-Dist: bandit==1.7.10; extra == 'dev'
@@ -50,7 +47,7 @@ Requires-Dist: pre-commit==4.0.1; extra == 'dev'
 Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
 Requires-Dist: pylint==3.3.1; extra == 'dev'
 Requires-Dist: python-dotenv==1.0.1; extra == 'dev'
-Requires-Dist: ruff==0.7.2; extra == 'dev'
+Requires-Dist: ruff==0.7.4; extra == 'dev'
 Requires-Dist: types-pyyaml==6.0.12; extra == 'dev'
 Requires-Dist: yamllint==1.35.1; extra == 'dev'
 Provides-Extra: docs
@@ -58,11 +55,11 @@ 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.5.43; extra == 'docs'
+Requires-Dist: mkdocs-material==9.5.45; extra == 'docs'
 Requires-Dist: mkdocs-minify-html-plugin==0.2.3; extra == 'docs'
 Requires-Dist: mkdocs==1.6.1; extra == 'docs'
 Requires-Dist: mkdocstrings-python==1.12.2; extra == 'docs'
-Requires-Dist: mkdocstrings[crystal,python]==0.26.2; extra == 'docs'
+Requires-Dist: mkdocstrings[crystal,python]==0.27.0; extra == 'docs'
 Provides-Extra: test
 Requires-Dist: pytest-cov==6.0.0; extra == 'test'
 Requires-Dist: pytest-html==4.1.1; extra == 'test'
@@ -74,20 +71,20 @@ Description-Content-Type: text/markdown
 
 # Waldiez
 
-![CI Build](https://github.com/waldiez/py/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/py/badge.svg)](https://coveralls.io/github/waldiez/py) [![PyPI version](https://badge.fury.io/py/waldiez.svg)](https://badge.fury.io/py/waldiez)
+![CI Build](https://github.com/waldiez/waldiez/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/waldiez/badge.svg)](https://coveralls.io/github/waldiez/waldiez) [![PyPI version](https://badge.fury.io/waldiez/waldiez.svg)](https://badge.fury.io/py/waldiez)
 
 Translate a Waldiez flow:
 
-![Flow](https://raw.githubusercontent.com/waldiez/py/refs/heads/main/docs/flow.png)
+![Flow](https://raw.githubusercontent.com/waldiez/waldiez/refs/heads/main/docs/static/images/overview.webp)
 
-To a python script or a jupyter notebook with the corresponding [autogen](https://github.com/microsoft/autogen/) agents and chats.
+To a python script or a jupyter notebook with the corresponding [ag2](https://github.com/ag2ai/ag2/) agents and chats.
 
 ## Features
 
 - Export .waldiez flows to .py or .ipynb
 - Run a .waldiez flow
 - Include a `logs` folder with the logs of the flow in csv format
-- Provide a custom [IOSStream](https://autogen-ai.github.io/autogen/docs/reference/io/base#iostream) to handle input and output.
+- Provide a custom [IOSStream](https://ag2ai.github.io/ag2/docs/reference/io/base#iostream) to handle input and output.
 
 ## Installation
 
@@ -100,7 +97,7 @@ python -m pip install waldiez
 From the repository:
 
 ```bash
-python -m pip install git+https://github.com/waldiez/py.git
+python -m pip install git+https://github.com/waldiez/waldiez.git
 ```
 
 ## Usage
@@ -119,13 +116,13 @@ waldiez /path/to/a/flow.waldiez --output /path/to/an/output/flow[.py] [--force]
 ```shell
 CONTAINER_COMMAND=docker # or podman
 # pull the image
-$CONTAINER_COMMAND pull waldiez/py
+$CONTAINER_COMMAND pull waldiez/waldiez
 # Export a Waldiez flow to a python script or a jupyter notebook
 $CONTAINER_COMMAND run \
   --rm \
   -v /path/to/a/flow.waldiez:/flow.waldiez \
   -v /path/to/an/output:/output \
-  waldiez/py --export /flow.waldiez --output /output/flow[.py|.ipynb]
+  waldiez/waldiez --export /flow.waldiez --output /output/flow[.py|.ipynb]
 
 # with selinux and/or podman, you might get permission (or file not found) errors, so you can try:
 $CONTAINER_COMMAND run \
@@ -134,12 +131,12 @@ $CONTAINER_COMMAND run \
   -v /path/to/an/output:/output \
   --userns=keep-id \
   --security-opt label=disable \
-  waldiez/py --export /flow.waldiez --output /output/flow[.py|.ipynb]
+  waldiez/waldiez --export /flow.waldiez --output /output/flow[.py|.ipynb]
 ```
 
 ```shell
 # Export and run the script
-$CONTAINER_COMMAND run --rm -v /path/to/a/flow.waldiez:/flow.waldiez -v /path/to/an/output:/output waldiez/py /flow.waldiez --output /output/output[.py]
+$CONTAINER_COMMAND run --rm -v /path/to/a/flow.waldiez:/flow.waldiez -v /path/to/an/output:/output waldiez/waldiez /flow.waldiez --output /output/output[.py]
 ```
 
 ### As a library
@@ -170,40 +167,75 @@ runner.run(output_path=output_path)
 
 ```python
 # Run the flow with a custom IOStream
+# In case the standard 'input' and 'print' functions cannot be used
+import time
+import threading
+
+from typing import Any
+
 from waldiez import WaldiezRunner
 from waldiez.io import WaldiezIOStream
 
 flow_path = "/path/to/a/flow.waldiez"
 output_path = "/path/to/an/output.py"
 
-def print_function(*values, **args) -> None:
-    """A custom print function."""
-    print(values)
 
-def on_prompt_input(prompt: str) -> str:
-    """A custom input function."""
-    return input(prompt)
+def custom_print_function(*args: Any, sep: str = " ", **kwargs: Any) -> None:
+    """Custom print function."""
+    print(*args, sep=sep, **kwargs)
+
+
+# Custom input handler
+class InputProcessorWrapper:
+    """Wrapper input processor.
+    
+    To manage the interaction between the custom input processor and IOStream.
+    """
+
+    def __init__(self):
+        self.stream = None  # Placeholder for the WaldiezIOStream instance
+        self.lock = threading.Lock()  # Ensure thread-safe operations
+
+    def custom_input_processor(self, prompt: str) -> None:
+        """Simulate external input and send it back to the IOStream."""
+        def external_input_simulation():
+            with self.lock:  # Ensure thread-safe access
+                time.sleep(2)  # Simulate delay for network input
+                if self.stream:
+                    self.stream.set_input("Simulated external input")
+                else:
+                    raise RuntimeError("Stream reference not set!")
 
-io_stream = WaldiezIOStream(
-    print_function=print_function,
-    on_prompt_input=on_prompt_input,
+        threading.Thread(target=external_input_simulation, daemon=True).start()
+
+    def set_stream(self, stream: "WaldiezIOStream"):
+        """Set the WaldiezIOStream instance."""
+        with self.lock:  # Ensure thread-safe setting of the stream
+            self.stream = stream
+
+processor_wrapper = InputProcessorWrapper()
+
+stream = WaldiezIOStream(
     input_timeout=30,
+    print_function=
+    on_prompt_input=processor_wrapper.custom_input_processor,
 )
+
+# Link the processor wrapper to the WaldiezIOStream instance
+processor_wrapper.set_stream(custom_stream)
+
 with WaldiezIOStream.set_default(io_stream):
     runner = WaldiezRunner.load(flow_path)
     runner.run(stream=io_stream, output_path=output_path)
 
-io_stream.close()
-
 ```
 
 ### Tools
 
-- [autogen](https://github.com/microsoft/autogen/)
+- [ag2 (formerly AutoGen)](https://github.com/ag2ai/ag2)
 - [juptytext](https://github.com/mwouts/jupytext)
-- [twisted](https://github.com/twisted/twisted)
 - [pydantic](https://github.com/pydantic/pydantic)
 
 ## License
 
-This project is licensed under the MIT License - see the [LICENSE](https://github.com/waldiez/py/blob/main/LICENSE) file for details.
+This project is licensed under the MIT License - see the [LICENSE](https://github.com/waldiez/waldiez/blob/main/LICENSE) file for details.

{waldiez-0.1.7.dist-info → waldiez-0.1.9.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=BAPW_qv-vw1iBQm2bCpQfS6VnM42uoEaWNsSV8dCyr8,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=JZJwISn75Hklm50lqKdGNRHuMg81AnDNTMqoI-TJLRk,4702
 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/exporting/utils/path_check.py,sha256=aO49sbk6hUHEr65UjNNUSKO5WCGnQitiT733W-kGVtI,1062
+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.9.dist-info/METADATA,sha256=tCHxrAXmDgRKloonwepcZQO-9X4MXYpQ5Hq523L4NEY,8679
+waldiez-0.1.9.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+waldiez-0.1.9.dist-info/entry_points.txt,sha256=5Po4yQXPa_QEdtTevpEBgr3rGoIvDMeQuJR2zqwBLBo,45
+waldiez-0.1.9.dist-info/licenses/LICENSE,sha256=VQEHM6WMQLRu1qaGl3GWsoOknDwro-69eGo4NLIJPIM,1064
+waldiez-0.1.9.dist-info/RECORD,,

{waldiez-0.1.7.dist-info → waldiez-0.1.9.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