waldiez 0.4.8__py3-none-any.whl → 0.4.11__py3-none-any.whl

{waldiez-0.4.8.dist-info → waldiez-0.4.11.dist-info}/RECORD

@@ -1,11 +1,11 @@
-waldiez/__init__.py,sha256=JVVh6d5Zw3ch4GYtIwXZEBpMEyJZAc5t0mtu08tLbZs,992
+waldiez/__init__.py,sha256=GdbXP8mS6FUiljxvd-eH_J2gEztoHCQpD8Hq51GKUSs,943
 waldiez/__main__.py,sha256=0dYzNrQbovRwQQvmZC6_1FDR1m71SUIOkTleO5tBnBw,203
-waldiez/_version.py,sha256=Nlo6ovErTlkRrYOk191gkWS6guip7lym_CNGuVCS1nI,249
-waldiez/cli.py,sha256=LAxEdPMBP7Un0hUGUmW7U9KaSgzJxtjQejS9CZ8LPKk,8804
-waldiez/exporter.py,sha256=Gl23KC-pF2986gnHLqGf1-7RAPfvbf2sz4VX236gXO8,6134
+waldiez/_version.py,sha256=QWIp4Z7OVTDhuUBWdL2fCZ0bU8ohzZDEUGN0OpjYUGQ,250
+waldiez/cli.py,sha256=yVa82t0YRLzif9S8yTDrwuMdCpfHwWsLnfi8zl3MDyg,8850
+waldiez/exporter.py,sha256=IEk8LM-tqjLxyi7R8M9wmbFXf4kZiqmiOMmsZumkgns,8347
 waldiez/logger.py,sha256=UFdPwS2AOl2LkFQVonDSET9Dguq5jsn2mV817iTZFJc,16375
 waldiez/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-waldiez/runner.py,sha256=mzMsk-YKhgsKpIPW5DW1nUv-wXo_P56k8QkKpntlw5A,13875
+waldiez/runner.py,sha256=ZeT_ERfzcm7mhLGH0po5Of0ecq9fQ_0c3ECdNRgq7g0,15619
 waldiez/exporting/__init__.py,sha256=xVOqb3gADkrTmp_BVrQ-dMPiySPUDU4xZmqYqhzL8BI,1026
 waldiez/exporting/agent/__init__.py,sha256=GpNfVQLAvGoJP_U4WkMMIGpEB8k7vEV-FVR446s2zhY,260
 waldiez/exporting/agent/code_execution.py,sha256=sxjulr2OmeSg0SrYz7rgpBgF_AfqpHfV86kmtGU49Yk,3795
@@ -28,7 +28,7 @@ waldiez/exporting/agent/extras/handoffs/condition.py,sha256=kURtSLS5B0WbXPARff3t
 waldiez/exporting/agent/extras/handoffs/handoff.py,sha256=Xp0sYpXIoXQ6GbLrLPxylDpyNPiE2w1oXmQ-UqfnK7U,5918
 waldiez/exporting/agent/extras/handoffs/target.py,sha256=R7KerfkL1kXkGMnoy1GXaOTH6DM6LSKXl7ohpTcInCE,6527
 waldiez/exporting/agent/extras/rag/__init__.py,sha256=jkKSaHLfwA8Tj7QJH0MSdAphaP4HowuhQXDj7V122Ek,259
-waldiez/exporting/agent/extras/rag/chroma_extras.py,sha256=m6PMx4opQ9VJPHdEW-kOmOZeqibLBeVDkkLxiZFkUTY,5018
+waldiez/exporting/agent/extras/rag/chroma_extras.py,sha256=vf4EuikCZPtqFEOCoB01YcD85Hi_SA37b1JO869DqSA,5555
 waldiez/exporting/agent/extras/rag/mongo_extras.py,sha256=bn_5bfmyfdW5eQCOjcCRYPKLIJ7vZXKEiUsa4zPx7Sk,2822
 waldiez/exporting/agent/extras/rag/pgvector_extras.py,sha256=Y08rU8Z3-oHUbh7huO1Jo_PZLlOW-NRa1Gz3-d9WXso,2961
 waldiez/exporting/agent/extras/rag/qdrant_extras.py,sha256=hTzY7GUL8RBMFNEGefLDAxKFMSlD5Qmkzm3Ducgp9vU,3633
