unique_toolkit 2026.36.0.dev6__py3-none-any.whl → 2026.36.0.dev7__py3-none-any.whl

unique_toolkit/experimental/integrations/openai/streaming/event_routing/_model_params.py

@@ -0,0 +1,133 @@
+"""Apply LanguageModelInfo constraints to experimental streaming requests."""
+
+from __future__ import annotations
+
+import logging
+from collections.abc import Mapping, Sequence
+from typing import Any
+
+from openai.types.shared_params import Reasoning
+
+from unique_toolkit.chat.responses_api import _attempt_extract_reasoning_from_options
+from unique_toolkit.language_model.infos import LanguageModelInfo, LanguageModelName
+from unique_toolkit.language_model.schemas import reasoning_effort_to_openai
+
+_LOGGER = logging.getLogger(__name__)
+
+_REASONING_OPTION_KEYS = ("reasoning", "reasoning_effort", "reasoningEffort")
+
+
+def prepare_chat_completions_model_params(
+    model_name: LanguageModelName | str,
+    temperature: float,
+    other_options: dict[str, Any] | None,
+) -> dict[str, Any]:
+    """Prepare Chat Completions create() options including clamped temperature.
+
+    Copies ``other_options`` so the caller dict is not mutated, applies
+    ``LanguageModelInfo`` temperature/reasoning rules, and writes the clamped
+    temperature onto the returned options dict.
+
+    Args:
+        model_name (LanguageModelName | str): Model identifier from the caller.
+        temperature (float): Requested sampling temperature.
+        other_options (dict[str, Any] | None): Extra OpenAI create() kwargs.
+
+    Returns:
+        dict[str, Any]: Sanitized create() options including ``temperature``.
+    """
+    options = dict(other_options) if other_options else {}
+    requested_effort = options.get("reasoning_effort")
+    temperature, resolved_effort = LanguageModelInfo.from_name(
+        model_name
+    ).resolve_temp_and_reasoning(temperature, requested_effort)
+    if resolved_effort is not None:
+        options["reasoning_effort"] = resolved_effort
+    else:
+        options.pop("reasoning_effort", None)
+    options["temperature"] = temperature
+    return options
+
+
+def prepare_responses_model_params(
+    model_name: LanguageModelName | str,
+    temperature: float,
+    reasoning: Reasoning | None,
+    other_options: dict[str, Any] | None,
+    tools: Sequence[Mapping[str, Any]] | None = None,
+) -> dict[str, Any]:
+    """Prepare Responses API create() options including clamped temperature.
+
+    Extracts reasoning from ``other_options`` when the explicit argument is
+    missing (UniqueAI passes ``reasoning`` that way), applies
+    ``LanguageModelInfo`` bounds, and bumps ``minimal`` effort to ``low`` when
+    a code-interpreter tool is present. Clamped temperature and resolved
+    reasoning are written onto the returned options dict.
+
+    Args:
+        model_name (LanguageModelName | str): Model identifier from the caller.
+        temperature (float): Requested sampling temperature.
+        reasoning (Reasoning | None): Explicit Responses ``reasoning`` payload.
+        other_options (dict[str, Any] | None): Extra OpenAI create() kwargs.
+        tools (Sequence[Mapping[str, Any]] | None): Converted Responses tools.
+
+    Returns:
+        dict[str, Any]: Sanitized create() options including ``temperature``.
+    """
+    options = dict(other_options) if other_options else {}
+    if reasoning is None:
+        reasoning = _attempt_extract_reasoning_from_options(options)
+    for key in _REASONING_OPTION_KEYS:
+        options.pop(key, None)
+
+    requested_effort = reasoning.get("effort") if reasoning is not None else None
+    temperature, resolved_effort = LanguageModelInfo.from_name(
+        model_name
+    ).resolve_temp_and_reasoning(temperature, requested_effort)
+    reasoning = _bump_code_interpreter_effort(
+        _apply_resolved_reasoning(reasoning, resolved_effort),
+        tools,
+    )
+    options["temperature"] = temperature
+    if reasoning is not None:
+        options["reasoning"] = reasoning
+    return options
+
+
+def _apply_resolved_reasoning(
+    reasoning: Reasoning | None,
+    resolved_effort: str | None,
+) -> Reasoning | None:
+    """Write the resolved effort into a new Responses ``reasoning`` payload."""
+    if resolved_effort is not None:
+        reasoning = Reasoning(**(reasoning or {}))
+        reasoning["effort"] = reasoning_effort_to_openai(resolved_effort)
+        return reasoning
+    if reasoning is None or "effort" not in reasoning:
+        return reasoning
+    reasoning = Reasoning(**reasoning)
+    del reasoning["effort"]
+    if not reasoning:
+        return None
+    return reasoning
+
+
+def _bump_code_interpreter_effort(
+    reasoning: Reasoning | None,
+    tools: Sequence[Mapping[str, Any]] | None,
+) -> Reasoning | None:
+    """Code interpreter cannot run at ``minimal`` effort; bump to ``low``."""
+    if (
+        reasoning is None
+        or tools is None
+        or not any(tool["type"] == "code_interpreter" for tool in tools)
+        or reasoning.get("effort") != "minimal"
+    ):
+        return reasoning
+
+    _LOGGER.warning(
+        "Code interpreter cannot be used with `minimal` effort. Switching to `low`."
+    )
+    reasoning = Reasoning(**reasoning)
+    reasoning["effort"] = "low"
+    return reasoning

