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

{waldiez-0.3.11.dist-info → waldiez-0.3.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.3.11
+Version: 0.3.12
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/waldiez/python
 Project-URL: repository, https://github.com/waldiez/python.git
@@ -20,7 +20,7 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: <3.13,>=3.10
 Requires-Dist: aiocsv==1.3.2
 Requires-Dist: aiofiles==24.1.0
-Requires-Dist: aiosqlite==0.20.0
+Requires-Dist: aiosqlite==0.21.0
 Requires-Dist: asyncer==0.0.8
 Requires-Dist: graphviz==0.20.3
 Requires-Dist: httpx<1
@@ -31,27 +31,36 @@ Requires-Dist: pyautogen==0.7.3
 Requires-Dist: pydantic<3,>=2.6.1
 Requires-Dist: typer<0.16,>=0.9
 Provides-Extra: ag2-extras
-Requires-Dist: autogen[captainagent]==0.7.3; (platform_system == 'Linux') and extra == 'ag2-extras'
-Requires-Dist: chromadb; (platform_system != 'Linux') and extra == 'ag2-extras'
-Requires-Dist: huggingface-hub; (platform_system != 'Linux') and extra == 'ag2-extras'
+Requires-Dist: beautifulsoup4; extra == 'ag2-extras'
+Requires-Dist: chromadb>=0.5.23; extra == 'ag2-extras'
+Requires-Dist: crewai-tools==0.33.0; extra == 'ag2-extras'
+Requires-Dist: crewai>0.98.0; extra == 'ag2-extras'
+Requires-Dist: embedchain==0.1.126; extra == 'ag2-extras'
+Requires-Dist: huggingface-hub; extra == 'ag2-extras'
+Requires-Dist: ipython; extra == 'ag2-extras'
+Requires-Dist: langchain-community<1,>=0.3.12; extra == 'ag2-extras'
+Requires-Dist: markdownify; extra == 'ag2-extras'
 Requires-Dist: pgvector>=0.3.6; extra == 'ag2-extras'
+Requires-Dist: protobuf>=4.25.3; extra == 'ag2-extras'
 Requires-Dist: psycopg[binary]>=3.2.3; extra == 'ag2-extras'
+Requires-Dist: psycopg[binary]>=3.2.4; extra == 'ag2-extras'
 Requires-Dist: pyautogen[anthropic]==0.7.3; extra == 'ag2-extras'
 Requires-Dist: pyautogen[bedrock]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[cohere]==0.7.3; extra == 'ag2-extras'
 Requires-Dist: pyautogen[gemini]==0.7.3; extra == 'ag2-extras'
 Requires-Dist: pyautogen[groq]==0.7.3; extra == 'ag2-extras'
 Requires-Dist: pyautogen[lmm]==0.7.3; extra == 'ag2-extras'
 Requires-Dist: pyautogen[mistral]==0.7.3; extra == 'ag2-extras'
 Requires-Dist: pyautogen[neo4j]==0.7.3; extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat-mongodb]==0.7.3; extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat-pgvector]==0.7.3; extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat-qdrant]==0.7.3; (python_version < '3.13') and extra == 'ag2-extras'
-Requires-Dist: pyautogen[retrievechat]==0.7.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[ollama]==0.7.3; extra == 'ag2-extras'
 Requires-Dist: pyautogen[together]==0.7.3; extra == 'ag2-extras'
 Requires-Dist: pyautogen[websurfer]==0.7.3; extra == 'ag2-extras'
-Requires-Dist: pymongo>=4.10.1; extra == 'ag2-extras'
-Requires-Dist: qdrant-client[fastembed]; (python_version >= '3.13') and extra == 'ag2-extras'
-Requires-Dist: sentence-transformers; (platform_system != 'Linux') and extra == 'ag2-extras'
+Requires-Dist: pydantic-ai==0.0.21; extra == 'ag2-extras'
+Requires-Dist: pymongo>=4.11; extra == 'ag2-extras'
+Requires-Dist: pypdf; extra == 'ag2-extras'
+Requires-Dist: qdrant-client[fastembed]; extra == 'ag2-extras'
+Requires-Dist: sentence-transformers; extra == 'ag2-extras'
+Requires-Dist: weaviate-client==4.10.2; extra == 'ag2-extras'
 Provides-Extra: dev
 Requires-Dist: autoflake==2.3.1; extra == 'dev'
 Requires-Dist: bandit==1.8.2; extra == 'dev'
