unique_toolkit 2026.44.0.dev9__py3-none-any.whl → 2026.46.0.dev0__py3-none-any.whl

unique_toolkit/chat/functions.py

@@ -35,6 +35,9 @@ from unique_toolkit.language_model.constants import (
     DEFAULT_COMPLETE_TEMPERATURE,
     DEFAULT_COMPLETE_TIMEOUT,
 )
+from unique_toolkit.language_model.errors import (
+    _reraise_as_language_model_call_error,
+)
 from unique_toolkit.language_model.functions import (
     _prepare_all_completions_params_util,
 )
@@ -954,7 +957,7 @@ def stream_complete_with_references(
         return LanguageModelStreamResponse(**response)
     except Exception as e:
         _LOGGER.error(f"Error streaming completion: {e}")
-        raise e
+        _reraise_as_language_model_call_error(e)
 
 
 @deprecated("Use stream_complete_with_references_async instead")
@@ -1051,7 +1054,7 @@ async def stream_complete_with_references_async(
         return LanguageModelStreamResponse(**response)
     except Exception as e:
         _LOGGER.error(f"Error streaming completion: {e}")
-        raise e
+        _reraise_as_language_model_call_error(e)
 
 
 def _get_model_dump_or_none(model: BaseModel | None) -> dict[str, Any] | None:

unique_toolkit/chat/responses_api.py

@@ -1,7 +1,7 @@
 import json
 import logging
 import random
-from collections.abc import Awaitable, Callable
+from collections.abc import Awaitable, Callable, Mapping
 from typing import Any, NamedTuple, Sequence
 
 import unique_sdk
@@ -30,6 +30,9 @@ from unique_toolkit.content.schemas import ContentChunk
 from unique_toolkit.language_model.constants import (
     DEFAULT_COMPLETE_TEMPERATURE,
 )
+from unique_toolkit.language_model.errors import (
+    _reraise_as_language_model_call_error,
+)
 from unique_toolkit.language_model.functions import (
     SearchContext,
     _to_search_context,
@@ -407,8 +410,26 @@ class RateLimitRetryConfig(BaseSettings):
 rate_limit_retry_config = RateLimitRetryConfig()
 
 
+def _retry_after_seconds(exc: BaseException | None) -> float | None:
+    params = getattr(exc, "params", None)
+    if not isinstance(params, Mapping):
+        return None
+    retry_after = params.get("retryAfterSeconds")
+    if (
+        isinstance(retry_after, (int, float))
+        and not isinstance(retry_after, bool)
+        and retry_after >= 0
+    ):
+        return float(retry_after)
+    return None
+
+
 def _rate_limit_wait(retry_state: RetryCallState) -> float:
-    """Exponential backoff with up to 10% jitter: initial_delay → 2x → 4x."""
+    """Wait ``retryAfterSeconds`` when the API sent one, else exponential backoff."""
+    exc = retry_state.outcome.exception() if retry_state.outcome else None
+    retry_after = _retry_after_seconds(exc)
+    if retry_after is not None:
+        return retry_after
     n = retry_state.attempt_number - 1  # 0-indexed
     base = rate_limit_retry_config.initial_delay_seconds * (2.0**n)
     return base + random.uniform(0.0, base * 0.1)
@@ -421,9 +442,8 @@ async def _responses_stream_with_rate_limit_retry(
 ) -> "unique_sdk.Integrated.ResponsesStreamResult":
     """Wrap responses_stream_async with exponential backoff for rate-limit errors.
 
-    The SDK already retries 3× (1s/2s/4s) before raising. This layer adds further
-    retries with longer delays, which are needed for models with tighter RPM limits
-    (e.g. GPT-4o) when large-token requests (web search, code execution) are in the loop.
+    The SDK does not retry 429s. This layer retries a retryable ``RateLimitError``,
+    waiting ``params.retryAfterSeconds`` when the API sent one.
 
     Args:
         responses_args: Keyword arguments forwarded to responses_stream_async.
@@ -434,6 +454,11 @@ async def _responses_stream_with_rate_limit_retry(
     """
 
     def _is_rate_limit(exc: BaseException) -> bool:
+        if isinstance(exc, unique_sdk.RateLimitError):
+            params = getattr(exc, "params", None)
+            if isinstance(params, Mapping) and params.get("retryable") is False:
+                return False
+            return True
         return isinstance(exc, unique_sdk.APIError) and any(
             kw in str(exc).lower() for kw in _RATE_LIMIT_KEYWORDS
         )
@@ -539,11 +564,11 @@ def stream_responses_with_references(
         other_options=other_options,
     )
 
-    return ResponsesLanguageModelStreamResponse.model_validate(
-        unique_sdk.Integrated.responses_stream(
-            **responses_args,
-        )
-    )
+    try:
+        streamed = unique_sdk.Integrated.responses_stream(**responses_args)
+    except unique_sdk.UniqueError as error:
+        _reraise_as_language_model_call_error(error)
+    return ResponsesLanguageModelStreamResponse.model_validate(streamed)
 
 
 async def stream_responses_with_references_async(
@@ -608,10 +633,12 @@ async def stream_responses_with_references_async(
         "Calling responses_stream_async model=%s",
         responses_params.model_name,
     )
-    return ResponsesLanguageModelStreamResponse.model_validate(
-        await _responses_stream_with_rate_limit_retry(
+    try:
+        streamed = await _responses_stream_with_rate_limit_retry(
             responses_args=responses_args,
             model_name=responses_params.model_name,
             on_rate_limit_retry=on_rate_limit_retry,
         )
-    )
+    except unique_sdk.UniqueError as error:
+        _reraise_as_language_model_call_error(error)
+    return ResponsesLanguageModelStreamResponse.model_validate(streamed)

unique_toolkit/language_model/__init__.py

@@ -2,6 +2,7 @@ from typing import TYPE_CHECKING, Any
 
 from .constants import DOMAIN_NAME
 from .default_language_model import DEFAULT_LANGUAGE_MODEL, DEFAULT_GPT_4o
+from .errors import LanguageModelCallError as LanguageModelCallError
 from .functions import (
     stream_complete_with_references_openai as stream_complete_with_references_openai,
 )
@@ -59,6 +60,7 @@ def __getattr__(name: str) -> Any:
 
 __all__ = [
     "LanguageModel",
+    "LanguageModelCallError",
     "LanguageModelName",
     "Prompt",
     "LanguageModelAssistantMessage",

unique_toolkit/language_model/errors.py

@@ -0,0 +1,90 @@
+from collections.abc import Mapping
+from typing import NoReturn
+
+import unique_sdk
+
+
+class LanguageModelCallError(unique_sdk.APIError):
+    """LLM call rejected with node-chat's error contract.
+
+    ``user_message`` is the text the API curated for the user. Display that
+    field. ``str(self)`` is the same text, without the SDK's
+    ``(Original error)`` suffix.
+
+    It subclasses ``APIError`` so callers that catch the SDK error around
+    toolkit calls keep catching it.
+
+    Args:
+        code (str): ``error.code`` from the API, for example ``content_filter``.
+        user_message (str): Text to show the user.
+        retryable (bool): Whether the API marked the call safe to retry.
+        correlation_id (str | None): Request correlation id, when the API sent one.
+        http_status (int | None): HTTP status of the failed call.
+    """
+
+    def __init__(
+        self,
+        *,
+        code: str,
+        user_message: str,
+        retryable: bool,
+        correlation_id: str | None,
+        http_status: int | None = None,
+    ) -> None:
+        super().__init__(message=user_message, http_status=http_status, code=code)
+        self.user_message = user_message
+        self.retryable = retryable
+        self.correlation_id = correlation_id
+
+
+def _language_model_call_error_from(
+    error: unique_sdk.UniqueError,
+) -> LanguageModelCallError | None:
+    """Build a ``LanguageModelCallError`` when ``error`` carries the contract.
+
+    Returns None for legacy SDK errors that have no ``code`` or no boolean
+    ``params.retryable``. Callers re-raise those unchanged.
+
+    Args:
+        error (unique_sdk.UniqueError): Error raised by ``unique_sdk``.
+
+    Returns:
+        LanguageModelCallError | None: The converted error, or None.
+    """
+    if not isinstance(error.code, str) or error.code == "":
+        return None
+    params = getattr(error, "params", None)
+    if not isinstance(params, Mapping):
+        return None
+    retryable = params.get("retryable")
+    if not isinstance(retryable, bool):
+        return None
+    user_message = error.args[0] if error.args else None
+    if not isinstance(user_message, str) or user_message == "":
+        return None
+    correlation_id = params.get("correlationId")
+    if not isinstance(correlation_id, str):
+        correlation_id = None
+    return LanguageModelCallError(
+        code=error.code,
+        user_message=user_message,
+        retryable=retryable,
+        correlation_id=correlation_id,
+        http_status=error.http_status,
+    )
+
+
+def _reraise_as_language_model_call_error(error: BaseException) -> NoReturn:
+    """Raise ``LanguageModelCallError`` for a contract error, else re-raise ``error``.
+
+    The SDK wraps the last error in ``APIError("Failed after N attempts")`` once
+    retries run out, so the contract is also read from ``error.__cause__``.
+    """
+    if isinstance(error, LanguageModelCallError):
+        raise error
+    for candidate in (error, error.__cause__):
+        if isinstance(candidate, unique_sdk.UniqueError):
+            converted = _language_model_call_error_from(candidate)
+            if converted is not None:
+                raise converted from error
+    raise error

unique_toolkit/language_model/functions.py

@@ -16,6 +16,9 @@ from pydantic import BaseModel
 
 from unique_toolkit.chat.schemas import ChatMessage, ChatMessageRole
 from unique_toolkit.content.schemas import ContentChunk, ContentReference
+from unique_toolkit.language_model.errors import (
+    _reraise_as_language_model_call_error,
+)
 from unique_toolkit.language_model.infos import (
     LanguageModelInfo,
     LanguageModelName,
@@ -133,7 +136,7 @@ def complete(
         return LanguageModelResponse(**response)
     except Exception as e:
         logger.error(f"Error completing: {e}")
-        raise e
+        _reraise_as_language_model_call_error(e)
 
 
 async def complete_async(
@@ -173,8 +176,8 @@ async def complete_async(
 
     Raises:
     ------
-        Exception: If an error occurs during the request, an exception is raised
-        and logged.
+        LanguageModelCallError: The API returned the LLM error contract.
+        Exception: Any other request failure is logged and re-raised.
 
     """
     if not user_id:
@@ -217,7 +220,7 @@ async def complete_async(
         return LanguageModelResponse(**response)
     except Exception as e:
         logger.exception(f"Error completing: {e}")
-        raise e
+        _reraise_as_language_model_call_error(e)
 
 
 def _add_tools_to_options(

{unique_toolkit-2026.44.0.dev9.dist-info → unique_toolkit-2026.46.0.dev0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.44.0.dev9
+Version: 2026.46.0.dev0
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>
@@ -16,7 +16,7 @@ Requires-Dist: sseclient>=0.0.27,<0.0.28
 Requires-Dist: openai>=1.105.0,<3
 Requires-Dist: platformdirs>=4.0.0,<5
 Requires-Dist: pillow>=12.1.1
-Requires-Dist: unique-sdk>=2026.44.0.dev13,<2026.44.0rc0
+Requires-Dist: unique-sdk>=2026.46.0.dev0,<2026.46.0rc0
 Requires-Dist: jambo>=0.1.2,<0.2
 Requires-Dist: jinja2>=3.1.6,<4
 Requires-Dist: docxtpl>=0.20.1,<0.21

{unique_toolkit-2026.44.0.dev9.dist-info → unique_toolkit-2026.46.0.dev0.dist-info}/RECORD

@@ -264,9 +264,9 @@ unique_toolkit/chat/cancellation.py,sha256=k5HMjpeBmofIiInwq5TjOR_5k91oq1bWG00lJ
 unique_toolkit/chat/constants.py,sha256=05kq6zjqUVB2d6_P7s-90nbljpB3ryxwCI-CAz0r2O4,83
 unique_toolkit/chat/deprecated/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/chat/deprecated/service.py,sha256=8_XoopUWHT396Z_5gwMmsI6v4NGVgQirSJCYUW4Gx8c,9339
-unique_toolkit/chat/functions.py,sha256=-XNlQHO4UAdDwwnZUXFUAlV-lKh-7oqyJO6ZKyr_4Jc,57846
+unique_toolkit/chat/functions.py,sha256=XCAErF-gRBTETx2QW7EpOyGpzI1QUh_7oMKVPgDnaG0,58008
 unique_toolkit/chat/rendering.py,sha256=c8YiV9oADRrJQ5A_QBJ4_UFc0NZ-2vVaa7tupoMusso,880
-unique_toolkit/chat/responses_api.py,sha256=Swp6wi10vCPkBiSSj8NN58gpjTgYVkqRlwdOJRQHVjo,21135
+unique_toolkit/chat/responses_api.py,sha256=8RBErUBtIvjBW2ZP644g7LqtAGiAVZlZ2ahh6tXfrbQ,22179
 unique_toolkit/chat/schemas.py,sha256=tRU8vcT1SQWtktkHQmDO26CNbOgSynFHfFB9FR6D4OQ,8372
 unique_toolkit/chat/service.py,sha256=6D00OL4QrGafbOhTaC5zNXaNgg7gS5W_2ePVa4LhqpE,4439
 unique_toolkit/chat/state.py,sha256=Cjgwv_2vhDFbV69xxsn7SefhaoIAEqLx3ferdVFCnOg,1445
@@ -392,12 +392,13 @@ unique_toolkit/framework_utilities/openai/__init__.py,sha256=CrHYoC7lv2pBscitLer
 unique_toolkit/framework_utilities/openai/client.py,sha256=Ft-oLiS6BTaMiuCdObdaMLkiEPE3RYjtc9VPnqQk3W8,2449
 unique_toolkit/framework_utilities/openai/message_builder.py,sha256=RT1pZjxH42TFZlAxQ5zlqdKPvHKVTjc5t3JDUy58I7Q,6887
 unique_toolkit/framework_utilities/utils.py,sha256=JK7g2yMfEx3eMprug26769xqNpS5WJcizf8n2zWMBng,789
-unique_toolkit/language_model/__init__.py,sha256=dr1O-m9irgQi2RxNz5g6sykhu9E2Z8-cs-_MMAD62WU,2711
+unique_toolkit/language_model/__init__.py,sha256=mgfm7S-BK4pZtz1dXykmfKAKtIBIDS2atiXtUIIioUg,2810
 unique_toolkit/language_model/_responses_api_utils.py,sha256=w2ZyleOQcWLpOWGJNNb2-uBYApUNlqS-ABqXoMFySlg,3084
 unique_toolkit/language_model/builder.py,sha256=KwBzsD6-fvQnB4p2I7M-5Tk6j4tmQ973MYwU1ulXFb4,4526
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
 unique_toolkit/language_model/default_language_model.py,sha256=mcyuFs_V6tp52C8xJSHiEHO4kFGj6xed1gvozvCRD7k,1181
-unique_toolkit/language_model/functions.py,sha256=sjuzKe6xAq6NFwwgIRvgAuU8t6apq1tcdGSkcYCl8EQ,27905
+unique_toolkit/language_model/errors.py,sha256=0nm4psGCGJM2B7LIyKXy7WuswRzwARIgIK5xoC4xo34,3186
+unique_toolkit/language_model/functions.py,sha256=6Le6dDm9su6xk3sQ8IPiyUn2tf1qvzxHVUhyTTgP8Kw,28109
 unique_toolkit/language_model/infos.py,sha256=_fbEwGA9AjBm7TF1FTsihtKzi4SybTwVN7QyHRacMrQ,183899
 unique_toolkit/language_model/invocation_stats.py,sha256=CQbpUz394IsVd8eNTB1gPSMt5f0UXmkN1amNsqtAW6U,5857
 unique_toolkit/language_model/model_costs.py,sha256=ShuDAzMew1fUkJ3tzPgU5Zqjp6ivUZaG97BFh93tEn4,6358
@@ -428,6 +429,6 @@ unique_toolkit/short_term_memory/service.py,sha256=2JMBEBn6GoyK-a0xvdcWpPft-3C74
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=vRe0aMJWuVIK-VsqomgxQImzke7MDmwgSxxJfYCa7zw,1386
 unique_toolkit/test_utilities/events.py,sha256=FLrhMh0A_mQutAUjoD0FdMcVQ4Gq31r35OVGzRjvLro,6760
-unique_toolkit-2026.44.0.dev9.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.44.0.dev9.dist-info/METADATA,sha256=g0ZlAxgIaNzapDN-UIvQYQ8S16AJWjJlofHlc5yzSm8,7822
-unique_toolkit-2026.44.0.dev9.dist-info/RECORD,,
+unique_toolkit-2026.46.0.dev0.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.46.0.dev0.dist-info/METADATA,sha256=yT51EF64Rcbqx5fwB5-Is09Or6rCvCMLHmT3OswAObo,7821
+unique_toolkit-2026.46.0.dev0.dist-info/RECORD,,