unique_toolkit/experimental/integrations/openai/streaming/event_routing/chat_completions/complete_with_references.py

@@ -44,6 +44,7 @@ from unique_toolkit.language_model.schemas import (
 from unique_toolkit.protocols.support import SupportCompleteWithReferences
 
 from .._async_bridge import run_async_from_sync
+from .._model_params import prepare_chat_completions_model_params
 from ..events import (
     StreamEnded,
     StreamEventBus,
@@ -378,6 +379,10 @@ class ChatCompletionsCompleteWithReferences(SupportCompleteWithReferences):
         message_id = chat.last_assistant_message_id
         chat_id = chat.chat_id
 
+        other_options = prepare_chat_completions_model_params(
+            model_name, temperature, other_options
+        )
+
         gpt_messages = (
             messages.to_openai(mode="completions")
             if isinstance(messages, LanguageModelMessages)
@@ -389,9 +394,8 @@ class ChatCompletionsCompleteWithReferences(SupportCompleteWithReferences):
             optional_create_kwargs["tools"] = converted_tools
         if tool_choice is not None:
             optional_create_kwargs["tool_choice"] = tool_choice
-        if other_options:
-            for k, v in other_options.items():
-                optional_create_kwargs.setdefault(k, v)
+        for k, v in other_options.items():
+            optional_create_kwargs.setdefault(k, v)
 
         gpt_request: list[dict[str, Any]] = [
             dict(cast("Mapping[str, Any]", message)) for message in gpt_messages
@@ -444,7 +448,6 @@ class ChatCompletionsCompleteWithReferences(SupportCompleteWithReferences):
                     messages=gpt_messages,
                     stream=True,
                     stream_options=stream_options,
-                    temperature=temperature,
                     **optional_create_kwargs,
                 )
 

unique_toolkit/experimental/integrations/openai/streaming/event_routing/responses/complete_with_references.py

@@ -39,6 +39,7 @@ from unique_toolkit.language_model.schemas import (
 from unique_toolkit.protocols.support import ResponsesSupportCompleteWithReferences
 
 from .._async_bridge import run_async_from_sync
+from .._model_params import prepare_responses_model_params
 from ..events import (
     ActivityProgress,
     StreamEnded,
@@ -512,8 +513,16 @@ class ResponsesCompleteWithReferences(ResponsesSupportCompleteWithReferences):
             )
 
         converted_messages = input_messages(messages)
-
         converted_tools = _convert_tools(tools)
+
+        other_options = prepare_responses_model_params(
+            model_name,
+            temperature,
+            reasoning,
+            other_options,
+            converted_tools,
+        )
+
         create_kwargs: dict[str, Any] = {}
         if converted_tools:
             create_kwargs["tools"] = converted_tools
@@ -533,16 +542,12 @@ class ResponsesCompleteWithReferences(ResponsesSupportCompleteWithReferences):
             create_kwargs["tool_choice"] = tool_choice
         if top_p is not None:
             create_kwargs["top_p"] = top_p
-        if reasoning is not None:
-            create_kwargs["reasoning"] = reasoning
-        if other_options:
-            for k, v in other_options.items():
-                create_kwargs.setdefault(k, v)
+        for k, v in other_options.items():
+            create_kwargs.setdefault(k, v)
         gpt_request: dict[str, Any] = {
             "model": model,
             "input": converted_messages,
             "stream": True,
-            "temperature": temperature,
             **create_kwargs,
         }
 
@@ -580,7 +585,6 @@ class ResponsesCompleteWithReferences(ResponsesSupportCompleteWithReferences):
                     model=model,
                     input=converted_messages,
                     stream=True,
-                    temperature=temperature,
                     **create_kwargs,
                 )
                 async with stream:

{unique_toolkit-2026.36.0.dev6.dist-info → unique_toolkit-2026.36.0.dev7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.36.0.dev6
+Version: 2026.36.0.dev7
 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-2026.36.0.dev6.dist-info → unique_toolkit-2026.36.0.dev7.dist-info}/RECORD

@@ -331,8 +331,9 @@ unique_toolkit/experimental/integrations/openai/__init__.py,sha256=1vh39WXdUMQXS
 unique_toolkit/experimental/integrations/openai/streaming/__init__.py,sha256=JhsXd_NZaCDY5XAT-qY0kumpmpYJ-Nws0ZWo6MGj2sI,47
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/__init__.py,sha256=-_SsMgmlySX1ae5XSKcoNjiJ1FRgepooJW49NaEhCHs,3217
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/_async_bridge.py,sha256=j7fK4Q555pXUpSPXdBllKk__b-susz_Jhmh5NqdM_EA,1377
+unique_toolkit/experimental/integrations/openai/streaming/event_routing/_model_params.py,sha256=K-aBUzLbIVOJprafZl6UD-gsJZW2acbY0O8NRFQxocY,4981
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/chat_completions/__init__.py,sha256=sv9Ldr0GeAwc0I7tyM4NbOf_sYJl602S0T2Z0oHb1nE,578
-unique_toolkit/experimental/integrations/openai/streaming/event_routing/chat_completions/complete_with_references.py,sha256=XDQtVQ2vpX-_9bnBCAVP5nY_vZqQZA18f5yO_l2hlUI,21172
+unique_toolkit/experimental/integrations/openai/streaming/event_routing/chat_completions/complete_with_references.py,sha256=YvYIGaHjqeUL-C6MJMKNhTKTinvbZ09mBHvLTRaq4sA,21284
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/chat_completions/stream_event_router.py,sha256=XNByWclJXeYOeeDBGkUBVY5r1aNckg4B5xf27UybPYc,6199
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/chat_completions/text_event_handler.py,sha256=1g_EVQ4IpWeorZ3cm9YuN4NMdg19Zv64Ts0ABUzWzR4,5664
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/chat_completions/tool_call_event_handler.py,sha256=Dxg8odtkSO8ErqLLhh0OEObsB-mSUVJ98eE2huTJUUg,2965
@@ -342,7 +343,7 @@ unique_toolkit/experimental/integrations/openai/streaming/event_routing/protocol
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/protocols/responses.py,sha256=bM5qSZjJOTWQyPDUPyMasMwZ-Wu7LQKR64OcKVpUY5E,3294
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/responses/__init__.py,sha256=-8Ey_srOpxX4B_HJhh27h4nZaNC6Y4mk5o8wSLelPwA,847
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/responses/code_interpreter_event_handler.py,sha256=3WFT-zuVCrTP01riJY9uuu5U8-tmFUD-XYTZYDrGx4w,6013
-unique_toolkit/experimental/integrations/openai/streaming/event_routing/responses/complete_with_references.py,sha256=rVcIY7CLPJUVWJrtZDZR5RLd3MnuXbsyL-FdpXoVcjQ,26126
+unique_toolkit/experimental/integrations/openai/streaming/event_routing/responses/complete_with_references.py,sha256=LAIt3xqroByvTw2DQrFs38Tods3ZjBnpOwzliAg17eM,26176
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/responses/completed_event_handler.py,sha256=BqsJC14KOk_cwelXqyMWlBrFwj5fugkr2gzvwYb3aYM,1360
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/responses/stream_event_router.py,sha256=hvQHDPmfYD-ZDRR5W079TFYEgQ-vhIGMNRyagiG6RzM,8280
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/responses/text_delta_event_handler.py,sha256=JB-7W89K4pT6K8A9Z_f_Pow5p-EIGwLoLUSb9BI3qW4,3634
@@ -421,6 +422,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.36.0.dev6.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.36.0.dev6.dist-info/METADATA,sha256=pZseZiZpalFBDEY17j10JOFZrZ51yy8Y9ByMQ5LEdAw,7821
-unique_toolkit-2026.36.0.dev6.dist-info/RECORD,,
+unique_toolkit-2026.36.0.dev7.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.36.0.dev7.dist-info/METADATA,sha256=Wqg6T1X6FZR7xev8-poWDfWwRIcdTFIH6hwdfRRB9XE,7821
+unique_toolkit-2026.36.0.dev7.dist-info/RECORD,,