unique_toolkit 1.70.0__tar.gz → 1.70.1__tar.gz

{unique_toolkit-1.70.0 → unique_toolkit-1.70.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.70.0
+Version: 1.70.1
 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>

{unique_toolkit-1.70.0 → unique_toolkit-1.70.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.70.0"
+version = "1.70.1"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"

{unique_toolkit-1.70.0 → unique_toolkit-1.70.1}/unique_toolkit/chat/responses_api.py

@@ -33,7 +33,6 @@ from unique_toolkit.language_model.constants import (
 )
 from unique_toolkit.language_model.functions import (
     SearchContext,
-    _clamp_temperature,
     _to_search_context,
 )
 from unique_toolkit.language_model.infos import (
@@ -51,6 +50,7 @@ from unique_toolkit.language_model.schemas import (
     LanguageModelToolMessage,
     LanguageModelUserMessage,
     ResponsesLanguageModelStreamResponse,
+    reasoning_effort_to_openai,
 )
 
 logger = logging.getLogger(__name__)
@@ -161,16 +161,19 @@ def _prepare_responses_params_util(
 
     if isinstance(model_name, LanguageModelName):
         model_info = LanguageModelInfo.from_name(model_name)
-
-        if model_info.temperature_bounds is not None and temperature is not None:
-            temperature = _clamp_temperature(temperature, model_info.temperature_bounds)
-
-        if (
-            reasoning is None
-            and model_info.default_options is not None
-            and "reasoning_effort" in model_info.default_options
-        ):
-            reasoning = Reasoning(effort=model_info.default_options["reasoning_effort"])
+        requested_effort = reasoning.get("effort") if reasoning is not None else None
+        temperature, resolved_effort = model_info.resolve_temp_and_reasoning(
+            temperature,
+            requested_effort,
+        )
+        if resolved_effort is not None:
+            reasoning = Reasoning(**(reasoning or {}))
+            reasoning["effort"] = reasoning_effort_to_openai(resolved_effort)
+        else:
+            if reasoning is not None and "effort" in reasoning:
+                del reasoning["effort"]
+                if not reasoning:
+                    reasoning = None
 
         if (
             reasoning is not None

{unique_toolkit-1.70.0 → unique_toolkit-1.70.1}/unique_toolkit/language_model/functions.py

@@ -19,7 +19,6 @@ from unique_toolkit.content.schemas import ContentChunk, ContentReference
 from unique_toolkit.language_model.infos import (
     LanguageModelInfo,
     LanguageModelName,
-    TemperatureBounds,
 )
 from unique_toolkit.language_model.reference import (
     add_references_to_message,
@@ -344,14 +343,6 @@ def __camelize_keys(data):
     return data
 
 
-def _clamp_temperature(
-    temperature: float, temperature_bounds: TemperatureBounds
-) -> float:
-    temperature = max(temperature_bounds.min_temperature, temperature)
-    temperature = min(temperature_bounds.max_temperature, temperature)
-    return round(temperature, 2)
-
-
 def _prepare_other_options(
     other_options: dict | None,
     default_options: dict,
@@ -417,14 +408,16 @@ def _prepare_all_completions_params_util(
         )
         messages_dict = __camelize_keys(messages.copy())
 
-    if (
-        model_info is not None
-        and model_info.temperature_bounds is not None
-        and "temperature" in options
-    ):
-        options["temperature"] = _clamp_temperature(
-            temperature, model_info.temperature_bounds
+    if model_info is not None:
+        reasoning_effort = options.get("reasoning_effort")
+        resolved_temp, resolved_effort = model_info.resolve_temp_and_reasoning(
+            temperature, reasoning_effort=reasoning_effort
         )
+        options["temperature"] = resolved_temp
+        if resolved_effort is not None:
+            options["reasoning_effort"] = resolved_effort
+        elif reasoning_effort is not None:
+            options.pop("reasoning_effort", None)
 
     integrated_messages = cast(
         "list[unique_sdk.Integrated.ChatCompletionRequestMessage]",