waldiez 0.3.7__py3-none-any.whl → 0.3.9__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.7"
+__version__ = "0.3.9"

waldiez/exporting/flow/utils/flow_content.py

@@ -1,6 +1,7 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 # flake8: noqa E501
+# pylint: disable=line-too-long
 """Utils to generate the content of a flow."""
 
 from typing import Callable, Dict, List, Optional
@@ -56,9 +57,8 @@ def get_ipynb_content_start(
     content += "import sys\n"
     requirements = " ".join(waldiez.requirements)
     if requirements:
-        # pylint: disable=line-too-long
         # fmt: off
-        content += "# " + f"!{{sys.executable}} -m pip install -q {requirements}" + "\n"  # noqa: E501
+        content += "# " + f"!{{sys.executable}} -m pip install -q {requirements}" + "\n"
         # fmt: on
     content += "# flake8: noqa E501"
     content += get_pylint_ignore_comment(True)
@@ -149,5 +149,12 @@ def get_after_run_content(
 {space}{space}{space}os.rename("tree_of_thoughts.png", new_name)
 {space}except BaseException:
 {space}{space}pass
+{space}# save the tree to json
+{space}try:
+{space}{space}data = {agent_name}._root.to_dict()  # pylint: disable=protected-access
+{space}{space}with open("{agent_name}_reasoning_tree.json", "w", encoding="utf-8") as f:
+{space}{space}{space}json.dump(data, f)
+{space}except BaseException:
+{space}{space}pass
 """
     return content

waldiez/models/model/model.py

@@ -12,6 +12,7 @@ from ..common import WaldiezBase, now
 from .model_data import WaldiezModelAPIType, WaldiezModelData
 
 DEFAULT_BASE_URLS: Dict[WaldiezModelAPIType, str] = {
+    "deepseek": "https://api.deepseek.com/v1",
     "google": "https://generativelanguage.googleapis.com/v1beta",
     "anthropic": "https://api.anthropic.com/v1",
     "mistral": "https://api.mistral.ai/v1",
@@ -110,6 +111,7 @@ class WaldiezModel(WaldiezBase):
 
         - openai: 'OPENAI_API_KEY',
         - azure: 'AZURE_API_KEY',
+        - deepseek: 'DEEPSEEK_API_KEY',
         - google: 'GOOGLE_GEMINI_API_KEY',
         - anthropic: 'ANTHROPIC_API_KEY',
         - mistral: 'MISTRAL_API_KEY',
@@ -133,6 +135,7 @@ class WaldiezModel(WaldiezBase):
 
             - openai: 'OPENAI_API_KEY',
             - azure: 'AZURE_API_KEY',
+            - deepseek: 'DEEPSEEK_API_KEY',
             - google: 'GOOGLE_GEMINI_API_KEY',
             - anthropic: 'ANTHROPIC_API_KEY',
             - mistral: 'MISTRAL_API_KEY',

waldiez/models/model/model_data.py

@@ -13,6 +13,7 @@ from ..common import WaldiezBase
 WaldiezModelAPIType = Literal[
     "openai",
     "azure",
+    "deepseek",
     "google",
     "anthropic",
     "mistral",
@@ -52,7 +53,7 @@ class WaldiezModelData(WaldiezBase):
         The base url of the model, by default None.
     api_key : Optional[str]
         The api key to use with the model, by default None.
-    api_type : Literal["openai","azure","google","anthropic","mistral","groq","together","nim","other"]
+    api_type : Literal["openai","azure","deepseek","google","anthropic","mistral","groq","together","nim","other"]
         The api type of the model.
     api_version : Optional[str]
         The api version of the model, by default None.

waldiez/running/running.py

@@ -259,7 +259,9 @@ def copy_results(
         if item.is_file():
             # let's also copy the tree of thoughts image
             # to the output directory
-            if item.name.endswith("tree_of_thoughts.png"):
+            if item.name.endswith("tree_of_thoughts.png") or item.name.endswith(
+                "reasoning_tree.json"
+            ):
                 shutil.copy(item, output_dir / item.name)
             shutil.copy(item, destination_dir)
         else:

waldiez/utils/cli_extras/studio.py

@@ -13,7 +13,7 @@ HAVE_STUDIO = False
 studio_app: Callable[..., Any] | None = None
 
 try:
-    from waldiez_studio.cli import run  # type: ignore[unused-ignore, import-untyped, import-not-found]  # noqa
+    from waldiez_studio.cli import run  # type: ignore[import-untyped, unused-ignore, import-not-found]  # noqa
 
     studio_app = run
 

{waldiez-0.3.7.dist-info → waldiez-0.3.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.3.7
+Version: 0.3.9
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/waldiez/python
 Project-URL: repository, https://github.com/waldiez/python.git
@@ -81,9 +81,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.7; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.3.9; extra == 'jupyter'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.3.7; extra == 'studio'
+Requires-Dist: waldiez-studio==0.3.9; 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.7.dist-info → waldiez-0.3.9.dist-info}/RECORD

@@ -1,6 +1,6 @@
 waldiez/__init__.py,sha256=cXWdy5P9i_x6fHwlL5DRaIhmpfnhabo5GjASStlcLWw,819
 waldiez/__main__.py,sha256=0dYzNrQbovRwQQvmZC6_1FDR1m71SUIOkTleO5tBnBw,203
-waldiez/_version.py,sha256=5BKC4jK5nNYNJAtQJvJbXLE-1WfATcL42nqWx0jzOXg,155
+waldiez/_version.py,sha256=HFkN__4uUqsN_dhEndMd-v-igGH08Wp2T_r192TNq90,155
 waldiez/cli.py,sha256=ZhcXOH5WtGVncQpoOix9yHXctQKBwckgiIdYLaRj40c,7108
 waldiez/exporter.py,sha256=8EqLya3UqCA_F2TwtfdeRCt7ILnWxBcQvkMyX8no0BU,6501
 waldiez/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -49,7 +49,7 @@ waldiez/exporting/flow/utils/__init__.py,sha256=Fe0sg5i1K2Vu0ruWsIuPj7oTRHoJTW4K
 waldiez/exporting/flow/utils/agent_utils.py,sha256=QL1nzlgYeh6Y-l9R3hmCsrxSzluvBVciuuEIFdBXXKM,6109
 waldiez/exporting/flow/utils/chat_utils.py,sha256=q6HS4JTNQJe15eh78H1BlN-wId9XzvVQziGhxsUIEFs,1882
 waldiez/exporting/flow/utils/def_main.py,sha256=7yrxYizssTBYvcsXj3p-LIP-29-KpCRAV7I3_bhWbXE,2440
-waldiez/exporting/flow/utils/flow_content.py,sha256=37aCeUvWUlKH3O9aljoNYI4QAa6KrMS8wwIUVbNdXrs,4231
+waldiez/exporting/flow/utils/flow_content.py,sha256=kAEXS5rWGw0Lvc4R5fosRZZ89SOlFbLsp7qk10pirPU,4515
 waldiez/exporting/flow/utils/flow_names.py,sha256=5i1qjSrkbIFtnE13sv_GjOTyRnmTzPMys1Ecd5au7mg,3463
 waldiez/exporting/flow/utils/importing_utils.py,sha256=5jeuo36cHSCyXRsi9OchEastP4O92lMStCqlQ2WlaI0,5415
 waldiez/exporting/flow/utils/logging_utils.py,sha256=m3Ja00Qb_07rLVgwSOFSBaJk48SfCNsyPuQbRy5VMNU,10963
@@ -114,25 +114,25 @@ waldiez/models/flow/flow.py,sha256=kM8K7dl8wZIP0GEryELC2yZ0t0blEFFkNOvx7VEMW4o,1
 waldiez/models/flow/flow_data.py,sha256=9oF2_bd-p2gmnv-rNDKeP_xh8zLiOY095dS0rig5JD4,4969
 waldiez/models/flow/utils.py,sha256=Bd8dWYlCenuU4VeusAL_qxkXzYSuxl9e3a7FhHs6UCY,5435
 waldiez/models/model/__init__.py,sha256=1cOyDThhZV8o7pZN1DaIhh12jWty_YMegoQ4Jg7uKyI,383
-waldiez/models/model/model.py,sha256=fRPJxvlINYbKzzDr5oWe9EKIU60w3YDMcdY9MWpFFOg,6853
-waldiez/models/model/model_data.py,sha256=XrMBbnCRkJBlk46xJMoBJrcTX9qnViJZTn9_MS9ujPw,3792
+waldiez/models/model/model.py,sha256=0EqHSMIBun_Owy-v0cbv39sRP8if1LlLKeMOD_XA3Ow,6984
+waldiez/models/model/model_data.py,sha256=EwObYtzPyxZqADdTeWjULNmb3dLgF91mzK3r1nxxJJk,3819
 waldiez/models/skill/__init__.py,sha256=kAyp9-LwBacI3ndQpU-dCGWjsjojRQy73zXsTddGAmg,306
 waldiez/models/skill/skill.py,sha256=6r8heHUgRYwfcBJbdsV5DywvTXcEeWPuJN4jzCh3-O4,4255
 waldiez/models/skill/skill_data.py,sha256=Spk3YxK0IgkXpJgk05DJwvZwUPrh-2l95o9PnS1WnoQ,908
 waldiez/running/__init__.py,sha256=hptO_sDFv0Alg-YJmLN-A49SeQ4YyWUMP8nBH21qO-o,633
 waldiez/running/environment.py,sha256=q4h3GWBSGR79F9lqLTqgGYh3xyt2xL-kPYDc65orryY,2540
 waldiez/running/gen_seq_diagram.py,sha256=wIsbAddASKNb-pRQhCIhouWGLuhX2Xja5x5xHCCSqHk,5716
-waldiez/running/running.py,sha256=tYZuXMfISgatSmAAxxokfi6QbTyWK2a99Z0_bhGWdbw,10016
+waldiez/running/running.py,sha256=Uqe4CCt_Rple4OGpOqLDeSGNXbqssXtdIezm0_Sm-PA,10091
 waldiez/utils/__init__.py,sha256=M7ZqIInj2371ojAO73OuEPZ3fF17WZVbGE-37oyJwl0,415
 waldiez/utils/conflict_checker.py,sha256=v7BNDenJSHfPBfqb4PnxaVZwjPqzWlh5w31WC9uOJL4,1369
 waldiez/utils/flaml_warnings.py,sha256=jgdS8zwzuWPodTgm9nnx3wO5MrOUbR7P7IRj9-aq2qk,601
 waldiez/utils/pysqlite3_checker.py,sha256=LypcHlCMbzdC0QUfrr68gzmrEbwzvKLUn-riYiCKu70,7636
 waldiez/utils/cli_extras/__init__.py,sha256=FQ6biuz6Ig-U3xkmgac3WMUynH_7XWjeGI-DqWFTC8c,625
 waldiez/utils/cli_extras/jupyter.py,sha256=9WlDxhn4LkoSa72PKyPAWsBUGwUez3YAk6IvJnMD7nc,2993
-waldiez/utils/cli_extras/studio.py,sha256=Irl34rshvCQYAC_62dAH-_OQxCDeQ8aNgJDw3ggTuyI,871
-waldiez-0.3.7.dist-info/METADATA,sha256=a8BHJOg27xJWK7hvZiMegFFToquiGdV-qrtO40g9RKg,9451
-waldiez-0.3.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-waldiez-0.3.7.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
-waldiez-0.3.7.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
-waldiez-0.3.7.dist-info/licenses/NOTICE.md,sha256=lrKsUNrpE18LPSLPJY_kt2ZJFZJEwH55wPv5Axcheyc,133
-waldiez-0.3.7.dist-info/RECORD,,
+waldiez/utils/cli_extras/studio.py,sha256=oF1O2pmAUZudNou7jzx57uQoKsH-0uib71ZtyT149zM,871
+waldiez-0.3.9.dist-info/METADATA,sha256=z-65ty-ZetUYGEoWkvC7Q_Ul5gW0mnC35IEX6x_soiI,9451
+waldiez-0.3.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+waldiez-0.3.9.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
+waldiez-0.3.9.dist-info/licenses/LICENSE,sha256=bWinEG_ynCS8eSj1vhAmYDWwYODhBOF0AUEjOgZ3kmU,11355
+waldiez-0.3.9.dist-info/licenses/NOTICE.md,sha256=lrKsUNrpE18LPSLPJY_kt2ZJFZJEwH55wPv5Axcheyc,133
+waldiez-0.3.9.dist-info/RECORD,,