waldiez 0.5.3__tar.gz → 0.5.5__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 (232) hide show
  1. {waldiez-0.5.3 → waldiez-0.5.5}/PKG-INFO +59 -62
  2. {waldiez-0.5.3 → waldiez-0.5.5}/pyproject.toml +74 -72
  3. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/_version.py +1 -1
  4. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/cli.py +3 -27
  5. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporter.py +0 -13
  6. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/exporter.py +38 -0
  7. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/__init__.py +2 -0
  8. waldiez-0.5.5/waldiez/exporting/agent/extras/doc_agent_extras.py +366 -0
  9. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/group_member_extras.py +3 -2
  10. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/processor.py +113 -15
  11. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/chats/processor.py +2 -21
  12. waldiez-0.5.5/waldiez/exporting/chats/utils/common.py +112 -0
  13. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/chats/utils/group.py +6 -3
  14. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/chats/utils/nested.py +1 -1
  15. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/chats/utils/sequential.py +25 -9
  16. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/chats/utils/single.py +8 -6
  17. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/context.py +0 -12
  18. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/agent_extras/standard_extras.py +3 -1
  19. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/base.py +20 -17
  20. waldiez-0.5.5/waldiez/exporting/core/extras/path_resolver.py +91 -0
  21. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/serializer.py +16 -1
  22. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/protocols.py +17 -0
  23. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/types.py +6 -9
  24. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/execution_generator.py +56 -21
  25. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/exporter.py +1 -4
  26. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/factory.py +0 -9
  27. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/file_generator.py +6 -0
  28. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/orchestrator.py +27 -21
  29. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/utils/__init__.py +0 -2
  30. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/utils/common.py +15 -96
  31. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/utils/importing.py +4 -0
  32. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/mqtt.py +33 -14
  33. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/redis.py +18 -13
  34. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/structured.py +9 -4
  35. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/utils.py +32 -0
  36. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/ws.py +8 -2
  37. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/__init__.py +6 -0
  38. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/__init__.py +8 -0
  39. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/agent/agent.py +136 -38
  40. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/agent/agent_type.py +3 -2
  41. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/agents.py +10 -0
  42. waldiez-0.5.5/waldiez/models/agents/doc_agent/__init__.py +13 -0
  43. waldiez-0.5.5/waldiez/models/agents/doc_agent/doc_agent.py +126 -0
  44. waldiez-0.5.5/waldiez/models/agents/doc_agent/doc_agent_data.py +149 -0
  45. waldiez-0.5.5/waldiez/models/agents/doc_agent/rag_query_engine.py +127 -0
  46. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/flow/flow.py +13 -2
  47. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/model/__init__.py +2 -2
  48. waldiez-0.5.5/waldiez/models/model/_aws.py +75 -0
  49. waldiez-0.5.5/waldiez/models/model/_llm.py +516 -0
  50. waldiez-0.5.5/waldiez/models/model/_price.py +30 -0
  51. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/model/model.py +45 -2
  52. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/model/model_data.py +2 -83
  53. waldiez-0.5.5/waldiez/models/tool/predefined/_duckduckgo.py +123 -0
  54. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/predefined/_google.py +31 -9
  55. waldiez-0.5.5/waldiez/models/tool/predefined/_perplexity.py +161 -0
  56. waldiez-0.5.5/waldiez/models/tool/predefined/_searxng.py +152 -0
  57. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/predefined/_tavily.py +46 -9
  58. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/predefined/_wikipedia.py +26 -6
  59. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/predefined/_youtube.py +36 -8
  60. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/predefined/registry.py +6 -0
  61. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/waldiez.py +12 -0
  62. waldiez-0.5.5/waldiez/runner.py +298 -0
  63. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/running/__init__.py +2 -4
  64. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/running/base_runner.py +100 -112
  65. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/running/environment.py +29 -4
  66. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/running/post_run.py +0 -1
  67. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/running/protocol.py +36 -48
  68. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/running/run_results.py +5 -5
  69. waldiez-0.5.5/waldiez/running/standard_runner.py +429 -0
  70. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/running/timeline_processor.py +0 -82
  71. waldiez-0.5.3/waldiez/exporting/chats/utils/common.py +0 -47
  72. waldiez-0.5.3/waldiez/exporting/core/extras/path_resolver.py +0 -93
  73. waldiez-0.5.3/waldiez/runner.py +0 -529
  74. waldiez-0.5.3/waldiez/running/import_runner.py +0 -437
  75. waldiez-0.5.3/waldiez/running/subprocess_runner.py +0 -104
  76. {waldiez-0.5.3 → waldiez-0.5.5}/.gitignore +0 -0
  77. {waldiez-0.5.3 → waldiez-0.5.5}/LICENSE +0 -0
  78. {waldiez-0.5.3 → waldiez-0.5.5}/NOTICE.md +0 -0
  79. {waldiez-0.5.3 → waldiez-0.5.5}/README.md +0 -0
  80. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/__init__.py +0 -0
  81. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/__main__.py +0 -0
  82. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/cli_extras/__init__.py +0 -0
  83. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/cli_extras/jupyter.py +0 -0
  84. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/cli_extras/runner.py +0 -0
  85. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/cli_extras/studio.py +0 -0
  86. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/__init__.py +0 -0
  87. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/__init__.py +0 -0
  88. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/code_execution.py +0 -0
  89. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/captain_agent_extras.py +0 -0
  90. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/group/target.py +0 -0
  91. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/group_manager_agent_extas.py +0 -0
  92. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/handoffs/__init__.py +0 -0
  93. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/handoffs/after_work.py +0 -0
  94. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/handoffs/available.py +0 -0
  95. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/handoffs/condition.py +0 -0
  96. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/handoffs/handoff.py +0 -0
  97. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/handoffs/target.py +0 -0
  98. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/rag/__init__.py +0 -0
  99. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/rag/chroma_extras.py +0 -0
  100. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/rag/mongo_extras.py +0 -0
  101. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/rag/pgvector_extras.py +0 -0
  102. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/rag/qdrant_extras.py +0 -0
  103. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/rag/vector_db_extras.py +0 -0
  104. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/rag_user_proxy_agent_extras.py +0 -0
  105. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/extras/reasoning_agent_extras.py +0 -0
  106. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/factory.py +0 -0
  107. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/system_message.py +0 -0
  108. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/agent/termination.py +0 -0
  109. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/chats/__init__.py +0 -0
  110. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/chats/exporter.py +0 -0
  111. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/chats/factory.py +0 -0
  112. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/chats/utils/__init__.py +0 -0
  113. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/__init__.py +0 -0
  114. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/constants.py +0 -0
  115. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/content.py +0 -0
  116. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/enums.py +0 -0
  117. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/errors.py +0 -0
  118. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/exporter.py +0 -0
  119. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/exporters.py +0 -0
  120. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/__init__.py +0 -0
  121. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/agent_extras/__init__.py +0 -0
  122. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/agent_extras/captain_extras.py +0 -0
  123. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/agent_extras/group_manager_extras.py +0 -0
  124. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/agent_extras/rag_user_extras.py +0 -0
  125. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/agent_extras/reasoning_extras.py +0 -0
  126. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/chat_extras.py +0 -0
  127. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/flow_extras.py +0 -0
  128. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/model_extras.py +0 -0
  129. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/extras/tool_extras.py +0 -0
  130. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/result.py +0 -0
  131. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/utils/__init__.py +0 -0
  132. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/utils/comment.py +0 -0
  133. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/utils/llm_config.py +0 -0
  134. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/core/validation.py +0 -0
  135. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/__init__.py +0 -0
  136. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/merger.py +0 -0
  137. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/utils/linting.py +0 -0
  138. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/flow/utils/logging.py +0 -0
  139. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/models/__init__.py +0 -0
  140. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/models/exporter.py +0 -0
  141. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/models/factory.py +0 -0
  142. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/models/processor.py +0 -0
  143. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/tools/__init__.py +0 -0
  144. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/tools/exporter.py +0 -0
  145. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/tools/factory.py +0 -0
  146. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/tools/processor.py +0 -0
  147. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/exporting/tools/registration.py +0 -0
  148. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/__init__.py +0 -0
  149. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/_ws.py +0 -0
  150. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/__init__.py +0 -0
  151. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/base.py +0 -0
  152. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/constants.py +0 -0
  153. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/content/__init__.py +0 -0
  154. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/content/audio.py +0 -0
  155. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/content/base.py +0 -0
  156. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/content/file.py +0 -0
  157. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/content/image.py +0 -0
  158. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/content/text.py +0 -0
  159. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/content/video.py +0 -0
  160. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/user_input.py +0 -0
  161. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/io/models/user_response.py +0 -0
  162. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/logger.py +0 -0
  163. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/agent/__init__.py +0 -0
  164. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/agent/agent_data.py +0 -0
  165. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/agent/code_execution.py +0 -0
  166. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/agent/human_input_mode.py +0 -0
  167. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/agent/linked_tool.py +0 -0
  168. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/agent/nested_chat.py +0 -0
  169. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/agent/termination_message.py +0 -0
  170. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/agent/update_system_message.py +0 -0
  171. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/assistant/__init__.py +0 -0
  172. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/assistant/assistant.py +0 -0
  173. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/assistant/assistant_data.py +0 -0
  174. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/captain/__init__.py +0 -0
  175. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/captain/captain_agent.py +0 -0
  176. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/captain/captain_agent_data.py +0 -0
  177. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/captain/captain_agent_lib_entry.py +0 -0
  178. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/extra_requirements.py +0 -0
  179. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/group_manager/__init__.py +0 -0
  180. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/group_manager/group_manager.py +0 -0
  181. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/group_manager/group_manager_data.py +0 -0
  182. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/group_manager/speakers.py +0 -0
  183. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/rag_user_proxy/__init__.py +0 -0
  184. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/rag_user_proxy/rag_user_proxy.py +0 -0
  185. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/rag_user_proxy/rag_user_proxy_data.py +0 -0
  186. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/rag_user_proxy/retrieve_config.py +0 -0
  187. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/rag_user_proxy/vector_db_config.py +0 -0
  188. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/reasoning/__init__.py +0 -0
  189. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/reasoning/reasoning_agent.py +0 -0
  190. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/reasoning/reasoning_agent_data.py +0 -0
  191. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/reasoning/reasoning_agent_reason_config.py +0 -0
  192. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/user_proxy/__init__.py +0 -0
  193. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/user_proxy/user_proxy.py +0 -0
  194. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/agents/user_proxy/user_proxy_data.py +0 -0
  195. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/chat/__init__.py +0 -0
  196. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/chat/chat.py +0 -0
  197. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/chat/chat_data.py +0 -0
  198. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/chat/chat_message.py +0 -0
  199. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/chat/chat_nested.py +0 -0
  200. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/chat/chat_summary.py +0 -0
  201. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/common/__init__.py +0 -0
  202. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/common/ag2_version.py +0 -0
  203. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/common/base.py +0 -0
  204. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/common/date_utils.py +0 -0
  205. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/common/dict_utils.py +0 -0
  206. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/common/handoff.py +0 -0
  207. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/common/id_generator.py +0 -0
  208. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/common/method_utils.py +0 -0
  209. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/common/naming.py +0 -0
  210. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/common/waldiez_version.py +0 -0
  211. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/flow/__init__.py +0 -0
  212. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/flow/connection.py +0 -0
  213. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/flow/flow_data.py +0 -0
  214. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/flow/info.py +0 -0
  215. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/flow/naming.py +0 -0
  216. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/model/extra_requirements.py +0 -0
  217. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/__init__.py +0 -0
  218. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/extra_requirements.py +0 -0
  219. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/predefined/__init__.py +0 -0
  220. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/predefined/_config.py +0 -0
  221. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/predefined/protocol.py +0 -0
  222. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/tool.py +0 -0
  223. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/tool_data.py +0 -0
  224. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/models/tool/tool_type.py +0 -0
  225. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/py.typed +0 -0
  226. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/running/gen_seq_diagram.py +0 -0
  227. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/running/patch_io_stream.py +0 -0
  228. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/running/pre_run.py +0 -0
  229. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/running/utils.py +0 -0
  230. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/utils/__init__.py +0 -0
  231. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/utils/conflict_checker.py +0 -0
  232. {waldiez-0.5.3 → waldiez-0.5.5}/waldiez/utils/version.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: waldiez
