waldiez 0.4.0__tar.gz → 0.4.1__tar.gz

{waldiez-0.4.0 → waldiez-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.4.0
+Version: 0.4.1
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/waldiez/python
 Project-URL: repository, https://github.com/waldiez/python.git
@@ -90,12 +90,12 @@ Requires-Dist: mkdocs-material==9.6.4; extra == 'docs'
 Requires-Dist: mkdocs-minify-html-plugin==0.2.4; extra == 'docs'
 Requires-Dist: mkdocs==1.6.1; extra == 'docs'
 Requires-Dist: mkdocstrings-python==1.15.0; extra == 'docs'
-Requires-Dist: mkdocstrings[crystal,python]==0.28.0; extra == 'docs'
+Requires-Dist: mkdocstrings[crystal,python]==0.28.1; extra == 'docs'
 Provides-Extra: jupyter
 Requires-Dist: jupyterlab>=4.3.0; extra == 'jupyter'
-Requires-Dist: waldiez-jupyter==0.4.0; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.4.1; extra == 'jupyter'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.4.0; extra == 'studio'
+Requires-Dist: waldiez-studio==0.4.1; extra == 'studio'
 Provides-Extra: test
 Requires-Dist: pytest-asyncio==0.25.3; extra == 'test'
 Requires-Dist: pytest-cov==6.0.0; extra == 'test'

{waldiez-0.4.0 → waldiez-0.4.1}/pyproject.toml

@@ -51,10 +51,10 @@ exclude = [ "**/example.py" ]
 
 [project.optional-dependencies]
 studio = [
-    "waldiez_studio==0.4.0",
+    "waldiez_studio==0.4.1",
 ]
 jupyter = [
-    "waldiez_jupyter==0.4.0",
+    "waldiez_jupyter==0.4.1",
     "jupyterlab>=4.3.0",
 ]
 dev = [
@@ -129,7 +129,7 @@ docs = [
   'mkdocs-macros-plugin==1.3.7',
   'mkdocs-material==9.6.4',
   'mkdocs-minify-html-plugin==0.2.4',
-  'mkdocstrings[crystal,python]==0.28.0',
+  'mkdocstrings[crystal,python]==0.28.1',
   'mkdocstrings-python==1.15.0'
 ]
 

{waldiez-0.4.0 → waldiez-0.4.1}/waldiez/_version.py

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

{waldiez-0.4.0 → waldiez-0.4.1}/waldiez/exporting/agent/utils/captain_agent.py

@@ -142,10 +142,11 @@ def get_llm_config(
     max_tokens: Optional[int] = 2048
     if agent.data.model_ids:
         waldiez_model = get_waldiez_model(agent.data.model_ids[0], all_models)
-        model_name = waldiez_model.name
-        temperature = waldiez_model.data.temperature
-        top_p = waldiez_model.data.top_p
-        max_tokens = waldiez_model.data.max_tokens
+        llm_config = waldiez_model.get_llm_config(skip_price=True)
+        for key in ["temperature", "top_p", "max_tokens"]:
+            if key not in llm_config:
+                llm_config[key] = None
+        return llm_config
     config_dict = {
         "model": model_name,
         "temperature": temperature,

{waldiez-0.4.0 → waldiez-0.4.1}/waldiez/exporting/flow/flow_exporter.py

@@ -213,7 +213,7 @@ class FlowExporter(BaseExporter, ExporterMixin):
         content += imports[0] + "\n"
         content += get_np_no_nep50_handle() + "\n"
         content += self.get_comment("logging", self.for_notebook) + "\n"
-        content += get_start_logging(tabs=0) + "\n"
+        content += get_start_logging(is_async=is_async, tabs=0) + "\n"
         content += "start_logging()\n\n"
         if models_output:
             content += self.get_comment("models", self.for_notebook) + "\n"

{waldiez-0.4.0 → waldiez-0.4.1}/waldiez/exporting/flow/utils/importing_utils.py

@@ -127,7 +127,6 @@ def get_the_imports_string(
             "\nimport anyio"
             "\nimport nest_asyncio"
             "\nfrom aiocsv import AsyncDictWriter"
-            "\nfrom asyncer import asyncify"
         )
     if got_import_autogen:
         final_string += "\nimport autogen  # type: ignore\n"

{waldiez-0.4.0 → waldiez-0.4.1}/waldiez/exporting/flow/utils/logging_utils.py

@@ -16,11 +16,13 @@ get_sqlite_to_csv_call_string
 """
 
 
-def get_start_logging(tabs: int = 0) -> str:
+def get_start_logging(is_async: bool, tabs: int = 0) -> str:
     """Get the logging start call string.
 
     Parameters
     ----------
+    is_async : bool
+        Whether to use async mode.
     tabs : int, optional
         The number of tabs to use for indentation, by default 0
 
@@ -41,7 +43,8 @@ def get_start_logging(tabs: int = 0) -> str:
         )
     """
     tab = "    " * tabs
-    content = f'''
+    if is_async is False:
+        return f'''
 {tab}def start_logging() -> None:
 {tab}    """Start logging."""
 {tab}    runtime_logging.start(
@@ -49,7 +52,20 @@ def get_start_logging(tabs: int = 0) -> str:
 {tab}        config={{"dbname": "flow.db"}},
 {tab}    )
 '''
-    return content
+    return f'''
+{tab}def start_logging() -> None:
+{tab}    """Start logging."""
+{tab}    # pylint: disable=import-outside-toplevel
+{tab}    from anyio.from_thread import start_blocking_portal
+
+{tab}    with start_blocking_portal(backend="asyncio") as portal:
+{tab}        portal.call(
+{tab}            runtime_logging.start,
+{tab}            None,
+{tab}            "sqlite",
+{tab}            {{"dbname": "flow.db"}},
+{tab}        )
+'''
 
 
 # pylint: disable=differing-param-doc,differing-type-doc
@@ -336,6 +352,11 @@ def get_stop_logging(tabs: int, is_async: bool) -> str:
         content += "async "
     content += "def stop_logging() -> None:\n"
     content += '    """Stop logging."""\n'
-    content += f"{tab}    runtime_logging.stop()\n"
+    if is_async:
+        content += f"{tab}    # pylint: disable=import-outside-toplevel\n"
+        content += f"{tab}    from asyncer import asyncify\n\n"
+        content += f"{tab}    await asyncify(runtime_logging.stop)()\n"
+    else:
+        content += f"{tab}    runtime_logging.stop()\n"
     content += get_sqlite_out_call(tabs + 1, is_async)
     return content

{waldiez-0.4.0 → waldiez-0.4.1}/waldiez/models/model/__init__.py

@@ -3,12 +3,13 @@
 """Waldiez model."""
 
 from .extra_requirements import get_models_extra_requirements
-from .model import DEFAULT_BASE_URLS, WaldiezModel
+from .model import DEFAULT_BASE_URLS, MODEL_NEEDS_BASE_URL, WaldiezModel
 from .model_data import WaldiezModelAPIType, WaldiezModelData, WaldiezModelPrice
 
 __all__ = [
     "get_models_extra_requirements",
     "DEFAULT_BASE_URLS",
+    "MODEL_NEEDS_BASE_URL",
     "WaldiezModel",
     "WaldiezModelData",
     "WaldiezModelPrice",

{waldiez-0.4.0 → waldiez-0.4.1}/waldiez/models/model/model.py

@@ -23,6 +23,22 @@ DEFAULT_BASE_URLS: Dict[WaldiezModelAPIType, str] = {
 }
 
 
+# we can omit the base_url for these models
+MODEL_NEEDS_BASE_URL: Dict[WaldiezModelAPIType, bool] = {
+    "openai": False,
+    "azure": False,
+    "google": False,
+    "anthropic": False,
+    "cohere": False,
+    "other": False,  # falls back to openai
+    "deepseek": True,
+    "mistral": True,
+    "groq": True,
+    "together": True,
+    "nim": True,
+}
+
+
 class WaldiezModel(WaldiezBase):
     """Waldiez Model class.
 
@@ -225,10 +241,13 @@ def set_default_base_url(
     Dict[str, Any]
         The llm config dictionary with the default base url set.
     """
-    if api_type in ("openai", "other", "azure", "cohere"):
-        return llm_config
+    dict_copy = llm_config.copy()
     if "base_url" not in llm_config or not llm_config["base_url"]:
-        dict_copy = llm_config.copy()
-        dict_copy["base_url"] = DEFAULT_BASE_URLS.get(api_type, "")
-        return dict_copy
-    return llm_config
+        if MODEL_NEEDS_BASE_URL.get(api_type, True):
+            dict_copy["base_url"] = DEFAULT_BASE_URLS.get(api_type, "")
+    if (
+        not llm_config.get("base_url", "")
+        and MODEL_NEEDS_BASE_URL.get(api_type, True) is False
+    ):
+        dict_copy.pop("base_url", None)
+    return dict_copy