waldiez 0.4.5__tar.gz → 0.4.6__tar.gz

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.

Files changed (151) hide show
  1. {waldiez-0.4.5 → waldiez-0.4.6}/PKG-INFO +36 -31
  2. {waldiez-0.4.5 → waldiez-0.4.6}/README.md +10 -6
  3. {waldiez-0.4.5 → waldiez-0.4.6}/pyproject.toml +28 -31
  4. waldiez-0.4.6/waldiez/__init__.py +38 -0
  5. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/_version.py +1 -1
  6. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/common/ag2_version.py +2 -2
  7. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/model/extra_requirements.py +2 -2
  8. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/skill/extra_requirements.py +2 -6
  9. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/waldiez.py +4 -3
  10. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/running/__init__.py +2 -0
  11. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/running/environment.py +20 -0
  12. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/running/running.py +3 -2
  13. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/utils/__init__.py +2 -0
  14. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/utils/conflict_checker.py +3 -3
  15. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/utils/pysqlite3_checker.py +63 -21
  16. waldiez-0.4.6/waldiez/utils/rdps_checker.py +122 -0
  17. waldiez-0.4.5/waldiez/__init__.py +0 -49
  18. waldiez-0.4.5/waldiez/utils/check_rdps.py +0 -18
  19. {waldiez-0.4.5 → waldiez-0.4.6}/.gitignore +0 -0
  20. {waldiez-0.4.5 → waldiez-0.4.6}/LICENSE +0 -0
  21. {waldiez-0.4.5 → waldiez-0.4.6}/NOTICE.md +0 -0
  22. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/__main__.py +0 -0
  23. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/cli.py +0 -0
  24. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporter.py +0 -0
  25. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/__init__.py +0 -0
  26. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/__init__.py +0 -0
  27. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/agent_exporter.py +0 -0
  28. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/__init__.py +0 -0
  29. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/captain_agent.py +0 -0
  30. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/code_execution.py +0 -0
  31. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/group_manager.py +0 -0
  32. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/rag_user/__init__.py +0 -0
  33. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/rag_user/chroma_utils.py +0 -0
  34. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/rag_user/mongo_utils.py +0 -0
  35. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/rag_user/pgvector_utils.py +0 -0
  36. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/rag_user/qdrant_utils.py +0 -0
  37. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/rag_user/rag_user.py +0 -0
  38. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/rag_user/vector_db.py +0 -0
  39. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/reasoning.py +0 -0
  40. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/swarm_agent.py +0 -0
  41. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/teachability.py +0 -0
  42. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/agent/utils/termination_message.py +0 -0
  43. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/base/__init__.py +0 -0
  44. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/base/agent_position.py +0 -0
  45. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/base/base_exporter.py +0 -0
  46. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/base/export_position.py +0 -0
  47. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/base/import_position.py +0 -0
  48. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/base/mixin.py +0 -0
  49. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/base/utils/__init__.py +0 -0
  50. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/base/utils/comments.py +0 -0
  51. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/base/utils/naming.py +0 -0
  52. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/base/utils/path_check.py +0 -0
  53. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/base/utils/to_string.py +0 -0
  54. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/chats/__init__.py +0 -0
  55. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/chats/chats_exporter.py +0 -0
  56. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/chats/utils/__init__.py +0 -0
  57. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/chats/utils/common.py +0 -0
  58. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/chats/utils/nested.py +0 -0
  59. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/chats/utils/sequential.py +0 -0
  60. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/chats/utils/single_chat.py +0 -0
  61. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/chats/utils/swarm.py +0 -0
  62. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/flow/__init__.py +0 -0
  63. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/flow/flow_exporter.py +0 -0
  64. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/flow/utils/__init__.py +0 -0
  65. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/flow/utils/agent_utils.py +0 -0
  66. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/flow/utils/chat_utils.py +0 -0
  67. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/flow/utils/def_main.py +0 -0
  68. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/flow/utils/flow_content.py +0 -0
  69. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/flow/utils/flow_names.py +0 -0
  70. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/flow/utils/importing_utils.py +0 -0
  71. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/flow/utils/logging_utils.py +0 -0
  72. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/models/__init__.py +0 -0
  73. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/models/models_exporter.py +0 -0
  74. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/models/utils.py +0 -0
  75. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/skills/__init__.py +0 -0
  76. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/skills/skills_exporter.py +0 -0
  77. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/exporting/skills/utils.py +0 -0
  78. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/__init__.py +0 -0
  79. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/__init__.py +0 -0
  80. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/agent/__init__.py +0 -0
  81. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/agent/agent.py +0 -0
  82. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/agent/agent_data.py +0 -0
  83. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/agent/agent_type.py +0 -0
  84. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/agent/code_execution.py +0 -0
  85. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/agent/linked_skill.py +0 -0
  86. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/agent/nested_chat.py +0 -0
  87. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/agent/teachability.py +0 -0
  88. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/agent/termination_message.py +0 -0
  89. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/agents.py +0 -0
  90. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/assistant/__init__.py +0 -0
  91. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/assistant/assistant.py +0 -0
  92. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/assistant/assistant_data.py +0 -0
  93. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/captain_agent/__init__.py +0 -0
  94. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/captain_agent/captain_agent.py +0 -0
  95. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/captain_agent/captain_agent_data.py +0 -0
  96. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/captain_agent/captain_agent_lib_entry.py +0 -0
  97. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/extra_requirements.py +0 -0
  98. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/group_manager/__init__.py +0 -0
  99. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/group_manager/group_manager.py +0 -0
  100. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/group_manager/group_manager_data.py +0 -0
  101. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/group_manager/speakers.py +0 -0
  102. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/rag_user/__init__.py +0 -0
  103. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/rag_user/rag_user.py +0 -0
  104. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/rag_user/rag_user_data.py +0 -0
  105. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/rag_user/retrieve_config.py +0 -0
  106. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/rag_user/vector_db_config.py +0 -0
  107. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/reasoning/__init__.py +0 -0
  108. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/reasoning/reasoning_agent.py +0 -0
  109. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/reasoning/reasoning_agent_data.py +0 -0
  110. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/reasoning/reasoning_agent_reason_config.py +0 -0
  111. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/swarm_agent/__init__.py +0 -0
  112. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/swarm_agent/after_work.py +0 -0
  113. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/swarm_agent/on_condition.py +0 -0
  114. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/swarm_agent/on_condition_available.py +0 -0
  115. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/swarm_agent/on_condition_target.py +0 -0
  116. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/swarm_agent/swarm_agent.py +0 -0
  117. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/swarm_agent/swarm_agent_data.py +0 -0
  118. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/swarm_agent/update_system_message.py +0 -0
  119. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/user_proxy/__init__.py +0 -0
  120. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/user_proxy/user_proxy.py +0 -0
  121. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/agents/user_proxy/user_proxy_data.py +0 -0
  122. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/chat/__init__.py +0 -0
  123. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/chat/chat.py +0 -0
  124. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/chat/chat_data.py +0 -0
  125. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/chat/chat_message.py +0 -0
  126. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/chat/chat_nested.py +0 -0
  127. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/chat/chat_summary.py +0 -0
  128. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/common/__init__.py +0 -0
  129. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/common/base.py +0 -0
  130. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/common/date_utils.py +0 -0
  131. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/common/dict_utils.py +0 -0
  132. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/common/method_utils.py +0 -0
  133. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/flow/__init__.py +0 -0
  134. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/flow/flow.py +0 -0
  135. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/flow/flow_data.py +0 -0
  136. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/flow/utils.py +0 -0
  137. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/model/__init__.py +0 -0
  138. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/model/model.py +0 -0
  139. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/model/model_data.py +0 -0
  140. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/skill/__init__.py +0 -0
  141. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/skill/skill.py +0 -0
  142. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/skill/skill_data.py +0 -0
  143. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/models/skill/skill_type.py +0 -0
  144. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/py.typed +0 -0
  145. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/runner.py +0 -0
  146. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/running/gen_seq_diagram.py +0 -0
  147. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/utils/cli_extras/__init__.py +0 -0
  148. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/utils/cli_extras/jupyter.py +0 -0
  149. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/utils/cli_extras/runner.py +0 -0
  150. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/utils/cli_extras/studio.py +0 -0
  151. {waldiez-0.4.5 → waldiez-0.4.6}/waldiez/utils/flaml_warnings.py +0 -0