3
- Version: 0.5.3
3
+ Version: 0.5.5
4
4
  Dynamic: Keywords
5
5
  Summary: Make AG2 Agents Collaborate: Drag, Drop, and Orchestrate with Waldiez.
6
6
  Project-URL: Homepage, https://waldiez.io
@@ -27,7 +27,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
27
27
  Classifier: Topic :: Software Development :: Code Generators
28
28
  Classifier: Typing :: Typed
29
29
  Requires-Python: <3.14,>=3.10
30
- Requires-Dist: ag2[openai]==0.9.5
30
+ Requires-Dist: ag2[openai]==0.9.6
31
31
  Requires-Dist: aiocsv==1.3.2
32
32
  Requires-Dist: aiofiles==24.1.0
33
33
  Requires-Dist: aiosqlite==0.21.0
@@ -41,53 +41,53 @@ Requires-Dist: numpy<=2.3.1
41
41
  Requires-Dist: pandas>=2
42
42
  Requires-Dist: parso==0.8.4
43
43
  Requires-Dist: pillow
44
- Requires-Dist: pip>=25
44
+ Requires-Dist: pip>=25.1.1
45
+ Requires-Dist: platformdirs==4.3.8
45
46
  Requires-Dist: pydantic<3,>=2.10.2
46
47
  Requires-Dist: rpds-py==0.26.0