@@ -46,19 +46,19 @@ waldiez/exporting/chats/utils/single.py,sha256=oOlqDxRcws0Dp7ignu-7tqzMBFgshKX-0
 waldiez/exporting/core/__init__.py,sha256=XwXYuiWyaPfTirac_3wVlNTBa5n0X09V752Bc4Dd0cg,3930
 waldiez/exporting/core/constants.py,sha256=PcJQftkXX7FsDkilH_jihje_9j9SenCjUa5Hj-95akQ,492
 waldiez/exporting/core/content.py,sha256=uxOnySvI30a2zgzkvdzkzhppZ-S1QnSxWPSJhy9wPMg,2063
-waldiez/exporting/core/context.py,sha256=Xy0iCbVb2fLew3AI2AYLNx_jkq1HM3gYZacMkhgYQu8,7862
+waldiez/exporting/core/context.py,sha256=p6wVl_wEALOoUdJUrMCNZGbcA7myG3XLJ8HmyFqOfyQ,8375
 waldiez/exporting/core/enums.py,sha256=CvmtPRv-4GwKpO2l0wXxxoV8Q5Oy5L5t0gnEMK8MVws,2464
 waldiez/exporting/core/errors.py,sha256=xLfq3Z4Kg2aqlu7zYOtBd5ces7RLG-o_nF82ITFsaJg,535
 waldiez/exporting/core/exporter.py,sha256=G-riIXeI6Ukr30mhpNNzDwbHvztZDbOhtUO5eTUYl5M,11173
 waldiez/exporting/core/exporters.py,sha256=tlX8e-8EeLNRh49gFPDswM4EQ8nrxnXtLUn4OmmuoyM,1795
 waldiez/exporting/core/protocols.py,sha256=xRnhET6WXc4vYbA-MoeUNRoxqt1MViSvTjwCaZLSvX4,7269
 waldiez/exporting/core/result.py,sha256=DiN-doVKKt_FTzB4l4oNPBXbWbVDYD0nDYcNBeWC5dE,20984
-waldiez/exporting/core/types.py,sha256=XR14B1Hjghi72EKJvqpvQCBUohZ0nVAE2jTSSlJTHmM,9512
+waldiez/exporting/core/types.py,sha256=b4G9qZ5r6q8-ap2zAVcvL7vHpynbY5pcG8QQWXofL4g,10307
 waldiez/exporting/core/validation.py,sha256=YlYpv4-VnBwmllznB1uijexPHaYDOrtnNwkS5QlQKF8,2728
 waldiez/exporting/core/extras/__init__.py,sha256=I9t7fmPpcLTwe0bErsfyoLcC6VrimJ2fw52K-JuAUYU,962
 waldiez/exporting/core/extras/base.py,sha256=ldmYmw3uCz3A-FQNoprdYiM1_45PYzm_oGtUR9OtFkI,7593
 waldiez/exporting/core/extras/chat_extras.py,sha256=rGbkmP6uj4bteyS9RvHrEdguNYYb32SOzVeqXuv1x5s,3435
-waldiez/exporting/core/extras/flow_extras.py,sha256=jjo42V9IEnwnW73MwbCY-1D0iTlWREC_n6XugQUfScc,2072
+waldiez/exporting/core/extras/flow_extras.py,sha256=8iktgmIMQ9mWMhGPvsFn1dKzflrQlscInE3zlA9ya3M,1661
 waldiez/exporting/core/extras/model_extras.py,sha256=bXbfgHEr9eD2KkarowzRCuYaH0qFzv4Bd0lcwn5JjzA,2069
 waldiez/exporting/core/extras/path_resolver.py,sha256=jongTqCMJFQ2rombuDenGwtYVEh-yrDMDgze2rOBKmQ,2140
 waldiez/exporting/core/extras/serializer.py,sha256=mFkvTHkEF73V4CtUfxN9b6tczzCyvTHMvDIxA4PcVWY,3567
@@ -74,16 +74,16 @@ waldiez/exporting/core/utils/comment.py,sha256=0RDl48DQsOoL4V8J2334LsjMU-dHtTgUT
 waldiez/exporting/core/utils/llm_config.py,sha256=NYsHimHk3c2Azjr7N4gMqjlSZO5Vw4FdFPgTxzv55iE,3959
 waldiez/exporting/flow/__init__.py,sha256=bsKIPmNhhNb7MPnimwL2X2fe8L4tXa_dH3CCDCNJ_6g,255
 waldiez/exporting/flow/execution_generator.py,sha256=uDlQrOh5z6SQS2ZcgBMqZ4gd0zxW0QgvlU9qOG9c4i8,6268
