unique_toolkit 1.38.2__py3-none-any.whl → 1.38.4__py3-none-any.whl

unique_toolkit/agentic/evaluation/evaluation_manager.py

@@ -2,15 +2,15 @@ import asyncio
 from abc import ABC
 from logging import Logger
 
+from unique_toolkit._common.execution import (
+    Result,
+    SafeTaskExecutor,
+)
 from unique_toolkit.agentic.evaluation.schemas import (
     EvaluationAssessmentMessage,
     EvaluationMetricName,
     EvaluationMetricResult,
 )
-from unique_toolkit.agentic.tools.utils.execution.execution import (
-    Result,
-    SafeTaskExecutor,
-)
 from unique_toolkit.chat.schemas import (
     ChatMessageAssessmentStatus,
     ChatMessageAssessmentType,

unique_toolkit/agentic/loop_runner/__init__.py

@@ -1,25 +1,35 @@
+from unique_toolkit.agentic.loop_runner._iteration_handler_utils import (
+    handle_forced_tools_iteration,
+    handle_last_iteration,
+    handle_normal_iteration,
+)
 from unique_toolkit.agentic.loop_runner.base import LoopIterationRunner
 from unique_toolkit.agentic.loop_runner.middleware import (
-    QWEN_FORCED_TOOL_CALL_PROMPT_INSTRUCTION,
     PlanningConfig,
     PlanningMiddleware,
     PlanningSchemaConfig,
-    QwenForcedToolCallMiddleware,
-    is_qwen_model,
 )
 from unique_toolkit.agentic.loop_runner.runners import (
+    QWEN_FORCED_TOOL_CALL_INSTRUCTION,
+    QWEN_LAST_ITERATION_INSTRUCTION,
     BasicLoopIterationRunner,
     BasicLoopIterationRunnerConfig,
+    QwenLoopIterationRunner,
+    is_qwen_model,
 )
 
 __all__ = [
     "LoopIterationRunner",
-    "QWEN_FORCED_TOOL_CALL_PROMPT_INSTRUCTION",
     "PlanningConfig",
     "PlanningMiddleware",
     "PlanningSchemaConfig",
-    "QwenForcedToolCallMiddleware",
+    "QwenLoopIterationRunner",
     "is_qwen_model",
     "BasicLoopIterationRunnerConfig",
     "BasicLoopIterationRunner",
+    "handle_forced_tools_iteration",
+    "handle_last_iteration",
+    "handle_normal_iteration",
+    "QWEN_FORCED_TOOL_CALL_INSTRUCTION",
+    "QWEN_LAST_ITERATION_INSTRUCTION",
 ]

unique_toolkit/agentic/loop_runner/_iteration_handler_utils.py

@@ -0,0 +1,95 @@
+import logging
+from typing import Protocol, Unpack, cast
+
+from unique_toolkit.agentic.loop_runner._stream_handler_utils import stream_response
+from unique_toolkit.agentic.loop_runner.base import (
+    _LoopIterationRunnerKwargs,
+)
+from unique_toolkit.chat.functions import LanguageModelStreamResponse
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class PrepareForcedToolIterationKwargs(Protocol):
+    def __call__(
+        self,
+        func_name: str | None,
+        per_choice_kwargs: _LoopIterationRunnerKwargs,
+    ) -> _LoopIterationRunnerKwargs: ...
+
+
+async def handle_last_iteration(
+    **kwargs: Unpack[_LoopIterationRunnerKwargs],
+) -> LanguageModelStreamResponse:
+    _LOGGER.info("Reached last iteration, removing tools and producing final response")
+
+    return await stream_response(
+        loop_runner_kwargs=kwargs,
+        tools=None,
+    )
+
+
+async def handle_normal_iteration(
+    **kwargs: Unpack[_LoopIterationRunnerKwargs],
+) -> LanguageModelStreamResponse:
+    _LOGGER.info("Running loop iteration %d", kwargs["iteration_index"])
+
+    return await stream_response(loop_runner_kwargs=kwargs)
+
+
+async def handle_forced_tools_iteration(
+    **kwargs: Unpack[_LoopIterationRunnerKwargs],
+) -> LanguageModelStreamResponse:
+    return await run_forced_tools_iteration(loop_runner_kwargs=kwargs)
+
+
+async def run_forced_tools_iteration(
+    *,
+    loop_runner_kwargs: _LoopIterationRunnerKwargs,
+    prepare_loop_runner_kwargs: PrepareForcedToolIterationKwargs | None = None,
+) -> LanguageModelStreamResponse:
+    """
+    Execute a "forced tools" iteration by running one stream_response per tool choice,
+    then merging tool calls and references into a single response.
+
+    Some models (e.g. Qwen) need per-tool-choice message rewriting; this can be done
+    via prepare_loop_runner_kwargs (called once per tool choice, on a copy of kwargs).
+    """
+    assert "tool_choices" in loop_runner_kwargs
+
+    tool_choices = loop_runner_kwargs["tool_choices"]
+    assert len(tool_choices) > 0, (
+        "run_forced_tools_iteration requires at least one tool choice"
+    )
+    _LOGGER.info("Forcing tools calls: %s", tool_choices)
+
+    responses: list[LanguageModelStreamResponse] = []
+
+    available_tools = {t.name: t for t in loop_runner_kwargs.get("tools") or []}
+
+    for opt in tool_choices:
+        func_name = opt.get("function", {}).get("name")
+
+        per_choice_kwargs = cast(_LoopIterationRunnerKwargs, dict(loop_runner_kwargs))
+        if prepare_loop_runner_kwargs:
+            per_choice_kwargs = prepare_loop_runner_kwargs(func_name, per_choice_kwargs)
+
+        limited_tool = available_tools.get(func_name) if func_name else None
+        stream_kwargs = {"loop_runner_kwargs": per_choice_kwargs, "tool_choice": opt}
+        if limited_tool:
+            stream_kwargs["tools"] = [limited_tool]
+        responses.append(await stream_response(**stream_kwargs))
+
+    # Merge responses and refs:
+    tool_calls = []
+    references = []
+    for r in responses:
+        if r.tool_calls:
+            tool_calls.extend(r.tool_calls)
+        references.extend(r.message.references)
+
+    response = responses[0]
+    response.tool_calls = tool_calls if len(tool_calls) > 0 else None
+    response.message.references = references
+
+    return response

unique_toolkit/agentic/loop_runner/middleware/__init__.py

@@ -3,17 +3,9 @@ from unique_toolkit.agentic.loop_runner.middleware.planning import (
     PlanningMiddleware,
     PlanningSchemaConfig,
 )
-from unique_toolkit.agentic.loop_runner.middleware.qwen_forced_tool_call import (
-    QWEN_FORCED_TOOL_CALL_PROMPT_INSTRUCTION,
-    QwenForcedToolCallMiddleware,
-    is_qwen_model,
-)
 
 __all__ = [
     "PlanningConfig",
     "PlanningMiddleware",
     "PlanningSchemaConfig",
-    "QWEN_FORCED_TOOL_CALL_PROMPT_INSTRUCTION",
-    "QwenForcedToolCallMiddleware",
-    "is_qwen_model",
 ]

unique_toolkit/agentic/loop_runner/middleware/planning/planning.py

@@ -5,6 +5,7 @@ from typing import Unpack
 from pydantic import BaseModel, Field
 
 from unique_toolkit import LanguageModelService
+from unique_toolkit._common.execution import failsafe_async
 from unique_toolkit._common.pydantic_helpers import get_configuration_dict
 from unique_toolkit.agentic.history_manager.history_manager import HistoryManager
 from unique_toolkit.agentic.loop_runner.base import (
@@ -15,7 +16,6 @@ from unique_toolkit.agentic.loop_runner.middleware.planning.schema import (
     PlanningSchemaConfig,
     get_planning_schema,
 )
-from unique_toolkit.agentic.tools.utils import failsafe_async
 from unique_toolkit.chat.service import LanguageModelStreamResponse
 from unique_toolkit.language_model import (
     LanguageModelAssistantMessage,

unique_toolkit/agentic/loop_runner/runners/__init__.py

@@ -2,5 +2,18 @@ from unique_toolkit.agentic.loop_runner.runners.basic import (
     BasicLoopIterationRunner,
     BasicLoopIterationRunnerConfig,
 )
+from unique_toolkit.agentic.loop_runner.runners.qwen import (
+    QWEN_FORCED_TOOL_CALL_INSTRUCTION,
+    QWEN_LAST_ITERATION_INSTRUCTION,
+    QwenLoopIterationRunner,
+    is_qwen_model,
+)
 
-__all__ = ["BasicLoopIterationRunnerConfig", "BasicLoopIterationRunner"]
+__all__ = [
+    "BasicLoopIterationRunnerConfig",
+    "BasicLoopIterationRunner",
+    "QwenLoopIterationRunner",
+    "QWEN_FORCED_TOOL_CALL_INSTRUCTION",
+    "QWEN_LAST_ITERATION_INSTRUCTION",
+    "is_qwen_model",
+]

unique_toolkit/agentic/loop_runner/runners/basic.py

@@ -4,7 +4,11 @@ from typing import Unpack, override
 from pydantic import BaseModel
 
 from unique_toolkit._common.pydantic_helpers import get_configuration_dict
-from unique_toolkit.agentic.loop_runner._stream_handler_utils import stream_response
+from unique_toolkit.agentic.loop_runner._iteration_handler_utils import (
+    handle_forced_tools_iteration,
+    handle_last_iteration,
+    handle_normal_iteration,
+)
 from unique_toolkit.agentic.loop_runner.base import (
     LoopIterationRunner,
     _LoopIterationRunnerKwargs,
@@ -26,60 +30,6 @@ class BasicLoopIterationRunner(LoopIterationRunner):
     def __init__(self, config: BasicLoopIterationRunnerConfig) -> None:
         self._config = config
 
-    async def _handle_last_iteration(
-        self, **kwargs: Unpack[_LoopIterationRunnerKwargs]
-    ) -> LanguageModelStreamResponse:
-        _LOGGER.info(
-            "Reached last iteration, removing tools and producing final response"
-        )
-
-        return await stream_response(
-            loop_runner_kwargs=kwargs,
-            tools=None,
-        )
-
-    async def _handle_normal_iteration(
-        self, **kwargs: Unpack[_LoopIterationRunnerKwargs]
-    ) -> LanguageModelStreamResponse:
-        _LOGGER.info("Running loop iteration %d", kwargs["iteration_index"])
-
-        return await stream_response(loop_runner_kwargs=kwargs)
-
-    async def _handle_forced_tools_iteration(
-        self,
-        **kwargs: Unpack[_LoopIterationRunnerKwargs],
-    ) -> LanguageModelStreamResponse:
-        assert "tool_choices" in kwargs
-
-        tool_choices = kwargs["tool_choices"]
-        _LOGGER.info("Forcing tools calls: %s", tool_choices)
-
-        responses: list[LanguageModelStreamResponse] = []
-
-        available_tools = {t.name: t for t in kwargs.get("tools") or []}
-
-        for opt in tool_choices:
-            func_name = opt.get("function", {}).get("name")
-            limited_tool = available_tools.get(func_name) if func_name else None
-            stream_kwargs = {"loop_runner_kwargs": kwargs, "tool_choice": opt}
-            if limited_tool:
-                stream_kwargs["tools"] = [limited_tool]
-            responses.append(await stream_response(**stream_kwargs))
-
-        # Merge responses and refs:
-        tool_calls = []
-        references = []
-        for r in responses:
-            if r.tool_calls:
-                tool_calls.extend(r.tool_calls)
-            references.extend(r.message.references)
-
-        response = responses[0]
-        response.tool_calls = tool_calls if len(tool_calls) > 0 else None
-        response.message.references = references
-
-        return response
-
     @override
     async def __call__(
         self,
@@ -89,8 +39,8 @@ class BasicLoopIterationRunner(LoopIterationRunner):
         iteration_index = kwargs["iteration_index"]
 
         if len(tool_choices) > 0 and iteration_index == 0:
-            return await self._handle_forced_tools_iteration(**kwargs)
+            return await handle_forced_tools_iteration(**kwargs)
         elif iteration_index == self._config.max_loop_iterations - 1:
-            return await self._handle_last_iteration(**kwargs)
+            return await handle_last_iteration(**kwargs)
         else:
-            return await self._handle_normal_iteration(**kwargs)
+            return await handle_normal_iteration(**kwargs)

unique_toolkit/agentic/loop_runner/runners/qwen/__init__.py

@@ -0,0 +1,15 @@
+from unique_toolkit.agentic.loop_runner.runners.qwen.helpers import (
+    is_qwen_model,
+)
+from unique_toolkit.agentic.loop_runner.runners.qwen.qwen_runner import (
+    QWEN_FORCED_TOOL_CALL_INSTRUCTION,
+    QWEN_LAST_ITERATION_INSTRUCTION,
+    QwenLoopIterationRunner,
+)
+
+__all__ = [
+    "QwenLoopIterationRunner",
+    "is_qwen_model",
+    "QWEN_FORCED_TOOL_CALL_INSTRUCTION",
+    "QWEN_LAST_ITERATION_INSTRUCTION",
+]

unique_toolkit/agentic/loop_runner/{middleware/qwen_forced_tool_call → runners/qwen}/helpers.py

@@ -1,5 +1,6 @@
 from unique_toolkit.language_model.infos import LanguageModelInfo
 from unique_toolkit.language_model.schemas import (
+    LanguageModelAssistantMessage,
     LanguageModelMessageRole,
     LanguageModelMessages,
 )
@@ -31,3 +32,17 @@ def append_qwen_forced_tool_call_instruction(
             )
             break
     return LanguageModelMessages(root=messages_list)
+
+
+def append_qwen_last_iteration_assistant_message(
+    *,
+    messages: LanguageModelMessages,
+    last_iteration_instruction: str,
+) -> LanguageModelMessages:
+    """Append an assistant message at the end to indicate no further tool calls are allowed."""
+    messages_list = list(messages)
+    assistant_message = LanguageModelAssistantMessage(
+        content=last_iteration_instruction,
+    )
+    messages_list.append(assistant_message)
+    return LanguageModelMessages(root=messages_list)

unique_toolkit/agentic/loop_runner/runners/qwen/qwen_runner.py

@@ -0,0 +1,118 @@
+import logging
+from typing import Unpack
+
+from unique_toolkit.agentic.loop_runner._iteration_handler_utils import (
+    handle_last_iteration,
+    handle_normal_iteration,
+    run_forced_tools_iteration,
+)
+from unique_toolkit.agentic.loop_runner.base import (
+    LoopIterationRunner,
+    _LoopIterationRunnerKwargs,
+)
+from unique_toolkit.agentic.loop_runner.runners.qwen.helpers import (
+    append_qwen_forced_tool_call_instruction,
+    append_qwen_last_iteration_assistant_message,
+)
+from unique_toolkit.chat.service import ChatService, LanguageModelStreamResponse
+
+_LOGGER = logging.getLogger(__name__)
+
+QWEN_FORCED_TOOL_CALL_INSTRUCTION = (
+    "**Tool Call Instruction:** \nYou MUST call the tool {TOOL_NAME}. "
+    "You must start the response with <tool_call>. "
+    "Do NOT provide natural language explanations, summaries, or any text outside the <tool_call> block."
+)
+
+QWEN_LAST_ITERATION_INSTRUCTION = "The maximum number of loop iteration have been reached. Not further tool calls are allowed. Based on the found information, an answer should be generated"
+
+
+class QwenLoopIterationRunner(LoopIterationRunner):
+    def __init__(
+        self,
+        *,
+        qwen_forced_tool_call_instruction: str,
+        qwen_last_iteration_instruction: str,
+        max_loop_iterations: int,
+        chat_service: ChatService,
+    ) -> None:
+        self._qwen_forced_tool_call_instruction = qwen_forced_tool_call_instruction
+        self._qwen_last_iteration_instruction = qwen_last_iteration_instruction
+        self._max_loop_iterations = max_loop_iterations
+        self._chat_service = chat_service
+
+    async def __call__(
+        self, **kwargs: Unpack[_LoopIterationRunnerKwargs]
+    ) -> LanguageModelStreamResponse:
+        tool_choices = kwargs.get("tool_choices") or []
+        iteration_index = kwargs["iteration_index"]
+
+        if len(tool_choices) > 0 and iteration_index == 0:
+            return await self._qwen_handle_forced_tools_iteration(**kwargs)
+        elif iteration_index == self._max_loop_iterations - 1:
+            return await self._qwen_handle_last_iteration(**kwargs)
+        else:
+            return await self._qwen_handle_normal_iteration(**kwargs)
+
+    async def _qwen_handle_forced_tools_iteration(
+        self, **kwargs: Unpack[_LoopIterationRunnerKwargs]
+    ) -> LanguageModelStreamResponse:
+        # For Qwen models, append tool call instruction to the last user message. These models ignore the parameter tool_choice.
+        # As the message has to be modified for each tool call instruction, the function from the basic runner cant be used.
+        original_messages = kwargs["messages"].model_copy(deep=True)
+
+        def _prepare(
+            func_name: str | None,
+            per_choice_kwargs: _LoopIterationRunnerKwargs,
+        ) -> _LoopIterationRunnerKwargs:
+            prompt_instruction = self._qwen_forced_tool_call_instruction.format(
+                TOOL_NAME=func_name or ""
+            )
+            per_choice_kwargs["messages"] = append_qwen_forced_tool_call_instruction(
+                messages=original_messages,
+                forced_tool_call_instruction=prompt_instruction,
+            )
+            return per_choice_kwargs
+
+        response = await run_forced_tools_iteration(
+            loop_runner_kwargs=kwargs,
+            prepare_loop_runner_kwargs=_prepare,
+        )
+        return self._process_response(response)
+
+    async def _qwen_handle_last_iteration(
+        self, **kwargs: Unpack[_LoopIterationRunnerKwargs]
+    ) -> LanguageModelStreamResponse:
+        # For Qwen models, append an assistant message with instructions to not call any tool in this iteration.
+        _LOGGER.info(
+            "Reached last iteration, removing tools. Appending assistant message with instructions to not call any tool in this iteration."
+        )
+        kwargs["messages"] = append_qwen_last_iteration_assistant_message(
+            messages=kwargs["messages"],
+            last_iteration_instruction=self._qwen_last_iteration_instruction,
+        )
+
+        response = await handle_last_iteration(
+            **kwargs,
+        )
+
+        return self._process_response(response)
+
+    async def _qwen_handle_normal_iteration(
+        self, **kwargs: Unpack[_LoopIterationRunnerKwargs]
+    ) -> LanguageModelStreamResponse:
+        response = await handle_normal_iteration(**kwargs)
+        return self._process_response(response)
+
+    def _process_response(
+        self, response: LanguageModelStreamResponse
+    ) -> LanguageModelStreamResponse:
+        # Check if content of response is </tool_call>
+        if "</tool_call>" == response.message.text.strip():
+            _LOGGER.warning(
+                "Response contains only <tool_call>. This is not allowed. Returning empty response."
+            )
+            self._chat_service.modify_assistant_message(content="")
+            response.message.text = ""
+
+        return response

unique_toolkit/agentic/postprocessor/postprocessor_manager.py

@@ -2,7 +2,7 @@ import asyncio
 from abc import ABC
 from logging import Logger
 
-from unique_toolkit.agentic.tools.utils.execution.execution import SafeTaskExecutor
+from unique_toolkit._common.execution import SafeTaskExecutor
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.language_model.schemas import (
     LanguageModelStreamResponse,

unique_toolkit/agentic/responses_api/postprocessors/generated_files.py

@@ -9,6 +9,7 @@ from openai.types.responses.response_output_text import AnnotationContainerFileC
 from pydantic import BaseModel, Field, RootModel
 
 from unique_toolkit import ChatService
+from unique_toolkit._common.execution import failsafe_async
 from unique_toolkit.agentic.postprocessor.postprocessor_manager import (
     ResponsesApiPostprocessor,
 )
@@ -16,7 +17,6 @@ from unique_toolkit.agentic.short_term_memory_manager.persistent_short_term_memo
     PersistentShortMemoryManager,
 )
 from unique_toolkit.agentic.tools.config import get_configuration_dict
-from unique_toolkit.agentic.tools.utils import failsafe_async
 from unique_toolkit.content.schemas import ContentReference
 from unique_toolkit.content.service import ContentService
 from unique_toolkit.language_model.schemas import ResponsesLanguageModelStreamResponse

unique_toolkit/agentic/short_term_memory_manager/persistent_short_term_memory_manager.py

@@ -5,7 +5,7 @@ from typing import Generic, Type, TypeVar
 
 from pydantic import BaseModel
 
-from unique_toolkit.agentic.tools.utils.execution.execution import SafeTaskExecutor
+from unique_toolkit._common.execution import SafeTaskExecutor
 from unique_toolkit.short_term_memory.schemas import ShortTermMemory
 from unique_toolkit.short_term_memory.service import ShortTermMemoryService
 

unique_toolkit/agentic/tools/a2a/evaluation/evaluator.py

@@ -5,6 +5,7 @@ import unique_sdk
 from jinja2 import Template
 from pydantic import BaseModel
 
+from unique_toolkit._common.execution import failsafe
 from unique_toolkit.agentic.evaluation.evaluation_manager import Evaluation
 from unique_toolkit.agentic.evaluation.schemas import (
     EvaluationAssessmentMessage,
@@ -24,7 +25,6 @@ from unique_toolkit.agentic.tools.a2a.response_watcher import (
     SubAgentResponse,
     SubAgentResponseWatcher,
 )
-from unique_toolkit.agentic.tools.utils import failsafe
 from unique_toolkit.chat.schemas import (
     ChatMessageAssessmentLabel,
     ChatMessageAssessmentStatus,

unique_toolkit/agentic/tools/tool_manager.py

@@ -11,6 +11,10 @@ from openai.types.responses import (
 )
 from pydantic import BaseModel, Field
 
+from unique_toolkit._common.execution import (
+    Result,
+    SafeTaskExecutor,
+)
 from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricName
 from unique_toolkit.agentic.tools.a2a import A2AManager, SubAgentTool
 from unique_toolkit.agentic.tools.config import ToolBuildConfig
@@ -24,10 +28,6 @@ from unique_toolkit.agentic.tools.openai_builtin.manager import OpenAIBuiltInToo
 from unique_toolkit.agentic.tools.schemas import ToolCallResponse, ToolPrompts
 from unique_toolkit.agentic.tools.tool import Tool
 from unique_toolkit.agentic.tools.tool_progress_reporter import ToolProgressReporter
-from unique_toolkit.agentic.tools.utils.execution.execution import (
-    Result,
-    SafeTaskExecutor,
-)
 from unique_toolkit.app.schemas import ChatEvent
 from unique_toolkit.language_model.schemas import (
     LanguageModelFunction,

unique_toolkit/agentic/tools/utils/__init__.py

@@ -1,19 +0,0 @@
-"""Utilities for tools."""
-
-from unique_toolkit.agentic.tools.utils.execution.execution import (
-    Result,
-    SafeTaskExecutor,
-    failsafe,
-    failsafe_async,
-    safe_execute,
-    safe_execute_async,
-)
-
-__all__ = [
-    "failsafe",
-    "failsafe_async",
-    "safe_execute",
-    "safe_execute_async",
-    "SafeTaskExecutor",
-    "Result",
-]

unique_toolkit/chat/responses_api.py

@@ -14,7 +14,7 @@ from openai.types.responses import (
 from openai.types.shared_params import Metadata, Reasoning
 from pydantic import BaseModel, TypeAdapter, ValidationError
 
-from unique_toolkit.agentic.tools.utils.execution.execution import (
+from unique_toolkit._common.execution import (
     failsafe,
 )
 from unique_toolkit.content.schemas import ContentChunk

{unique_toolkit-1.38.2.dist-info → unique_toolkit-1.38.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.38.2
+Version: 1.38.4
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -121,6 +121,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.38.4] - 2025-12-17
+- Improving handling of tool calls with Qwen models
+
+## [1.38.3] - 2025-12-17
+- Move the failsafe exception to root folder of unique_toolkit from agentic tools
+
 ## [1.38.2] - 2025-12-17
 - Fixing bug that language model infos were not loaded correctly
 

{unique_toolkit-1.38.2.dist-info → unique_toolkit-1.38.4.dist-info}/RECORD

@@ -17,6 +17,7 @@ unique_toolkit/_common/docx_generator/template/Doc Template.docx,sha256=USnCg8h6
 unique_toolkit/_common/endpoint_builder.py,sha256=pEDwgeDzt67qbyaM98u8X7UAy29mQIw9Qufjz2bxgEA,11410
 unique_toolkit/_common/endpoint_requestor.py,sha256=TBLYUVpfdYIPpeKfEVYfH8BrNxnn4yAcSUjfdVDkT78,15443
 unique_toolkit/_common/exception.py,sha256=ho0uBcPeZXU2w15IrSBhO5w7KUgxp1HcKAQrf2uin-w,1243
+unique_toolkit/_common/execution.py,sha256=ocPGGfUwa851207HNTLYiBJ1pNzJp4VhMZ49OPP33gU,8022
 unique_toolkit/_common/experimental/endpoint_builder.py,sha256=pEDwgeDzt67qbyaM98u8X7UAy29mQIw9Qufjz2bxgEA,11410
 unique_toolkit/_common/experimental/endpoint_requestor.py,sha256=YnDr8wASAEjZjLAeBmOWuFn4wUIZslTHBN_aApWeJBA,16079
 unique_toolkit/_common/feature_flags/schema.py,sha256=X32VqH4VMK7bhEfSd8Wbddl8FVs7Gh7ucuIEbmqc4Kw,268
@@ -47,7 +48,7 @@ unique_toolkit/agentic/evaluation/config.py,sha256=zcW7m63Yt5G39hN2If8slBl6Eu3jT
 unique_toolkit/agentic/evaluation/context_relevancy/prompts.py,sha256=EdHFUOB581yVxcOL8482KUv_LzaRjuiem71EF8udYMc,1331
 unique_toolkit/agentic/evaluation/context_relevancy/schema.py,sha256=lZd0TPzH43ifgWWGg3WO6b1AQX8aK2R9y51yH0d1DHM,2919
 unique_toolkit/agentic/evaluation/context_relevancy/service.py,sha256=2NM1_PCP6fXsRm0r-MGrUg5Z3WO00FBCqmiU8f5Kagg,9661
-unique_toolkit/agentic/evaluation/evaluation_manager.py,sha256=IPx4BVUgkjFOP1BGLi0BlB6UujpXlZ0KGuSXDRemQhY,8143
+unique_toolkit/agentic/evaluation/evaluation_manager.py,sha256=wDN_Uuut9kEGek8JY3QeInKpF-ukbvOSKOVd7DHFT3Q,8121
 unique_toolkit/agentic/evaluation/exception.py,sha256=7lcVbCyoN4Md1chNJDFxpUYyWbVrcr9dcc3TxWykJTc,115
 unique_toolkit/agentic/evaluation/hallucination/constants.py,sha256=SoGmoYti2J33tSmmOC1BSF6Pkh8DQvbQAU9xIZFQZRs,2070
 unique_toolkit/agentic/evaluation/hallucination/hallucination_evaluation.py,sha256=x5ta2Fum4fE5ySgIXPKlnbTtmV140z0IazSATd0-REg,4092
@@ -62,27 +63,28 @@ unique_toolkit/agentic/history_manager/history_construction_with_contents.py,sha
 unique_toolkit/agentic/history_manager/history_manager.py,sha256=7V7_173XkAjc8otBACF0G3dbqRs34FSlURbBPrE95Wk,9537
 unique_toolkit/agentic/history_manager/loop_token_reducer.py,sha256=3c-uonDovtanEJUpAO4zlA4-n9MS_Ws_V0Yb6G7hPM0,20172
 unique_toolkit/agentic/history_manager/utils.py,sha256=VIn_UmcR3jHtpux0qp5lQQzczgAm8XYSeQiPo87jC3A,3143
-unique_toolkit/agentic/loop_runner/__init__.py,sha256=Kg-6Zgt--iIYk-biAkvnuwbButxPRPoVlrpppUblm0s,721
+unique_toolkit/agentic/loop_runner/__init__.py,sha256=3Ddm4WXa5KBTwScvmn6INmBBcuFVfJGriOE6VUfGVkM,1037
+unique_toolkit/agentic/loop_runner/_iteration_handler_utils.py,sha256=czwTbMMmikppRjAA2wpx-UZGub8KrpSoWfojklr-5sE,3261
 unique_toolkit/agentic/loop_runner/_stream_handler_utils.py,sha256=FTGc5y8wkDnwnRVSYEdandgKz-FiySOsrTFFMadwP6E,1706
 unique_toolkit/agentic/loop_runner/base.py,sha256=3g4PalzV00o8kcRwHds2c2rtxW4idD7_7vS2Z7GkMvQ,1370
-unique_toolkit/agentic/loop_runner/middleware/__init__.py,sha256=r9c_Ml2g7obglnEC7gDihSTUrZ6s1sNGCMuMXR0Yl90,520
+unique_toolkit/agentic/loop_runner/middleware/__init__.py,sha256=5yhFZ14_C1qAt-Mb3u3nZ1h6gxuSZ5Ts90-rbk2jjUM,232
 unique_toolkit/agentic/loop_runner/middleware/planning/__init__.py,sha256=Y9MlihNA8suNREixW98RF45bj0EMtD_tQuDrO2MEML4,304
-unique_toolkit/agentic/loop_runner/middleware/planning/planning.py,sha256=5d8kyipuFyI_1SQG49f165eOwSHeSG1qjbJQ7laeTsk,3218
+unique_toolkit/agentic/loop_runner/middleware/planning/planning.py,sha256=s6SAP3BCCExgwnyRj_bZTaWgTOiNVju5qcJA0WFUUoE,3216
 unique_toolkit/agentic/loop_runner/middleware/planning/schema.py,sha256=76C36CWCLfDAYYqtaQlhXsmkWM1fCqf8j-l5afQREKA,2869
-unique_toolkit/agentic/loop_runner/middleware/qwen_forced_tool_call/__init__.py,sha256=lP8N8XLvV1irvGC6Q0FedAlBx-T2UPKotDRwkdx7neA,417
-unique_toolkit/agentic/loop_runner/middleware/qwen_forced_tool_call/helpers.py,sha256=DGKW9i7mCXNTejO2fotCmqSzI2b5k89ybkJA0QQ75qU,1234
-unique_toolkit/agentic/loop_runner/middleware/qwen_forced_tool_call/qwen_forced_tool_call.py,sha256=dp08YgL4UwVDTsJB-z7eiJn7zWEbqG8eiWfMOcvKZdI,1955
-unique_toolkit/agentic/loop_runner/runners/__init__.py,sha256=raaNpHcTfXkYURy0ysyacispSdQzYPDoG17PyR57uK4,205
-unique_toolkit/agentic/loop_runner/runners/basic.py,sha256=SQzwkLEiraU8neXvPEc_uOBzC17PkCpOEsFrZ79YGCY,3379
+unique_toolkit/agentic/loop_runner/runners/__init__.py,sha256=9aUSdHgwAfwEmvE_xL7luxmaL2HaU-fG_x-HLbHls9k,536
+unique_toolkit/agentic/loop_runner/runners/basic.py,sha256=PKg7AXp5_2OJpJX8b7_5u0jBKffX5mg8X6DFk98sSwA,1535
+unique_toolkit/agentic/loop_runner/runners/qwen/__init__.py,sha256=b2zAQIjtPy_BBQeDHsdfDLTKigI11w7tXHgUGNnFIiU,419
+unique_toolkit/agentic/loop_runner/runners/qwen/helpers.py,sha256=JBnxeYKu8HiUq3VHjnb8XdHCe1c3cJ3OwagckF4UvnU,1763
+unique_toolkit/agentic/loop_runner/runners/qwen/qwen_runner.py,sha256=7tYfTkNKwxYVn6C1Htya5QEIK2BfWkxCPNkdMoaZGq0,4920
 unique_toolkit/agentic/message_log_manager/__init__.py,sha256=3-KY_sGkPbNoSnrzwPY0FQIJNnsz4NHXvocXgGRUeuE,169
 unique_toolkit/agentic/message_log_manager/service.py,sha256=AiuIq2dKQg9Y8bEYgGcve1X8-WRRdqPZXaZXXLJxfFM,3057
-unique_toolkit/agentic/postprocessor/postprocessor_manager.py,sha256=s6HFhA61TE05aAay15NFTWI1JvdSlxmGpEVfpBbGFyM,7684
+unique_toolkit/agentic/postprocessor/postprocessor_manager.py,sha256=CoKzVFeLIr1eRP3ZLnmUJ8KNsFLyvK5iuvUilbcGAm0,7662
 unique_toolkit/agentic/reference_manager/reference_manager.py,sha256=x51CT0D8HHu2LzgXdHGy0leOYpjnsxVbPZ2nc28G9mA,4005
 unique_toolkit/agentic/responses_api/__init__.py,sha256=9WTO-ef7fGE9Y1QtZJFm8Q_jkwK8Srtl-HWvpAD2Wxs,668
 unique_toolkit/agentic/responses_api/postprocessors/code_display.py,sha256=h6ZqPR0kPQnxM0ynshYQTa1BrcN8XGbUz9p03m8rOj0,2339
-unique_toolkit/agentic/responses_api/postprocessors/generated_files.py,sha256=ZdxJQQlD5K2Arn_NJVQNzIUwp-q5SmIE17o-UquTa0A,11691
+unique_toolkit/agentic/responses_api/postprocessors/generated_files.py,sha256=Bywzf0SVwdzvWbNJbsMU4T43IclIqlUO_adCo7xaU7U,11689
 unique_toolkit/agentic/responses_api/stream_handler.py,sha256=Y1IM0uiPBdlab5UuOTCsHTaVX-fd9MxfS3xkwhdFie4,647
-unique_toolkit/agentic/short_term_memory_manager/persistent_short_term_memory_manager.py,sha256=uF3HSoZF0hBfuNhIE9N8KRtuwDfpoeXUFVrv_cyZ3Sw,5839
+unique_toolkit/agentic/short_term_memory_manager/persistent_short_term_memory_manager.py,sha256=g8I64dKkpwWIXfwpxD1-rLte00hh_PoQ9-fXUAcNQCo,5817
 unique_toolkit/agentic/thinking_manager/thinking_manager.py,sha256=41QWFsdRrbWlQHBfYCFv726UDom4WbcvaRfjCmoUOQI,4183
 unique_toolkit/agentic/tools/__init__.py,sha256=-ToY9-Xiz0K7qCUydH1h1yG6n4h1hQS8sBuSVPNEq2Y,43
 unique_toolkit/agentic/tools/a2a/__init__.py,sha256=41DntFL-YKOQFBxObfrhlKmMj2BD0Zx611U1qSSyL9E,1329
@@ -90,7 +92,7 @@ unique_toolkit/agentic/tools/a2a/config.py,sha256=6diTTSiS2prY294LfYozB-db2wmJ6j
 unique_toolkit/agentic/tools/a2a/evaluation/__init__.py,sha256=Efso468EQ4UANv140qcrRPCdX98-OQJfnrLqhgJ9pfE,412
 unique_toolkit/agentic/tools/a2a/evaluation/_utils.py,sha256=FO5_us6mC4t_X4OVtNei1Ife4SjMjCiFOsPhUHUsY-s,1878
 unique_toolkit/agentic/tools/a2a/evaluation/config.py,sha256=Ra5rzJArS3r8C7RTmzKB8cLEYh4w-u3pe_XLgul3GOA,2355
-unique_toolkit/agentic/tools/a2a/evaluation/evaluator.py,sha256=AJvXu0UJKHe72nRmFZQjmBqxglSNDpYk_Tup3jJ8irg,9388
+unique_toolkit/agentic/tools/a2a/evaluation/evaluator.py,sha256=74ceLV-2zye0MANxhsEt7oG8KSxYT1_HMnlwUAi7tXw,9386
 unique_toolkit/agentic/tools/a2a/evaluation/summarization_user_message.j2,sha256=acP1YqD_sCy6DT0V2EIfhQTmaUKeqpeWNJ7RGgceo8I,271
 unique_toolkit/agentic/tools/a2a/manager.py,sha256=pk06UUXKQdIUY-PyykYiItubBjmIydOaqWvBBDwhMN4,1939
 unique_toolkit/agentic/tools/a2a/postprocessing/__init__.py,sha256=aVtUBPN7kDrqA6Bze34AbqQpcBBqpvfyJG-xF65w7R0,659
@@ -128,11 +130,9 @@ unique_toolkit/agentic/tools/schemas.py,sha256=TXshRvivr2hD-McXHumO0bp-Z0mz_GnAm
 unique_toolkit/agentic/tools/test/test_mcp_manager.py,sha256=VpB4k4Dh0lQWakilJMQSzO8sBXapuEC26cub_lorl-M,19221
 unique_toolkit/agentic/tools/test/test_tool_progress_reporter.py,sha256=XHNezB8itj9KzpQgD0cwRtp2AgRUfUkQsHc3bTyPj6c,15801
 unique_toolkit/agentic/tools/tool.py,sha256=YmhGZ-NkPlgQ4X1hA6IVod10LNM5YErzbd0Zn6ANA98,6529
-unique_toolkit/agentic/tools/tool_manager.py,sha256=ZwzCcjzHYZgB1FxrpKRmarepfaZknIhLilxeyKoe4BY,17541
+unique_toolkit/agentic/tools/tool_manager.py,sha256=zwpIxPUl1SamqGUUwgl0X4H-IR0ZpGjaqvszDLYy7ig,17519
 unique_toolkit/agentic/tools/tool_progress_reporter.py,sha256=GaR0oqDUJZvBB9WCUVYYh0Zvs6U-LMygJCCrqPlitgA,10296
-unique_toolkit/agentic/tools/utils/__init__.py,sha256=s75sjY5nrJchjLGs3MwSIqhDW08fFXIaX7eRQjFIA4s,346
-unique_toolkit/agentic/tools/utils/execution/__init__.py,sha256=OHiKpqBnfhBiEQagKVWJsZlHv8smPp5OI4dFIexzibw,37
-unique_toolkit/agentic/tools/utils/execution/execution.py,sha256=ocPGGfUwa851207HNTLYiBJ1pNzJp4VhMZ49OPP33gU,8022
+unique_toolkit/agentic/tools/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/agentic/tools/utils/source_handling/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/agentic/tools/utils/source_handling/schema.py,sha256=iHBKuks6tUy8tvian4Pd0B6_-8__SehVVNcxIUAUjEA,882
 unique_toolkit/agentic/tools/utils/source_handling/source_formatting.py,sha256=uZ0QXqrPWgId3ZA67dvjHQ6xrW491LK1xxx_sVJmFHg,9160
@@ -153,7 +153,7 @@ unique_toolkit/chat/constants.py,sha256=05kq6zjqUVB2d6_P7s-90nbljpB3ryxwCI-CAz0r
 unique_toolkit/chat/deprecated/service.py,sha256=CYwzXi7OB0RjHd73CO2jq8SlpdBmDYLatzPFkb5sA0k,6529
 unique_toolkit/chat/functions.py,sha256=VdehD26NnsEhBsV5tD0tgUFD4LuEqIYgz8wWSsxYFgA,46078
 unique_toolkit/chat/rendering.py,sha256=c8YiV9oADRrJQ5A_QBJ4_UFc0NZ-2vVaa7tupoMusso,880
-unique_toolkit/chat/responses_api.py,sha256=MCI1MR_4wlo9xY1ifH2daNz4JvjX18uPdryQlemaeLw,14488
+unique_toolkit/chat/responses_api.py,sha256=K_3lY8PUmey2io_enx_WY3hxE7TmMZueeIBG2hYa8KY,14466
 unique_toolkit/chat/schemas.py,sha256=Zh903Xt9PhugvwzoC_k5KXQWAqllWHvj9DmqDVom2yk,6890
 unique_toolkit/chat/service.py,sha256=6D00OL4QrGafbOhTaC5zNXaNgg7gS5W_2ePVa4LhqpE,4439
 unique_toolkit/chat/state.py,sha256=Cjgwv_2vhDFbV69xxsn7SefhaoIAEqLx3ferdVFCnOg,1445
@@ -212,7 +212,7 @@ unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBu
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=Ozhh70qCn2yOzRWr9d8WmJeTo7AQurwd3tStgBMPFLA,1246
 unique_toolkit/test_utilities/events.py,sha256=_mwV2bs5iLjxS1ynDCjaIq-gjjKhXYCK-iy3dRfvO3g,6410
-unique_toolkit-1.38.2.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.38.2.dist-info/METADATA,sha256=2ZglK0rXlPIPEkPJIOui1aQlBND2oCzd-C72_7z1gN0,46611
-unique_toolkit-1.38.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.38.2.dist-info/RECORD,,
+unique_toolkit-1.38.4.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.38.4.dist-info/METADATA,sha256=fRZxzhXh974Ju0N6ugjU2PqqrdDbf22IL3sjyzHoElI,46797
+unique_toolkit-1.38.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.38.4.dist-info/RECORD,,

unique_toolkit/agentic/loop_runner/middleware/qwen_forced_tool_call/__init__.py

@@ -1,13 +0,0 @@
-from unique_toolkit.agentic.loop_runner.middleware.qwen_forced_tool_call.helpers import (
-    is_qwen_model,
-)
-from unique_toolkit.agentic.loop_runner.middleware.qwen_forced_tool_call.qwen_forced_tool_call import (
-    QWEN_FORCED_TOOL_CALL_PROMPT_INSTRUCTION,
-    QwenForcedToolCallMiddleware,
-)
-
-__all__ = [
-    "QwenForcedToolCallMiddleware",
-    "QWEN_FORCED_TOOL_CALL_PROMPT_INSTRUCTION",
-    "is_qwen_model",
-]

unique_toolkit/agentic/loop_runner/middleware/qwen_forced_tool_call/qwen_forced_tool_call.py

@@ -1,50 +0,0 @@
-import logging
-from typing import Unpack
-
-from unique_toolkit.agentic.loop_runner.base import (
-    LoopIterationRunner,
-    _LoopIterationRunnerKwargs,
-)
-from unique_toolkit.agentic.loop_runner.middleware.qwen_forced_tool_call.helpers import (
-    append_qwen_forced_tool_call_instruction,
-)
-from unique_toolkit.chat.service import LanguageModelStreamResponse
-
-_LOGGER = logging.getLogger(__name__)
-
-QWEN_FORCED_TOOL_CALL_PROMPT_INSTRUCTION = (
-    "Tool Call Instruction: \nYou always have to return a tool call. "
-    "You must start the response with <tool_call> and end with </tool_call>. "
-    "Do NOT provide natural language explanations, summaries, or any text outside the <tool_call> block."
-)
-
-
-class QwenForcedToolCallMiddleware(LoopIterationRunner):
-    def __init__(
-        self,
-        *,
-        loop_runner: LoopIterationRunner,
-        qwen_forced_tool_call_prompt_instruction: str,
-    ) -> None:
-        self._qwen_forced_tool_call_prompt_instruction = (
-            qwen_forced_tool_call_prompt_instruction
-        )
-        self._loop_runner = loop_runner
-
-    async def __call__(
-        self, **kwargs: Unpack[_LoopIterationRunnerKwargs]
-    ) -> LanguageModelStreamResponse:
-        tool_choices = kwargs.get("tool_choices") or []
-        iteration_index = kwargs["iteration_index"]
-
-        # For Qwen models, append tool call instruction to the last user message. These models ignore the parameter tool_choice.
-        if len(tool_choices) > 0 and iteration_index == 0 and kwargs.get("messages"):
-            _LOGGER.info(
-                "Appending tool call instruction to the last user message for Qwen models to force tool calls."
-            )
-            kwargs["messages"] = append_qwen_forced_tool_call_instruction(
-                messages=kwargs["messages"],
-                forced_tool_call_instruction=self._qwen_forced_tool_call_prompt_instruction,
-            )
-
-        return await self._loop_runner(**kwargs)

unique_toolkit/agentic/tools/utils/execution/__init__.py

@@ -1 +0,0 @@
-"""Execution utilities for tools."""