47
48
  Requires-Dist: typer<1,>=0.9.0
48
49
  Provides-Extra: ag2-extras
49
- Requires-Dist: ag2[anthropic]==0.9.5; extra == 'ag2-extras'
50
- Requires-Dist: ag2[bedrock]==0.9.5; extra == 'ag2-extras'
51
- Requires-Dist: ag2[cohere]==0.9.5; extra == 'ag2-extras'
52
- Requires-Dist: ag2[gemini]==0.9.5; (sys_platform != 'win32') and extra == 'ag2-extras'
53
- Requires-Dist: ag2[gemini]==0.9.5; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
54
- Requires-Dist: ag2[groq]==0.9.5; extra == 'ag2-extras'
55
- Requires-Dist: ag2[interop-crewai]==0.9.5; extra == 'ag2-extras'
56
- Requires-Dist: ag2[interop-langchain]==0.9.5; extra == 'ag2-extras'
57
- Requires-Dist: ag2[lmm]==0.9.5; extra == 'ag2-extras'
58
- Requires-Dist: ag2[mistral]==0.9.5; extra == 'ag2-extras'
59
- Requires-Dist: ag2[neo4j]==0.9.5; (sys_platform != 'win32') and extra == 'ag2-extras'
60
- Requires-Dist: ag2[neo4j]==0.9.5; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
61
- Requires-Dist: ag2[ollama]==0.9.5; extra == 'ag2-extras'
62
- Requires-Dist: ag2[redis]==0.9.5; extra == 'ag2-extras'
63
- Requires-Dist: ag2[together]==0.9.5; (sys_platform != 'win32') and extra == 'ag2-extras'
64
- Requires-Dist: ag2[together]==0.9.5; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
65
- Requires-Dist: ag2[websockets]==0.9.5; extra == 'ag2-extras'
66
- Requires-Dist: ag2[websurfer]==0.9.5; extra == 'ag2-extras'
67
- Requires-Dist: beautifulsoup4; extra == 'ag2-extras'
50
+ Requires-Dist: ag2[anthropic]==0.9.6; extra == 'ag2-extras'
51
+ Requires-Dist: ag2[autobuild]==0.9.6; extra == 'ag2-extras'
52
+ Requires-Dist: ag2[bedrock]==0.9.6; extra == 'ag2-extras'
53
+ Requires-Dist: ag2[blendsearch]==0.9.6; extra == 'ag2-extras'
54
+ Requires-Dist: ag2[cohere]==0.9.6; extra == 'ag2-extras'
55
+ Requires-Dist: ag2[commsagent-discord]==0.9.6; extra == 'ag2-extras'
56
+ Requires-Dist: ag2[commsagent-slack]==0.9.6; extra == 'ag2-extras'
57
+ Requires-Dist: ag2[commsagent-telegram]==0.9.6; extra == 'ag2-extras'
58
+ Requires-Dist: ag2[crawl4ai]==0.9.6; extra == 'ag2-extras'
59
+ Requires-Dist: ag2[duckduckgo]==0.9.6; extra == 'ag2-extras'
60
+ Requires-Dist: ag2[gemini-realtime]==0.9.6; (sys_platform != 'win32') and extra == 'ag2-extras'
61
+ Requires-Dist: ag2[gemini-realtime]==0.9.6; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
62
+ Requires-Dist: ag2[gemini]==0.9.6; (sys_platform != 'win32') and extra == 'ag2-extras'
63
+ Requires-Dist: ag2[gemini]==0.9.6; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
64
+ Requires-Dist: ag2[google-api]==0.9.6; extra == 'ag2-extras'
65
+ Requires-Dist: ag2[google-client]==0.9.6; extra == 'ag2-extras'
66
+ Requires-Dist: ag2[google-search]==0.9.6; extra == 'ag2-extras'
67
+ Requires-Dist: ag2[groq]==0.9.6; extra == 'ag2-extras'
68
+ Requires-Dist: ag2[interop-crewai]==0.9.6; extra == 'ag2-extras'
69
+ Requires-Dist: ag2[interop-langchain]==0.9.6; extra == 'ag2-extras'
70
+ Requires-Dist: ag2[lmm]==0.9.6; extra == 'ag2-extras'
71
+ Requires-Dist: ag2[mcp]==0.9.6; extra == 'ag2-extras'
72
+ Requires-Dist: ag2[mistral]==0.9.6; extra == 'ag2-extras'
73
+ Requires-Dist: ag2[neo4j]==0.9.6; (sys_platform != 'win32') and extra == 'ag2-extras'
74
+ Requires-Dist: ag2[neo4j]==0.9.6; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
75
+ Requires-Dist: ag2[ollama]==0.9.6; extra == 'ag2-extras'
76
+ Requires-Dist: ag2[openai-realtime]==0.9.6; extra == 'ag2-extras'
77
+ Requires-Dist: ag2[rag]==0.9.6; extra == 'ag2-extras'
78
+ Requires-Dist: ag2[redis]==0.9.6; extra == 'ag2-extras'
79
+ Requires-Dist: ag2[tavily]==0.9.6; extra == 'ag2-extras'
80
+ Requires-Dist: ag2[together]==0.9.6; (sys_platform != 'win32') and extra == 'ag2-extras'
81
+ Requires-Dist: ag2[together]==0.9.6; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
82
+ Requires-Dist: ag2[twilio]==0.9.6; extra == 'ag2-extras'
83
+ Requires-Dist: ag2[websockets]==0.9.6; extra == 'ag2-extras'
84
+ Requires-Dist: ag2[websurfer]==0.9.6; extra == 'ag2-extras'
85
+ Requires-Dist: ag2[wikipedia]==0.9.6; extra == 'ag2-extras'
68
86
  Requires-Dist: chromadb>=0.5.10; (sys_platform != 'win32') and extra == 'ag2-extras'
