waldiez 0.3.7__tar.gz → 0.3.8__tar.gz

{waldiez-0.3.7 → waldiez-0.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.3.7
+Version: 0.3.8
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/waldiez/python
 Project-URL: repository, https://github.com/waldiez/python.git

{waldiez-0.3.7 → waldiez-0.3.8}/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.8"

{waldiez-0.3.7 → waldiez-0.3.8}/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-0.3.7 → waldiez-0.3.8}/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-0.3.7 → waldiez-0.3.8}/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-0.3.7 → waldiez-0.3.8}/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-0.3.7 → waldiez-0.3.8}/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