waldiez 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of waldiez might be problematic. Click here for more details.

waldiez/_version.py CHANGED
@@ -2,4 +2,4 @@
2
2
  # Copyright (c) 2024 - 2025 Waldiez and contributors.
3
3
  """Version information for Waldiez."""
4
4
 
5
- __version__ = "0.3.0"
5
+ __version__ = "0.3.2"
@@ -15,11 +15,14 @@ BUILTIN_IMPORTS = [
15
15
  "from pprint import pprint",
16
16
  ]
17
17
  TYPING_IMPORTS = [
18
+ "Annotated",
18
19
  "Any",
19
20
  "Callable",
20
21
  "Dict",
21
22
  "List",
23
+ "Literal",
22
24
  "Optional",
25
+ "Set",
23
26
  "Tuple",
24
27
  "Union",
25
28
  ]
@@ -237,11 +237,19 @@ class WaldiezFlow(WaldiezBase):
237
237
  ):
238
238
  return [(chat, source, target)]
239
239
  if source.agent_type == "swarm" and target.agent_type == "swarm":
240
- if isinstance(source, WaldiezSwarmAgent) and source.is_initial:
241
- return [(chat, source, target)]
242
240
  valid_chats.append(chat)
243
241
  if not valid_chats:
244
242
  return []
243
+ for valid_chat in valid_chats:
244
+ source = self.get_agent_by_id(valid_chat.source)
245
+ if isinstance(source, WaldiezSwarmAgent) and source.is_initial:
246
+ return [
247
+ (
248
+ valid_chat,
249
+ source,
250
+ self.get_agent_by_id(valid_chat.target),
251
+ )
252
+ ]
245
253
  first_chat: Optional[WaldiezChat] = None
246
254
  # first check the order