69
87
  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'
70
- Requires-Dist: crewai-tools<=0.46.0; (python_version < '3.13') and extra == 'ag2-extras'
71
- Requires-Dist: crewai<1,>=0.76; extra == 'ag2-extras'
72
- Requires-Dist: embedchain; (python_version < '3.13.3' and sys_platform != 'win32') and extra == 'ag2-extras'
73
- Requires-Dist: embedchain; (python_version < '3.13.3' and 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: google-api-python-client<3.0,>=2.163.0; extra == 'ag2-extras'
75
- Requires-Dist: google-auth-httplib2<0.3,>=0.2.0; extra == 'ag2-extras'
76
- Requires-Dist: google-auth-oauthlib<2.0,>=1.2.1; extra == 'ag2-extras'
77
- Requires-Dist: hf-xet<2.0.0,>=1.1.2; (platform_machine == 'x86_64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'ARM64' or platform_machine == 'aarch64') and extra == 'ag2-extras'
78
- Requires-Dist: huggingface-hub; extra == 'ag2-extras'
79
- Requires-Dist: ipython; extra == 'ag2-extras'
80
- Requires-Dist: langchain-community<1,>=0.3.12; extra == 'ag2-extras'
81
- Requires-Dist: litellm; extra == 'ag2-extras'
82
- Requires-Dist: markdownify; extra == 'ag2-extras'
83
- Requires-Dist: mcp<2,>=1.4.0; extra == 'ag2-extras'
84
- Requires-Dist: mistralai>=1.8.1; extra == 'ag2-extras'
85
- Requires-Dist: networkx; (python_version < '3.11') and extra == 'ag2-extras'
86
- Requires-Dist: networkx>=3.5; (python_version >= '3.11') and extra == 'ag2-extras'
87
- Requires-Dist: opentelemetry-api>=1.34.0; extra == 'ag2-extras'
88
- Requires-Dist: opentelemetry-sdk>=1.34.0; extra == 'ag2-extras'
88
+ Requires-Dist: couchbase>=4.3.0; extra == 'ag2-extras'
89
+ Requires-Dist: fastembed>=0.3.1; extra == 'ag2-extras'
89
90
  Requires-Dist: pgvector>=0.4.0; extra == 'ag2-extras'
90
- Requires-Dist: protobuf>=5.29.3; extra == 'ag2-extras'
91
91
  Requires-Dist: psycopg>=3.2.6; (sys_platform == 'linux') and extra == 'ag2-extras'
92
92
  Requires-Dist: psycopg>=3.2.6; (sys_platform == 'win32' and platform_machine == 'AARCH64') and extra == 'ag2-extras'
93
93
  Requires-Dist: psycopg>=3.2.6; (sys_platform == 'win32' and platform_machine == 'ARM64') and extra == 'ag2-extras'
@@ -96,15 +96,11 @@ Requires-Dist: psycopg>=3.2.6; (sys_platform == 'win32' and platform_machine ==
96
96
  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'
97
97
  Requires-Dist: pydantic-ai>=0.0.21; extra == 'ag2-extras'
98
98
  Requires-Dist: pymongo>=4.11; extra == 'ag2-extras'
99
- Requires-Dist: pypdf; extra == 'ag2-extras'
100
99
  Requires-Dist: qdrant-client[fastembed]; (sys_platform != 'win32') and extra == 'ag2-extras'
101
100
  Requires-Dist: qdrant-client[fastembed]; (sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
102
- Requires-Dist: sentence-transformers; (sys_platform == 'linux') and extra == 'ag2-extras'
103
- Requires-Dist: weaviate-client<5,>=4; extra == 'ag2-extras'
104
- Requires-Dist: wikipedia-api<1.0,>=0.8.1; extra == 'ag2-extras'
105
101
  Provides-Extra: dev
106
- Requires-Dist: ag2[redis]==0.9.5; extra == 'dev'
107
- Requires-Dist: ag2[websockets]==0.9.5; extra == 'dev'
102
+ Requires-Dist: ag2[redis]==0.9.6; extra == 'dev'
103
+ Requires-Dist: ag2[websockets]==0.9.6; extra == 'dev'
108
104
  Requires-Dist: autoflake==2.3.1; extra == 'dev'
109
105
  Requires-Dist: bandit==1.8.6; extra == 'dev'
110
106
  Requires-Dist: black[jupyter]==25.1.0; extra == 'dev'
@@ -113,11 +109,11 @@ Requires-Dist: fakeredis==2.30.1; extra == 'dev'
113
109
  Requires-Dist: fastjsonschema>=2.21.1; extra == 'dev'
114
110
  Requires-Dist: flake8==7.3.0; extra == 'dev'
115
111
  Requires-Dist: hatchling==1.27.0; extra == 'dev'
116
- Requires-Dist: jsonschema==4.24.0; extra == 'dev'
112
+ Requires-Dist: jsonschema==4.25.0; extra == 'dev'
117
113
  Requires-Dist: jupyter-server==2.16.0; extra == 'dev'
118
114
  Requires-Dist: jupyterlab<5.0,>=4.4.0; extra == 'dev'
119
115
  Requires-Dist: mypy-extensions>=1.1.0; extra == 'dev'
120
- Requires-Dist: mypy==1.16.1; extra == 'dev'
116
+ Requires-Dist: mypy==1.17.0; extra == 'dev'
121
117
  Requires-Dist: nbclient>=0.10.2; extra == 'dev'
122
118
  Requires-Dist: nbconvert>=7.16.6; extra == 'dev'
123
119
  Requires-Dist: nbformat>=5.10.4; extra == 'dev'
@@ -128,10 +124,11 @@ Requires-Dist: pandas-stubs==2.3.0.250703; extra == 'dev'
128
124
  Requires-Dist: pre-commit==4.2.0; extra == 'dev'
129
125
  Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
130
126
  Requires-Dist: pylint==3.3.7; extra == 'dev'
131
- Requires-Dist: python-dotenv>=1.1.0; extra == 'dev'
132
- Requires-Dist: ruff==0.12.2; extra == 'dev'
127
+ Requires-Dist: python-dotenv>=1.1.1; extra == 'dev'
128
+ Requires-Dist: ruff==0.12.4; extra == 'dev'
133
129
  Requires-Dist: toml==0.10.2; (python_version <= '3.10') and extra == 'dev'
134
- Requires-Dist: types-jsonschema==4.24.0.20250528; extra == 'dev'
130
+ Requires-Dist: types-aiofiles==24.1.0.20250708; extra == 'dev'
131
+ Requires-Dist: types-jsonschema==4.25.0.20250720; extra == 'dev'
135
132
  Requires-Dist: types-pyyaml==6.0.12.20250516; extra == 'dev'
136
133
  Requires-Dist: types-redis==4.6.0.20241004; extra == 'dev'
137
134
  Requires-Dist: types-requests==2.32.4.20250611; extra == 'dev'
@@ -152,26 +149,26 @@ Requires-Dist: mkdocs-open-in-new-tab==1.0.8; extra == 'docs'
152
149
  Requires-Dist: mkdocs==1.6.1; extra == 'docs'
153
150
  Requires-Dist: mkdocstrings-crystal==0.3.7; extra == 'docs'
154
151
  Requires-Dist: mkdocstrings-python==1.16.12; extra == 'docs'
155
- Requires-Dist: mkdocstrings[crystal,python]==0.29.1; extra == 'docs'
152
+ Requires-Dist: mkdocstrings[crystal,python]==0.30.0; extra == 'docs'
156
153
  Requires-Dist: natsort==8.4.0; extra == 'docs'
157
154
  Provides-Extra: jupyter
158
155
  Requires-Dist: jupyter-server==2.16.0; extra == 'jupyter'
159
156
  Requires-Dist: jupyterlab<5.0,>=4.3.0; extra == 'jupyter'
160
- Requires-Dist: waldiez-jupyter==0.5.3; extra == 'jupyter'
157
+ Requires-Dist: waldiez-jupyter==0.5.5; extra == 'jupyter'
161
158
  Provides-Extra: mqtt
162
159
  Requires-Dist: paho-mqtt<3.0,>=2.1.0; extra == 'mqtt'
163
160
  Provides-Extra: redis
164
- Requires-Dist: ag2[redis]==0.9.5; extra == 'redis'
161
+ Requires-Dist: ag2[redis]==0.9.6; extra == 'redis'
165
162
  Provides-Extra: runner
166
- Requires-Dist: waldiez-runner==0.5.3; (python_version >= '3.11') and extra == 'runner'
163
+ Requires-Dist: waldiez-runner==0.5.5; (python_version >= '3.11') and extra == 'runner'
167
164
  Provides-Extra: studio
168
- Requires-Dist: waldiez-studio==0.5.3; extra == 'studio'
165
+ Requires-Dist: waldiez-studio==0.5.5; extra == 'studio'
169
166
  Provides-Extra: test
170
- Requires-Dist: ag2[redis]==0.9.5; extra == 'test'
171
- Requires-Dist: ag2[websockets]==0.9.5; extra == 'test'
167
+ Requires-Dist: ag2[redis]==0.9.6; extra == 'test'
168
+ Requires-Dist: ag2[websockets]==0.9.6; extra == 'test'
172
169
  Requires-Dist: fakeredis==2.30.1; extra == 'test'
173
170
  Requires-Dist: paho-mqtt<3.0,>=2.1.0; extra == 'test'
174
- Requires-Dist: pytest-asyncio==1.0.0; extra == 'test'
171
+ Requires-Dist: pytest-asyncio==1.1.0; extra == 'test'
175
172
  Requires-Dist: pytest-cov==6.2.1; extra == 'test'
176
173
  Requires-Dist: pytest-env==1.1.5; extra == 'test'
177
174
  Requires-Dist: pytest-html==4.1.1; extra == 'test'
@@ -180,7 +177,7 @@ Requires-Dist: pytest-timeout==2.4.0; extra == 'test'
180
177
  Requires-Dist: pytest-xdist==3.8.0; extra == 'test'
181
178
  Requires-Dist: pytest==8.4.1; extra == 'test'
182
179
  Provides-Extra: websockets
183
- Requires-Dist: ag2[websockets]==0.9.5; extra == 'websockets'
180
+ Requires-Dist: ag2[websockets]==0.9.6; extra == 'websockets'
184
181
  Description-Content-Type: text/markdown
185
182
 
186
183
  # Waldiez
@@ -27,7 +27,7 @@ classifiers = [
27
27
  ]
28
28
 
29
29
  dependencies = [
30
- "ag2[openai]==0.9.5",
30
+ "ag2[openai]==0.9.6",
31
31
  "aiocsv==1.3.2",
32
32
  "aiofiles==24.1.0",
33
33
  "aiosqlite==0.21.0",
@@ -37,12 +37,13 @@ dependencies = [
37
37
  "graphviz<=0.21",
38
38
  "pandas>=2",
39
39
  "pillow",
40
- "pip>=25",
40
+ "pip>=25.1.1",
41
41
  "jupytext",
42
42
  "parso==0.8.4",
43
43
  "rpds-py==0.26.0",
44
+ "platformdirs==4.3.8",
44
45
  "pydantic>=2.10.2,<3",
45
- # ag2[together]==0.9.5 depends on together>=1.2
46
+ # ag2[together]==0.9.6 depends on together>=1.2
46
47
  # together>=1.2 depends on typer>=0.9,<0.16
47
48
  "typer>=0.9.0,<1",
48
49
  "click<8.2",
@@ -71,59 +72,73 @@ waldiez = "waldiez.cli:app"
71
72
  [project.optional-dependencies]
72
73
  # stream-io-extensions: redis, websockets, mqtt
73
74
  redis = [
74
- "ag2[redis]==0.9.5",
75
+ "ag2[redis]==0.9.6",
75
76
  ]
76
77
  websockets = [
77
- "ag2[websockets]==0.9.5",
78
+ "ag2[websockets]==0.9.6",
78
79
  ]
79
80
  mqtt = [
80
81
  "paho-mqtt>=2.1.0,<3.0",
81
82
  ]
82
83
  # jupyterlab extension
83
84
  jupyter = [
84
- "waldiez_jupyter==0.5.3",
85
+ "waldiez_jupyter==0.5.5",
85
86
  "jupyterlab>=4.3.0,<5.0",
86
87
  "jupyter_server==2.16.0",
87
88
  ]
88
89
  runner = [
89
- "waldiez_runner==0.5.3; python_version >= '3.11'",
90
+ "waldiez_runner==0.5.5; python_version >= '3.11'",
90
91
  ]
91
92
  studio = [
92
- "waldiez_studio==0.5.3",
93
+ "waldiez_studio==0.5.5",
93
94
  ]
94
95
  ag2_extras = [
95
- # models
96
- "ag2[anthropic]==0.9.5",
97
- "ag2[bedrock]==0.9.5",
98
- "ag2[cohere]==0.9.5",
99
- "ag2[gemini]==0.9.5;sys_platform != 'win32'",
100
- "ag2[gemini]==0.9.5;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
101
- "ag2[groq]==0.9.5",
102
- "ag2[lmm]==0.9.5",
103
- "ag2[mistral]==0.9.5",
104
- "mistralai>=1.8.1",
105
- "ag2[ollama]==0.9.5",
106
- "ag2[together]==0.9.5;sys_platform != 'win32'",
107
- "ag2[together]==0.9.5;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'", # utils
108
- # utils
109
- "ag2[websockets]==0.9.5",
110
- "ag2[redis]==0.9.5",
111
- "crewai>=0.76,<1",
112
- "opentelemetry-api>=1.34.0",
113
- "opentelemetry-sdk>=1.34.0",
114
- # "crewai-tools>=0.47.0;python_version >= '3.13'",
115
- "crewai-tools<=0.46.0;python_version<'3.13'",
116
- "ag2[interop-crewai]==0.9.5",
117
- # "crewai>=0.126.0;python_version >= '3.13'",
118
- # "crewai-tools>=0.47.0;python_version >= '3.13'",
119
- "litellm",
120
- "weaviate-client>=4,<5",
121
- "ag2[interop-langchain]==0.9.5",
122
- "ag2[neo4j]==0.9.5;sys_platform != 'win32'",
123
- "ag2[neo4j]==0.9.5;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
124
- "ag2[websurfer]==0.9.5",
96
+ "ag2[anthropic]==0.9.6",
97
+ "ag2[autobuild]==0.9.6",
98
+ "ag2[bedrock]==0.9.6",
99
+ "ag2[blendsearch]==0.9.6",
100
+ "ag2[cohere]==0.9.6",
101
+ "ag2[commsagent-discord]==0.9.6",
102
+ "ag2[commsagent-slack]==0.9.6",
103
+ "ag2[commsagent-telegram]==0.9.6",
104
+ "ag2[crawl4ai]==0.9.6",
105
+ "ag2[duckduckgo]==0.9.6",
106
+ "ag2[gemini-realtime]==0.9.6;sys_platform != 'win32'",
107
+ "ag2[gemini-realtime]==0.9.6;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
108
+ "ag2[gemini]==0.9.6;sys_platform != 'win32'",
109
+ "ag2[gemini]==0.9.6;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
110
+ "ag2[google-api]==0.9.6",
111
+ "ag2[google-client]==0.9.6",
112
+ "ag2[google-search]==0.9.6",
113
+ "ag2[groq]==0.9.6",
114
+ "ag2[interop-crewai]==0.9.6",
115
+ "ag2[interop-langchain]==0.9.6",
116
+ "ag2[lmm]==0.9.6",
117
+ # "ag2[mcp-proxy-gen]==0.9.6",
118
+ "ag2[mcp]==0.9.6",
119
+ "ag2[mistral]==0.9.6",
120
+ "ag2[neo4j]==0.9.6;sys_platform != 'win32'",
121
+ "ag2[neo4j]==0.9.6;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
122
+ "ag2[ollama]==0.9.6",
123
+ "ag2[openai-realtime]==0.9.6",
124
+ "ag2[rag]==0.9.6",
125
+ "ag2[redis]==0.9.6",
126
+ "ag2[tavily]==0.9.6",
127
+ # "ag2[retrievechat-couchbase]==0.9.6",
128
+ # "ag2[retrievechat-mongodb]==0.9.6",
129
+ # "ag2[retrievechat-pgvector]==0.9.6",
130
+ # "ag2[retrievechat-qdrant]==0.9.6",
131
+ # "ag2[retrievechat]==0.9.6",
132
+ "ag2[together]==0.9.6;sys_platform != 'win32'",
133
+ "ag2[together]==0.9.6;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
134
+ "ag2[twilio]==0.9.6",
135
+ "ag2[websockets]==0.9.6",
136
+ "ag2[websurfer]==0.9.6",
137
+ "ag2[wikipedia]==0.9.6",
125
138
  "chromadb>=0.5.10;sys_platform != 'win32'",
126
139
  "chromadb>=0.5.10;sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
140
+ "couchbase>=4.3.0",
141
+ "fastembed>=0.3.1",
127
142
  "pgvector>=0.4.0",
128
143
  "psycopg>=3.2.6;sys_platform == 'linux'",
129
144
  "psycopg>=3.2.6;sys_platform == 'win32' and platform_machine == 'arm64'",
@@ -131,41 +146,21 @@ ag2_extras = [
131
146
  "psycopg>=3.2.6;sys_platform == 'win32' and platform_machine == 'aarch64'",
132
147
  "psycopg>=3.2.6;sys_platform == 'win32' and platform_machine == 'AARCH64'",
133
148
  "psycopg[binary]>=3.2.6;sys_platform != 'linux' and platform_machine != 'arm64' and platform_machine != 'ARM64' and platform_machine != 'aarch64' and platform_machine != 'AARCH64'",
134
- "protobuf>=5.29.3",
135
- "pypdf",
136
- "ipython",
137
- "markdownify",
138
- "beautifulsoup4",
139
- "sentence-transformers;sys_platform == 'linux'",
140
- "huggingface-hub",
141
- "hf-xet>=1.1.2,<2.0.0;platform_machine == 'x86_64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'ARM64' or platform_machine == 'aarch64'",
142
- "qdrant-client[fastembed];sys_platform != 'win32'",
143
- "qdrant-client[fastembed];sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
144
- "pgvector>=0.4.0",
145
- # for some reason embedchain has: python = ">=3.9,<=3.13.2"
146
- "embedchain;python_version < '3.13.3' and sys_platform != 'win32'",
147
- "embedchain;python_version < '3.13.3' and sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
148
- "langchain-community>=0.3.12,<1",
149
- "networkx>=3.5;python_version >= '3.11'",
150
- "networkx;python_version < '3.11'",
151
149
  "pydantic-ai>=0.0.21",
152
150
  "pymongo>=4.11",
153
- "wikipedia-api>=0.8.1, <1.0",
154
- "google-api-python-client>=2.163.0,<3.0",
155
- "google-auth-httplib2>=0.2.0,<0.3",
156
- "google-auth-oauthlib>=1.2.1,<2.0",
157
- "mcp>=1.4.0,<2",
151
+ "qdrant-client[fastembed];sys_platform != 'win32'",
152
+ "qdrant-client[fastembed];sys_platform == 'win32' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
158
153
  ]
159
154
  dev = [
160
- "ag2[redis]==0.9.5",
161
- "ag2[websockets]==0.9.5",
155
+ "ag2[redis]==0.9.6",
156
+ "ag2[websockets]==0.9.6",
162
157
  "autoflake==2.3.1",
163
158
  "bandit==1.8.6",
164
159
  "black[jupyter]==25.1.0",
165
160
  "build==1.2.2.post1",
166
161
  "fakeredis==2.30.1",
167
162
  "flake8==7.3.0",
168
- "jsonschema==4.24.0",
163
+ "jsonschema==4.25.0",
169
164
  "jupyterlab>=4.4.0,<5.0",
170
165
  "notebook-shim>=0.2.4",
171
166
  "fastjsonschema>=2.21.1",
@@ -174,7 +169,7 @@ dev = [
174
169
  "nbformat>=5.10.4",
175
170
  "nbconvert>=7.16.6",
176
171
  "hatchling==1.27.0",
177
- "mypy==1.16.1",
172
+ "mypy==1.17.0",
178
173
  "mypy-extensions>=1.1.0",
179
174
  "paho-mqtt>=2.1.0,<3.0",
180
175
  "pandas-stubs==2.3.0.250703",
@@ -182,10 +177,11 @@ dev = [
182
177
  "pre-commit==4.2.0",
183
178
  "pydocstyle==6.3.0",
184
179
  "pylint==3.3.7",
185
- "python-dotenv>=1.1.0",
186
- "ruff==0.12.2",
180
+ "python-dotenv>=1.1.1",
181
+ "ruff==0.12.4",
182
+ "types-aiofiles==24.1.0.20250708",
187
183
  "types-pyyaml==6.0.12.20250516",
188
- "types-jsonschema==4.24.0.20250528",
184
+ "types-jsonschema==4.25.0.20250720",
189
185
  "types-redis==4.6.0.20241004",
190
186
  "types-toml==0.10.8.20240310",
191
187
  "types-requests==2.32.4.20250611",
@@ -203,7 +199,7 @@ docs = [
203
199
  "mkdocs-macros-plugin==1.3.7",
204
200
  "mkdocs-material==9.6.15",
205
201
  "mkdocs-minify-html-plugin==0.3.1",
206
- "mkdocstrings[crystal,python]==0.29.1",
202
+ "mkdocstrings[crystal,python]==0.30.0",
207
203
  "mkdocstrings-crystal==0.3.7",
208
204
  "mkdocstrings-python==1.16.12",
209
205
  "mkdocs-awesome-nav==3.1.2",
@@ -211,12 +207,12 @@ docs = [
211
207
  "natsort==8.4.0",
212
208
  ]
213
209
  test = [
214
- "ag2[redis]==0.9.5",
215
- "ag2[websockets]==0.9.5",
210
+ "ag2[redis]==0.9.6",
211
+ "ag2[websockets]==0.9.6",
216
212
  "fakeredis==2.30.1",
217
213
  "paho-mqtt>=2.1.0,<3.0",
218
214
  "pytest==8.4.1",
219
- "pytest-asyncio==1.0.0",
215
+ "pytest-asyncio==1.1.0",
220
216
  "pytest-cov==6.2.1",
221
217
  "pytest-env==1.1.5",
222
218
  "pytest-html==4.1.1",
@@ -261,7 +257,9 @@ fields = ["description", "authors", "urls"]
261
257
  [tool.hatch.envs.default]
262
258
  python = "3.13"
263
259
  installer = "uv"
264
- dependencies = ["waldiez[dev,test,docs,ag2_extras,redis,websockets,mqtt]"]
260
+ post-install-commands = [
261
+ "pip install .[dev,test,docs,ag2_extras,redis,websockets,mqtt]"
262
+ ]
265
263
 
266
264
  [tool.hatch.envs.default.scripts]
267
265
  pre-test = [
@@ -288,6 +286,10 @@ test-running = [
288
286
  "hatch run pre-test",
289
287
  "pytest -c pyproject.toml --cov=waldiez/running --cov-branch --cov-report=term-missing:skip-covered -vv tests/running"
290
288
  ]
289
+ test-storage = [
290
+ "hatch run pre-test",
291
+ "pytest -c pyproject.toml --cov=waldiez/storage --cov-branch --cov-report=term-missing:skip-covered -vv tests/storage"
292
+ ]
291
293
 
292
294
  format-black = "black --config pyproject.toml waldiez tests scripts"
293
295
  format-autoflake = "autoflake --in-place --remove-all-unused-imports --remove-unused-variables --recursive waldiez tests scripts"
@@ -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.5.3"
8
+ __version__ = VERSION = "0.5.5"
@@ -92,21 +92,8 @@ def run(
92
92
  structured: bool = typer.Option( # noqa: B008
93
93
  False,
94
94
  help=(
95
- "If set, running the flow will use structured io stream instead of the default 'input/print' "
96
- ),
97
- ),
98
- threaded: bool = typer.Option( # noqa: B008
99
- False,
100
- help=(
101
- "If set, the flow will be run in a separate thread. "
102
- "This is useful for running flows that require user input or print output."
103
- ),
104
- ),
105
- patch_io: bool = typer.Option( # noqa: B008
106
- False,
107
- help=(
108
- "If set, the flow will patch ag2's IOStream to safe print and input methods. "
109
- "This is useful for running flows that require user input or print output."
95
+ "If set, the output will be structured as a directory with "
96
+ "the flow file and any additional generated files in it."
110
97
  ),
111
98
  ),
112
99
  force: bool = typer.Option( # noqa: B008
@@ -136,8 +123,7 @@ def run(
136
123
  runner.a_run,
137
124
  output_path,
138
125
  uploads_root,
139
- structured,
140
- not patch_io, # skip_patch_io
126
+ structured, # structured_io
141
127
  False, # skip_mmd
142
128
  False, # skip_timeline
143
129
  )
@@ -146,8 +132,6 @@ def run(
146
132
  output_path=output_path,
147
133
  uploads_root=uploads_root,
148
134
  structured_io=structured,
149
- threaded=threaded,
150
- skip_patch_io=not patch_io,
151
135
  skip_mmd=False,
152
136
  skip_timeline=False,
153
137
  )
@@ -184,13 +168,6 @@ def convert(
184
168
  False,
185
169
  help="Override the output file if it already exists.",
186
170
  ),
187
- patch_io: bool = typer.Option( # noqa: B008
188
- False,
189
- help=(
190
- "If set, the exported script will inlclude a snippet to patch ag2's IOStream "
191
- "to safe print and input methods. "
192
- ),
193
- ),
194
171
  debug: bool = typer.Option(
195
172
  False,
196
173
  "--debug",
@@ -219,7 +196,6 @@ def convert(
219
196
  exporter.export(
220
197
  output,
221
198
  force=force,
222
- skip_patch_io=not patch_io,
223
199
  debug=debug,
224
200
  )
225
201
  generated = str(output).replace(os.getcwd(), ".")
@@ -63,7 +63,6 @@ class WaldiezExporter:
63
63
  path: str | Path,
64
64
  structured_io: bool = False,
65
65
  uploads_root: Path | None = None,
66
- skip_patch_io: bool = True,
67
66
  force: bool = False,
68
67
  debug: bool = False,
69
68
  ) -> None:
@@ -78,8 +77,6 @@ class WaldiezExporter:
78
77
  by default False.
79
78
  uploads_root : str | Path | None, (optional)
80
79
  The uploads root, to get user-uploaded files, by default None.
81
- skip_patch_io : bool, (optional)
82
- Whether to skip patching I/O, by default True.
83
80
  force : bool, (optional)
84
81
  Override the output file if it already exists, by default False.
85
82
  debug : bool, (optional)
@@ -112,7 +109,6 @@ class WaldiezExporter:
112
109
  path,
113
110
  structured_io=structured_io,
114
111
  uploads_root=uploads_root,
115
- skip_patch_io=skip_patch_io,
116
112
  debug=debug,
117
113
  )
118
114
  elif extension == ".ipynb":
@@ -120,7 +116,6 @@ class WaldiezExporter:
120
116
  path,
121
117
  structured_io=structured_io,
122
118
  uploads_root=uploads_root,
123
- skip_patch_io=skip_patch_io,
124
119
  debug=debug,
125
120
  )
126
121
  else:
@@ -131,7 +126,6 @@ class WaldiezExporter:
131
126
  path: str | Path,
132
127
  structured_io: bool = False,
133
128
  uploads_root: Path | None = None,
134
- skip_patch_io: bool = True,
135
129
  debug: bool = False,
136
130
  ) -> None:
137
131
  """Export flow to jupyter notebook.
@@ -145,8 +139,6 @@ class WaldiezExporter:
145
139
  by default False.
146
140
  uploads_root : Path | None, optional
147
141
  The uploads root, to get user-uploaded files, by default None.
148
- skip_patch_io : bool, optional
149
- Whether to skip patching I/O, by default True.
150
142
  debug : bool, optional
151
143
  Whether to enable debug mode, by default False.
152
144
 
@@ -165,7 +157,6 @@ class WaldiezExporter:
165
157
  uploads_root=uploads_root,
166
158
  structured_io=structured_io,
167
159
  for_notebook=True,
168
- skip_patch_io=skip_patch_io,
169
160
  debug=debug,
170
161
  )
171
162
  self.flow_extras = exporter.extras
@@ -202,7 +193,6 @@ class WaldiezExporter:
202
193
  path: str | Path,
203
194
  structured_io: bool = False,
204
195
  uploads_root: Path | None = None,
205
- skip_patch_io: bool = True,
206
196
  debug: bool = False,
207
197
  ) -> None:
208
198
  """Export waldiez flow to a python script.
@@ -216,8 +206,6 @@ class WaldiezExporter:
216
206
  by default False.
217
207
  uploads_root : Path | None, optional
218
208
  The uploads root, to get user-uploaded files, by default None.
219
- skip_patch_io : bool, optional
220
- Whether to skip patching I/O, by default True.
221
209
  debug : bool, optional
222
210
  Whether to enable debug mode, by default False.
223
211
 
@@ -234,7 +222,6 @@ class WaldiezExporter:
234
222
  for_notebook=False,
235
223
  uploads_root=uploads_root,
236
224
  structured_io=structured_io,
237
- skip_patch_io=skip_patch_io,
238
225
  debug=debug,
239
226
  )
240
227
  self.flow_extras = exporter.extras