-waldiez/exporting/flow/exporter.py,sha256=rnwM04FtDMumjzlaHvRCEdzbISEKX9FGmZBP2bMGVSM,3331
-waldiez/exporting/flow/factory.py,sha256=U-c1UVnFhQCkqn9LjsVCRvPtxX1AwQs-v5KxJ45kYYc,2962
+waldiez/exporting/flow/exporter.py,sha256=8wRTCiAR4JZdxT4c22cRa6E2xFx0xs34vsZZIMZblRQ,3486
+waldiez/exporting/flow/factory.py,sha256=GbUb8ou4pfLh2QQt17I6mFSbc3Nwuh6RY1bJqCjbHuw,3640
 waldiez/exporting/flow/file_generator.py,sha256=7Wj2magf2J9N_jd0JzzNLuhvNBv77VhPxJX7xAZRdfE,6762
 waldiez/exporting/flow/merger.py,sha256=F9VNFDqic4cKRAxne5kFNQT0Jn_BccVvJuOQF365ld4,12247
-waldiez/exporting/flow/orchestrator.py,sha256=TR_mRnDmOOVrHU4UBD4SIXYRd2AZrq21o3q0aJe3508,14981
-waldiez/exporting/flow/utils/__init__.py,sha256=3cdbesV0G6ZXR5yHLSIkqaGFEoF5YyjKWlqQIPKNzbs,584
-waldiez/exporting/flow/utils/common.py,sha256=VX6Bs6Pew9Y-6MpSZ1TciSrkk0aCNhcyCagNL_oa2K8,5413
+waldiez/exporting/flow/orchestrator.py,sha256=2x2nTITgb0Q9OZLufDpTBMteu3PNFQq1qQt5KwdFq1s,15523
+waldiez/exporting/flow/utils/__init__.py,sha256=Wh-ptZOkqt8FdRnHNfSqeJJzKS6LL9TP_VbToWnhHo4,632
+waldiez/exporting/flow/utils/common.py,sha256=YY-4pN1zgD34d0ZHDBSZ3tteIZQyO4AUJe3bUPjNILc,7782
 waldiez/exporting/flow/utils/importing.py,sha256=EnKeHZbD3H-xXYTZYaFJGIY6mAolVP-NNCasFHsYX9I,11119
 waldiez/exporting/flow/utils/linting.py,sha256=N5WHCn0vANMWzl4Wzi5KA_2SwfIeVdig4_jalC5gh6I,4686
-waldiez/exporting/flow/utils/logging.py,sha256=Y7UkaIPNORbYai9sB3R_cGbt65-8lusWiSe_rxLy_Dk,11943
+waldiez/exporting/flow/utils/logging.py,sha256=4Nlj95TFfpHfgjYWOOKEIipYppJdFyGpVgpZszuC47A,11965
 waldiez/exporting/models/__init__.py,sha256=xYq6kUfoeE5Jyml0PJJiviayEPFh82kxbk9C208BvbQ,279
 waldiez/exporting/models/exporter.py,sha256=4rocKI0cTh-njbZhZdbOrRODGK-0L-xumfFgeM_ndRk,7220
 waldiez/exporting/models/factory.py,sha256=d6DeI2aeY64zrjnmdG43bfBkfwTBBHOcS-4t0p43GqE,2053
@@ -95,9 +95,9 @@ waldiez/exporting/tools/processor.py,sha256=g9T60wucRHp91kTnc0Z3_ulg6Mn0LNhXWfL-
 waldiez/exporting/tools/registration.py,sha256=Jf9QSQIDPCJoePc86zEMyxWUF-4a04qA2jQemodGmEA,3896
 waldiez/io/__init__.py,sha256=x4-LxBgHd0ekLXyccgcgV8ERwKRENaJQxaegULkRW3g,3361
 waldiez/io/_ws.py,sha256=uyuIfCrBK7LDXowTQ2fsbP4jvjUcIQZqPQefx2UXmp8,5652