247
255
  by_order = sorted(
@@ -141,11 +141,13 @@ class WaldiezModel(WaldiezBase):
141
141
  - nim: 'NIM_API_KEY',
142
142
  - other: 'OPENAI_API_KEY'
143
143
  """
144
- if self.data.api_key:
144
+ if self.data.api_key and self.data.api_key != "REPLACE_ME":
145
145
  return self.data.api_key
146
146
  env_key = self.api_key_env_key
147
- api_key = os.environ.get(env_key, "")
148
- return api_key
147
+ api_key = os.environ.get(
148
+ env_key, getattr(self.data, "api_key", "REPLACE_ME")
149
+ )
150
+ return api_key or "REPLACE_ME"
149
151
 
150
152
  @property
151
153
  def price(self) -> Optional[List[float]]:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: waldiez
3
- Version: 0.3.0
3
+ Version: 0.3.2
4
4
  Summary: waldiez
5
5
  Project-URL: homepage, https://waldiez.github.io/waldiez/python
6
6
  Project-URL: repository, https://github.com/waldiez/python.git
@@ -18,11 +18,12 @@ Classifier: Programming Language :: Python :: 3.11
18
18
  Classifier: Programming Language :: Python :: 3.12
19
19
  Requires-Python: <3.13,>=3.10
20
20
  Requires-Dist: asyncer==0.0.8
21
+ Requires-Dist: httpx<0.28.0,>=0.27.0
21
22
  Requires-Dist: jupytext
22
23
  Requires-Dist: pandas>=2
23
24
  Requires-Dist: parso==0.8.4
24
25
  Requires-Dist: pyautogen==0.7.1
25
- Requires-Dist: pydantic>=2.0
26
+ Requires-Dist: pydantic<3,>=2.6.1
26
27
  Requires-Dist: typer<0.13,>=0.9
27
28
  Provides-Extra: ag2-extras
28
29
  Requires-Dist: autogen[captainagent]==0.7.1; (platform_system == 'Linux') and extra == 'ag2-extras'
@@ -51,7 +52,7 @@ Requires-Dist: flake8==7.1.1; extra == 'dev'
51
52
  Requires-Dist: isort==5.13.2; extra == 'dev'
52
53
  Requires-Dist: mypy==1.14.1; extra == 'dev'
53
54
  Requires-Dist: pandas-stubs; extra == 'dev'
54
- Requires-Dist: pre-commit==4.0.1; extra == 'dev'
55
+ Requires-Dist: pre-commit==4.1.0; extra == 'dev'
55
56
  Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
56
57
  Requires-Dist: pylint==3.3.3; extra == 'dev'
57
58
  Requires-Dist: python-dotenv==1.0.1; extra == 'dev'
@@ -72,9 +73,9 @@ Requires-Dist: mkdocstrings-python==1.13.0; extra == 'docs'
72
73
  Requires-Dist: mkdocstrings[crystal,python]==0.27.0; extra == 'docs'
73
74
  Provides-Extra: jupyter
74
75
  Requires-Dist: jupyterlab>=4.3.0; extra == 'jupyter'
75
- Requires-Dist: waldiez-jupyter==0.3.0; extra == 'jupyter'
76
+ Requires-Dist: waldiez-jupyter==0.3.1; extra == 'jupyter'
76
77
  Provides-Extra: studio
77
- Requires-Dist: waldiez-studio==0.3.0; extra == 'studio'
78
+ Requires-Dist: waldiez-studio==0.3.1; extra == 'studio'
78
79
  Provides-Extra: test
79
80
  Requires-Dist: pytest-asyncio==0.25.2; extra == 'test'
80
81
  Requires-Dist: pytest-cov==6.0.0; extra == 'test'
@@ -1,6 +1,6 @@
1
1
  waldiez/__init__.py,sha256=WrnR3f-FQ0mK2orSfyApwRs9UGfq1nYiohQTyLrX_hs,2149
2
2
  waldiez/__main__.py,sha256=0dYzNrQbovRwQQvmZC6_1FDR1m71SUIOkTleO5tBnBw,203
3
- waldiez/_version.py,sha256=hYl_P0IvGsJ8GonDWm3yBqvX4ADcfHQKDot-6Tl9sPc,155
3
+ waldiez/_version.py,sha256=cK77Z-sXaRYFTxQXqYv2zorH0l3iirkzDAVkJ1XdW8o,155
4
4
  waldiez/cli.py,sha256=SUdYLwNfG2TRnZr_E2gZ6SGczSQMIUdeIDzSZEYaw7o,6947
5
5
  waldiez/cli_extras.py,sha256=cnMJnznNgTjtI_qKAlgLOzXaJSuzx7D_XNDzC1faKzw,3018
6
6
  waldiez/conflict_checker.py,sha256=oF7PT2bdMGeIoj13cE2F6hn5dxhMhOutq34z8hCkZYg,976
@@ -52,7 +52,7 @@ waldiez/exporting/flow/utils/chat_utils.py,sha256=q6HS4JTNQJe15eh78H1BlN-wId9Xzv
52
52
  waldiez/exporting/flow/utils/def_main.py,sha256=k9yxlc-Ct84PYpYNvzKjL16vZirC6N7Gq7vlOdG-NHw,2068
53
53
  waldiez/exporting/flow/utils/flow_content.py,sha256=F5x22HVPEzpGRV7kRQrY_oE8nvmqOud5Qih963Z4zyA,2997
54
54
  waldiez/exporting/flow/utils/flow_names.py,sha256=5i1qjSrkbIFtnE13sv_GjOTyRnmTzPMys1Ecd5au7mg,3463
55
- waldiez/exporting/flow/utils/importing_utils.py,sha256=SidCqVVafdfkvPbDJR2CaoRklPaKHNd__sEwSvUXATk,5187
55
+ waldiez/exporting/flow/utils/importing_utils.py,sha256=ZcmupLDeN8KqH0id_Hw5iPpyERkm-UYG-5KxnUiC4i0,5230
56
56
  waldiez/exporting/flow/utils/logging_utils.py,sha256=QREMpEDiwo-V1UCWJ6t_J09AYOyubTcwBYGRIcg2dWw,6169
57
57
  waldiez/exporting/models/__init__.py,sha256=QbyYuS-Px-M9vMrdL-UKqnDzhYWRkNmRGBbc1IBuwpM,212
58
58
  waldiez/exporting/models/models_exporter.py,sha256=W4EDMY79RNSqZKBd2wYGFRQnt-OvbixXMXiAeTjNdU0,5994
@@ -105,11 +105,11 @@ waldiez/models/common/__init__.py,sha256=OpQTQ-wNA56hZTKZfBSj4joKRzX5n2uNGhL1jKo
105
105
  waldiez/models/common/base.py,sha256=SwnnuFTJ0C3PfBoRilqE0uZv3j5aelw7-s-79KcxJNA,1840
106
106
  waldiez/models/common/method_utils.py,sha256=jT6Cpu_XskroLlMwLGCY4HNptQtnccvMOa2CwRwT6xc,6944
107
107
  waldiez/models/flow/__init__.py,sha256=3UdwoUZdkLr6G78tNTyaijoiwvmHZ-N0tY3G7S_xbRs,255
108
- waldiez/models/flow/flow.py,sha256=AIguDr9pd6N9csyKnnkHyJ0-g_4AcAT4TLcOOXTLDiU,14436
108
+ waldiez/models/flow/flow.py,sha256=kM8K7dl8wZIP0GEryELC2yZ0t0blEFFkNOvx7VEMW4o,14700
109
109
  waldiez/models/flow/flow_data.py,sha256=5Mn27AMzd44rrcg2Toa6wZC4cX5oCOsCUhuoE0giQsE,4949
110
110
  waldiez/models/flow/utils.py,sha256=Bd8dWYlCenuU4VeusAL_qxkXzYSuxl9e3a7FhHs6UCY,5435
111
111
  waldiez/models/model/__init__.py,sha256=1cOyDThhZV8o7pZN1DaIhh12jWty_YMegoQ4Jg7uKyI,383
112
- waldiez/models/model/model.py,sha256=PJCV0OJMDCCoy20voGZj2pFVZjjJw86ozKiMRsX6oSY,6736
112
+ waldiez/models/model/model.py,sha256=fRPJxvlINYbKzzDr5oWe9EKIU60w3YDMcdY9MWpFFOg,6853
113
113
  waldiez/models/model/model_data.py,sha256=XrMBbnCRkJBlk46xJMoBJrcTX9qnViJZTn9_MS9ujPw,3792
114
114
  waldiez/models/skill/__init__.py,sha256=kAyp9-LwBacI3ndQpU-dCGWjsjojRQy73zXsTddGAmg,306
115
115
  waldiez/models/skill/skill.py,sha256=6r8heHUgRYwfcBJbdsV5DywvTXcEeWPuJN4jzCh3-O4,4255
@@ -118,8 +118,8 @@ waldiez/running/__init__.py,sha256=hptO_sDFv0Alg-YJmLN-A49SeQ4YyWUMP8nBH21qO-o,6
118
118
  waldiez/running/environment.py,sha256=8IgEmm1MJrfwOlxyMAdoYtH-KRmLtlVROOWzlMtSbps,2268
119
119
  waldiez/running/gen_seq_diagram.py,sha256=y9O5HFe25YvfNc78MrWTwaSfFQEDu6-buq7lja8PYgg,5639
120
120
  waldiez/running/running.py,sha256=CpgiK9E-4T7ruVwB2baSv6qTipgTcdR9IulE8db5V4o,8561
121
- waldiez-0.3.0.dist-info/METADATA,sha256=MXqBcs79PwxYwc3nwQnvYT-tAM8b_n-fKvKXjVtTDSE,8984
122
- waldiez-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
123
- waldiez-0.3.0.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
124
- waldiez-0.3.0.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
125
- waldiez-0.3.0.dist-info/RECORD,,
121
+ waldiez-0.3.2.dist-info/METADATA,sha256=93egHoY93Jbsi1ja39S2rFzroUWZ_r2CSnB3CEYZh9s,9026
122
+ waldiez-0.3.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
123
+ waldiez-0.3.2.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
124
+ waldiez-0.3.2.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
125
+ waldiez-0.3.2.dist-info/RECORD,,