waldiez 0.3.10__py3-none-any.whl → 0.3.12__py3-none-any.whl

{waldiez-0.3.10.dist-info → waldiez-0.3.12.dist-info}/RECORD

@@ -1,16 +1,15 @@
 waldiez/__init__.py,sha256=cXWdy5P9i_x6fHwlL5DRaIhmpfnhabo5GjASStlcLWw,819
 waldiez/__main__.py,sha256=0dYzNrQbovRwQQvmZC6_1FDR1m71SUIOkTleO5tBnBw,203
-waldiez/_version.py,sha256=Dlx5WyLmhau88DdjdwOZD6NwqTOnOOMu9xNL-3soMP8,156
+waldiez/_version.py,sha256=8g1vIaGTCQVgdq8Z3RCbEIHIp4T5eWWWaTpPmb13Ui0,156
 waldiez/cli.py,sha256=ZhcXOH5WtGVncQpoOix9yHXctQKBwckgiIdYLaRj40c,7108
-waldiez/exporter.py,sha256=8EqLya3UqCA_F2TwtfdeRCt7ILnWxBcQvkMyX8no0BU,6501
+waldiez/exporter.py,sha256=ZuMF8cra0Shw4nkDiFuPqaQKFYrF0vPj_I3EhPyesPw,4788
 waldiez/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 waldiez/runner.py,sha256=rFuCPFSmdOxujEWdEtCrGv8FpZ0cZhPagIslPs-2DZU,11317
 waldiez/exporting/__init__.py,sha256=nlMWBnn_MViGBJmDhrTCYpPv19fEP_WixvmmY9jt5c8,399
 waldiez/exporting/agent/__init__.py,sha256=GOgri16QNICpx3YNZPzVn_KYVxuQACYKQyazhJNtYK0,188
-waldiez/exporting/agent/agent_exporter.py,sha256=ZvEcBG1kNlaAitTZbqIfxF7wqnxUtobDvAC1gPcLME8,10060
-waldiez/exporting/agent/utils/__init__.py,sha256=Pg8Gwj9lA8MAb1SkkFDn6PRa2wtVJh4pABOmEC_ClRI,909
-waldiez/exporting/agent/utils/agent_class_name.py,sha256=vzeAafjj1Sb5-u-LziTgbMnnY2s7D6Z9yu0GUBqiAao,999
-waldiez/exporting/agent/utils/agent_imports.py,sha256=UgP3hWvKngMWUk2S6n2J2mhIsnfFCkEKHVVdXLoPu3M,1775
+waldiez/exporting/agent/agent_exporter.py,sha256=mLRocSnhXMTK0KPXN_F_tKTrSXBkg9uyAGDkhPPsPhs,10329
+waldiez/exporting/agent/utils/__init__.py,sha256=UlyOZCm3jLz5PHOJ9fhZuCqbNBBlomzvRd7K_tKR-No,844
+waldiez/exporting/agent/utils/captain_agent.py,sha256=jA-UjyK0AMr4yTjlXRIv9CQPytzNOGyR44iZ6iZ84Vo,7355
 waldiez/exporting/agent/utils/code_execution.py,sha256=sRHczOmm12U9E-Q1v9UQRwao1LUEve4fnKvJwsokNNk,2352
 waldiez/exporting/agent/utils/group_manager.py,sha256=VCqd-lsCertwVTWVez9Z1HS_H-UmBGyNGizvC736VnI,7360
 waldiez/exporting/agent/utils/reasoning.py,sha256=EBE42Dar9n6fBrhg9D9wch0KptsP_m3Dm2WNEDuKb7g,999
@@ -36,35 +35,36 @@ waldiez/exporting/base/utils/naming.py,sha256=G4e9MPnbg9W8QLVblQj31sV46GKmAOqEA8
 waldiez/exporting/base/utils/path_check.py,sha256=5XGKmHA3oi9kf2gI4ZHB3K9gnjDP0NEfpn0tP8cme-8,1425
 waldiez/exporting/base/utils/to_string.py,sha256=fo8i6QaLZJsym3biN2IhVooTWlXiXVCzuU8FNFJVXME,2183
 waldiez/exporting/chats/__init__.py,sha256=BS8F-hgrzsKPtWEyEMPE-e0JL6q5478i76vR1bKjXXY,195