@@ -1,8 +1,8 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: waldiez
3
- Version: 0.4.5
3
+ Version: 0.4.6
4
4
  Dynamic: Keywords
5
- Summary: Make AI Agents Collaborate: Drag, Drop, and Orchestrate with Waldiez.
5
+ Summary: Make AG2 Agents Collaborate: Drag, Drop, and Orchestrate with Waldiez.
6
6
  Project-URL: Homepage, https://waldiez.io
7
7
  Project-URL: Bug Tracker, https://github.com/waldiez/waldiez/issues
8
8
  Project-URL: Repository, https://github.com/waldiez/waldiez.git
@@ -21,6 +21,7 @@ Classifier: Programming Language :: Python :: 3.11
21
21
  Classifier: Programming Language :: Python :: 3.12
22
22
  Classifier: Programming Language :: Python :: 3.13
23
23
  Requires-Python: <3.14,>=3.10
24
+ Requires-Dist: ag2[openai]==0.8.7
24
25
  Requires-Dist: aiocsv==1.3.2
25
26
  Requires-Dist: aiofiles==24.1.0
26
27
  Requires-Dist: aiosqlite==0.21.0
@@ -32,10 +33,26 @@ Requires-Dist: nest-asyncio==1.6.0
32
33
  Requires-Dist: numpy<=2.2.5
33
34
  Requires-Dist: pandas>=2
34
35
  Requires-Dist: parso==0.8.4
35
- Requires-Dist: pyautogen[openai]==0.8.7
36
+ Requires-Dist: pillow>=10.0.0
36
37
  Requires-Dist: pydantic<3,>=2.10.2
37
38
  Requires-Dist: typer<0.16,>=0.9
38
39
  Provides-Extra: ag2-extras