-waldiez/io/mqtt.py,sha256=T01XCGyoWYdIEhzxS5M9igbYLgZqhF-K63j3JQDmj5E,22418
+waldiez/io/mqtt.py,sha256=bLEH5vWuQXjtLxqyDYHI-WjZs7fhH5QraENuaBi8YXc,22660
 waldiez/io/redis.py,sha256=diz0UqgcpIejBGFb9TOoT2iVzGtHWFnhg0cfEOox25Q,25523
-waldiez/io/structured.py,sha256=n-YeW-BFgNzFJU7QXbG3rG6v6R58ZltttouYZm2rZ6E,13717
+waldiez/io/structured.py,sha256=RiR9231ayskt0RvEyENhcHuFcfYFi08807PGXjGWJzw,14622
 waldiez/io/utils.py,sha256=J2I0cuBMMjJadM8OjP7kiqFjoZp8hc9rXF9n5fdltEE,4488
 waldiez/io/ws.py,sha256=aiORAYjvSA57ZqI1mXINkkHnZzqkibaICTj2bZ9Yze8,9107
 waldiez/io/models/__init__.py,sha256=N21ET1_QEkCsJNrax8y5cDDiTrGj2rv9BfEsVvUY1m0,1240
@@ -163,7 +163,7 @@ waldiez/models/common/date_utils.py,sha256=NdVmShZjvhDVT-Y5rT8FgwQTHcDoljBJQauTs
 waldiez/models/common/dict_utils.py,sha256=_Csan1HqlGFo0jVRxLQ2-2y5jzL-9qD6ggXn10nb6Vo,1958
 waldiez/models/common/handoff.py,sha256=lJosPlt-N2RSLdbK60h77AEi9h5y6rRMMtkj-CaPp6U,12172
 waldiez/models/common/id_generator.py,sha256=XoAzdZ0Os5ds4HFQMT_Dd53ryrmG-tB4rWKrz1D-j9A,421
-waldiez/models/common/method_utils.py,sha256=sNyB1jYE0E-3n_JFf99a2sd3qj8IQesARk-Tn4c_aHU,11856
+waldiez/models/common/method_utils.py,sha256=S2jCxxWdfhVIxJN3bj5Uo0HtVuYQufvc-jC88ffQ6a8,11866
 waldiez/models/common/naming.py,sha256=w5k6q7NPsBHQTAQBwlA07D_3q2Qn6w_V0pcqxNLLyVk,3534
 waldiez/models/common/waldiez_version.py,sha256=0IDG6cIBgBjzaVSLhthhL-K5dYt3tC2rfW_mLBQDYUc,1216
 waldiez/models/flow/__init__.py,sha256=-EdEvNRRqcRuK1JAqV9glg31e8Ic7_Tx4QCzJUdEq40,580
@@ -178,23 +178,31 @@ waldiez/models/model/model.py,sha256=3yMlwdlagDU1jeIdij-esnZfUvlhyPiBp-rz3PWeFbE
 waldiez/models/model/model_data.py,sha256=QJ-mNneo-w8Q5ZtZx6eQ4WAtS5fxSQ7BcABKtmQRQwo,6140
 waldiez/models/tool/__init__.py,sha256=6-oOUyHO13MUj-Kqiw-A-2a0KahxRlIdMHgEOHXoTC0,456
 waldiez/models/tool/extra_requirements.py,sha256=kNrmWxS96FhO93NV3B5Hr5wAJrjuUWbBrclpCikW4ng,988
-waldiez/models/tool/tool.py,sha256=Mzst4x9tS-1exryTYZSAlr6B3E7ba7Q5yF9WLxXp5G8,8987
+waldiez/models/tool/tool.py,sha256=GOqtq21ZOi-go2DXmBik4hJAc_MPZ6uSo5ZrM2rgboY,9008
 waldiez/models/tool/tool_data.py,sha256=lOtIetRM35tUvDCtjiHjddVSoqlwV4NFkz_496GLunA,1305
 waldiez/models/tool/tool_type.py,sha256=VYcgnAlj1n6ZqXA8D0VdbRfZ-_U89CPSDuNgk822_UA,268