-waldiez/exporting/chats/chats_exporter.py,sha256=6nfLxvS3UWnhe0aq0l6FACJQIrdENCDZlyAl6qIAV84,8515
+waldiez/exporting/chats/chats_exporter.py,sha256=xbVSa9HJ5X8tU197XqAeWfORdtSxcZf9V9scbO_6cNg,8515
 waldiez/exporting/chats/utils/__init__.py,sha256=FMS94ObfHIlH_Pg_5s5PR-QaXYzcpZErP1Vut2PYaTY,442
 waldiez/exporting/chats/utils/common.py,sha256=kZ9xO26Oo3tKSxZescFfB2laE9I8F5Hnq3c8oojEVGo,2410
 waldiez/exporting/chats/utils/nested.py,sha256=evjvJeWHATumNNffAPZUInlyQuzXLmBWnlHfYQaub0I,9227
-waldiez/exporting/chats/utils/sequential.py,sha256=kKdfxpt-ZCDeXqwbWRDydkp5BwCuk2BJdG_6DMnK8TM,7787
-waldiez/exporting/chats/utils/single_chat.py,sha256=DvB2ODOATHdJ1sw0QIuqoH-2jRim7Y22xWfCdMr6GCc,9416
+waldiez/exporting/chats/utils/sequential.py,sha256=TYAte2hwpGkEENjCMer-6PcL2r8PE-o95hgoi_AX2eg,7841
+waldiez/exporting/chats/utils/single_chat.py,sha256=FYz0jvtO0yFYPk5CBuXb7rOW8VxB0NJWvWY1PEHPIAQ,9535
 waldiez/exporting/chats/utils/swarm.py,sha256=KJTtgQFt8c_jR800vC8R14ArYVm6ZgpsUyGbiOykmMI,6929
 waldiez/exporting/flow/__init__.py,sha256=TAJ8lq9saP1IX0U-A6yGKCfvbQnEocmRAX4Hc60f3yY,183
-waldiez/exporting/flow/flow_exporter.py,sha256=vI0dKd2jNPGoDTaah_4zPbXKewvJtKB8vbyRRYXh36I,17769
+waldiez/exporting/flow/flow_exporter.py,sha256=Kvaei1uhMfBBdzOTsnxEW-f-Q41OKj9kpB4e5rCjiso,18117
 waldiez/exporting/flow/utils/__init__.py,sha256=Fe0sg5i1K2Vu0ruWsIuPj7oTRHoJTW4KdKmktKRxyrc,1389
 waldiez/exporting/flow/utils/agent_utils.py,sha256=QL1nzlgYeh6Y-l9R3hmCsrxSzluvBVciuuEIFdBXXKM,6109
 waldiez/exporting/flow/utils/chat_utils.py,sha256=q6HS4JTNQJe15eh78H1BlN-wId9XzvVQziGhxsUIEFs,1882
-waldiez/exporting/flow/utils/def_main.py,sha256=7yrxYizssTBYvcsXj3p-LIP-29-KpCRAV7I3_bhWbXE,2440
-waldiez/exporting/flow/utils/flow_content.py,sha256=kAEXS5rWGw0Lvc4R5fosRZZ89SOlFbLsp7qk10pirPU,4515
+waldiez/exporting/flow/utils/def_main.py,sha256=GxEBfX6cUGFCMdxiu_hp5TLIhIC4ohczxA1A01mHnOo,2746
+waldiez/exporting/flow/utils/flow_content.py,sha256=RIMtpBFZCJf2G67ipOa3Co5OQinaobYvthqisgir5sQ,4505
 waldiez/exporting/flow/utils/flow_names.py,sha256=GtfVRF8WYUP8Xx47AiyZ5FuuCUb22jhJn3diblbbBfM,3463
