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

unique_toolkit/agentic/loop_runner/_responses_iteration_handler_utils.py

@@ -1,6 +1,9 @@
 import logging
+from collections.abc import Sequence
 from typing import Unpack
 
+from openai.types.responses import ToolParam, response_create_params
+
 from unique_toolkit.agentic.loop_runner._responses_stream_handler_utils import (
     responses_stream_response,
 )
@@ -9,12 +12,44 @@ from unique_toolkit.agentic.loop_runner.base import (
 )
 from unique_toolkit.language_model.schemas import (
     LanguageModelTokenUsage,
+    LanguageModelToolDescription,
     ResponsesLanguageModelStreamResponse,
 )
 
 _LOGGER = logging.getLogger(__name__)
 
 
+def _is_forced_function_choice(
+    tool_choice: response_create_params.ToolChoice,
+) -> bool:
+    return isinstance(tool_choice, dict) and tool_choice.get("type") == "function"
+
+
+def _is_hosted_tool(tool: LanguageModelToolDescription | ToolParam) -> bool:
+    """Return whether OpenAI executes the tool."""
+    if isinstance(tool, LanguageModelToolDescription):
+        return False
+    return tool.get("type") != "function"
+
+
+def _tools_for_forced_choice(
+    tools: Sequence[LanguageModelToolDescription | ToolParam] | None,
+    tool_choice: response_create_params.ToolChoice,
+) -> list[LanguageModelToolDescription | ToolParam] | None:
+    """Drop hosted tools from requests that force a function tool."""
+    if tools is None or not _is_forced_function_choice(tool_choice):
+        return list(tools) if tools is not None else None
+
+    remaining = [tool for tool in tools if not _is_hosted_tool(tool)]
+    dropped = len(tools) - len(remaining)
+    if dropped > 0:
+        _LOGGER.info(
+            "Dropped %d hosted tool(s) from the forced function request.",
+            dropped,
+        )
+    return remaining
+
+
 async def handle_responses_last_iteration(
     **kwargs: Unpack[_ResponsesLoopIterationRunnerKwargs],
 ) -> ResponsesLanguageModelStreamResponse:
@@ -47,9 +82,16 @@ async def handle_responses_forced_tools_iteration(
     responses: list[ResponsesLanguageModelStreamResponse] = []
 
     for opt in tool_choices:
-        responses.append(
-            await responses_stream_response(loop_runner_kwargs=kwargs, tool_choice=opt)
-        )
+        tools = _tools_for_forced_choice(kwargs.get("tools"), opt)
+        if tools is None:
+            response = await responses_stream_response(
+                loop_runner_kwargs=kwargs, tool_choice=opt
+            )
+        else:
+            response = await responses_stream_response(
+                loop_runner_kwargs=kwargs, tool_choice=opt, tools=tools
+            )
+        responses.append(response)
 
     # Merge responses and refs:
     tool_calls = []

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

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

@@ -120,7 +120,7 @@ unique_toolkit/agentic/history_manager/settings.py,sha256=Lrx3a8yXeU0MXrWlV6RZSH
 unique_toolkit/agentic/history_manager/utils.py,sha256=DiBip8xtah3I9-G-_Ul3OjlYKunrXR-5RJtzREHWV_0,6579
 unique_toolkit/agentic/loop_runner/__init__.py,sha256=bhwOIloPn2MZyCw60FkHy330twCCRovg3bgye9OjZRE,1496
 unique_toolkit/agentic/loop_runner/_iteration_handler_utils.py,sha256=y3tjTAtpkyL_Y7279_3v0fGaq9FUYN3UquwdIB3ZhgI,3831
-unique_toolkit/agentic/loop_runner/_responses_iteration_handler_utils.py,sha256=DuQRsToV0KLs7YcFjJVnEmrryhlX5q4isztx2jUxZcI,2064
+unique_toolkit/agentic/loop_runner/_responses_iteration_handler_utils.py,sha256=SbfV5KCZ-3QiMRYjaDDUnY0JuBtfBCSBofF6sxTOugI,3615
 unique_toolkit/agentic/loop_runner/_responses_stream_handler_utils.py,sha256=6HXpSPGUp096a175-NvXqU2Gw7e8Msjo60CP4wGFOoM,2340
 unique_toolkit/agentic/loop_runner/_stream_handler_utils.py,sha256=ERk-qS8MG8FM2F2AEzaub_XsuZQD1tsD8-li692LKp0,1708
 unique_toolkit/agentic/loop_runner/base.py,sha256=OddFwxT7zOOLd2MlAef63BNhzH-NNET8RAsfaMcsHQE,3307
@@ -426,6 +426,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.42.0.dev6.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.42.0.dev6.dist-info/METADATA,sha256=9WKUOCtvXNQS7Svf4qdFEfym23otCvkrLHRhQscwag0,7821
-unique_toolkit-2026.42.0.dev6.dist-info/RECORD,,
+unique_toolkit-2026.42.0.dev7.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.42.0.dev7.dist-info/METADATA,sha256=lerw6mdP9Cp9f4mxf3nMFsJEZg9R_iak69i2o-uZcwU,7821
+unique_toolkit-2026.42.0.dev7.dist-info/RECORD,,