-waldiez/running/__init__.py,sha256=EqH4knSyNxYjN31YBifoLfXS5nEW4ItWMA6-CNpZszk,662
-waldiez/running/environment.py,sha256=DxiI9GqFrFiF6p5icrpCFkbsvq6zgACoqXYyHZniffQ,3472
+waldiez/running/__init__.py,sha256=4A3FHJcN4a9P8OWlKblikl89O4bzKa6JcTAPNBuwTh0,370
+waldiez/running/base_runner.py,sha256=tp_2fFU-LFORYq0wENVFi0ZVNwm6jrZ5F1Y_oy2aSvs,29828
+waldiez/running/environment.py,sha256=RMlX_yekAQKhexAHjWO1oAhJVRj88Qz01ZDLxfzYYWQ,5729
 waldiez/running/gen_seq_diagram.py,sha256=wsF9b9uG0NA5iVAlW-wOtYn0w_9-bySBO9nWkUIBNUc,5943
-waldiez/running/running.py,sha256=0nZ33hwvUAVGcemT2C2Rec8fqHkB5xU4TRiCzs1fAKE,11750
-waldiez/utils/__init__.py,sha256=ITQcsqe2jhO376VSeZGfcf0n0kUhMw-4rKxUWtFB8SY,345
+waldiez/running/import_runner.py,sha256=fW8LdhymRvYUD3Pc0DpcDkozQHC9QQNXlUVMCl3MYj8,15687
+waldiez/running/patch_io_stream.py,sha256=HcbS_580xSa1YD6qy6FalbuJcpMGh9Qc87MSybtGh2c,6808
+waldiez/running/post_run.py,sha256=kD5VIjfrqeXqRqq6DhkuZcXvh7kVaiWZ_T9-KA9D0nU,3980
+waldiez/running/pre_run.py,sha256=ASQ5uQx6lvI67kwyLVNJm6j2ZCCeCBBl40Z3t-Ra7lk,3667
+waldiez/running/protocol.py,sha256=nAIz219Hv_0o03TNzimRzA9c0Yk7LX5csdJyJb0zRjo,8452
+waldiez/running/run_results.py,sha256=c1fp5WTi8CEXIZpZRRH4nkJKW6RidhzIfIatjJVe9Kw,518
+waldiez/running/subprocess_runner.py,sha256=S_B0QhH4vSRW9qyOcIy7s0jEcT3dAen4N4sPAk4tazI,2776
+waldiez/running/utils.py,sha256=7DnuL-FNkraDhu3QUV3GEZC_dNgo3IMlptBzG9tMUKA,3002
+waldiez/utils/__init__.py,sha256=pci9Z4pO6nEnoqCPfJBDIISYDuGH65NTRC_evZbV7ao,336
 waldiez/utils/conflict_checker.py,sha256=bskxC2KmPrAOjjYfgQhGXgNxdU5Z10ZZmx-_c0W8I7I,1493
-waldiez/utils/flaml_warnings.py,sha256=gN1yzlAAHor04w0-_xXUK0wID6aXCizKYElaNHflkg8,630
+waldiez/utils/version.py,sha256=qaduzoxI1qBEtDLdM5Gw0V76YgF1WABPcDC95VL8HLQ,1452
 waldiez/utils/cli_extras/__init__.py,sha256=ZvuLaN3IGuffWMh7mladTGkJxx3kn5IepUF3jk4ZAWY,684
 waldiez/utils/cli_extras/jupyter.py,sha256=nOQraO7Xg1G8SCP93OpIVheXieXW5ireSC9Oir-dXzA,3047
 waldiez/utils/cli_extras/runner.py,sha256=V_aoTKZsvtPgjszvWlPamKjnOG93L2aY6sBcHnIh9XI,983
 waldiez/utils/cli_extras/studio.py,sha256=p4CMDCYW6lxrPvHvWX2kpsE1Ix7NnKZg1Jf15weWirE,984
-waldiez-0.4.8.dist-info/METADATA,sha256=3Ta0RN91r4JKHXSf8hZ7P4KekETUeJeLgquleK0pS6w,22620
-waldiez-0.4.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-waldiez-0.4.8.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
-waldiez-0.4.8.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-waldiez-0.4.8.dist-info/licenses/NOTICE.md,sha256=L7xtckFRYvYJjhjQNtFpURWCiAvEuq4ePvxJsC-XAdk,785
-waldiez-0.4.8.dist-info/RECORD,,
+waldiez-0.4.11.dist-info/METADATA,sha256=OiHsNqVC0_0UYWvpWcsC8K5vXD3BM4OEJM6atU8XvPI,22638
+waldiez-0.4.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+waldiez-0.4.11.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
+waldiez-0.4.11.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+waldiez-0.4.11.dist-info/licenses/NOTICE.md,sha256=L7xtckFRYvYJjhjQNtFpURWCiAvEuq4ePvxJsC-XAdk,785
+waldiez-0.4.11.dist-info/RECORD,,