-waldiez/exporting/flow/utils/importing_utils.py,sha256=j0E6gyDl0-l1oQUoSd8tRb64TfAet9HEtnQrnGLIDmE,5519
+waldiez/exporting/flow/utils/importing_utils.py,sha256=h8FqyAU0ApSFaie2lipxk69rNWNGwp_xAuhPpT991-M,5552
 waldiez/exporting/flow/utils/logging_utils.py,sha256=m3Ja00Qb_07rLVgwSOFSBaJk48SfCNsyPuQbRy5VMNU,10963
 waldiez/exporting/models/__init__.py,sha256=QbyYuS-Px-M9vMrdL-UKqnDzhYWRkNmRGBbc1IBuwpM,212
-waldiez/exporting/models/models_exporter.py,sha256=W4EDMY79RNSqZKBd2wYGFRQnt-OvbixXMXiAeTjNdU0,5994
-waldiez/exporting/models/utils.py,sha256=atxJ1OxjlSIxdyfBCmr_U7aXjHcZ82IQmjgw8QB4XLM,4849
+waldiez/exporting/models/models_exporter.py,sha256=i9aSnL7JNU8iWJotfSxWbm1CRjUP6yqd9CtfXWiRbV4,6289
+waldiez/exporting/models/utils.py,sha256=g15N_JA21Ly5nzCfburEbHf6fTWX7hG5rxyTvAwyXl0,4997
 waldiez/exporting/skills/__init__.py,sha256=sJm4ZnN3haldxRBgNlLK0qMK5YtydF2XnSaqnxu2hQo,195
 waldiez/exporting/skills/skills_exporter.py,sha256=o_16s0VjMoaDwe-JJdbqDP2BzQJV9SAM57k_4KxAix0,5290
 waldiez/exporting/skills/utils.py,sha256=SAJ55fpNlU9DmmAcSHafq1ddgg4A-bjBxbL3aaWps50,8120
-waldiez/models/__init__.py,sha256=R_1yLo1mSJHMyrxokXuO-24PVxE3sA0ep2Tj5EG-mTM,3816
-waldiez/models/waldiez.py,sha256=Db63-G-fcssUOtOB0ePVvZcLxMr2f6d_UJVrn7hgBwk,10450
-waldiez/models/agents/__init__.py,sha256=HOY3IG1WWNqepRkETjFZihHRZHgipxWO5cxeVWm1Rv0,4553
-waldiez/models/agents/agents.py,sha256=2eN0S9Zjm5pJgv-0JOl8HtcsurbTx01kbUprlMSGWq8,4324
+waldiez/models/__init__.py,sha256=OsLOhFRoKYgB-7zNEdTG8BmHPiSRIlZv8NKZ6c2bVJs,3996
+waldiez/models/waldiez.py,sha256=V6rrdD9P9mD8bVYWldha53gfZocaWqaadQTUI6IoQT0,8928
+waldiez/models/agents/__init__.py,sha256=bB1TKT8rqb_-SfCIIdjNi3Xkof4IVBtuhCubg2UEmXI,4970
+waldiez/models/agents/agents.py,sha256=kwUOlV4VJIgqhYBQHbAZhWzx3iENOvYilN3nDonZTIU,4605
+waldiez/models/agents/extra_requirements.py,sha256=6v61EsAUCKAiUi52_p4kanB9EgwGN9nRy7qV-f13tyc,2681
 waldiez/models/agents/agent/__init__.py,sha256=us1qPNCwKdUao9UUVXDHbcjuUUA5yexWEQkozvWLUDY,1044
-waldiez/models/agents/agent/agent.py,sha256=qZLvi3kAW13DiOzL6IqsjrgkA4q65gTJY9ymC0Pf044,5803
+waldiez/models/agents/agent/agent.py,sha256=Qfyl1VibDyg7Zz22gjQ6vot702TtnEHjdlKmEX9jpRQ,8269
 waldiez/models/agents/agent/agent_data.py,sha256=rqRRMdqOzUF5jjdVoYika-tqe0YUWHQ4at5fNL9GAYM,5437
 waldiez/models/agents/agent/code_execution.py,sha256=lBfJMRvzl1QfNIHwQXxY2ote2Kjh9iT3twz-Tj8JrZA,2035
 waldiez/models/agents/agent/linked_skill.py,sha256=VpVDtI13bUPRUDk8usoIJVHs7ZLPWS3pp5GvuzIiZec,760
