waldiez 0.3.1__py3-none-any.whl → 0.3.3__py3-none-any.whl

waldiez/_version.py

@@ -2,4 +2,4 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Version information for Waldiez."""
 
-__version__ = "0.3.1"
+__version__ = "0.3.3"

waldiez/exporting/agent/agent_exporter.py

@@ -56,10 +56,8 @@ class AgentExporter(BaseExporter, ExporterMixin):
             The model ids to names mapping.
         skill_names : Dict[str, str]
             The skill ids to names mapping.
-        all_chats : List[WaldiezChat]
-            All the chats in the flow.
-        chat_names : Dict[str, str]
-            The chat ids to names mapping.
+        chats : Tuple[List[WaldiezChat], Dict[str, str]]
+            All the chats and the chat ids to names mapping.
         is_async : bool
             Whether the whole flow is async.
         for_notebook : bool

waldiez/models/flow/flow.py

@@ -237,11 +237,19 @@ class WaldiezFlow(WaldiezBase):
             ):
                 return [(chat, source, target)]
             if source.agent_type == "swarm" and target.agent_type == "swarm":
-                if isinstance(source, WaldiezSwarmAgent) and source.is_initial:
-                    return [(chat, source, target)]
                 valid_chats.append(chat)
         if not valid_chats:
             return []
+        for valid_chat in valid_chats:
+            source = self.get_agent_by_id(valid_chat.source)
+            if isinstance(source, WaldiezSwarmAgent) and source.is_initial:
+                return [
+                    (
+                        valid_chat,
+                        source,
+                        self.get_agent_by_id(valid_chat.target),
+                    )
+                ]
         first_chat: Optional[WaldiezChat] = None
         # first check the order
         by_order = sorted(

{waldiez-0.3.1.dist-info → waldiez-0.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.3.1
+Version: 0.3.3
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/waldiez/python
 Project-URL: repository, https://github.com/waldiez/python.git
@@ -73,9 +73,9 @@ Requires-Dist: mkdocstrings-python==1.13.0; extra == 'docs'
 Requires-Dist: mkdocstrings[crystal,python]==0.27.0; extra == 'docs'
 Provides-Extra: jupyter
 Requires-Dist: jupyterlab>=4.3.0; extra == 'jupyter'
-Requires-Dist: waldiez-jupyter==0.3.1; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.3.3; extra == 'jupyter'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.3.1; extra == 'studio'
+Requires-Dist: waldiez-studio==0.3.3; extra == 'studio'
 Provides-Extra: test
 Requires-Dist: pytest-asyncio==0.25.2; extra == 'test'
 Requires-Dist: pytest-cov==6.0.0; extra == 'test'

{waldiez-0.3.1.dist-info → waldiez-0.3.3.dist-info}/RECORD

@@ -1,6 +1,6 @@
 waldiez/__init__.py,sha256=WrnR3f-FQ0mK2orSfyApwRs9UGfq1nYiohQTyLrX_hs,2149
 waldiez/__main__.py,sha256=0dYzNrQbovRwQQvmZC6_1FDR1m71SUIOkTleO5tBnBw,203
-waldiez/_version.py,sha256=u6Jrpa5kGNsojGMAJ432UdSS_MdnGqYkbkOOcIp533s,155
+waldiez/_version.py,sha256=CyMN539kB8LnRXcUgDMXLFAZiap2g-3YrgLFQfYH6Wk,155
 waldiez/cli.py,sha256=SUdYLwNfG2TRnZr_E2gZ6SGczSQMIUdeIDzSZEYaw7o,6947
 waldiez/cli_extras.py,sha256=cnMJnznNgTjtI_qKAlgLOzXaJSuzx7D_XNDzC1faKzw,3018
 waldiez/conflict_checker.py,sha256=oF7PT2bdMGeIoj13cE2F6hn5dxhMhOutq34z8hCkZYg,976
@@ -9,7 +9,7 @@ waldiez/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 waldiez/runner.py,sha256=5J8bAdh2XaovM6xA8OW2jtfHXckjVUIE_9GtHn1UqzI,11038
 waldiez/exporting/__init__.py,sha256=7dMoWMJ-Hv48ZmM4m7QaqYvZ4G1o9uDJnklI6EU4r50,345
 waldiez/exporting/agent/__init__.py,sha256=GOgri16QNICpx3YNZPzVn_KYVxuQACYKQyazhJNtYK0,188
-waldiez/exporting/agent/agent_exporter.py,sha256=1fjf8yJdSDnYAeEmEUVL955f2ztlMHq33I8Tek9qYQo,9823
+waldiez/exporting/agent/agent_exporter.py,sha256=Xg6XY7fmJbBrvX_s3YS8ABIxM368jNYQTCGUnMuvcao,9785
 waldiez/exporting/agent/utils/__init__.py,sha256=G0iJgfI8tKrNufawPmG2um-hijUS4TZEn7Xwk2DkC88,825
 waldiez/exporting/agent/utils/agent_class_name.py,sha256=V3EB5VhMT5xDC0u92YB7_p_nEL6rO1Jbp63DKG60kVY,927
 waldiez/exporting/agent/utils/agent_imports.py,sha256=X8aX0ThYmBVYzqzAdKX1U4IttqneJuYANi3eAj7mGiE,1549
@@ -105,7 +105,7 @@ waldiez/models/common/__init__.py,sha256=OpQTQ-wNA56hZTKZfBSj4joKRzX5n2uNGhL1jKo
 waldiez/models/common/base.py,sha256=SwnnuFTJ0C3PfBoRilqE0uZv3j5aelw7-s-79KcxJNA,1840
 waldiez/models/common/method_utils.py,sha256=jT6Cpu_XskroLlMwLGCY4HNptQtnccvMOa2CwRwT6xc,6944
 waldiez/models/flow/__init__.py,sha256=3UdwoUZdkLr6G78tNTyaijoiwvmHZ-N0tY3G7S_xbRs,255
-waldiez/models/flow/flow.py,sha256=AIguDr9pd6N9csyKnnkHyJ0-g_4AcAT4TLcOOXTLDiU,14436
+waldiez/models/flow/flow.py,sha256=kM8K7dl8wZIP0GEryELC2yZ0t0blEFFkNOvx7VEMW4o,14700
 waldiez/models/flow/flow_data.py,sha256=5Mn27AMzd44rrcg2Toa6wZC4cX5oCOsCUhuoE0giQsE,4949
 waldiez/models/flow/utils.py,sha256=Bd8dWYlCenuU4VeusAL_qxkXzYSuxl9e3a7FhHs6UCY,5435
 waldiez/models/model/__init__.py,sha256=1cOyDThhZV8o7pZN1DaIhh12jWty_YMegoQ4Jg7uKyI,383
@@ -118,8 +118,8 @@ waldiez/running/__init__.py,sha256=hptO_sDFv0Alg-YJmLN-A49SeQ4YyWUMP8nBH21qO-o,6
 waldiez/running/environment.py,sha256=8IgEmm1MJrfwOlxyMAdoYtH-KRmLtlVROOWzlMtSbps,2268
 waldiez/running/gen_seq_diagram.py,sha256=y9O5HFe25YvfNc78MrWTwaSfFQEDu6-buq7lja8PYgg,5639
 waldiez/running/running.py,sha256=CpgiK9E-4T7ruVwB2baSv6qTipgTcdR9IulE8db5V4o,8561
-waldiez-0.3.1.dist-info/METADATA,sha256=OMQRrzPdM5OV5pGQodRxwafDHJpgjNAKFTVbQxP-o2Q,9026
-waldiez-0.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-waldiez-0.3.1.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
-waldiez-0.3.1.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
-waldiez-0.3.1.dist-info/RECORD,,
+waldiez-0.3.3.dist-info/METADATA,sha256=4PiL-W28dG9VdC-8O2ZpwK4Plxty8lcWE7HIzVIflLg,9026
+waldiez-0.3.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+waldiez-0.3.3.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
+waldiez-0.3.3.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
+waldiez-0.3.3.dist-info/RECORD,,