waldiez 0.5.8__py3-none-any.whl → 0.5.10__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of waldiez might be problematic. Click here for more details.
- waldiez/_version.py +1 -1
- waldiez/cli.py +112 -24
- waldiez/exporting/agent/exporter.py +3 -0
- waldiez/exporting/agent/extras/captain_agent_extras.py +44 -7
- waldiez/exporting/agent/extras/handoffs/condition.py +3 -1
- waldiez/exporting/chats/utils/common.py +25 -23
- waldiez/exporting/core/__init__.py +0 -2
- waldiez/exporting/core/context.py +13 -13
- waldiez/exporting/core/protocols.py +0 -141
- waldiez/exporting/core/result.py +5 -5
- waldiez/exporting/flow/merger.py +2 -2
- waldiez/exporting/flow/orchestrator.py +1 -0
- waldiez/exporting/flow/utils/common.py +2 -2
- waldiez/exporting/flow/utils/importing.py +1 -0
- waldiez/exporting/flow/utils/logging.py +6 -7
- waldiez/exporting/tools/exporter.py +5 -0
- waldiez/exporting/tools/factory.py +4 -0
- waldiez/exporting/tools/processor.py +5 -1
- waldiez/io/_ws.py +13 -5
- waldiez/io/models/content/image.py +1 -0
- waldiez/io/models/user_input.py +4 -4
- waldiez/io/models/user_response.py +1 -0
- waldiez/io/mqtt.py +1 -1
- waldiez/io/structured.py +17 -17
- waldiez/io/utils.py +1 -1
- waldiez/io/ws.py +9 -11
- waldiez/logger.py +180 -63
- waldiez/models/agents/agent/update_system_message.py +0 -2
- waldiez/models/agents/doc_agent/doc_agent.py +8 -1
- waldiez/models/common/dict_utils.py +169 -40
- waldiez/models/flow/flow.py +6 -6
- waldiez/models/flow/info.py +5 -1
- waldiez/models/model/_llm.py +28 -14
- waldiez/models/model/model.py +4 -1
- waldiez/models/model/model_data.py +18 -5
- waldiez/models/tool/predefined/_config.py +5 -1
- waldiez/models/tool/predefined/_duckduckgo.py +4 -0
- waldiez/models/tool/predefined/_email.py +474 -0
- waldiez/models/tool/predefined/_google.py +8 -6
- waldiez/models/tool/predefined/_perplexity.py +3 -0
- waldiez/models/tool/predefined/_searxng.py +3 -0
- waldiez/models/tool/predefined/_tavily.py +4 -1
- waldiez/models/tool/predefined/_wikipedia.py +4 -1
- waldiez/models/tool/predefined/_youtube.py +4 -1
- waldiez/models/tool/predefined/protocol.py +3 -0
- waldiez/models/tool/tool.py +22 -4
- waldiez/models/waldiez.py +12 -0
- waldiez/runner.py +37 -54
- waldiez/running/__init__.py +6 -0
- waldiez/running/base_runner.py +310 -353
- waldiez/running/environment.py +1 -0
- waldiez/running/exceptions.py +9 -0
- waldiez/running/post_run.py +4 -4
- waldiez/running/pre_run.py +51 -40
- waldiez/running/protocol.py +21 -101
- waldiez/running/run_results.py +1 -1
- waldiez/running/standard_runner.py +84 -277
- waldiez/running/step_by_step/__init__.py +46 -0
- waldiez/running/step_by_step/breakpoints_mixin.py +188 -0
- waldiez/running/step_by_step/step_by_step_models.py +224 -0
- waldiez/running/step_by_step/step_by_step_runner.py +745 -0
- waldiez/running/subprocess_runner/__base__.py +282 -0
- waldiez/running/subprocess_runner/__init__.py +16 -0
- waldiez/running/subprocess_runner/_async_runner.py +362 -0
- waldiez/running/subprocess_runner/_sync_runner.py +455 -0
- waldiez/running/subprocess_runner/runner.py +561 -0
- waldiez/running/timeline_processor.py +1 -1
- waldiez/running/utils.py +376 -1
- waldiez/utils/version.py +2 -6
- waldiez/ws/__init__.py +70 -0
- waldiez/ws/__main__.py +15 -0
- waldiez/ws/_file_handler.py +201 -0
- waldiez/ws/cli.py +211 -0
- waldiez/ws/client_manager.py +835 -0
- waldiez/ws/errors.py +416 -0
- waldiez/ws/models.py +971 -0
- waldiez/ws/reloader.py +342 -0
- waldiez/ws/server.py +469 -0
- waldiez/ws/session_manager.py +393 -0
- waldiez/ws/session_stats.py +83 -0
- waldiez/ws/utils.py +385 -0
- {waldiez-0.5.8.dist-info → waldiez-0.5.10.dist-info}/METADATA +74 -74
- {waldiez-0.5.8.dist-info → waldiez-0.5.10.dist-info}/RECORD +87 -65
- waldiez/running/patch_io_stream.py +0 -210
- {waldiez-0.5.8.dist-info → waldiez-0.5.10.dist-info}/WHEEL +0 -0
- {waldiez-0.5.8.dist-info → waldiez-0.5.10.dist-info}/entry_points.txt +0 -0
- {waldiez-0.5.8.dist-info → waldiez-0.5.10.dist-info}/licenses/LICENSE +0 -0
- {waldiez-0.5.8.dist-info → waldiez-0.5.10.dist-info}/licenses/NOTICE.md +0 -0
|
@@ -1,11 +1,11 @@
|
|
|
1
1
|
waldiez/__init__.py,sha256=GdbXP8mS6FUiljxvd-eH_J2gEztoHCQpD8Hq51GKUSs,943
|
|
2
2
|
waldiez/__main__.py,sha256=0dYzNrQbovRwQQvmZC6_1FDR1m71SUIOkTleO5tBnBw,203
|
|
3
|
-
waldiez/_version.py,sha256=
|
|
4
|
-
waldiez/cli.py,sha256=
|
|
3
|
+
waldiez/_version.py,sha256=F6MOiQ6KLN0ozUof4ZB080jTwzMVUNQe0AnZTNDypc4,250
|
|
4
|
+
waldiez/cli.py,sha256=J62MP4VCLvUFcT3aupOLytqQ0gC_GqQzQMst3mqycpo,10100
|
|
5
5
|
waldiez/exporter.py,sha256=A-j2CjQIz0sRi_lRoUcZFGsoHK_PmorGzMPffVtn7Fo,7766
|
|
6
|
-
waldiez/logger.py,sha256=
|
|
6
|
+
waldiez/logger.py,sha256=FXJDMThkKYO18f_t1gncurhTNorNM59btQTmmFWPiMQ,20362
|
|
7
7
|
waldiez/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
8
|
-
waldiez/runner.py,sha256=
|
|
8
|
+
waldiez/runner.py,sha256=RcO1WQnSc1I3nhX02TfsbpgInuijG44dv2Uxzn_kbA0,9077
|
|
9
9
|
waldiez/cli_extras/__init__.py,sha256=ZvuLaN3IGuffWMh7mladTGkJxx3kn5IepUF3jk4ZAWY,684
|
|
10
10
|
waldiez/cli_extras/jupyter.py,sha256=nOQraO7Xg1G8SCP93OpIVheXieXW5ireSC9Oir-dXzA,3047
|
|
11
11
|
waldiez/cli_extras/runner.py,sha256=V_aoTKZsvtPgjszvWlPamKjnOG93L2aY6sBcHnIh9XI,983
|
|
@@ -13,13 +13,13 @@ waldiez/cli_extras/studio.py,sha256=p4CMDCYW6lxrPvHvWX2kpsE1Ix7NnKZg1Jf15weWirE,
|
|
|
13
13
|
waldiez/exporting/__init__.py,sha256=xVOqb3gADkrTmp_BVrQ-dMPiySPUDU4xZmqYqhzL8BI,1026
|
|
14
14
|
waldiez/exporting/agent/__init__.py,sha256=GpNfVQLAvGoJP_U4WkMMIGpEB8k7vEV-FVR446s2zhY,260
|
|
15
15
|
waldiez/exporting/agent/code_execution.py,sha256=Oc-jfsJG0WC80MCNBohYCw3TOiDrbfHfwVrTkrXLEow,3957
|
|
16
|
-
waldiez/exporting/agent/exporter.py,sha256=
|
|
16
|
+
waldiez/exporting/agent/exporter.py,sha256=VnZ8Z_LRlynLC68tbhZrBnSZriYlVDNscMkAcS_CdNw,14583
|
|
17
17
|
waldiez/exporting/agent/factory.py,sha256=x0oJCyF4OHH1CM9rs0EQOHlQKPBPrhbk60e3sDFoRfM,3094
|
|
18
18
|
waldiez/exporting/agent/processor.py,sha256=tv2lqhP2AiDaen63BN49b9BzKn41yTQUTE7Wl2rotZY,7519
|
|
19
19
|
waldiez/exporting/agent/system_message.py,sha256=HEVcfpCC4NEB4XP6eFhZcosiZBMCuDhjqywIXIb0seM,941
|
|
20
20
|
waldiez/exporting/agent/termination.py,sha256=wL9Bbd2yDyi90Dd4P9lVb5fMhKPBhqFO9yZpKVQKYto,1566
|
|
21
21
|
waldiez/exporting/agent/extras/__init__.py,sha256=o9q-PTzagOC1mPlSWTlyJU1jNnQQob3n2pfYVU3gxJU,570
|
|
22
|
-
waldiez/exporting/agent/extras/captain_agent_extras.py,sha256=
|
|
22
|
+
waldiez/exporting/agent/extras/captain_agent_extras.py,sha256=j7K2tmCDccXTUwdT6w_nTPs8exwCsHj_tUJ1NdhcFb0,11846
|
|
23
23
|
waldiez/exporting/agent/extras/doc_agent_extras.py,sha256=jLyjfPkLWieIItHw66HKFiM080j1e0cIYnu1oEQGEm0,12838
|
|
24
24
|
waldiez/exporting/agent/extras/group_manager_agent_extas.py,sha256=xRZ7lmMAb2wWRVZoUqwsYcLRtAth7XyrWYDN0pm1FbA,18068
|
|
25
25
|
waldiez/exporting/agent/extras/group_member_extras.py,sha256=Heo3Gkr8QiihkWxoJKs8dXEEK5myKvu97AwN15Dslsw,5680
|
|
@@ -28,7 +28,7 @@ waldiez/exporting/agent/extras/reasoning_agent_extras.py,sha256=9Mf5bz_h99AHWRMk
|
|
|
28
28
|
waldiez/exporting/agent/extras/handoffs/__init__.py,sha256=az26bx7kl3Iy0UsUiJoruuBQTJetkQpp4k_zCw1HZ-o,504
|
|
29
29
|
waldiez/exporting/agent/extras/handoffs/after_work.py,sha256=xWG0NCI7R_Yb5p9lFsFnLzEJzOecSDkiJ8M-hwaxPVM,2429
|
|
30
30
|
waldiez/exporting/agent/extras/handoffs/available.py,sha256=7e6NkmyDPWCAu00BDhJ_dmf-aNdUJOuddnf71uOaKOA,2375
|
|
31
|
-
waldiez/exporting/agent/extras/handoffs/condition.py,sha256=
|
|
31
|
+
waldiez/exporting/agent/extras/handoffs/condition.py,sha256=Ma2F4H7rWwbzBAioW73jz7sHymjrKiCMdhgivjtNzhY,5457
|
|
32
32
|
waldiez/exporting/agent/extras/handoffs/handoff.py,sha256=Xp0sYpXIoXQ6GbLrLPxylDpyNPiE2w1oXmQ-UqfnK7U,5918
|
|
33
33
|
waldiez/exporting/agent/extras/handoffs/target.py,sha256=GNeq0RsY-0i2-6NkwVzwMm6osPzCQa_olBwSbcLiPco,6638
|
|
34
34
|
waldiez/exporting/agent/extras/rag/__init__.py,sha256=jkKSaHLfwA8Tj7QJH0MSdAphaP4HowuhQXDj7V122Ek,259
|
|
@@ -42,21 +42,21 @@ waldiez/exporting/chats/exporter.py,sha256=LvWnmigUrXYhuTSSGB-O1AZTjc0Rwe98kwqLD
|
|
|
42
42
|
waldiez/exporting/chats/factory.py,sha256=j10j3AKXN1iin-y4w7ytVpM8ocdwEM3ZpQSu8wRCzS4,1963
|
|
43
43
|
waldiez/exporting/chats/processor.py,sha256=ARckmLAFOhm_H5nS-nrXdayekBinRp3RcL1aqVojJ_g,7741
|
|
44
44
|
waldiez/exporting/chats/utils/__init__.py,sha256=N5lfl_veJH7smiKXxmaBDew1cgUNcX-PxnWl3u4tj_U,509
|
|
45
|
-
waldiez/exporting/chats/utils/common.py,sha256=
|
|
45
|
+
waldiez/exporting/chats/utils/common.py,sha256=uRs73RarQjgW9NCTYsGKjcyAO1I48RmT71XMxSUzz3M,5872
|
|
46
46
|
waldiez/exporting/chats/utils/group.py,sha256=qaBIyiUg96i7Y0e-HBTSXHY8EGoB52dFWTxsovpuLDI,1606
|
|
47
47
|
waldiez/exporting/chats/utils/nested.py,sha256=lCzq6IdYzgOG7XyJwJuCgR8sMUOxshjcT1Nc-JU7Sek,8703
|
|
48
48
|
waldiez/exporting/chats/utils/sequential.py,sha256=S9bFvxIVb1QOwuyao9VzFQ56flKiFIwOMdwZ8ohkfp8,6488
|
|
49
49
|
waldiez/exporting/chats/utils/single.py,sha256=Ii5IEcKx926LzVInV-n_ke87yXr69ErHw95xLtE7vPM,8365
|
|
50
|
-
waldiez/exporting/core/__init__.py,sha256=
|
|
50
|
+
waldiez/exporting/core/__init__.py,sha256=o5y0r1rWH8olKptHNkY1b7uy7beL1wNCi-I4h_EBosg,3886
|
|
51
51
|
waldiez/exporting/core/constants.py,sha256=PcJQftkXX7FsDkilH_jihje_9j9SenCjUa5Hj-95akQ,492
|
|
52
52
|
waldiez/exporting/core/content.py,sha256=uxOnySvI30a2zgzkvdzkzhppZ-S1QnSxWPSJhy9wPMg,2063
|
|
53
|
-
waldiez/exporting/core/context.py,sha256=
|
|
53
|
+
waldiez/exporting/core/context.py,sha256=FOtO7n7XZ1WtdgrGEI3gODjFkpIjLMeG7vV88gRTis8,7727
|
|
54
54
|
waldiez/exporting/core/enums.py,sha256=CvmtPRv-4GwKpO2l0wXxxoV8Q5Oy5L5t0gnEMK8MVws,2464
|
|
55
55
|
waldiez/exporting/core/errors.py,sha256=xLfq3Z4Kg2aqlu7zYOtBd5ces7RLG-o_nF82ITFsaJg,535
|
|
56
56
|
waldiez/exporting/core/exporter.py,sha256=aDYSsd6oErKkJSBMIQhhy07Bc1Hoi8Gw6qB6G9VibuM,11182
|
|
57
57
|
waldiez/exporting/core/exporters.py,sha256=nQ8LgPdLqFrFTRC-99wwvV3Va_HNgWTwksqlzTPepjc,2818
|
|
58
|
-
waldiez/exporting/core/protocols.py,sha256=
|
|
59
|
-
waldiez/exporting/core/result.py,sha256=
|
|
58
|
+
waldiez/exporting/core/protocols.py,sha256=N1tdY5OWnkDWPP6BhxSRdItDjwDspd7QvAP9KsUi3iU,3428
|
|
59
|
+
waldiez/exporting/core/result.py,sha256=jadI-HLRz1HqtRARZaqgGiCa_AumphgHBEVLGy4Rhag,21109
|
|
60
60
|
waldiez/exporting/core/types.py,sha256=nG7BzcLX8OIKgYflpSZR4RFQINvu_S8u10gB0_uf0R0,10184
|
|
61
61
|
waldiez/exporting/core/validation.py,sha256=YlYpv4-VnBwmllznB1uijexPHaYDOrtnNwkS5QlQKF8,2728
|
|
62
62
|
waldiez/exporting/core/extras/__init__.py,sha256=I9t7fmPpcLTwe0bErsfyoLcC6VrimJ2fw52K-JuAUYU,962
|
|
@@ -81,43 +81,43 @@ waldiez/exporting/flow/execution_generator.py,sha256=NGB33uNFR6ELPfUyOMZ4aDj3w62
|
|
|
81
81
|
waldiez/exporting/flow/exporter.py,sha256=e0APVleLgIvfnr-OXMsptjPgR9JWLX4lnVCItfs7jLc,3392
|
|
82
82
|
waldiez/exporting/flow/factory.py,sha256=qySg6cG3ZyGrGFsJlyIyXjqOYgsNjyEFWtC-rmjJ_3E,3246
|
|
83
83
|
waldiez/exporting/flow/file_generator.py,sha256=Y0711XMorEKLP9q5bFkuCTlLL0u4zo8w4JgWsnFTd5E,7009
|
|
84
|
-
waldiez/exporting/flow/merger.py,sha256=
|
|
85
|
-
waldiez/exporting/flow/orchestrator.py,sha256=
|
|
84
|
+
waldiez/exporting/flow/merger.py,sha256=uD0LlRchGVLCS-H_PGvl3O4YnBFk9abQ_mPk8Tu0igI,12268
|
|
85
|
+
waldiez/exporting/flow/orchestrator.py,sha256=8TB94Win-Z3wXfF29an_NHpob0VkQQP2Xxb1WBBzaps,16155
|
|
86
86
|
waldiez/exporting/flow/utils/__init__.py,sha256=1QpDJvQD5g1Z6FTm_QeNKyxcUayM-JuSnAfS2Q7yaog,646
|
|
87
|
-
waldiez/exporting/flow/utils/common.py,sha256=
|
|
88
|
-
waldiez/exporting/flow/utils/importing.py,sha256=
|
|
87
|
+
waldiez/exporting/flow/utils/common.py,sha256=ilRIBZW6SUjPn-7OA-fHoTABdrxrkTh29QLNhP442Yc,5780
|
|
88
|
+
waldiez/exporting/flow/utils/importing.py,sha256=OLMJw6fz3dnS_jO5Kg36SLZRs1ln6btHHxucSKmawU0,11499
|
|
89
89
|
waldiez/exporting/flow/utils/linting.py,sha256=N5WHCn0vANMWzl4Wzi5KA_2SwfIeVdig4_jalC5gh6I,4686
|
|
90
|
-
waldiez/exporting/flow/utils/logging.py,sha256
|
|
90
|
+
waldiez/exporting/flow/utils/logging.py,sha256=oLmaO6B0AgMF188JNyKoEI0YAciseTGzvmOxfo9zRlY,12023
|
|
91
91
|
waldiez/exporting/models/__init__.py,sha256=xYq6kUfoeE5Jyml0PJJiviayEPFh82kxbk9C208BvbQ,279
|
|
92
92
|
waldiez/exporting/models/exporter.py,sha256=4vKwys8VcqA-H-35WjwZa-fCFfqRT9-CUFHzQxU-dcA,7226
|
|
93
93
|
waldiez/exporting/models/factory.py,sha256=z0L4ZVtVWfZR8zbpq-rEol9gvIUHl5Kc1HZiKMN5eKA,1988
|
|
94
94
|
waldiez/exporting/models/processor.py,sha256=LxJDnwoZz6VIw2PjynPBeYkx037hhidtKX-YW5R8SoQ,4331
|
|
95
95
|
waldiez/exporting/tools/__init__.py,sha256=sKodShu-YROgOak5a3tf0f60Ab3LQDJQXO3BMFFtwKQ,257
|
|
96
|
-
waldiez/exporting/tools/exporter.py,sha256=
|
|
97
|
-
waldiez/exporting/tools/factory.py,sha256=
|
|
98
|
-
waldiez/exporting/tools/processor.py,sha256=
|
|
96
|
+
waldiez/exporting/tools/exporter.py,sha256=CEsM2ghQUgI5EbJayn7IrhMUzHUkl2uB0eBKhdNMo9c,6898
|
|
97
|
+
waldiez/exporting/tools/factory.py,sha256=ViSEmo081fcsafc2V167lod6Als_IEMpO8Bdg6Rn68M,1708
|
|
98
|
+
waldiez/exporting/tools/processor.py,sha256=vt0vQSaJf1EXNJ6Dz54ebXXP99vnp64v5lZQSis3dDM,8031
|
|
99
99
|
waldiez/exporting/tools/registration.py,sha256=cnbtMt0YlSjTF6x4q7YIMNNBU9FALcdZ0y9IZCNtcts,3935
|
|
100
100
|
waldiez/io/__init__.py,sha256=x4-LxBgHd0ekLXyccgcgV8ERwKRENaJQxaegULkRW3g,3361
|
|
101
|
-
waldiez/io/_ws.py,sha256=
|
|
102
|
-
waldiez/io/mqtt.py,sha256=
|
|
101
|
+
waldiez/io/_ws.py,sha256=n7g7hwqjZKJBm7FJZpO-3htq4742hH-IZZNdulFeGb4,6047
|
|
102
|
+
waldiez/io/mqtt.py,sha256=EFjkYrMwRT2Tq_DvAJ70EOsAo06AswhVMQHxCWG-Bm4,23600
|
|
103
103
|
waldiez/io/redis.py,sha256=q6qK1Q_dSgPrMRSjIike4NQS3nwqx8_ulM-es2mTKnw,25800
|
|
104
|
-
waldiez/io/structured.py,sha256=
|
|
105
|
-
waldiez/io/utils.py,sha256=
|
|
106
|
-
waldiez/io/ws.py,sha256=
|
|
104
|
+
waldiez/io/structured.py,sha256=67WtZU67tUq0-pgZHpxZmLdGml1JESeIOJ4qhPne0ds,14866
|
|
105
|
+
waldiez/io/utils.py,sha256=Ag9nQ7OC5hPnxC43O9xD0E_QfElgE7-05FvAyf2Wmzs,5628
|
|
106
|
+
waldiez/io/ws.py,sha256=o4lkQBwAdf60XP-ifzYlboVC29iT2hyIXFbeoGFupS8,9339
|
|
107
107
|
waldiez/io/models/__init__.py,sha256=N21ET1_QEkCsJNrax8y5cDDiTrGj2rv9BfEsVvUY1m0,1240
|
|
108
108
|
waldiez/io/models/base.py,sha256=zrnjEgHXlMvohT4q1lCQSlcEOIi3MjeF4zZ5Rv-6c0c,1715
|
|
109
109
|
waldiez/io/models/constants.py,sha256=e4MJ4sAwadCF9zRRYb93dMozwhK5Z3YfVJVQtAL_P5Y,2011
|
|
110
|
-
waldiez/io/models/user_input.py,sha256=
|
|
111
|
-
waldiez/io/models/user_response.py,sha256=
|
|
110
|
+
waldiez/io/models/user_input.py,sha256=bWFAoRyyQl-DE0UNU--3fajAmzAbPzNjOAvOD8bTx_8,8342
|
|
111
|
+
waldiez/io/models/user_response.py,sha256=bZI_6WmfVRVitSoeKCfxSU6Ik4-Pw77ZSrqRvJLRRJ4,7075
|
|
112
112
|
waldiez/io/models/content/__init__.py,sha256=bCDezBSeOKFPKhH689p7cWQx0CbbenlnZYml9_wrPB4,652
|
|
113
113
|
waldiez/io/models/content/audio.py,sha256=neBHLUyEaYjHdrlv8lDuqxyCLdtWa94tdxdttsKD_jo,1150
|
|
114
114
|
waldiez/io/models/content/base.py,sha256=gFKBzoS0VkvX4HbZOyjKkBwenNxeMsSF892J-u5hLuI,1073
|
|
115
115
|
waldiez/io/models/content/file.py,sha256=TKoAoBVruzR83z7aLna1kKlLc4PbLnd6MfnidPk9SHo,1155
|
|
116
|
-
waldiez/io/models/content/image.py,sha256=
|
|
116
|
+
waldiez/io/models/content/image.py,sha256=xmNTAI0Kjgwv9NUmqd3I65eXQiQWWXXNU9sX4BSgKTE,2681
|
|
117
117
|
waldiez/io/models/content/text.py,sha256=DR6qDp2gWulBmzfxWLYTWV18wteY1l8Cyf3neetageY,932
|
|
118
118
|
waldiez/io/models/content/video.py,sha256=7t5pj4IKrHTl6FH-8bZOOFsNN23oBdp4AaajzlIHtuQ,1171
|
|
119
119
|
waldiez/models/__init__.py,sha256=4d8WtmNCiLV6g6QYr4vnBnivdQJaylSGs_JNYXuGCdA,5653
|
|
120
|
-
waldiez/models/waldiez.py,sha256=
|
|
120
|
+
waldiez/models/waldiez.py,sha256=aU_jpeUWG2h5p9YRgEVjERfGKy08XJus4WFEzYxVHLY,10952
|
|
121
121
|
waldiez/models/agents/__init__.py,sha256=0nTNPK3izYPf6NarJq89mHqDbn2k3J5-WGYx7lf3ZVg,4078
|
|
122
122
|
waldiez/models/agents/agents.py,sha256=2Zb8mwbef7vOXG1sCiPbj_J7jNVgfHmy7TFVm7Qn4G4,5110
|
|
123
123
|
waldiez/models/agents/extra_requirements.py,sha256=XfHSRseqimGz3knOd3CX44OCZwuJzRsQZLuga1WS6xA,2420
|
|
@@ -130,7 +130,7 @@ waldiez/models/agents/agent/human_input_mode.py,sha256=nYoRXsoNSmwxKeyuY9c7De9xy
|
|
|
130
130
|
waldiez/models/agents/agent/linked_tool.py,sha256=_ccap59KO6AAZ0TizfItV4tPUlHAZlzCQ9jd0Uljaws,753
|
|
131
131
|
waldiez/models/agents/agent/nested_chat.py,sha256=I77QdvJuwEILvI8riF-S8IhgwUt0V2XcibvRxMZB-ZI,2655
|
|
132
132
|
waldiez/models/agents/agent/termination_message.py,sha256=Q0YPj-W3X6KqnxcZGc_mivyOa5Z-8bgieyvPIfxRO7w,6979
|
|
133
|
-
waldiez/models/agents/agent/update_system_message.py,sha256=
|
|
133
|
+
waldiez/models/agents/agent/update_system_message.py,sha256=I4xJHJIK8BgUEcisEHmr7JUpViEzBXz7V_tjpBbkOyw,4665
|
|
134
134
|
waldiez/models/agents/assistant/__init__.py,sha256=y2s2F2e7njMcmwye1UOUULHSvU3VsDLreTLvRU1I4M0,268
|
|
135
135
|
waldiez/models/agents/assistant/assistant.py,sha256=Abp7x-U0LSXGr6ZVQa7rF_JVt1GpjW7ThfuDpG3i9ns,1259
|
|
136
136
|
waldiez/models/agents/assistant/assistant_data.py,sha256=qAsurUJ9KNbPQTFDvgK_gV3y1lRDzmE_RxGgb69HtOI,1281
|
|
@@ -139,7 +139,7 @@ waldiez/models/agents/captain/captain_agent.py,sha256=Fz_YfraDHrWf0SBUvSyml8F-Vx
|
|
|
139
139
|
waldiez/models/agents/captain/captain_agent_data.py,sha256=3G3RkFrNisd1onuBAOy3FEaL74v4-OMYlkeBdbwhlHs,1741
|
|
140
140
|
waldiez/models/agents/captain/captain_agent_lib_entry.py,sha256=oxuKB6w8LuCUPqNcc78NsdVeiAMM_BnvuFtUOlZfeG0,868
|
|
141
141
|
waldiez/models/agents/doc_agent/__init__.py,sha256=ffxeWkjpxploI1KW3OE_b9tjfa_KFBMZlDLdJ6BL7zE,365
|
|
142
|
-
waldiez/models/agents/doc_agent/doc_agent.py,sha256=
|
|
142
|
+
waldiez/models/agents/doc_agent/doc_agent.py,sha256=z9dUntH-p7KhbdXt2tTI0JnMQlex1LPBeeKkpdnNGcM,3802
|
|
143
143
|
waldiez/models/agents/doc_agent/doc_agent_data.py,sha256=eh2_uW37GN9zm_LL5qaKlCMaJ9J6EExu_ttvQkGqtNs,4847
|
|
144
144
|
waldiez/models/agents/doc_agent/rag_query_engine.py,sha256=22wIcshA0CHdbwtCiJ2wD19edFmhdCAD168fkr3LaWI,3551
|
|
145
145
|
waldiez/models/agents/group_manager/__init__.py,sha256=MlMzrAfiSRprVJvAVNLUuulckNsBuWsiHXa0-tcUCWM,894
|
|
@@ -168,7 +168,7 @@ waldiez/models/common/__init__.py,sha256=ASr7j6CrAQ5RVs2Jh6gW2vsrfQvlouNbnD91ums
|
|
|
168
168
|
waldiez/models/common/ag2_version.py,sha256=WqaRaKuKjXYeQwRq04iOdSNt08TYpZCNdYsq2VNEjzE,736
|
|
169
169
|
waldiez/models/common/base.py,sha256=Jnq1d56rt37_WnZAABMVzWu_mpB4W0nycqVQSYjB0vU,3409
|
|
170
170
|
waldiez/models/common/date_utils.py,sha256=NdVmShZjvhDVT-Y5rT8FgwQTHcDoljBJQauTsE7gwbE,437
|
|
171
|
-
waldiez/models/common/dict_utils.py,sha256=
|
|
171
|
+
waldiez/models/common/dict_utils.py,sha256=FFm6FCx2a6Rf8VUjEBe0Lv4UnO4B0US103g2tuEWju8,5116
|
|
172
172
|
waldiez/models/common/handoff.py,sha256=BYozmXOUCUPU7LSL5lw2yonXNwybqagtyjZ2CyI1QQo,12467
|
|
173
173
|
waldiez/models/common/id_generator.py,sha256=XoAzdZ0Os5ds4HFQMT_Dd53ryrmG-tB4rWKrz1D-j9A,421
|
|
174
174
|
waldiez/models/common/method_utils.py,sha256=S2jCxxWdfhVIxJN3bj5Uo0HtVuYQufvc-jC88ffQ6a8,11866
|
|
@@ -176,51 +176,73 @@ waldiez/models/common/naming.py,sha256=IwthgnUEuwbbdh6tmkKqg99aj9Lyn2xA0XbHXK4yy
|
|
|
176
176
|
waldiez/models/common/waldiez_version.py,sha256=0IDG6cIBgBjzaVSLhthhL-K5dYt3tC2rfW_mLBQDYUc,1216
|
|
177
177
|
waldiez/models/flow/__init__.py,sha256=-EdEvNRRqcRuK1JAqV9glg31e8Ic7_Tx4QCzJUdEq40,580
|
|
178
178
|
waldiez/models/flow/connection.py,sha256=0izOcOzNkmsiFuNfx9YFUHyHcHsV__fNfF4Ho_r_ulg,380
|
|
179
|
-
waldiez/models/flow/flow.py,sha256=
|
|
179
|
+
waldiez/models/flow/flow.py,sha256=rRFW5U-O2UBNbU0vMtBN4dAUhYwywnaBUjVp5HqfYhM,20380
|
|
180
180
|
waldiez/models/flow/flow_data.py,sha256=hAGG7p4dLzidT5UYBkSfde1QkuXavq1fVzJSviIjiho,10841
|
|
181
|
-
waldiez/models/flow/info.py,sha256
|
|
181
|
+
waldiez/models/flow/info.py,sha256=--C4W9EMbesQsyfbGdbrcpj7U53t18-R8y1GZro99m4,2581
|
|
182
182
|
waldiez/models/flow/naming.py,sha256=6v5BpNmBWuWxP_DuHoAGcSwQqwikz9S9k0T8QH3h72k,3845
|
|
183
183
|
waldiez/models/model/__init__.py,sha256=TpgMqB3b8aanVMAdm6DqG-pRtQU--3yUEZhsPtl0QGs,621
|
|
184
184
|
waldiez/models/model/_aws.py,sha256=YK2bHjujxPLj31N0CCsO-SiIgCR2SZAxSv0f85_ypCk,1866
|
|
185
|
-
waldiez/models/model/_llm.py,sha256=
|
|
185
|
+
waldiez/models/model/_llm.py,sha256=Owlsi6pPMv1n30vDgP2tXPb-Pj9R7FuJz8BD1yomCeA,16976
|
|
186
186
|
waldiez/models/model/_price.py,sha256=PbysFzmm2vKfT3qhcoOjNZgt9C0SWS3QfFWi2aizPc0,765
|
|
187
187
|
waldiez/models/model/extra_requirements.py,sha256=3sQrF_--pm4Gom5CS9BzWQwHdoTTozfQCgPpBWvgyQ0,1319
|
|
188
|
-
waldiez/models/model/model.py,sha256=
|
|
189
|
-
waldiez/models/model/model_data.py,sha256=
|
|
188
|
+
waldiez/models/model/model.py,sha256=xBDrN6SfX25LqaOGcByeb8sm3TZS9byZ3lQd_gWYPcQ,10849
|
|
189
|
+
waldiez/models/model/model_data.py,sha256=3QnyS7yST4YaBGuoKV8kPzWnOhR2BcOuovvVP5PldSQ,4484
|
|
190
190
|
waldiez/models/tool/__init__.py,sha256=6-oOUyHO13MUj-Kqiw-A-2a0KahxRlIdMHgEOHXoTC0,456
|
|
191
191
|
waldiez/models/tool/extra_requirements.py,sha256=6Zw1IQSbT0Njkmu2BEzLT7vZ75IVNih4FZe1R3eq68s,1174
|
|
192
|
-
waldiez/models/tool/tool.py,sha256=
|
|
192
|
+
waldiez/models/tool/tool.py,sha256=nrUrjqeTVzYOCgQro7Mim6-CDk1xetvvEpr9IwvmasM,12455
|
|
193
193
|
waldiez/models/tool/tool_data.py,sha256=dUwG7hgbMRufbaWEx68adOqQQGXxM5FHpKScXFF_mOo,2605
|
|
194
194
|
waldiez/models/tool/tool_type.py,sha256=Jlm0lMymnJsxbtuvGIHjIGrnWUMHlITyFIIwJ88ergg,288
|
|
195
195
|
waldiez/models/tool/predefined/__init__.py,sha256=LyTZzG2fSWEWwGxgTGGaM1r1LQu-EXRuoizeBEjAK_g,615
|
|
196
|
-
waldiez/models/tool/predefined/_config.py,sha256=
|
|
197
|
-
waldiez/models/tool/predefined/_duckduckgo.py,sha256=
|
|
198
|
-
waldiez/models/tool/predefined/
|
|
199
|
-
waldiez/models/tool/predefined/
|
|
200
|
-
waldiez/models/tool/predefined/
|
|
201
|
-
waldiez/models/tool/predefined/
|
|
202
|
-
waldiez/models/tool/predefined/
|
|
203
|
-
waldiez/models/tool/predefined/
|
|
204
|
-
waldiez/models/tool/predefined/
|
|
196
|
+
waldiez/models/tool/predefined/_config.py,sha256=dlPHA7Lu2iRwyMEbweUMcnBDfd1nfbcer8Ho8GBdLaA,1997
|
|
197
|
+
waldiez/models/tool/predefined/_duckduckgo.py,sha256=bGSC_7y3kUtoIttLtydiPdqdLZeW9md30BwQbaxzuaw,3695
|
|
198
|
+
waldiez/models/tool/predefined/_email.py,sha256=0BIFgB4gepubNnw003KmdCI-3lAZFOxZxi_H_1Lq4_M,14106
|
|
199
|
+
waldiez/models/tool/predefined/_google.py,sha256=bpv6jNwO9utY5CLi4aIOM534QA4gZ30ZfzEtNrA6L4Y,5761
|
|
200
|
+
waldiez/models/tool/predefined/_perplexity.py,sha256=RcxBCr7FFHE_YjZW9o8-1PqAl8tXt4NxBo2a_xLlBKg,5460
|
|
201
|
+
waldiez/models/tool/predefined/_searxng.py,sha256=Tr3KvrnazdZ5piP16R8XAc58KZJWoPAxBH9U7oM5Bqc,4674
|
|
202
|
+
waldiez/models/tool/predefined/_tavily.py,sha256=nrvo8aBkwECap-VdBn-UppnCxfbwQ-stjoPg2p1ztoU,4947
|
|
203
|
+
waldiez/models/tool/predefined/_wikipedia.py,sha256=CAhSkYdnQFZdSke3ccGq59MukXeTlMoVxiXWd8JNOmc,4785
|
|
204
|
+
waldiez/models/tool/predefined/_youtube.py,sha256=IN9po9BF3p8Ps1hTHI1FedhQPBHSxgCdGmsTl_k-Qh4,4394
|
|
205
|
+
waldiez/models/tool/predefined/protocol.py,sha256=0CX6VsB_EPF4uxces0gwPz4OqMUm7e8-3qqCDEOkEkU,2306
|
|
205
206
|
waldiez/models/tool/predefined/registry.py,sha256=_QQHSoKPiNhpjqolIzYrhPzSMgLO4d9TnU6YJXcfgGE,2465
|
|
206
|
-
waldiez/running/__init__.py,sha256=
|
|
207
|
-
waldiez/running/base_runner.py,sha256=
|
|
208
|
-
waldiez/running/environment.py,sha256=
|
|
207
|
+
waldiez/running/__init__.py,sha256=JBy1lV2FDOrqJE8l-NIvD88CZdz2RNAYx5eEltWmXGY,530
|
|
208
|
+
waldiez/running/base_runner.py,sha256=4lprLl9Qmp7IyZ116lzWc_ZTgjW7_LpUKA5XdbsCaXk,31363
|
|
209
|
+
waldiez/running/environment.py,sha256=akDiCLfnR13wM3_zS2UXeqGyCiP1A2cxTqfUnRnlPk0,7684
|
|
210
|
+
waldiez/running/exceptions.py,sha256=_UrxnF4EfBLyvFfxMxeuCJ9qAWEERsmg-kuxzACSiJU,275
|
|
209
211
|
waldiez/running/gen_seq_diagram.py,sha256=wsF9b9uG0NA5iVAlW-wOtYn0w_9-bySBO9nWkUIBNUc,5943
|
|
210
|
-
waldiez/running/
|
|
211
|
-
waldiez/running/
|
|
212
|
-
waldiez/running/
|
|
213
|
-
waldiez/running/
|
|
214
|
-
waldiez/running/
|
|
215
|
-
waldiez/running/
|
|
216
|
-
waldiez/running/
|
|
217
|
-
waldiez/running/
|
|
212
|
+
waldiez/running/post_run.py,sha256=m8HsizO0pV_A82O3FnC-7BwL86s4Ef2_pgkCYxVlpto,6075
|
|
213
|
+
waldiez/running/pre_run.py,sha256=JfQLR-qyJUaYWVmzfGyiay1iKXuujqAfOYaLrOxvkXw,5857
|
|
214
|
+
waldiez/running/protocol.py,sha256=DSVm0yLVD9C5QuUXwNWRc_Eo9YpbgxG3IlzldEaDr18,5749
|
|
215
|
+
waldiez/running/run_results.py,sha256=o-9bKSMxCht-j05x3KbjqOqlN6vH8tYQbAzcSgorAp8,330
|
|
216
|
+
waldiez/running/standard_runner.py,sha256=Xb3mvx7_E0JwOs6y4oayyJKLcBF0_9WXSofVp1BFbQM,9464
|
|
217
|
+
waldiez/running/timeline_processor.py,sha256=rsPT_asCZvWvM-wro-8Qe3HrkHwOuGGOFByF8hUXUx8,41272
|
|
218
|
+
waldiez/running/utils.py,sha256=S4MdVOESMao3_f8ileDPpPSwL6MZxV_MfrlYqcack5g,14179
|
|
219
|
+
waldiez/running/step_by_step/__init__.py,sha256=s1CURBTSHtRwDygsE1fLFChDjoCi_P2xlV9l5AA9HUE,1314
|
|
220
|
+
waldiez/running/step_by_step/breakpoints_mixin.py,sha256=hyzWZr85fQI-4XYjm9trPJgCG-0i24HfYQI-czm6i2Y,5663
|
|
221
|
+
waldiez/running/step_by_step/step_by_step_models.py,sha256=0T8aLu9c9i_DuOUihooOl-fyZwh_oHdIZ5U1AOYezPw,6413
|
|
222
|
+
waldiez/running/step_by_step/step_by_step_runner.py,sha256=wyc-HaqQQq3hFwAjsfETz15zSqSG24LHYZvvoWdrQU4,25538
|
|
223
|
+
waldiez/running/subprocess_runner/__base__.py,sha256=zi8CXTDxxXt4Ed8PtjZTQ1DD30nko-9RCcNftsAFNls,7720
|
|
224
|
+
waldiez/running/subprocess_runner/__init__.py,sha256=fwNnjmpVZtHyQmLuvvMzc1cfko5SDDG9icr_6_cc26w,440
|
|
225
|
+
waldiez/running/subprocess_runner/_async_runner.py,sha256=WkqAEV_2qGjQH03v1ftsz0hpcLGKoWxwnZLu4x4y6RU,12443
|
|
226
|
+
waldiez/running/subprocess_runner/_sync_runner.py,sha256=VRkXQY9mbhFG7wrHtDxAI8e46oIzbH5dUU_lsrO5HCE,15272
|
|
227
|
+
waldiez/running/subprocess_runner/runner.py,sha256=sVDWALaj3lDoa3YNfHeW83N0KKi5gyN0De2fnyIKU-8,19493
|
|
218
228
|
waldiez/utils/__init__.py,sha256=D2AY5iqhh95zarm-FHLXSOzNSV3DITJVm1spemtA9hw,275
|
|
219
229
|
waldiez/utils/conflict_checker.py,sha256=EPq_FVC-nt_LgmLpM-UoVDisEu4JlvhpvPUpdBgg-TQ,1488
|
|
220
|
-
waldiez/utils/version.py,sha256=
|
|
221
|
-
waldiez
|
|
222
|
-
waldiez
|
|
223
|
-
waldiez
|
|
224
|
-
waldiez
|
|
225
|
-
waldiez
|
|
226
|
-
waldiez
|
|
230
|
+
waldiez/utils/version.py,sha256=HXvLq9pPT6Rrwf3slzJ5_5wQuK_ahDNY5_0rsJxQqXE,1833
|
|
231
|
+
waldiez/ws/__init__.py,sha256=-Sh5jzpBrQ064BHL-e8x8cMuHhhkJfKoy0mszwK7jpg,1587
|
|
232
|
+
waldiez/ws/__main__.py,sha256=HMpmnJcQXLk16cdBfh7l5zp80685e8NArdekFmH5caU,452
|
|
233
|
+
waldiez/ws/_file_handler.py,sha256=FOJVXVW2AfXblVXbwfdfvTVmoMEQKr3gbj0B-BQ7tL0,6471
|
|
234
|
+
waldiez/ws/cli.py,sha256=eguiBICzU3TPEnSWtDUol9DdjhanxuDIL4wPdo8T5h8,5894
|
|
235
|
+
waldiez/ws/client_manager.py,sha256=hKTaJLiLaNZiE78mZjxSAj6g1cvwB6ijybGg0QSL3ck,29668
|
|
236
|
+
waldiez/ws/errors.py,sha256=H5eajKWPv7igLU_PkYR7ydaic-eraUiFvLnrsEyI5N4,11968
|
|
237
|
+
waldiez/ws/models.py,sha256=NffeB0gAiZQwSv8td2fNqmgZ4rtaCQC4XQnyV02O0BU,26034
|
|
238
|
+
waldiez/ws/reloader.py,sha256=eqw0bC99pPY0oC3fTKo6WtsgkDklB_pW4VoXo9cv8Y8,10164
|
|
239
|
+
waldiez/ws/server.py,sha256=vPQNyEiXzCH68Nk9UanklCxAaEfxoxcz_t86hIHZT-w,15847
|
|
240
|
+
waldiez/ws/session_manager.py,sha256=2WN3qZqstIx7uhXxTrnextMAU6Vbq8TNvjhSWkyg350,12444
|
|
241
|
+
waldiez/ws/session_stats.py,sha256=moGKtSkzTW2Ddh8pnr3Zd8z7nwbcItBu77XXnwDuk8Q,2846
|
|
242
|
+
waldiez/ws/utils.py,sha256=Axc3hI2hwOT38jM8rLAtLE5zshPMFYyqxGcElZG-t2Q,10835
|
|
243
|
+
waldiez-0.5.10.dist-info/METADATA,sha256=fJbHbFBDlvx1PtbxQcZv284_w0QOJfUU06GGLCKwjsc,20014
|
|
244
|
+
waldiez-0.5.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
|
245
|
+
waldiez-0.5.10.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
|
|
246
|
+
waldiez-0.5.10.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
|
|
247
|
+
waldiez-0.5.10.dist-info/licenses/NOTICE.md,sha256=L7xtckFRYvYJjhjQNtFpURWCiAvEuq4ePvxJsC-XAdk,785
|
|
248
|
+
waldiez-0.5.10.dist-info/RECORD,,
|
|
@@ -1,210 +0,0 @@
|
|
|
1
|
-
# SPDX-License-Identifier: Apache-2.0.
|
|
2
|
-
# Copyright (c) 2024 - 2025 Waldiez and contributors.
|
|
3
|
-
|
|
4
|
-
# pylint: disable=import-outside-toplevel, broad-exception-caught
|
|
5
|
-
# pylint: disable=too-many-try-statements, too-complex
|
|
6
|
-
# noqa: C901
|
|
7
|
-
"""Patch ag2's IOStream if a flow is async.
|
|
8
|
-
|
|
9
|
-
# let's keep an eye here:
|
|
10
|
-
# https://github.com/ag2ai/ag2/blob/main/autogen/agentchat/conversable_agent.py#L2973
|
|
11
|
-
# reply = await iostream.input(prompt) ???? (await???)
|
|
12
|
-
"""
|
|
13
|
-
|
|
14
|
-
import inspect
|
|
15
|
-
import sys
|
|
16
|
-
from typing import Any, Callable
|
|
17
|
-
|
|
18
|
-
from asyncer import syncify
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
def patch_io_stream(is_async: bool) -> None:
|
|
22
|
-
"""Patch the IOStream to handle async flows.
|
|
23
|
-
|
|
24
|
-
Parameters
|
|
25
|
-
----------
|
|
26
|
-
is_async : bool
|
|
27
|
-
Whether the flow is async or not.
|
|
28
|
-
"""
|
|
29
|
-
if is_async:
|
|
30
|
-
patch_async_io_stream()
|
|
31
|
-
else:
|
|
32
|
-
patch_sync_io_stream()
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
def patch_sync_io_stream() -> None:
|
|
36
|
-
"""Patch the IOStream to handle async flows."""
|
|
37
|
-
from autogen.io import IOStream # type: ignore
|
|
38
|
-
|
|
39
|
-
iostream = IOStream.get_default()
|
|
40
|
-
original_input = iostream.input
|
|
41
|
-
|
|
42
|
-
def _safe_input(prompt: str = "", *, password: bool = False) -> str:
|
|
43
|
-
"""Async input method."""
|
|
44
|
-
try:
|
|
45
|
-
input_or_coro = original_input(prompt, password=password)
|
|
46
|
-
if inspect.iscoroutine(input_or_coro):
|
|
47
|
-
|
|
48
|
-
async def _async_input() -> str:
|
|
49
|
-
reply = await input_or_coro
|
|
50
|
-
return reply
|
|
51
|
-
|
|
52
|
-
return syncify(_async_input)()
|
|
53
|
-
return input_or_coro
|
|
54
|
-
|
|
55
|
-
except EOFError:
|
|
56
|
-
# Handle EOFError gracefully
|
|
57
|
-
return ""
|
|
58
|
-
|
|
59
|
-
iostream.input = _safe_input # pyright: ignore
|
|
60
|
-
iostream.print = get_printer() # pyright: ignore
|
|
61
|
-
IOStream.set_default(iostream)
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
def patch_async_io_stream() -> None:
|
|
65
|
-
"""Patch the IOStream to handle async flows."""
|
|
66
|
-
from autogen.io import IOStream # pyright: ignore
|
|
67
|
-
|
|
68
|
-
iostream = IOStream.get_default()
|
|
69
|
-
original_input = iostream.input
|
|
70
|
-
|
|
71
|
-
async def _async_input(prompt: str = "", *, password: bool = False) -> str:
|
|
72
|
-
"""Async input method."""
|
|
73
|
-
try:
|
|
74
|
-
input_or_coro = original_input(prompt, password=password)
|
|
75
|
-
if inspect.iscoroutine(input_or_coro):
|
|
76
|
-
reply = await input_or_coro
|
|
77
|
-
else:
|
|
78
|
-
reply = input_or_coro
|
|
79
|
-
return reply
|
|
80
|
-
except EOFError:
|
|
81
|
-
# Handle EOFError gracefully
|
|
82
|
-
return ""
|
|
83
|
-
|
|
84
|
-
iostream.input = _async_input # pyright: ignore
|
|
85
|
-
iostream.print = get_printer() # pyright: ignore
|
|
86
|
-
IOStream.set_default(iostream)
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
def get_printer() -> Callable[..., None]: # noqa: C901
|
|
90
|
-
"""Get the printer function.
|
|
91
|
-
|
|
92
|
-
Returns
|
|
93
|
-
-------
|
|
94
|
-
Callable[..., None]
|
|
95
|
-
The printer function that handles Unicode encoding errors gracefully.
|
|
96
|
-
"""
|
|
97
|
-
try:
|
|
98
|
-
# noinspection PyUnresolvedReferences
|
|
99
|
-
from autogen.io import IOStream # pyright: ignore
|
|
100
|
-
|
|
101
|
-
printer = IOStream.get_default().print
|
|
102
|
-
except ImportError:
|
|
103
|
-
# Fallback to standard print if autogen is not available
|
|
104
|
-
printer = print
|
|
105
|
-
|
|
106
|
-
# noinspection PyBroadException
|
|
107
|
-
def safe_printer(*args: Any, **kwargs: Any) -> None: # noqa: C901
|
|
108
|
-
"""Safe printer that handles Unicode encoding errors.
|
|
109
|
-
|
|
110
|
-
Parameters
|
|
111
|
-
----------
|
|
112
|
-
*args : Any
|
|
113
|
-
Arguments to pass to the printer
|
|
114
|
-
**kwargs : Any
|
|
115
|
-
Keyword arguments to pass to the printer
|
|
116
|
-
"""
|
|
117
|
-
try:
|
|
118
|
-
printer(*args, **kwargs)
|
|
119
|
-
except UnicodeEncodeError:
|
|
120
|
-
# First fallback: try to get a safe string representation
|
|
121
|
-
try:
|
|
122
|
-
msg, flush = get_what_to_print(*args, **kwargs)
|
|
123
|
-
# Convert problematic characters to safe representations
|
|
124
|
-
safe_msg = msg.encode("utf-8", errors="replace").decode("utf-8")
|
|
125
|
-
printer(safe_msg, end="", flush=flush)
|
|
126
|
-
except (UnicodeEncodeError, UnicodeDecodeError):
|
|
127
|
-
# Second fallback: use built-in print with safe encoding
|
|
128
|
-
try:
|
|
129
|
-
# Convert args to safe string representations
|
|
130
|
-
safe_args: list[str] = []
|
|
131
|
-
for arg in args:
|
|
132
|
-
try:
|
|
133
|
-
safe_args.append(
|
|
134
|
-
str(arg)
|
|
135
|
-
.encode("utf-8", errors="replace")
|
|
136
|
-
.decode("utf-8")
|
|
137
|
-
)
|
|
138
|
-
except (UnicodeEncodeError, UnicodeDecodeError):
|
|
139
|
-
safe_args.append(repr(arg))
|
|
140
|
-
|
|
141
|
-
# Use built-in print instead of the custom printer
|
|
142
|
-
print(*safe_args, **kwargs)
|
|
143
|
-
|
|
144
|
-
except Exception:
|
|
145
|
-
# Final fallback: write directly to stderr buffer
|
|
146
|
-
try:
|
|
147
|
-
error_msg = (
|
|
148
|
-
"Could not print the message "
|
|
149
|
-
"due to encoding issues.\n"
|
|
150
|
-
)
|
|
151
|
-
if hasattr(sys.stderr, "buffer"):
|
|
152
|
-
sys.stderr.buffer.write(
|
|
153
|
-
error_msg.encode("utf-8", errors="replace")
|
|
154
|
-
)
|
|
155
|
-
sys.stderr.buffer.flush()
|
|
156
|
-
else:
|
|
157
|
-
sys.stderr.write(error_msg)
|
|
158
|
-
sys.stderr.flush()
|
|
159
|
-
except Exception:
|
|
160
|
-
# If even this fails, we're in a very bad state
|
|
161
|
-
pass
|
|
162
|
-
except Exception as e:
|
|
163
|
-
# Handle any other unexpected errors
|
|
164
|
-
try:
|
|
165
|
-
error_msg = f"Unexpected error in printer: {str(e)}\n"
|
|
166
|
-
if hasattr(sys.stderr, "buffer"):
|
|
167
|
-
sys.stderr.buffer.write(
|
|
168
|
-
error_msg.encode("utf-8", errors="replace")
|
|
169
|
-
)
|
|
170
|
-
sys.stderr.buffer.flush()
|
|
171
|
-
else:
|
|
172
|
-
sys.stderr.write(error_msg)
|
|
173
|
-
sys.stderr.flush()
|
|
174
|
-
except Exception:
|
|
175
|
-
pass
|
|
176
|
-
|
|
177
|
-
return safe_printer
|
|
178
|
-
|
|
179
|
-
|
|
180
|
-
def get_what_to_print(*args: Any, **kwargs: Any) -> tuple[str, bool]:
|
|
181
|
-
"""Extract message and flush flag from print arguments.
|
|
182
|
-
|
|
183
|
-
Parameters
|
|
184
|
-
----------
|
|
185
|
-
*args : Any
|
|
186
|
-
Arguments to print
|
|
187
|
-
**kwargs : Any
|
|
188
|
-
Keyword arguments for print function
|
|
189
|
-
|
|
190
|
-
Returns
|
|
191
|
-
-------
|
|
192
|
-
tuple[str, bool]
|
|
193
|
-
Message to print and flush flag
|
|
194
|
-
"""
|
|
195
|
-
# Convert all args to strings and join with spaces (like print does)
|
|
196
|
-
msg = " ".join(str(arg) for arg in args)
|
|
197
|
-
|
|
198
|
-
# Handle sep parameter
|
|
199
|
-
sep = kwargs.get("sep", " ")
|
|
200
|
-
if len(args) > 1:
|
|
201
|
-
msg = sep.join(str(arg) for arg in args)
|
|
202
|
-
|
|
203
|
-
# Handle end parameter
|
|
204
|
-
end = kwargs.get("end", "\n")
|
|
205
|
-
msg += end
|
|
206
|
-
|
|
207
|
-
# Handle flush parameter
|
|
208
|
-
flush = kwargs.get("flush", False)
|
|
209
|
-
|
|
210
|
-
return msg, flush
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|