40
+ Requires-Dist: ag2[anthropic]==0.8.7; extra == 'ag2-extras'
41
+ Requires-Dist: ag2[bedrock]==0.8.7; extra == 'ag2-extras'
42
+ Requires-Dist: ag2[cohere]==0.8.7; extra == 'ag2-extras'
43
+ Requires-Dist: ag2[gemini]==0.8.7; (sys_platform != 'win32') and extra == 'ag2-extras'
44
+ Requires-Dist: ag2[gemini]==0.8.7; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
45
+ Requires-Dist: ag2[groq]==0.8.7; extra == 'ag2-extras'
46
+ Requires-Dist: ag2[interop-crewai]==0.8.7; extra == 'ag2-extras'
47
+ Requires-Dist: ag2[interop-langchain]==0.8.7; extra == 'ag2-extras'
48
+ Requires-Dist: ag2[lmm]==0.8.7; extra == 'ag2-extras'
49
+ Requires-Dist: ag2[mistral]==0.8.7; extra == 'ag2-extras'
50
+ Requires-Dist: ag2[neo4j]==0.8.7; (sys_platform != 'win32') and extra == 'ag2-extras'
51
+ Requires-Dist: ag2[neo4j]==0.8.7; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
52
+ Requires-Dist: ag2[ollama]==0.8.7; extra == 'ag2-extras'
53
+ Requires-Dist: ag2[together]==0.8.7; (sys_platform != 'win32') and extra == 'ag2-extras'
54
+ Requires-Dist: ag2[together]==0.8.7; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
55
+ Requires-Dist: ag2[websurfer]==0.8.7; extra == 'ag2-extras'
39
56
  Requires-Dist: beautifulsoup4; extra == 'ag2-extras'
40
57
  Requires-Dist: chromadb>=0.5.10; (sys_platform != 'win32') and extra == 'ag2-extras'
41
58
  Requires-Dist: chromadb>=0.5.10; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