@@ -74,6 +74,10 @@ waldiez/models/agents/agent/termination_message.py,sha256=eKd_vvKx0Eegq-gU1o7BJx
 waldiez/models/agents/assistant/__init__.py,sha256=y2s2F2e7njMcmwye1UOUULHSvU3VsDLreTLvRU1I4M0,268
 waldiez/models/agents/assistant/assistant.py,sha256=POBoEv5iOCKF3s7a5Ww7OojM16PREye6r9jO7e6epd8,1188
 waldiez/models/agents/assistant/assistant_data.py,sha256=KFxgCXdcmr6v6kG13nLQYUrktIVV0DMNcpV-7kt-5rk,916
+waldiez/models/agents/captain_agent/__init__.py,sha256=xM2coa1L09hEgm36xh9GQJk_RPa-Ev8Jot_KuCYUaKQ,406
+waldiez/models/agents/captain_agent/captain_agent.py,sha256=ZjB0JYQ2SbWeP4QeR5OflKGvkoBxAezzMwfTtaWE_-A,1208
+waldiez/models/agents/captain_agent/captain_agent_data.py,sha256=o_MsCyg6tTRbRdofAGP6fJdS4F7yE6Q7aPgCOWLNk50,1723
+waldiez/models/agents/captain_agent/captain_agent_lib_entry.py,sha256=oxuKB6w8LuCUPqNcc78NsdVeiAMM_BnvuFtUOlZfeG0,868
 waldiez/models/agents/group_manager/__init__.py,sha256=MlMzrAfiSRprVJvAVNLUuulckNsBuWsiHXa0-tcUCWM,894
 waldiez/models/agents/group_manager/group_manager.py,sha256=LG5-PEi6Hf69fZSkwOBjHnhC_SikRUewqBU37x4DxXk,2780
 waldiez/models/agents/group_manager/group_manager_data.py,sha256=LZYG2RF6El041rIE0X_9tSfe7bElgixsSja04v4KiiA,2878
@@ -104,18 +108,20 @@ waldiez/models/chat/chat_data.py,sha256=a0ScDNtU_VgAMGzrR2Lj7mPZpY-yQKazYiiwDTfh
 waldiez/models/chat/chat_message.py,sha256=FyQCIVFl6Z1nlyCpDYHZ5YySN6pHSNbMSp4e56hodJQ,8306
 waldiez/models/chat/chat_nested.py,sha256=C1ygS7TeB5emRIIIs_xFabKgXzze0HTEO4CIexedGKw,4132
 waldiez/models/chat/chat_summary.py,sha256=RDeRw7vTV4Niw5_v_UZTOWQkIZUI0q9vdQ01Ja1isM4,2916
-waldiez/models/common/__init__.py,sha256=I_j63y68FCpSVDz9YXW7aKQsA-X3MOWDvNokyrVpv4w,493
-waldiez/models/common/base.py,sha256=SwnnuFTJ0C3PfBoRilqE0uZv3j5aelw7-s-79KcxJNA,1840
-waldiez/models/common/date_utils.py,sha256=hdnxYU8GKQTnucOnAfs1DAqER0oe59A5G3UVDktl-ro,344
-waldiez/models/common/dict_utils.py,sha256=ydUP9Am9QxQveqnDB3Rdt62FvTHeFfD4kTmCLXyutGY,1050
+waldiez/models/common/__init__.py,sha256=a61rn0Nl0V-BFvJJxx9CDKc2CTxSsErnFPQ4QE3IYwQ,565
+waldiez/models/common/ag2_version.py,sha256=p9EF8N2zsb14YXrbZtMIWYLUGFB_Jnjbm4o6tODCw-Q,744
+waldiez/models/common/base.py,sha256=caFhCnUH41OYyx19Zl2gHXzhlZbl8Ewmpc5t7LpYhjY,1961
+waldiez/models/common/date_utils.py,sha256=NdVmShZjvhDVT-Y5rT8FgwQTHcDoljBJQauTsE7gwbE,437
+waldiez/models/common/dict_utils.py,sha256=xfZ6fFJXGyW2yiPiBcGO7WKqhAlCH1sMRp_Z4__xyr0,1143
 waldiez/models/common/method_utils.py,sha256=CyTwRppegHviN-J_oj5XsPvX0edMSmEbLVeTkrG2HVo,7010