waldiez/running/running.py

@@ -1,388 +0,0 @@
-# SPDX-License-Identifier: Apache-2.0.
-# Copyright (c) 2024 - 2025 Waldiez and contributors.
-"""Utilities for running code."""
-
-import asyncio
-import datetime
-import io
-import os
-import re
-import shutil
-import subprocess
-import sys
-import tempfile
-from contextlib import asynccontextmanager, contextmanager
-from pathlib import Path
-from typing import (
-    Any,
-    AsyncIterator,
-    Callable,
-    Iterator,
-    Optional,
-    Set,
-    Union,
-)
-
-from .environment import in_virtualenv, is_root
-from .gen_seq_diagram import generate_sequence_diagram
-
-# pylint: disable=import-outside-toplevel
-
-
-@contextmanager
-def chdir(to: Union[str, Path]) -> Iterator[None]:
-    """Change the current working directory in a context.
-
-    Parameters
-    ----------
-    to : Union[str, Path]
-        The directory to change to.
-
-    Yields
-    ------
-    Iterator[None]
-        The context manager.
-    """
-    old_cwd = str(os.getcwd())
-    os.chdir(to)
-    try:
-        yield
-    finally:
-        os.chdir(old_cwd)
-
-
-@asynccontextmanager
-async def a_chdir(to: Union[str, Path]) -> AsyncIterator[None]:
-    """Asynchronously change the current working directory in a context.
-
-    Parameters
-    ----------
-    to : Union[str, Path]
-        The directory to change to.
-
-    Yields
-    ------
-    AsyncIterator[None]
-        The async context manager.
-    """
-    old_cwd = str(os.getcwd())
-    os.chdir(to)
-    try:
-        yield
-    finally:
-        os.chdir(old_cwd)
-
-
-def strip_ansi(text: str) -> str:
-    """Remove ANSI escape sequences from text.
-
-    Parameters
-    ----------
-    text : str
-        The text to strip.
-
-    Returns
-    -------
-    str
-        The text without ANSI escape sequences.
-    """
-    ansi_pattern = re.compile(r"\x1b\[[0-9;]*m|\x1b\[.*?[@-~]")
-    return ansi_pattern.sub("", text)
-
-
-def before_run(
-    output_path: Optional[Union[str, Path]],
-    uploads_root: Optional[Union[str, Path]],
-) -> str:
-    """Actions to perform before running the flow.
-
-    Parameters
-    ----------
-    output_path : Optional[Union[str, Path]]
-        The output path.
-    uploads_root : Optional[Union[str, Path]]
-        The runtime uploads root.
-
-    Returns
-    -------
-    str
-        The file name.
-    """
-    if not uploads_root:
-        uploads_root = Path(tempfile.mkdtemp())
-    else:
-        uploads_root = Path(uploads_root)
-    if not uploads_root.exists():
-        uploads_root.mkdir(parents=True)
-    file_name = "waldiez_flow.py" if not output_path else Path(output_path).name
-    if file_name.endswith((".json", ".waldiez")):
-        file_name = file_name.replace(".json", ".py").replace(".waldiez", ".py")
-    if not file_name.endswith(".py"):
-        file_name += ".py"
-    return file_name
-
-
-def install_requirements(
-    extra_requirements: Set[str], printer: Callable[..., None]
-) -> None:
-    """Install the requirements.
-
-    Parameters
-    ----------
-    extra_requirements : Set[str]
-        The extra requirements.
-    printer : Callable[..., None]
-        The printer function.
-    """
-    requirements_string = ", ".join(extra_requirements)
-    printer(f"Installing requirements: {requirements_string}")
-    pip_install = [sys.executable, "-m", "pip", "install"]
-    break_system_packages = ""
-    if not in_virtualenv():  # it should
-        # if not, let's try to install as user
-        # not sure if --break-system-packages is safe,
-        # but it might fail if we don't
-        break_system_packages = os.environ.get("PIP_BREAK_SYSTEM_PACKAGES", "")
-        os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = "1"
-        if not is_root():
-            pip_install.append("--user")
-    pip_install.extend(extra_requirements)
-    # pylint: disable=too-many-try-statements
-    try:
-        with subprocess.Popen(
-            pip_install,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE,
-        ) as proc:
-            if proc.stdout:
-                for line in io.TextIOWrapper(proc.stdout, encoding="utf-8"):
-                    printer(strip_ansi(line.strip()))
-            if proc.stderr:
-                for line in io.TextIOWrapper(proc.stderr, encoding="utf-8"):
-                    printer(strip_ansi(line.strip()))
-    finally:
-        if not in_virtualenv():
-            # restore the old env var
-            if break_system_packages:
-                os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = break_system_packages
-            else:
-                del os.environ["PIP_BREAK_SYSTEM_PACKAGES"]
-
-
-async def a_install_requirements(
-    extra_requirements: Set[str], printer: Callable[..., None]
-) -> None:
-    """Install the requirements asynchronously.
-
-    Parameters
-    ----------
-    extra_requirements : Set[str]
-        The extra requirements.
-    printer : Callable[..., None]
-        The printer function.
-    """
-    requirements_string = ", ".join(extra_requirements)
-    printer(f"Installing requirements: {requirements_string}")
-    pip_install = [sys.executable, "-m", "pip", "install"]
-    break_system_packages = ""
-    if not in_virtualenv():
-        break_system_packages = os.environ.get("PIP_BREAK_SYSTEM_PACKAGES", "")
-        os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = "1"
-        if not is_root():
-            pip_install.extend(["--user"])
-    pip_install.extend(extra_requirements)
-    # pylint: disable=too-many-try-statements
-    try:
-        proc = await asyncio.create_subprocess_exec(
-            *pip_install,
-            stdout=asyncio.subprocess.PIPE,
-            stderr=asyncio.subprocess.PIPE,
-        )
-        if proc.stdout:
-            async for line in proc.stdout:
-                printer(strip_ansi(line.decode().strip()))
-        if proc.stderr:
-            async for line in proc.stderr:
-                printer(strip_ansi(line.decode().strip()))
-    finally:
-        if not in_virtualenv():
-            if break_system_packages:
-                os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = break_system_packages
-            else:
-                del os.environ["PIP_BREAK_SYSTEM_PACKAGES"]
-
-
-def after_run(
-    temp_dir: Path,
-    output_path: Optional[Union[str, Path]],
-    printer: Callable[..., None],
-    flow_name: str,
-    skip_mmd: bool = False,
-) -> None:
-    """Actions to perform after running the flow.
-
-    Parameters
-    ----------
-    temp_dir : Path
-        The temporary directory.
-    output_path : Optional[Union[str, Path]]
-        The output path.
-    printer : Callable[..., None]
-        The printer function.
-    flow_name : str
-        The flow name.
-    skip_mmd : bool, optional
-        Whether to skip the mermaid sequence diagram generation,
-        by default, False
-    """
-    if isinstance(output_path, str):
-        output_path = Path(output_path)
-    output_dir = output_path.parent if output_path else Path.cwd()
-    if skip_mmd is False:
-        events_csv_path = temp_dir / "logs" / "events.csv"
-        if events_csv_path.exists():
-            printer("Generating mermaid sequence diagram...")
-            mmd_path = temp_dir / f"{flow_name}.mmd"
-            generate_sequence_diagram(events_csv_path, mmd_path)
-            if mmd_path.exists():
-                shutil.copyfile(mmd_path, output_dir / f"{flow_name}.mmd")
-    if output_path:
-        destination_dir = output_path.parent
-        destination_dir = (
-            destination_dir
-            / "waldiez_out"
-            / datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-        )
-        destination_dir.mkdir(parents=True, exist_ok=True)
-        # copy the contents of the temp dir to the destination dir
-        printer(f"Copying the results to {destination_dir}")
-        copy_results(
-            temp_dir=temp_dir,
-            output_path=output_path,
-            output_dir=output_dir,
-            destination_dir=destination_dir,
-        )
-    shutil.rmtree(temp_dir)
-
-
-def copy_results(
-    temp_dir: Path,
-    output_path: Path,
-    output_dir: Path,
-    destination_dir: Path,
-) -> None:
-    """Copy the results to the output directory.
-
-    Parameters
-    ----------
-    temp_dir : Path
-        The temporary directory.
-    output_path : Path
-        The output path.
-    output_dir : Path
-        The output directory.
-    destination_dir : Path
-        The destination directory.
-    """
-    temp_dir.mkdir(parents=True, exist_ok=True)
-    for item in temp_dir.iterdir():
-        # skip cache files
-        if (
-            item.name.startswith("__pycache__")
-            or item.name.endswith(".pyc")
-            or item.name.endswith(".pyo")
-            or item.name.endswith(".pyd")
-            or item.name == ".cache"
-        ):
-            continue
-        if item.is_file():
-            # let's also copy the "tree of thoughts" image
-            # to the output directory
-            if item.name.endswith("tree_of_thoughts.png") or item.name.endswith(
-                "reasoning_tree.json"
-            ):
-                shutil.copy(item, output_dir / item.name)
-            shutil.copy(item, destination_dir)
-        else:
-            shutil.copytree(item, destination_dir / item.name)
-    if output_path.is_file():
-        if output_path.suffix == ".waldiez":
-            output_path = output_path.with_suffix(".py")
-        if output_path.suffix == ".py":
-            src = temp_dir / output_path.name
-            if src.exists():
-                dst = destination_dir / output_path.name
-                if dst.exists():
-                    dst.unlink()
-                shutil.copyfile(src, output_dir / output_path.name)
-
-
-def get_printer() -> Callable[..., None]:
-    """Get the printer function.
-
-    Returns
-    -------
-    Callable[..., None]
-        The printer function.
-    """
-    from autogen.io import IOStream  # type: ignore
-
-    # noinspection PyUnboundLocalVariable
-    printer = IOStream.get_default().print
-
-    def safe_printer(*args: Any, **kwargs: Any) -> None:
-        try:
-            printer(*args, **kwargs)
-        except UnicodeEncodeError:
-            try:
-                # Use the helper to get safely encoded message
-                safe_msg, flush = get_what_to_print(*args, **kwargs)
-                # Try printing the safe message
-                # (without end since it's already included)
-                printer(safe_msg, end="", flush=flush)
-            except UnicodeEncodeError:
-                # pylint: disable=too-many-try-statements
-                try:
-                    # If that still fails, write directly to buffer
-                    msg, flush = get_what_to_print(*args, **kwargs)
-                    safe_msg_bytes = msg.encode("utf-8", errors="replace")
-                    sys.stdout.buffer.write(safe_msg_bytes)
-                    if flush:
-                        sys.stdout.buffer.flush()
-                except Exception:  # pylint: disable=broad-exception-caught
-                    error_msg = (
-                        b"Could not print message due to encoding issues.\n"
-                    )
-                    sys.stderr.buffer.write(error_msg)
-                    sys.stderr.buffer.flush()
-
-    return safe_printer
-
-
-def get_what_to_print(*args: object, **kwargs: object) -> tuple[str, bool]:
-    """Get what to print.
-
-    Parameters
-    ----------
-    args : object
-        The print arguments.
-    kwargs : object
-        The keyword arguments.
-
-    Returns
-    -------
-    tuple[str, bool]
-        The message and whether to flush.
-    """
-    sep = kwargs.get("sep", " ")
-    if not isinstance(sep, str):
-        sep = " "
-    end = kwargs.get("end", "\n")
-    if not isinstance(end, str):
-        end = "\n"
-    flush = kwargs.get("flush", False)
-    if not isinstance(flush, bool):
-        flush = False
-    msg = sep.join(str(arg) for arg in args) + end
-    utf8_msg = msg.encode("utf-8", errors="replace").decode("utf-8")
-    return utf8_msg, flush

waldiez/utils/flaml_warnings.py

@@ -1,17 +0,0 @@
-# SPDX-License-Identifier: Apache-2.0.
-# Copyright (c) 2024 - 2025 Waldiez and contributors.
-"""Try to suppress the warning about flaml.automl not being available."""
-
-import logging
-
-__waldiez_checked_flaml_warnings = False  # pylint: disable=invalid-name
-
-
-def check_flaml_warnings() -> None:  # pragma: no cover
-    """Check for flaml warnings once."""
-    # pylint: disable=global-statement
-    global __waldiez_checked_flaml_warnings
-    if __waldiez_checked_flaml_warnings is False:
-        flam_logger = logging.getLogger("flaml")
-        flam_logger.setLevel(logging.ERROR)
-        __waldiez_checked_flaml_warnings = True