@@ -59,22 +76,6 @@ Requires-Dist: psycopg==3.2.6; (sys_platform == 'win32' and platform_machine ==
59
76
  Requires-Dist: psycopg>=3.2.6; (sys_platform == 'win32') and extra == 'ag2-extras'
60
77
  Requires-Dist: psycopg[binary]==3.2.6; (sys_platform != 'linux' and platform_machine != 'arm64' and platform_machine != 'ARM64' and platform_machine != 'aarch64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
61
78
  Requires-Dist: psycopg[binary]>=3.2.6; (sys_platform != 'win32') and extra == 'ag2-extras'
62
- Requires-Dist: pyautogen[anthropic]==0.8.7; extra == 'ag2-extras'
63
- Requires-Dist: pyautogen[bedrock]==0.8.7; extra == 'ag2-extras'
64
- Requires-Dist: pyautogen[cohere]==0.8.7; extra == 'ag2-extras'
65
- Requires-Dist: pyautogen[gemini]==0.8.7; (sys_platform != 'win32') and extra == 'ag2-extras'
66
- Requires-Dist: pyautogen[gemini]==0.8.7; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
67
- Requires-Dist: pyautogen[groq]==0.8.7; extra == 'ag2-extras'
68
- Requires-Dist: pyautogen[interop-crewai]==0.8.7; extra == 'ag2-extras'
69
- Requires-Dist: pyautogen[interop-langchain]==0.8.7; extra == 'ag2-extras'
70
- Requires-Dist: pyautogen[lmm]==0.8.7; extra == 'ag2-extras'
71
- Requires-Dist: pyautogen[mistral]==0.8.7; extra == 'ag2-extras'
72
- Requires-Dist: pyautogen[neo4j]==0.8.7; (sys_platform != 'win32') and extra == 'ag2-extras'
73
- Requires-Dist: pyautogen[neo4j]==0.8.7; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
74
- Requires-Dist: pyautogen[ollama]==0.8.7; extra == 'ag2-extras'
75
- Requires-Dist: pyautogen[together]==0.8.7; (sys_platform != 'win32') and extra == 'ag2-extras'
76
- Requires-Dist: pyautogen[together]==0.8.7; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
77
- Requires-Dist: pyautogen[websurfer]==0.8.7; extra == 'ag2-extras'
78
79
  Requires-Dist: pydantic-ai>=0.0.21; extra == 'ag2-extras'
79
80
  Requires-Dist: pymongo>=4.11; extra == 'ag2-extras'
80
81
  Requires-Dist: pypdf; extra == 'ag2-extras'
@@ -97,12 +98,12 @@ Requires-Dist: pre-commit==4.2.0; extra == 'dev'
97
98
  Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
98
99
  Requires-Dist: pylint==3.3.6; extra == 'dev'
99
100
  Requires-Dist: python-dotenv==1.1.0; extra == 'dev'
100
- Requires-Dist: ruff==0.11.7; extra == 'dev'
101
+ Requires-Dist: ruff==0.11.8; extra == 'dev'
101
102
  Requires-Dist: toml==0.10.2; (python_version <= '3.10') and extra == 'dev'
102
103
  Requires-Dist: types-pyyaml==6.0.12.20250402; extra == 'dev'
103
104
  Requires-Dist: types-requests==2.32.0.20250328; extra == 'dev'
104
105
  Requires-Dist: types-toml==0.10.8.20240310; extra == 'dev'
105
- Requires-Dist: yamllint==1.37.0; extra == 'dev'
106
+ Requires-Dist: yamllint==1.37.1; extra == 'dev'
106
107
  Provides-Extra: docs
107
108
  Requires-Dist: mdx-include==1.4.2; extra == 'docs'
108
109
  Requires-Dist: mdx-truly-sane-lists==1.3; extra == 'docs'
@@ -118,28 +119,31 @@ Requires-Dist: mkdocstrings[crystal,python]==0.29.1; extra == 'docs'
118
119
  Provides-Extra: jupyter
119
120
  Requires-Dist: jupyter-server==2.15.0; extra == 'jupyter'
120
121
  Requires-Dist: jupyterlab>=4.3.0; extra == 'jupyter'
121
- Requires-Dist: waldiez-jupyter==0.4.5; extra == 'jupyter'
122
+ Requires-Dist: waldiez-jupyter==0.4.6; extra == 'jupyter'
122
123
  Provides-Extra: runner
123
- Requires-Dist: waldiez-runner==0.4.5; (python_version >= '3.11') and extra == 'runner'
124
+ Requires-Dist: waldiez-runner==0.4.6; (python_version >= '3.11') and extra == 'runner'
124
125
  Provides-Extra: studio
125
- Requires-Dist: waldiez-studio==0.4.5; extra == 'studio'
126
+ Requires-Dist: waldiez-studio==0.4.6; extra == 'studio'
126
127
  Provides-Extra: test
127
128
  Requires-Dist: pytest-asyncio==0.26.0; extra == 'test'
128
129
  Requires-Dist: pytest-cov==6.1.1; extra == 'test'
129
130
  Requires-Dist: pytest-html==4.1.1; extra == 'test'
130
131
  Requires-Dist: pytest-sugar==1.0.0; extra == 'test'
131
- Requires-Dist: pytest-timeout==2.3.1; extra == 'test'
132
+ Requires-Dist: pytest-timeout==2.4.0; extra == 'test'
132
133
  Requires-Dist: pytest-xdist==3.6.1; extra == 'test'
133
134
  Requires-Dist: pytest==8.3.5; extra == 'test'
134
135
  Description-Content-Type: text/markdown
135
136
 
136
137
  # Waldiez
137
138
 
138
- ![CI Build](https://github.com/waldiez/python/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/python/badge.svg)](https://coveralls.io/github/waldiez/python) [![PyPI version](https://badge.fury.io/py/waldiez.svg?icon=si%3Apython)](https://badge.fury.io/py/waldiez) [![npm version](https://badge.fury.io/js/@waldiez%2Freact.svg)](https://badge.fury.io/js/@waldiez%2Freact)
139
+ ![CI Build](https://github.com/waldiez/python/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/python/badge.svg)](https://coveralls.io/github/waldiez/python) [![PyPI Downloads](https://static.pepy.tech/badge/waldiez)](https://pepy.tech/projects/waldiez) [![PyPI version](https://badge.fury.io/py/waldiez.svg?icon=si%3Apython)](https://badge.fury.io/py/waldiez) [![npm version](https://badge.fury.io/js/@waldiez%2Freact.svg)](https://badge.fury.io/js/@waldiez%2Freact)
139
140
 
140
- Translate a Waldiez flow:
141
+ ## Make AG2 Agents Collaborate: Drag, Drop, and Orchestrate with Waldiez
141
142
 
142
- ![Flow](https://raw.githubusercontent.com/waldiez/python/refs/heads/main/docs/static/images/overview.webp)
143
+ Design AI Agents and translate a Waldiez flow to AG2:
144
+
145
+
146
+ https://github.com/user-attachments/assets/71d4b8d1-a24b-45ab-ab53-dfc193e8fcaa
143
147
 
144
148
  To a python script or a jupyter notebook with the corresponding [ag2](https://github.com/ag2ai/ag2/) agents and chats.
145
149
 
@@ -278,7 +282,7 @@ runner.run(output_path=output_path)
278
282
 
279
283
  ## Known Conflicts
280
284
 
281
- - **autogen-agentchat**: This package conflicts with `ag2` / `pyautogen`. Ensure that `autogen-agentchat` is uninstalled before installing `waldiez`. If you have already installed `autogen-agentchat`, you can uninstall it with the following command:
285
+ - **autogen-agentchat**: This package conflicts with `ag2`. Ensure that `autogen-agentchat` is uninstalled before installing `waldiez`. If you have already installed `autogen-agentchat`, you can uninstall it with the following command:
282
286
 
283
287
  ```shell
284
288
  pip uninstall autogen-agentchat -y
@@ -287,7 +291,7 @@ runner.run(output_path=output_path)
287
291
  If already installed waldiez you might need to reinstall it after uninstalling `autogen-agentchat`:
288
292
 
289
293
  ```shell
290
- pip install --force --no-cache waldiez pyautogen
294
+ pip install --force --no-cache waldiez ag2
291
295
  ```
292
296
 
293
297
  ## See also
@@ -295,7 +299,7 @@ runner.run(output_path=output_path)
295
299
  - [Waldiez Playground](https://waldiez.github.io)
296
300
  - [React Component](https://github.com/waldiez/waldiez/blob/main/README.npm.md)
297
301
  - [Waldiez Studio](https://github.com/waldiez/studio)
298
- - [Waldiez JupyterLab Extension](htttps://github.com/waldiez/jupyter)
302
+ - [Waldiez JupyterLab Extension](https://github.com/waldiez/jupyter)
299
303
  - [Waldiez VSCode Extension](https://github.com/waldiez/vscode)
300
304
 
301
305
  ## Contributors ✨
@@ -311,6 +315,7 @@ Thanks goes to these wonderful people ([emoji key](https://allcontributors.org/d
311
315
  <td align="center" valign="top" width="14.28%"><a href="https://scholar.google.com/citations?user=JmW9DwkAAAAJ"><img src="https://avatars.githubusercontent.com/u/29335277?v=4?s=100" width="100px;" alt="Panagiotis Kasnesis"/><br /><sub><b>Panagiotis Kasnesis</b></sub></a><br /><a href="#projectManagement-ounospanas" title="Project Management">📆</a> <a href="#research-ounospanas" title="Research">🔬</a></td>
312
316
  <td align="center" valign="top" width="14.28%"><a href="https://humancentered.gr/"><img src="https://avatars.githubusercontent.com/u/3456066?v=4?s=100" width="100px;" alt="Stella Ioannidou"/><br /><sub><b>Stella Ioannidou</b></sub></a><br /><a href="#promotion-siioannidou" title="Promotion">📣</a> <a href="#design-siioannidou" title="Design">🎨</a></td>
313
317
  <td align="center" valign="top" width="14.28%"><a href="https://github.com/lazToum"><img src="https://avatars.githubusercontent.com/u/4764837?v=4?s=100" width="100px;" alt="Lazaros Toumanidis"/><br /><sub><b>Lazaros Toumanidis</b></sub></a><br /><a href="https://github.com/waldiez/react/commits?author=lazToum" title="Code">💻</a></td>
318
+ <td align="center" valign="top" width="14.28%"><a href="https://github.com/amaliacontiero"><img src="https://avatars.githubusercontent.com/u/29499343?v=4?s=100" width="100px;" alt="Amalia Contiero"/><br /><sub><b>Amalia Contiero</b></sub></a><br /><a href="https://github.com/waldiez/vscode/commits?author=amaliacontiero" title="Code">💻</a> <a href="https://github.com/waldiez/vscode/issues?q=author%3Aamaliacontiero" title="Bug reports">🐛</a></td>
314
319
  </tr>
315
320
  </tbody>
316
321
  <tfoot>
@@ -1,10 +1,13 @@
1
1
  # Waldiez
2
2
 
3
- ![CI Build](https://github.com/waldiez/python/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/python/badge.svg)](https://coveralls.io/github/waldiez/python) [![PyPI version](https://badge.fury.io/py/waldiez.svg?icon=si%3Apython)](https://badge.fury.io/py/waldiez) [![npm version](https://badge.fury.io/js/@waldiez%2Freact.svg)](https://badge.fury.io/js/@waldiez%2Freact)
3
+ ![CI Build](https://github.com/waldiez/python/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/python/badge.svg)](https://coveralls.io/github/waldiez/python) [![PyPI Downloads](https://static.pepy.tech/badge/waldiez)](https://pepy.tech/projects/waldiez) [![PyPI version](https://badge.fury.io/py/waldiez.svg?icon=si%3Apython)](https://badge.fury.io/py/waldiez) [![npm version](https://badge.fury.io/js/@waldiez%2Freact.svg)](https://badge.fury.io/js/@waldiez%2Freact)
4
4
 
5
- Translate a Waldiez flow:
5
+ ## Make AG2 Agents Collaborate: Drag, Drop, and Orchestrate with Waldiez
6
6
 
7
- ![Flow](https://raw.githubusercontent.com/waldiez/python/refs/heads/main/docs/static/images/overview.webp)
7
+ Design AI Agents and translate a Waldiez flow to AG2:
8
+
9
+
10
+ https://github.com/user-attachments/assets/71d4b8d1-a24b-45ab-ab53-dfc193e8fcaa
8
11
 
9
12
  To a python script or a jupyter notebook with the corresponding [ag2](https://github.com/ag2ai/ag2/) agents and chats.
10
13
 
@@ -143,7 +146,7 @@ runner.run(output_path=output_path)
143
146
 
144
147
  ## Known Conflicts
145
148
 
146
- - **autogen-agentchat**: This package conflicts with `ag2` / `pyautogen`. Ensure that `autogen-agentchat` is uninstalled before installing `waldiez`. If you have already installed `autogen-agentchat`, you can uninstall it with the following command:
149
+ - **autogen-agentchat**: This package conflicts with `ag2`. Ensure that `autogen-agentchat` is uninstalled before installing `waldiez`. If you have already installed `autogen-agentchat`, you can uninstall it with the following command:
147
150
 
148
151
  ```shell
149
152
  pip uninstall autogen-agentchat -y
@@ -152,7 +155,7 @@ runner.run(output_path=output_path)
152
155
  If already installed waldiez you might need to reinstall it after uninstalling `autogen-agentchat`:
153
156
 
154
157
  ```shell
155
- pip install --force --no-cache waldiez pyautogen
158
+ pip install --force --no-cache waldiez ag2
156
159
  ```
157
160
 
158
161
  ## See also
@@ -160,7 +163,7 @@ runner.run(output_path=output_path)
160
163
  - [Waldiez Playground](https://waldiez.github.io)
161
164
  - [React Component](https://github.com/waldiez/waldiez/blob/main/README.npm.md)
162
165
  - [Waldiez Studio](https://github.com/waldiez/studio)
163
- - [Waldiez JupyterLab Extension](htttps://github.com/waldiez/jupyter)
166
+ - [Waldiez JupyterLab Extension](https://github.com/waldiez/jupyter)
164
167
  - [Waldiez VSCode Extension](https://github.com/waldiez/vscode)
165
168
 
166
169
  ## Contributors ✨
@@ -176,6 +179,7 @@ Thanks goes to these wonderful people ([emoji key](https://allcontributors.org/d
176
179
  <td align="center" valign="top" width="14.28%"><a href="https://scholar.google.com/citations?user=JmW9DwkAAAAJ"><img src="https://avatars.githubusercontent.com/u/29335277?v=4?s=100" width="100px;" alt="Panagiotis Kasnesis"/><br /><sub><b>Panagiotis Kasnesis</b></sub></a><br /><a href="#projectManagement-ounospanas" title="Project Management">📆</a> <a href="#research-ounospanas" title="Research">🔬</a></td>
177
180
  <td align="center" valign="top" width="14.28%"><a href="https://humancentered.gr/"><img src="https://avatars.githubusercontent.com/u/3456066?v=4?s=100" width="100px;" alt="Stella Ioannidou"/><br /><sub><b>Stella Ioannidou</b></sub></a><br /><a href="#promotion-siioannidou" title="Promotion">📣</a> <a href="#design-siioannidou" title="Design">🎨</a></td>
178
181
  <td align="center" valign="top" width="14.28%"><a href="https://github.com/lazToum"><img src="https://avatars.githubusercontent.com/u/4764837?v=4?s=100" width="100px;" alt="Lazaros Toumanidis"/><br /><sub><b>Lazaros Toumanidis</b></sub></a><br /><a href="https://github.com/waldiez/react/commits?author=lazToum" title="Code">💻</a></td>
182
+ <td align="center" valign="top" width="14.28%"><a href="https://github.com/amaliacontiero"><img src="https://avatars.githubusercontent.com/u/29499343?v=4?s=100" width="100px;" alt="Amalia Contiero"/><br /><sub><b>Amalia Contiero</b></sub></a><br /><a href="https://github.com/waldiez/vscode/commits?author=amaliacontiero" title="Code">💻</a> <a href="https://github.com/waldiez/vscode/issues?q=author%3Aamaliacontiero" title="Bug reports">🐛</a></td>
179
183
  </tr>
180
184
  </tbody>
181
185
  <tfoot>
@@ -20,24 +20,21 @@ classifiers = [
20
20
  "Intended Audience :: Developers",
21
21
  ]
22
22
  dependencies = [
23
+ "ag2[openai]==0.8.7",
23
24
  "aiocsv==1.3.2",
24
25
  "aiofiles==24.1.0",
25
26
  "aiosqlite==0.21.0",
26
27
  "asyncer==0.0.8",
27
- "numpy<=2.2.5",
28
- "nest_asyncio==1.6.0",
29
28
  "graphviz==0.20.3",
30
- "pandas>=2",
31
- "pyautogen[openai]==0.8.7",
29
+ "httpx<1",
32
30
  "jupytext",
33
- # "rpds-py @ git+https://github.com/crate-py/rpds.git@v0.24.0 ;sys_platform == "win32" and platform_machine == "arm64"",
34
- # "rpds-py @ git+https://github.com/crate-py/rpds.git@v0.24.0 ;sys_platform == "win32" and platform_machine == "ARM64"",
35
- # "rpds-py @ git+https://github.com/crate-py/rpds.git@v0.24.0 ;sys_platform == "win32" and platform_machine == "aarch64"",
36
- # "rpds-py @ git+https://github.com/crate-py/rpds.git@v0.24.0 ;sys_platform == "win32" and platform_machine == "AARCH64"",
31
+ "nest_asyncio==1.6.0",
32
+ "numpy<=2.2.5",
33
+ "pandas>=2",
37
34
  "parso==0.8.4",
35
+ "pillow>=10.0.0",
38
36
  "pydantic>=2.10.2,<3",
39
37
  "typer>=0.9,<0.16",
40
- "httpx<1",
41
38
  ]
42
39
  dynamic = ["version", "description", "authors", "urls", "keywords"]
43
40
  name = "waldiez"
@@ -78,22 +75,22 @@ fields = ["description", "authors", "urls"]
78
75
  [project.optional-dependencies]
79
76
  ag2_extras = [
80
77
  # models
81
- "pyautogen[anthropic]==0.8.7",
82
- "pyautogen[bedrock]==0.8.7",
83
- "pyautogen[cohere]==0.8.7",
84
- "pyautogen[gemini]==0.8.7 ;sys_platform != 'win32'",
85
- "pyautogen[gemini]==0.8.7 ;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
86
- "pyautogen[groq]==0.8.7",
87
- "pyautogen[lmm]==0.8.7",
88
- "pyautogen[mistral]==0.8.7",
89
- "pyautogen[ollama]==0.8.7",
90
- "pyautogen[together]==0.8.7 ;sys_platform != 'win32'",
91
- "pyautogen[together]==0.8.7 ;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'", # utils
92
- "pyautogen[interop-crewai]==0.8.7",
93
- "pyautogen[interop-langchain]==0.8.7",
94
- "pyautogen[neo4j]==0.8.7 ;sys_platform != 'win32'",
95
- "pyautogen[neo4j]==0.8.7 ;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
96
- "pyautogen[websurfer]==0.8.7",
78
+ "ag2[anthropic]==0.8.7",
79
+ "ag2[bedrock]==0.8.7",
80
+ "ag2[cohere]==0.8.7",
81
+ "ag2[gemini]==0.8.7 ;sys_platform != 'win32'",
82
+ "ag2[gemini]==0.8.7 ;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
83
+ "ag2[groq]==0.8.7",
84
+ "ag2[lmm]==0.8.7",
85
+ "ag2[mistral]==0.8.7",
86
+ "ag2[ollama]==0.8.7",
87
+ "ag2[together]==0.8.7 ;sys_platform != 'win32'",
88
+ "ag2[together]==0.8.7 ;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'", # utils
89
+ "ag2[interop-crewai]==0.8.7",
90
+ "ag2[interop-langchain]==0.8.7",
91
+ "ag2[neo4j]==0.8.7 ;sys_platform != 'win32'",
92
+ "ag2[neo4j]==0.8.7 ;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
93
+ "ag2[websurfer]==0.8.7",
97
94
  "chromadb>=0.5.10 ;sys_platform != 'win32'",
98
95
  "chromadb>=0.5.10 ;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
99
96
  "pgvector>=0.4.0",
@@ -141,9 +138,9 @@ dev = [
141
138
  "pydocstyle==6.3.0",
142
139
  "pylint==3.3.6",
143
140
  "python-dotenv==1.1.0",
144
- "ruff==0.11.7",
141
+ "ruff==0.11.8",
145
142
  "types-pyyaml==6.0.12.20250402",
146
- "yamllint==1.37.0",
143
+ "yamllint==1.37.1",
147
144
  "toml==0.10.2; python_version <= '3.10'",
148
145
  "types-toml==0.10.8.20240310",
149
146
  "types-requests==2.32.0.20250328",
@@ -162,15 +159,15 @@ docs = [
162
159
  "mkdocs-open-in-new-tab==1.0.8",
163
160
  ]
164
161
  jupyter = [
165
- "waldiez_jupyter==0.4.5",
162
+ "waldiez_jupyter==0.4.6",
166
163
  "jupyterlab>=4.3.0",
167
164
  "jupyter_server==2.15.0",
168
165
  ]
169
166
  runner = [
170
- "waldiez_runner==0.4.5; python_version >= '3.11'",
167
+ "waldiez_runner==0.4.6; python_version >= '3.11'",
171
168
  ]
172
169
  studio = [
173
- "waldiez_studio==0.4.5",
170
+ "waldiez_studio==0.4.6",
174
171
  ]
175
172
  test = [
176
173
  "pytest==8.3.5",
@@ -178,7 +175,7 @@ test = [
178
175
  "pytest-cov==6.1.1",
179
176
  "pytest-html==4.1.1",
180
177
  "pytest-sugar==1.0.0",
181
- "pytest-timeout==2.3.1",
178
+ "pytest-timeout==2.4.0",
182
179
  "pytest-xdist==3.6.1",
183
180
  ]
184
181
 
@@ -0,0 +1,38 @@
1
+ # SPDX-License-Identifier: Apache-2.0.
2
+ # Copyright (c) 2024 - 2025 Waldiez and contributors.
3
+ """Waldiez package."""
4
+
5
+ from .exporter import WaldiezExporter
6
+ from .models import Waldiez
7
+ from .runner import WaldiezRunner
8
+ from .utils import check_conflicts, check_flaml_warnings, check_rpds_py
9
+
10
+ # flake8: noqa: F401
11
+ # pylint: disable=import-error,line-too-long
12
+ # pyright: reportMissingImports=false
13
+ try:
14
+ # noqa: I001
15
+ from ._version import __version__ # type: ignore[unused-ignore, unused-import, import-not-found, import-untyped] # noqa
16
+ except ImportError: # pragma: no cover
17
+ import warnings
18
+
19
+ warnings.warn(
20
+ "Importing __version__ failed. Using 'dev' as version.",
21
+ stacklevel=2,
22
+ )
23
+ __version__ = "dev"
24
+
25
+ __WALDIEZ_INITIALIZED = False
26
+
27
+ if not __WALDIEZ_INITIALIZED:
28
+ __WALDIEZ_INITIALIZED = True
29
+ check_conflicts()
30
+ check_flaml_warnings()
31
+ check_rpds_py()
32
+
33
+ __all__ = [
34
+ "Waldiez",
35
+ "WaldiezExporter",
36
+ "WaldiezRunner",
37
+ "__version__",
38
+ ]
@@ -5,4 +5,4 @@
5
5
  This file is automatically generated by Hatchling.
6
6
  Do not edit this file directly.
7
7
  """
8
- __version__ = VERSION = "0.4.5"
8
+ __version__ = VERSION = "0.4.6"
@@ -18,7 +18,7 @@ def get_autogen_version() -> str:
18
18
  Raises
19
19
  ------
20
20
  ValueError
21
- If pyautogen is not installed.
21
+ If ag2 is not installed.
22
22
  """
23
23
  # pylint: disable=import-outside-toplevel
24
24
  with warnings.catch_warnings():
@@ -26,5 +26,5 @@ def get_autogen_version() -> str:
26
26
  try:
27
27
  from autogen.version import __version__ as ag2 # type: ignore
28
28
  except ImportError as error: # pragma: no cover
29
- raise ValueError("pyautogen is not installed.") from error
29
+ raise ValueError("ag2 is not installed.") from error
30
30
  return ag2
@@ -39,11 +39,11 @@ def get_models_extra_requirements(
39
39
  for requirement in model.requirements:
40
40
  model_requirements.add(requirement)
41
41
  if model.data.api_type == "google":
42
- model_requirements.add(f"pyautogen[gemini]=={autogen_version}")
42
+ model_requirements.add(f"ag2[gemini]=={autogen_version}")
43
43
  continue
44
44
  if model.data.api_type in models_with_additional_requirements:
45
45
  model_requirements.add(
46
- f"pyautogen[{model.data.api_type}]=={autogen_version}"
46
+ f"ag2[{model.data.api_type}]=={autogen_version}"
47
47
  )
48
48
  return model_requirements
49
49
 
@@ -28,13 +28,9 @@ def get_skills_extra_requirements(
28
28
  skill_requirements: Set[str] = set()
29
29
  for skill in skills:
30
30
  if skill.skill_type == "langchain":
31
- skill_requirements.add(
32
- f"pyautogen[interop-langchain]=={autogen_version}"
33
- )
31
+ skill_requirements.add(f"ag2[interop-langchain]=={autogen_version}")
34
32
  if skill.skill_type == "crewai":
35
- skill_requirements.add(
36
- f"pyautogen[interop-crewai]=={autogen_version}"
37
- )
33
+ skill_requirements.add(f"ag2[interop-crewai]=={autogen_version}")
38
34
  for requirement in skill.requirements:
39
35
  skill_requirements.add(requirement)
40
36
  return skill_requirements
@@ -230,8 +230,9 @@ class Waldiez:
230
230
  """Get the flow requirements."""
231
231
  autogen_version = get_autogen_version()
232
232
  requirements_list = filter(
233
- # we use the fixed "pyautogen=={autogen_version}" below
233
+ # we use the fixed "ag2=={autogen_version}" below
234
234
  lambda requirement: not (
235
+ # cspell:disable-next-line
235
236
  requirement.startswith("pyautogen")
236
237
  or requirement.startswith("ag2")
237
238
  or requirement.startswith("autogen")
@@ -239,12 +240,12 @@ class Waldiez:
239
240
  self.flow.requirements,
240
241
  )
241
242
  requirements = set(requirements_list)
242
- requirements.add(f"pyautogen[openai]=={autogen_version}")
243
+ requirements.add(f"ag2[openai]=={autogen_version}")
243
244
  if self.has_rag_agents:
244
245
  rag_extras = get_retrievechat_extra_requirements(self.agents)
245
246
  requirements.update(rag_extras)
246
247
  if self.has_multimodal_agents:
247
- requirements.add(f"pyautogen[lmm]=={autogen_version}")
248
+ requirements.add(f"ag2[lmm]=={autogen_version}")
248
249
  if self.has_captain_agents:
249
250
  # pysqlite3-binary might not get installed on windows
250
251
  captain_extras = [
@@ -4,6 +4,7 @@
4
4
 
5
5
  from .environment import (
6
6
  in_virtualenv,
7
+ is_root,
7
8
  refresh_environment,
8
9
  reset_env_vars,
9
10
  set_env_vars,
@@ -26,6 +27,7 @@ __all__ = [
26
27
  "chdir",
27
28
  "get_printer",
28
29
  "in_virtualenv",
30
+ "is_root",
29
31
  "install_requirements",
30
32
  "refresh_environment",
31
33
  "reset_env_vars",
@@ -25,6 +25,26 @@ def in_virtualenv() -> bool:
25
25
  )
26
26
 
27
27
 
28
+ def is_root() -> bool:
29
+ """Check if the script is running as root/administrator.
30
+
31
+ Returns
32
+ -------
33
+ bool
34
+ True if running as root/administrator, False otherwise.
35
+ """
36
+ # pylint: disable=import-outside-toplevel,line-too-long
37
+ if os.name == "nt":
38
+ try:
39
+ import ctypes
40
+
41
+ return ctypes.windll.shell32.IsUserAnAdmin() != 0 # type: ignore[unused-ignore,attr-defined] # noqa: E501
42
+ except Exception: # pylint: disable=broad-exception-caught
43
+ return False
44
+ else:
45
+ return os.getuid() == 0
46
+
47
+
28
48
  def refresh_environment() -> None:
29
49
  """Refresh the environment."""
30
50
  with warnings.catch_warnings():
@@ -23,7 +23,7 @@ from typing import (
23
23
  Union,
24
24
  )
25
25
 
26
- from .environment import in_virtualenv
26
+ from .environment import in_virtualenv, is_root
27
27
  from .gen_seq_diagram import generate_sequence_diagram
28
28
 
29
29
  # pylint: disable=import-outside-toplevel
@@ -126,7 +126,8 @@ def install_requirements(
126
126
  # but it might fail if we don't
127
127
  break_system_packages = os.environ.get("PIP_BREAK_SYSTEM_PACKAGES", "")
128
128
  os.environ["PIP_BREAK_SYSTEM_PACKAGES"] = "1"
129
- pip_install.append("--user")
129
+ if not is_root():
130
+ pip_install.append("--user")
130
131
  pip_install.extend(extra_requirements)
131
132
  # pylint: disable=too-many-try-statements
132
133
  try:
@@ -6,10 +6,12 @@ from .cli_extras import add_cli_extras
6
6
  from .conflict_checker import check_conflicts
7
7
  from .flaml_warnings import check_flaml_warnings
8
8
  from .pysqlite3_checker import check_pysqlite3
9
+ from .rdps_checker import check_rpds_py
9
10
 
10
11
  __all__ = [
11
12
  "check_conflicts",
12
13
  "check_flaml_warnings",
13
14
  "add_cli_extras",
14
15
  "check_pysqlite3",
16
+ "check_rpds_py",
15
17
  ]
@@ -17,11 +17,11 @@ def _check_conflicts() -> None: # pragma: no cover
17
17
  print(
18
18
  "Conflict detected: 'autogen-agentchat' is installed "
19
19
  "in the current environment, \n"
20
- "which conflicts with 'ag2' / 'pyautogen'.\n"
20
+ "which conflicts with 'ag2'.\n"
21
21
  "Please uninstall 'autogen-agentchat': \n"
22
22
  f"{sys.executable} -m pip uninstall -y autogen-agentchat" + "\n"
23
- "And install 'pyautogen' (and/or 'waldiez') again: \n"
24
- f"{sys.executable} -m pip install --force pyautogen waldiez",
23
+ "And install 'ag2' (and/or 'waldiez') again: \n"
24
+ f"{sys.executable} -m pip install --force ag2 waldiez",
25
25
  file=sys.stderr,
26
26
  )
27
27
  sys.exit(1)