-waldiez/models/flow/__init__.py,sha256=3UdwoUZdkLr6G78tNTyaijoiwvmHZ-N0tY3G7S_xbRs,255
-waldiez/models/flow/flow.py,sha256=kM8K7dl8wZIP0GEryELC2yZ0t0blEFFkNOvx7VEMW4o,14700
-waldiez/models/flow/flow_data.py,sha256=Mqa7NLZMfdiPknj-RhyDPgMZjkDTmcDFUpPKkXuYpTk,5034
-waldiez/models/flow/utils.py,sha256=Bd8dWYlCenuU4VeusAL_qxkXzYSuxl9e3a7FhHs6UCY,5435
-waldiez/models/model/__init__.py,sha256=1cOyDThhZV8o7pZN1DaIhh12jWty_YMegoQ4Jg7uKyI,383
-waldiez/models/model/model.py,sha256=0EqHSMIBun_Owy-v0cbv39sRP8if1LlLKeMOD_XA3Ow,6984
-waldiez/models/model/model_data.py,sha256=EwObYtzPyxZqADdTeWjULNmb3dLgF91mzK3r1nxxJJk,3819
+waldiez/models/flow/__init__.py,sha256=3UFbMMn_5c5AXs6ppkvhNeWb9jelru2lNHUjpEq0N68,309
+waldiez/models/flow/flow.py,sha256=Shyg-XwzloSzqLLFPwY9g_MjZHxAGy2h_qfPmK58UZs,16876
+waldiez/models/flow/flow_data.py,sha256=afD97Ff9Yd7QSXE4DphaENY7wqrq4QIl4EVgI8O4ZFk,5464
+waldiez/models/flow/utils.py,sha256=X_jelP5lHuQt1OZE6sxLlhEKbkkt8jZgbkF79e9rQig,7140
+waldiez/models/model/__init__.py,sha256=0klSsHb8poHgtnVdCi-rDiSOb_yOofk9RPehjLQnGFc,482
+waldiez/models/model/extra_requirements.py,sha256=XAjZs-P2g_FjU0Jb27NYrwLi8kyVn-XIvLCGr1UrKAU,1595
+waldiez/models/model/model.py,sha256=wYT_ajxIYY-wxJPRiZctAs78nazdUYkYSlCVw3ZE4Go,7283
+waldiez/models/model/model_data.py,sha256=8GZerOBu5eGvpIfZm4aypFiiCUxz-SQtIfVLr3E59rg,3753
 waldiez/models/skill/__init__.py,sha256=kAyp9-LwBacI3ndQpU-dCGWjsjojRQy73zXsTddGAmg,306
 waldiez/models/skill/skill.py,sha256=6r8heHUgRYwfcBJbdsV5DywvTXcEeWPuJN4jzCh3-O4,4255
 waldiez/models/skill/skill_data.py,sha256=Spk3YxK0IgkXpJgk05DJwvZwUPrh-2l95o9PnS1WnoQ,908
@@ -130,9 +136,9 @@ waldiez/utils/pysqlite3_checker.py,sha256=LypcHlCMbzdC0QUfrr68gzmrEbwzvKLUn-riYi
 waldiez/utils/cli_extras/__init__.py,sha256=FQ6biuz6Ig-U3xkmgac3WMUynH_7XWjeGI-DqWFTC8c,625
 waldiez/utils/cli_extras/jupyter.py,sha256=C4fOiS_PbU15X-6HUZBPHvfgEjOrY39e07mClKvswPI,2971
 waldiez/utils/cli_extras/studio.py,sha256=te6VmwYDeEEK6Jps-Ou84QPZmCYMsF-oE1hNu6ewFHg,887
