waldiez 0.1.19__py3-none-any.whl → 0.1.20__py3-none-any.whl

waldiez/__init__.py

@@ -1,6 +1,7 @@
 """Waldiez package."""
 
 import logging
+import warnings
 
 from ._version import __version__
 from .conflict_checker import check_conflicts
@@ -8,6 +9,8 @@ from .exporter import WaldiezExporter
 from .models import Waldiez
 from .runner import WaldiezRunner
 
+warnings.filterwarnings("ignore", "flaml.automl is not available")
+
 
 # pylint: disable=too-few-public-methods
 class FlamlFilter(logging.Filter):

waldiez/_version.py

@@ -1,3 +1,3 @@
 """Version information for Waldiez."""
 
-__version__ = "0.1.19"
+__version__ = "0.1.20"

waldiez/models/model/__init__.py

@@ -1,9 +1,10 @@
 """Waldiez model."""
 
-from .model import WaldiezModel
+from .model import DEFAULT_BASE_URLS, WaldiezModel
 from .model_data import WaldiezModelAPIType, WaldiezModelData, WaldiezModelPrice
 
 __all__ = [
+    "DEFAULT_BASE_URLS",
     "WaldiezModel",
     "WaldiezModelData",
     "WaldiezModelPrice",

waldiez/models/model/model.py

@@ -7,7 +7,16 @@ from pydantic import Field
 from typing_extensions import Annotated, Literal
 
 from ..common import WaldiezBase, now
-from .model_data import WaldiezModelData
+from .model_data import WaldiezModelAPIType, WaldiezModelData
+
+DEFAULT_BASE_URLS: Dict[WaldiezModelAPIType, str] = {
+    "google": "https://generativelanguage.googleapis.com/v1beta",
+    "anthropic": "https://api.anthropic.com/v1",
+    "mistral": "https://api.mistral.ai/v1",
+    "groq": "https://api.groq.com/openai/v1",
+    "together": "https://api.together.xyz/v1",
+    "nim": "https://integrate.api.nvidia.com/v1",
+}
 
 
 class WaldiezModel(WaldiezBase):
@@ -166,4 +175,30 @@ class WaldiezModel(WaldiezBase):
             value = getattr(self, attr)
             if value:
                 _llm_config[attr] = value
-        return _llm_config
+        return set_default_base_url(_llm_config, self.data.api_type)
+
+
+def set_default_base_url(
+    llm_config: Dict[str, Any], api_type: WaldiezModelAPIType
+) -> Dict[str, Any]:
+    """Set the default base url if not provided.
+
+    Parameters
+    ----------
+    llm_config : Dict[str, Any]
+        The llm config dictionary.
+    api_type : str
+        The api type.
+
+    Returns
+    -------
+    Dict[str, Any]
+        The llm config dictionary with the default base url set.
+    """
+    if api_type in ("openai", "other", "azure"):
+        return llm_config
+    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

waldiez/runner.py

@@ -18,6 +18,7 @@ import site
 import subprocess  # nosemgrep # nosec
 import sys
 import tempfile
+import warnings
 from contextlib import contextmanager
 from pathlib import Path
 from types import TracebackType
@@ -28,6 +29,7 @@ from typing import (
     Iterator,
     List,
     Optional,
+    Tuple,
     Type,
     Union,
 )
@@ -349,6 +351,7 @@ def refresh_environment() -> None:
     importlib.reload(autogen)
     # restore the default IOStream
     IOStream.set_global_default(default_io_stream)
+    warnings.filterwarnings("ignore", "flaml.automl is not available")
 
 
 def get_printer() -> Callable[..., None]:
@@ -361,4 +364,57 @@ def get_printer() -> Callable[..., None]:
     """
     from autogen.io import IOStream
 
-    return IOStream.get_default().print
+    printer = IOStream.get_default().print
+
+    def safe_printer(*args: object, **kwargs: object) -> None:
+        try:
+            printer(*args, **kwargs)
+        except UnicodeEncodeError:
+            # pylint: disable=too-many-try-statements
+            try:
+                msg, flush = get_what_to_print(*args, **kwargs)
+                printer(msg, end="", flush=flush)
+            except UnicodeEncodeError:
+                sys.stdout = io.TextIOWrapper(
+                    sys.stdout.buffer, encoding="utf-8"
+                )
+                sys.stderr = io.TextIOWrapper(
+                    sys.stderr.buffer, encoding="utf-8"
+                )
+                try:
+                    printer(*args, **kwargs)
+                except UnicodeEncodeError:
+                    sys.stderr.write(
+                        "Could not print the message due to encoding issues.\n"
+                    )
+
+    return safe_printer
+
+
+def get_what_to_print(*args: object, **kwargs: object) -> Tuple[str, bool]:
+    """Get what to print.
+
+    Parameters
+    ----------
+    args : object
+        The arguments.
+    kwargs : object
+        The keyword arguments.
+
+    Returns
+    -------
+    Tuple[str, bool]
+        The message and whether to flush.
+    """
+    sep = kwargs.get("sep", " ")
+    if not isinstance(sep, str):
+        sep = " "
+    end = kwargs.get("end", "\n")
+    if not isinstance(end, str):
+        end = "\n"
+    flush = kwargs.get("flush", False)
+    if not isinstance(flush, bool):
+        flush = False
+    msg = sep.join(str(arg) for arg in args) + end
+    utf8_msg = msg.encode("utf-8", errors="replace").decode("utf-8")
+    return utf8_msg, flush

{waldiez-0.1.19.dist-info → waldiez-0.1.20.dist-info}/METADATA

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

{waldiez-0.1.19.dist-info → waldiez-0.1.20.dist-info}/RECORD

@@ -1,11 +1,11 @@
-waldiez/__init__.py,sha256=9BLMLUURndkGEwnSNF1RXg1XKm4WTYM3cUeuYo-M0qY,1936
+waldiez/__init__.py,sha256=EhgIJb-VHvb74qIwO2BDME3CSvi4VEsn8LHeCa1SmEU,2020
 waldiez/__main__.py,sha256=mUQWu4CF03Jbbscvcfb_9EFTnIMliJJJmAuWf0sRRZU,110
-waldiez/_version.py,sha256=EX_WkICFRi_lmrwAjPX_8gLn_pS_uIsuK3aso-wK5Hk,63
+waldiez/_version.py,sha256=O-rY-SU1lOwACK5jnGmgJiS5lr3jJ-QamFbqNmBmLzM,63
 waldiez/cli.py,sha256=19LFqep8urzcLm6TzVSyJEOKI2Q8XXZbuuFsgV0k9LQ,6334
 waldiez/conflict_checker.py,sha256=E-w0TfTivDAVpNvjd_NSBeaaFsWtWAyxKoSz0-g6x2U,880
 waldiez/exporter.py,sha256=iKe-l_Me8NRWsXHIdBcrOrnLT9XIyp4iYi4HLuuj2jA,9342
 waldiez/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-waldiez/runner.py,sha256=10GoT27cxEJdrapmQ4ny9aN7ZcnmsNHApzwIft5ZxP0,11220
+waldiez/runner.py,sha256=rkRRFMLI2bM0LcZGG78qxGS2p67V2AvN0CsLMVA0Igo,12947
 waldiez/exporting/__init__.py,sha256=GMY7qTRpNmc7tpaCFNGLT5wX9eu26NvoNPuYX8MzP50,344
 waldiez/exporting/agents/__init__.py,sha256=v5KA112W_EFYwXE2TSBKYyO8rRKUOUAOpFS5CMSnfRs,110
 waldiez/exporting/agents/agent.py,sha256=d2ut2B90-aG4uBRhPWS7B4MKIvSJqQva6DjlXiqleVQ,7513
@@ -78,14 +78,14 @@ waldiez/models/common/method_utils.py,sha256=K7dNyvqHwvwuULLgn1_uK72Bfnm9d0CYHP3
 waldiez/models/flow/__init__.py,sha256=oy_G58xDkZk_LZEvqmr-0MJrpYy1JRf-U-F5-bI1944,162
 waldiez/models/flow/flow.py,sha256=1r0g_vBMd2PKcRC8P1ylWOUW3zwb529oOglf91SkILI,9663
 waldiez/models/flow/flow_data.py,sha256=m-RYaqlWjVpjyUzX9iAtXn1GOH0pFGKd2QD3e_e7yrk,2694
-waldiez/models/model/__init__.py,sha256=32Xox2Qw1jsf7ZCcoi0vMgZ5yxulLmtBrRRxq3FBAks,246
-waldiez/models/model/model.py,sha256=aG-9YAyLDYPqF-GWRTfuI7hP1Z7tfWWfj7Dhy1Y2GSc,4997
+waldiez/models/model/__init__.py,sha256=nT9jdKqeV4LPkqWi84Z7hCYZ1257Tj8YhUWbxhFlQl0,290
+waldiez/models/model/model.py,sha256=9A-pIufm9Sn4X9syphxtCOFfZ3BNo8ft07uxNXSRRiI,6130
 waldiez/models/model/model_data.py,sha256=VLPb60rJeZEgVZCjjkQGiwTrKz7OddVLrXCrEuteMcc,3679
 waldiez/models/skill/__init__.py,sha256=rU88bajKOGMYoHFcE8MP0jW9H0MswbQmvz5wxS35BYE,169
 waldiez/models/skill/skill.py,sha256=fhsAI413an2_d4DBIkf7dzEuWk6rGs2t4sl97a4dj20,3473
 waldiez/models/skill/skill_data.py,sha256=RTWn8Od6w7g-nRIpsS29sqZ8sPm5dCPiK7-qXmU-KD4,815
-waldiez-0.1.19.dist-info/METADATA,sha256=w6B5mTtRqZ80Y42gnIU1MbjKtCOWK94WdXKTbhgKpw0,8161
-waldiez-0.1.19.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-waldiez-0.1.19.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
-waldiez-0.1.19.dist-info/licenses/LICENSE,sha256=VQEHM6WMQLRu1qaGl3GWsoOknDwro-69eGo4NLIJPIM,1064
-waldiez-0.1.19.dist-info/RECORD,,
+waldiez-0.1.20.dist-info/METADATA,sha256=xTKC0uq5CGHR5EtIAx7vkKitjdeHmaVP6TxgrsUClCA,8161
+waldiez-0.1.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+waldiez-0.1.20.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
+waldiez-0.1.20.dist-info/licenses/LICENSE,sha256=VQEHM6WMQLRu1qaGl3GWsoOknDwro-69eGo4NLIJPIM,1064
+waldiez-0.1.20.dist-info/RECORD,,