@@ -74,16 +83,16 @@ 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.6.1; extra == 'docs'
-Requires-Dist: mkdocs-minify-html-plugin==0.2.3; extra == 'docs'
+Requires-Dist: mkdocs-material==9.6.2; extra == 'docs'
+Requires-Dist: mkdocs-minify-html-plugin==0.2.4; extra == 'docs'
 Requires-Dist: mkdocs==1.6.1; extra == 'docs'
-Requires-Dist: mkdocstrings-python==1.13.0; extra == 'docs'
-Requires-Dist: mkdocstrings[crystal,python]==0.27.0; extra == 'docs'
+Requires-Dist: mkdocstrings-python==1.14.1; extra == 'docs'
+Requires-Dist: mkdocstrings[crystal,python]==0.28.0; extra == 'docs'
 Provides-Extra: jupyter
 Requires-Dist: jupyterlab>=4.3.0; extra == 'jupyter'
-Requires-Dist: waldiez-jupyter==0.3.11; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.3.12; extra == 'jupyter'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.3.11; extra == 'studio'
+Requires-Dist: waldiez-studio==0.3.12; extra == 'studio'
 Provides-Extra: test
 Requires-Dist: pytest-asyncio==0.25.3; extra == 'test'
 Requires-Dist: pytest-cov==6.0.0; extra == 'test'

{waldiez-0.3.11.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=xakNljN5CllPqeJEkZbxbd1Lwzqsba0yrHbJfxNpzSU,156
+waldiez/_version.py,sha256=8g1vIaGTCQVgdq8Z3RCbEIHIp4T5eWWWaTpPmb13Ui0,156
 waldiez/cli.py,sha256=ZhcXOH5WtGVncQpoOix9yHXctQKBwckgiIdYLaRj40c,7108
 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
@@ -44,11 +43,11 @@ waldiez/exporting/chats/utils/sequential.py,sha256=TYAte2hwpGkEENjCMer-6PcL2r8PE
 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=DIiplGZaUretIpp6z2inEiiKIxNVhPw8-DF6tC23gjg,18280
+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=3sINpH9q5h4v14tsV5G1abUgm6-v5HdVwUyfsN8CRN4,2662
+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=h8FqyAU0ApSFaie2lipxk69rNWNGwp_xAuhPpT991-M,5552
@@ -59,12 +58,13 @@ waldiez/exporting/models/utils.py,sha256=g15N_JA21Ly5nzCfburEbHf6fTWX7hG5rxyTvAw
 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=Hc54e_aTzckmi8jsfgNP-an2ocx2HRQhTVhWaexNzBI,10737
-waldiez/models/agents/__init__.py,sha256=HOY3IG1WWNqepRkETjFZihHRZHgipxWO5cxeVWm1Rv0,4553
-waldiez/models/agents/agents.py,sha256=pSB3UjeUcYpmWqKoxKpT_CAp2YMzN6l2-ZEYh8Pdddc,4322
+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=WTqElKnOIoDUN9tfaVR4q3-U-DyyrkyV8x7296eZ8jE,1941
-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/__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=Bd8dWYlCenuU4VeusAL_qxkXzYSuxl9e3a7FhHs6UCY,5435
-waldiez/models/model/__init__.py,sha256=1cOyDThhZV8o7pZN1DaIhh12jWty_YMegoQ4Jg7uKyI,383
-waldiez/models/model/model.py,sha256=k31Tr5mjW4zmahBI52e0pBPnOSxXY0KgOgiNKvNVLQw,7139
-waldiez/models/model/model_data.py,sha256=EwObYtzPyxZqADdTeWjULNmb3dLgF91mzK3r1nxxJJk,3819
+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.11.dist-info/METADATA,sha256=olpy7txsUlYNHBlFo5-6fU9F1rRRk8_sgeNiw7hq0Hc,9422
-waldiez-0.3.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-waldiez-0.3.11.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
-waldiez-0.3.11.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
-waldiez-0.3.11.dist-info/licenses/NOTICE.md,sha256=lrKsUNrpE18LPSLPJY_kt2ZJFZJEwH55wPv5Axcheyc,133
-waldiez-0.3.11.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