-waldiez-0.3.10.dist-info/METADATA,sha256=oupQyYHDh2lDqTki8KTa5seNXAjOLCItAVc64lOB32Y,9438
-waldiez-0.3.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-waldiez-0.3.10.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
-waldiez-0.3.10.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
-waldiez-0.3.10.dist-info/licenses/NOTICE.md,sha256=lrKsUNrpE18LPSLPJY_kt2ZJFZJEwH55wPv5Axcheyc,133
-waldiez-0.3.10.dist-info/RECORD,,
+waldiez-0.3.12.dist-info/METADATA,sha256=ggItUOmudOaRVTAAGSn0ysxGF_i0TUe3ZmRbk4d8aKE,9661
+waldiez-0.3.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+waldiez-0.3.12.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
+waldiez-0.3.12.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
+waldiez-0.3.12.dist-info/licenses/NOTICE.md,sha256=lrKsUNrpE18LPSLPJY_kt2ZJFZJEwH55wPv5Axcheyc,133
+waldiez-0.3.12.dist-info/RECORD,,

waldiez/exporting/agent/utils/agent_class_name.py

@@ -1,36 +0,0 @@
-# SPDX-License-Identifier: Apache-2.0.
-# Copyright (c) 2024 - 2025 Waldiez and contributors.
-"""Get the agent class name."""
-
-from waldiez.models import WaldiezAgent
-
-
-# pylint: disable=too-many-return-statements
-def get_agent_class_name(agent: WaldiezAgent) -> str:
-    """Get the agent class name.
-
-    Parameters
-    ----------
-    agent : WaldiezAgent
-        The agent.
-
-    Returns
-    -------
-    str
-        The agent class name.
-    """
-    if agent.data.is_multimodal:
-        return "MultimodalConversableAgent"
-    if agent.agent_type == "assistant":
-        return "AssistantAgent"
-    if agent.agent_type == "user":
-        return "UserProxyAgent"
-    if agent.agent_type == "manager":
-        return "GroupChatManager"
-    if agent.agent_type == "rag_user":
-        return "RetrieveUserProxyAgent"
-    if agent.agent_type == "swarm":
-        return "SwarmAgent"
-    if agent.agent_type == "reasoning":
-        return "ReasoningAgent"
-    return "ConversableAgent"  # pragma: no cover

waldiez/exporting/agent/utils/agent_imports.py

@@ -1,55 +0,0 @@
-# SPDX-License-Identifier: Apache-2.0.
-# Copyright (c) 2024 - 2025 Waldiez and contributors.
-"""Get the imports needed for the agent."""
-
-from typing import Set
-
-
-def get_agent_imports(agent_class: str) -> Set[str]:
-    """Get the imports needed for the agent.
-
-    Parameters
-    ----------
-    agent_class : str
-        The agent class name.
-
-    Returns
-    -------
-    Set[str]
-        The imports needed for the agent.
-    """
-    imports = set(["import autogen"])
-    if agent_class == "AssistantAgent":
-        imports.add("from autogen import AssistantAgent")
-    elif agent_class == "UserProxyAgent":
-        imports.add("from autogen import UserProxyAgent")
-    elif agent_class == "GroupChatManager":
-        imports.add("from autogen import GroupChatManager")
-    elif agent_class == "RetrieveUserProxyAgent":
-        imports.add(
-            "from autogen.agentchat.contrib.retrieve_user_proxy_agent "
-            "import RetrieveUserProxyAgent"
-        )
-    elif agent_class == "MultimodalConversableAgent":
-        imports.add(
-            "from autogen.agentchat.contrib.multimodal_conversable_agent "
-            "import MultimodalConversableAgent"
-        )
-    elif agent_class == "SwarmAgent":
-        imports.add(
-            "from autogen import "
-            "AFTER_WORK, "
-            "ON_CONDITION, "
-            "UPDATE_SYSTEM_MESSAGE, "
-            "AfterWorkOption, "
-            "SwarmAgent, "
-            "SwarmResult"
-        )
-    elif agent_class == "ReasoningAgent":
-        # pylint: disable=line-too-long
-        imports.add(
-            "from autogen.agentchat.contrib.reasoning_agent import ReasoningAgent, visualize_tree"  # noqa: E501
-        )
-    else:
-        imports.add("from autogen import ConversableAgent")
-    return imports