unique_toolkit 2026.24.0__tar.gz → 2026.24.0.dev1__tar.gz

{unique_toolkit-2026.24.0 → unique_toolkit-2026.24.0.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.24.0
+Version: 2026.24.0.dev1
 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.24.0
+Requires-Dist: unique-sdk>=2026.24.0.dev2,<2026.24.0rc0
 Requires-Dist: jambo>=0.1.2,<0.2
 Requires-Dist: jinja2>=3.1.6,<4
 Requires-Dist: docxtpl>=0.20.1,<0.21
@@ -31,7 +31,6 @@ Requires-Dist: deepdiff>=8.6.1,<9
 Requires-Dist: pypandoc>=1.16.2,<2
 Requires-Dist: tenacity>=9.1.4
 Requires-Dist: pydantic-core>=2.41.5
-Requires-Dist: cachetools>=5.0,<6
 Requires-Dist: fastapi>=0.121.0,<0.122 ; extra == 'fastapi'
 Requires-Dist: starlette>=0.49.1 ; extra == 'fastapi'
 Requires-Dist: uvicorn>=0.37.0,<0.38 ; extra == 'fastapi'

{unique_toolkit-2026.24.0 → unique_toolkit-2026.24.0.dev1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.24.0"
+version = "2026.24.0.dev1"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -28,7 +28,7 @@ dependencies = [
     "openai>=1.105.0,<3",
     "platformdirs>=4.0.0,<5",
     "pillow>=12.1.1",
-    "unique-sdk>=2026.24.0",
+    "unique-sdk>=2026.24.0.dev2,<2026.24.0rc0",
     "jambo>=0.1.2,<0.2",
     "jinja2>=3.1.6,<4",
     "docxtpl>=0.20.1,<0.21",
@@ -43,7 +43,6 @@ dependencies = [
     "pypandoc>=1.16.2,<2",
     "tenacity>=9.1.4",
     "pydantic-core>=2.41.5",
-    "cachetools>=5.0,<6",
 ]
 
 [project.optional-dependencies]
@@ -61,6 +60,7 @@ fastapi = [
 monitoring = [
     "prometheus_client>=0.24.1,<1",
 ]
+
 [build-system]
 requires = ["uv_build>=0.8.14,<0.9.0"]
 build-backend = "uv_build"
@@ -69,15 +69,13 @@ build-backend = "uv_build"
 module-root = ""
 
 [dependency-groups]
-dev = [
-    "types-cachetools>=5.0,<6",
-]
+dev = []
 
 [tool.uv]
 exclude-newer = "2 weeks"
 constraint-dependencies = [
     "starlette>=0.49.1",
-    "lxml>=6.1.0",
+    "lxml>=5.0.0",
 ]
 
 [tool.uv.exclude-newer-package]

unique_toolkit-2026.24.0.dev1/unique_toolkit/__init__.py

@@ -0,0 +1,85 @@
+# Re-export commonly used classes for easier imports
+from unique_toolkit.chat import ChatService
+from unique_toolkit.content import ContentService
+from unique_toolkit.data_extraction import (
+    StructuredOutputDataExtractor,
+    StructuredOutputDataExtractorConfig,
+)
+from unique_toolkit.embedding import EmbeddingService
+from unique_toolkit.framework_utilities.openai.client import (
+    get_async_openai_client,
+    get_openai_client,
+)
+from unique_toolkit.language_model import (
+    LanguageModelMessages,
+    LanguageModelName,
+    LanguageModelService,
+    LanguageModelToolDescription,
+)
+from unique_toolkit.services.knowledge_base import KnowledgeBaseService
+from unique_toolkit.short_term_memory import ShortTermMemoryService
+
+# Conditionally import config_checker (requires pydantic-settings)
+_CONFIG_CHECKER_AVAILABLE: bool = False
+try:
+    from unique_toolkit._common.config_checker import (
+        ConfigDiffer,  # noqa: F401, I001
+        ConfigEntry,  # noqa: F401, I001
+        ConfigExporter,  # noqa: F401, I001
+        ConfigRegistry,  # noqa: F401, I001
+        ConfigValidator,  # noqa: F401, I001
+        DefaultChangeReport,  # noqa: F401, I001
+        ExportManifest,  # noqa: F401, I001
+        ValidationReport,  # noqa: F401, I001
+        register_config,  # noqa: F401, I001
+    )
+
+    _CONFIG_CHECKER_AVAILABLE = True  # pyright: ignore[reportConstantRedefinition]
+except ImportError:
+    pass
+
+# Conditionally import langchain utilities if langchain is installed
+_LANGCHAIN_AVAILABLE: bool = False
+try:
+    from unique_toolkit.framework_utilities.langchain.client import get_langchain_client  # noqa: F401, I001
+
+    _LANGCHAIN_AVAILABLE = True  # pyright: ignore[reportConstantRedefinition]
+except ImportError:
+    pass
+
+# You can add other classes you frequently use here as well
+
+__all__ = [
+    "ChatService",
+    "ContentService",
+    "EmbeddingService",
+    "get_openai_client",
+    "get_async_openai_client",
+    "KnowledgeBaseService",
+    "LanguageModelMessages",
+    "LanguageModelName",
+    "LanguageModelService",
+    "LanguageModelToolDescription",
+    "ShortTermMemoryService",
+    "StructuredOutputDataExtractor",
+    "StructuredOutputDataExtractorConfig",
+]
+
+# Add config_checker exports if available
+if _CONFIG_CHECKER_AVAILABLE:
+    __all__.extend(
+        [
+            "ConfigDiffer",
+            "ConfigEntry",
+            "ConfigExporter",
+            "ConfigRegistry",
+            "ConfigValidator",
+            "DefaultChangeReport",
+            "ExportManifest",
+            "ValidationReport",
+            "register_config",
+        ]
+    )
+# Add langchain-specific exports if available
+if _LANGCHAIN_AVAILABLE:
+    __all__.append("get_langchain_client")

{unique_toolkit-2026.24.0 → unique_toolkit-2026.24.0.dev1}/unique_toolkit/_common/metadata_filter_scope.py

@@ -8,8 +8,6 @@ from typing import Any
 from unique_toolkit.content.smart_rules import (
     Operator,
     Statement,
-    UniqueQL,
-    uniqueql_to_dict,
 )
 
 _UQL_AND = "and"
@@ -31,17 +29,16 @@ def build_folder_id_in_clause(scope_ids: list[str]) -> dict[str, Any]:
 
 def merge_scope_clause_into_metadata_filter(
     scope_clause: Mapping[str, Any],
-    metadata_filter: UniqueQL | Mapping[str, Any] | None,
+    metadata_filter: dict[str, Any] | None,
 ) -> dict[str, Any]:
     """AND-merge *scope_clause* with *metadata_filter*, flattening a top-level ``and``."""
     scope_dict = dict(scope_clause)
-    metadata_dict = uniqueql_to_dict(metadata_filter)
-    if not metadata_dict:
+    if not metadata_filter:
         return scope_dict
-    inner_and = metadata_dict.get(_UQL_AND)
+    inner_and = metadata_filter.get(_UQL_AND)
     if isinstance(inner_and, list):
         return {_UQL_AND: [scope_dict, *inner_and]}
-    return {_UQL_AND: [scope_dict, metadata_dict]}
+    return {_UQL_AND: [scope_dict, dict(metadata_filter)]}
 
 
 __all__ = [

{unique_toolkit-2026.24.0 → unique_toolkit-2026.24.0.dev1}/unique_toolkit/_common/pydantic/rjsf_tags.py

@@ -974,12 +974,7 @@ def ui_schema_for_model(
             node["items"] = item_node
 
         # Dict -> additionalProperties (value side)
-        # Skip when the node already carries a ui:widget (fully-specified
-        # widget) or an anyOf (fully-specified type branching): walking the
-        # dict value type in those cases adds a spurious "additionalProperties"
-        # node that confuses RJSF renderers.  Presentation-only meta such as
-        # ui:expandable or ui:title does NOT prevent the walk.
-        elif origin is dict and "ui:widget" not in node and "anyOf" not in node:
+        elif origin is dict:
             _key_t, val_t = get_args(base) or (Any, Any)
             val_base, val_meta = _walk_annotated_chain(val_t)
             val_base = _unwrap_optional(val_base)

unique_toolkit-2026.24.0.dev1/unique_toolkit/agentic/tools/a2a/config.py

@@ -0,0 +1,17 @@
+from pydantic import Field
+
+from unique_toolkit.agentic.tools.a2a.evaluation import SubAgentEvaluationConfig
+from unique_toolkit.agentic.tools.a2a.postprocessing import SubAgentDisplayConfig
+from unique_toolkit.agentic.tools.a2a.tool import SubAgentToolConfig
+
+
+# SubAgentToolConfig with display and evaluation configs
+class ExtendedSubAgentToolConfig(SubAgentToolConfig):
+    response_display_config: SubAgentDisplayConfig = Field(
+        default_factory=SubAgentDisplayConfig,
+        description="Configuration for how to display the sub-agent response.",
+    )
+    evaluation_config: SubAgentEvaluationConfig = Field(
+        default_factory=SubAgentEvaluationConfig,
+        description="Configuration for handling assessments of the sub-agent response.",
+    )

unique_toolkit-2026.24.0.dev1/unique_toolkit/agentic/tools/a2a/tool/__init__.py

@@ -0,0 +1,4 @@
+from unique_toolkit.agentic.tools.a2a.tool.config import SubAgentToolConfig
+from unique_toolkit.agentic.tools.a2a.tool.service import SubAgentTool
+
+__all__ = ["SubAgentTool", "SubAgentToolConfig"]

{unique_toolkit-2026.24.0 → unique_toolkit-2026.24.0.dev1}/unique_toolkit/agentic/tools/a2a/tool/config.py

@@ -1,9 +1,8 @@
-import logging
 import re
 from enum import StrEnum
-from typing import Annotated, Generic, Literal, Self, TypeVar
+from typing import Annotated, Generic, Literal, TypeVar
 
-from pydantic import Field, field_validator, model_validator
+from pydantic import Field, field_validator
 from pydantic.main import BaseModel
 
 from unique_toolkit._common.pydantic.rjsf_tags import RJSFMetaTag
@@ -27,9 +26,6 @@ class SystemReminderConfig(BaseModel, Generic[T]):
     type: T
 
 
-_LOGGER = logging.getLogger(__name__)
-
-
 _SYSTEM_REMINDER_FIELD_DESCRIPTION = """
 The reminder to add to the tool response. The reminder can be a Jinja template and can contain the following placeholders:
 - {{ display_name }}: The display name of the sub agent.
@@ -107,19 +103,6 @@ This is the message that will be sent to the sub-agent.
 """.strip()
 
 
-class SubAgentPassthroughConfig(BaseModel):
-    model_config = get_configuration_dict()
-
-    enabled: bool = Field(
-        default=False,
-        description="If set, this sub agent's response will be directly displayed to the user once the agent is called, ending the orchestrator loop.",
-    )
-    include_assessments: bool = Field(
-        default=True,
-        description="If set, this sub agent's assessments will be added to the main agent response.",
-    )
-
-
 class SubAgentToolConfig(BaseToolConfig):
     model_config = get_configuration_dict()
 
@@ -188,12 +171,10 @@ class SubAgentToolConfig(BaseToolConfig):
     poll_interval: Annotated[float, RJSFMetaTag.NumberWidget.updown()] = Field(
         default=1.0,
         description="Time interval in seconds between polling attempts when waiting for sub-agent response.",
-        gt=0,
     )
     max_wait: Annotated[float, RJSFMetaTag.NumberWidget.updown()] = Field(
         default=120.0,
         description="Maximum time in seconds to wait for the sub-agent response before timing out.",
-        gt=0,
     )
     stop_condition: Literal["stoppedStreamingAt", "completedAt"] = Field(
         default="completedAt",
@@ -216,11 +197,6 @@ class SubAgentToolConfig(BaseToolConfig):
         default=False,
         description="If set, the sub-agent response will be interpreted as a list of content chunks.",
     )
-    passthrough_config: SubAgentPassthroughConfig = Field(
-        default_factory=SubAgentPassthroughConfig,
-        description="Configuration for streaming the sub-agent response directly back to the user.",
-        title="Direct Streaming Configuration (Experimental)",
-    )
 
     system_reminders_config: list[
         Annotated[
@@ -231,28 +207,3 @@ class SubAgentToolConfig(BaseToolConfig):
         default=[],
         description="Configuration for the system reminders to add to the tool response.",
     )
-
-    @model_validator(mode="after")
-    def _ensure_response_passthrough_and_returns_content_chunks_mutually_exclusive(
-        self,
-    ) -> Self:
-        if self.passthrough_config.enabled and self.returns_content_chunks:
-            raise ValueError(
-                f"SubAgent (assistant_id={self.assistant_id!r}): "
-                "`passthrough_config.enabled` and `returns_content_chunks` are mutually exclusive."
-            )
-        return self
-
-    @model_validator(mode="after")
-    def _ensure_use_sub_agent_references_off_when_tool_response_passthrough(
-        self,
-    ) -> Self:
-        if self.passthrough_config.enabled and self.use_sub_agent_references:
-            _LOGGER.warning(
-                "SubAgent (assistant_id=%r): `passthrough_config.enabled=True` and `use_sub_agent_references=True`. "
-                "`use_sub_agent_references` will be overridden to False.",
-                self.assistant_id,
-            )
-            self.use_sub_agent_references = False
-
-        return self

{unique_toolkit-2026.24.0 → unique_toolkit-2026.24.0.dev1}/unique_toolkit/agentic/tools/a2a/tool/service.py

@@ -11,7 +11,6 @@ from pydantic import Field, TypeAdapter, create_model
 from unique_sdk.api_resources._space import Space
 from unique_sdk.utils.chat_in_space import send_message_and_wait_for_completion
 
-from unique_toolkit._common.execution import SafeTaskExecutor
 from unique_toolkit._common.referencing import (
     get_all_ref_numbers,
     get_detection_pattern_for_ref,
@@ -42,15 +41,9 @@ from unique_toolkit.agentic.tools.tool_progress_reporter import (
     ToolProgressReporter,
 )
 from unique_toolkit.app import ChatEvent
-from unique_toolkit.chat.schemas import (
-    ChatMessageAssessmentLabel,
-    ChatMessageAssessmentStatus,
-    ChatMessageAssessmentType,
-    MessageLog,
-    MessageLogStatus,
-)
+from unique_toolkit.chat.schemas import MessageLog, MessageLogStatus
 from unique_toolkit.chat.service import ChatService
-from unique_toolkit.content import ContentChunk, ContentReference
+from unique_toolkit.content import ContentChunk
 from unique_toolkit.language_model import (
     LanguageModelFunction,
     LanguageModelToolDescription,
@@ -153,10 +146,6 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
     def tool_format_information_for_user_prompt(self) -> str:
         return self.config.tool_format_information_for_user_prompt
 
-    @override
-    def takes_control(self) -> bool:
-        return self.config.passthrough_config.enabled
-
     @override
     def evaluation_check_list(self) -> list[EvaluationMetricName]:
         return [EvaluationMetricName.SUB_AGENT] if self._should_run_evaluation else []
@@ -226,7 +215,6 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
                 self._should_run_evaluation |= (
                     response["assessment"] is not None
                     and len(response["assessment"]) > 0
-                    and not self.config.passthrough_config.enabled
                 )  # Run evaluation if any sub agent returned an assessment
 
                 self._notify_watcher(response, sequence_number, timestamp)
@@ -285,9 +273,6 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
                     active_message_log=active_message_log,
                 )
 
-                if self.config.passthrough_config.enabled:
-                    await self._display_sub_agent_response(response)
-
                 return ToolCallResponse(
                     id=tool_call.id,
                     name=tool_call.name,
@@ -318,68 +303,6 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
             )
             raise e
 
-    async def _display_sub_agent_response(self, response: Space.Message) -> None:
-        await self._publish_sub_agent_response(response, set_completed_at=True)
-
-        if self.config.passthrough_config.include_assessments:
-            await self._forward_sub_agent_assessments(response["assessment"] or [])
-
-    async def _publish_sub_agent_response(
-        self,
-        response: Space.Message,
-        *,
-        set_completed_at: bool,
-    ) -> None:
-        await self._chat_service.modify_assistant_message_async(
-            content=response["text"],
-            original_content=response["originalText"],
-            references=[
-                ContentReference.from_sdk_reference(ref)
-                for ref in response["references"] or []
-            ],
-            set_completed_at=set_completed_at,
-        )
-
-    async def _publish_sub_agent_response_update(
-        self,
-        response: Space.Message,
-    ) -> None:
-        try:
-            await self._publish_sub_agent_response(response, set_completed_at=False)
-        except Exception:
-            logger.warning(
-                "Failed to publish streaming sub agent passthrough update",
-                exc_info=True,
-            )
-
-    async def _forward_sub_agent_assessments(
-        self,
-        assessments: list[unique_sdk.Space.Assessment],
-    ) -> None:
-        async def _create_one_assessment(
-            assessment: unique_sdk.Space.Assessment,
-        ) -> None:
-            await self._chat_service.create_message_assessment_async(
-                assistant_message_id=self._chat_service._assistant_message_id,
-                status=ChatMessageAssessmentStatus(assessment["status"]),
-                type=ChatMessageAssessmentType(assessment["type"]),
-                title=assessment["title"],
-                explanation=assessment["explanation"],
-                label=ChatMessageAssessmentLabel(assessment["label"]),
-                is_visible=assessment["isVisible"],
-            )
-
-        if not assessments:
-            return
-
-        task_executor = SafeTaskExecutor(logger=logger)
-        await asyncio.gather(
-            *(
-                task_executor.execute_async(_create_one_assessment, a)
-                for a in assessments
-            ),
-        )
-
     async def _get_chat_id(self) -> str | None:
         if not self.config.reuse_chat:
             return None
@@ -465,12 +388,6 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
         active_message_log: MessageLog | None = None,
     ) -> unique_sdk.Space.Message:
         try:
-            on_message_update = (
-                self._publish_sub_agent_response_update
-                if self.config.passthrough_config.enabled
-                else None
-            )
-
             return await send_message_and_wait_for_completion(
                 user_id=self._user_id,
                 assistant_id=self.config.assistant_id,
@@ -486,7 +403,6 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
                     parentChatId=self._event.payload.chat_id,
                     parentAssistantId=self.config.assistant_id,
                 ),
-                on_message_update=on_message_update,
             )
         except TimeoutError as e:
             await self._notify_progress(

{unique_toolkit-2026.24.0 → unique_toolkit-2026.24.0.dev1}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/config.py

@@ -161,7 +161,6 @@ Handling User Queries:
 - Whenever the user query requires using the python tool, you must always think first about the steps required.
 - CRITICAL: If any step generates a plot or visualization as a PNG, use **matplotlib** — that code block MUST include `plt.savefig('/mnt/data/<filename>.png', bbox_inches='tight')` and `plt.close()` as the final lines. Code that only calls `plt.show()` without saving will NOT display the image to the user.
 - CRITICAL: NEVER reference a `sandbox:/mnt/data/<filename>` link unless you have already executed code in this response that saved that exact file. Referencing a file that was not created by executed code will result in a broken link.
-  - Self-check before writing any sandbox link: look back at the code blocks you executed in this response. Is there a block that explicitly saves this exact filename to `/mnt/data/`? If not, execute the code first — then reference the file.
 - CRITICAL: The sandbox has NO internet access. NEVER use `requests`, `httpx`, `urllib`, or any other HTTP library to fetch data from the web — these calls will always fail. If the user's query requires live web data (e.g. market prices, news, APIs), you MUST retrieve it using the web search tool first and then pass the result into the code interpreter.
 - CRITICAL: For Excel files (`.xls`, `.xlsx`) and CSV files uploaded to the chat, ALWAYS use this tool to read and process them (e.g. `pandas.read_excel`, `pandas.read_csv`, `openpyxl`). Do NOT rely on the chat's text content or prior ingestion for spreadsheet data — the raw file contents are only accessible via code execution from `/mnt/data/<filename>`.
 - For calculation, retrieval, or exploratory-analysis answers, return the result concisely in the chat as markdown (key numbers inline, short tables where useful). Do NOT produce a separate artifact file (HTML, PDF, etc.) unless the user explicitly asked for a downloadable output or the request clearly calls for a dashboard/report.

{unique_toolkit-2026.24.0 → unique_toolkit-2026.24.0.dev1}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py

@@ -620,16 +620,13 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 self._log.info("Fence injection modified the message text")
 
         _warn_missing_content_ids(loop_response.message.text or "", self._content_map)
-        ci_ran = bool(loop_response.code_interpreter_calls)
         loop_response.message.text, dangling_replaced = _replace_dangling_sandbox_links(
-            loop_response.message.text, ci_ran=ci_ran
+            loop_response.message.text
         )
         changed |= dangling_replaced
         if dangling_replaced:
             self._log.warning(
-                "apply_postprocessing found and replaced dangling sandbox links "
-                "(ci_ran=%s)",
-                ci_ran,
+                "apply_postprocessing found and replaced dangling sandbox links"
             )
 
         apply_ms = (time.monotonic() - apply_t0) * 1000
@@ -1257,9 +1254,7 @@ _SANDBOX_MARKDOWN_LINK_RE = re.compile(r"!?\[.*?\]\(sandbox:/mnt/data/\S+?\)")
 _SANDBOX_FILENAME_RE = re.compile(r"sandbox:/mnt/data/([^)\s]+)")
 
 
-def _replace_dangling_sandbox_links(
-    text: str, *, ci_ran: bool = True
-) -> tuple[str, bool]:
+def _replace_dangling_sandbox_links(text: str) -> tuple[str, bool]:
     """Replace any remaining sandbox:/mnt/data/ markdown links with a per-file notice.
 
     A dangling link means either the LLM hallucinated a file reference (the sandbox
@@ -1268,12 +1263,6 @@ def _replace_dangling_sandbox_links(
     cases the user would see a broken link; this function replaces each such link with
     an actionable message that names the specific file, and logs a warning so the
     incident is visible in production logs.
-
-    Args:
-        text: The message text to scan for dangling sandbox links.
-        ci_ran: Whether the code interpreter was called at all this turn.  When
-            ``False`` the LLM emitted a sandbox link without invoking the tool
-            (hallucination); a more accurate message is shown to the user.
     """
     if not _SANDBOX_MARKDOWN_LINK_RE.search(text):
         return text, False
@@ -1282,17 +1271,11 @@ def _replace_dangling_sandbox_links(
         filename_match = _SANDBOX_FILENAME_RE.search(m.group())
         filename = filename_match.group(1) if filename_match else "the file"
         logger.warning(
-            "Dangling sandbox link found in final text: '%s'. ci_ran=%s. "
-            "The file was either never generated or the link format did not match "
+            "Dangling sandbox link found in final text: '%s'. "
+            "The file was either never uploaded or the link format did not match "
             "the expected pattern — replacing with per-file notice.",
             m.group(),
-            ci_ran,
         )
-        if not ci_ran:
-            return (
-                f"⚠️ The file *{filename}* was not generated. "
-                "Please ask me to create it."
-            )
         return f"⚠️ The file *{filename}* could not be retrieved. You can ask me to regenerate it."
 
     return _SANDBOX_MARKDOWN_LINK_RE.sub(_replacement, text), True

{unique_toolkit-2026.24.0 → unique_toolkit-2026.24.0.dev1}/unique_toolkit/agentic/tools/tool_manager.py

@@ -310,65 +310,20 @@ class _ToolManager(Generic[_ApiMode]):
         if tool.name not in self._tool_choices:
             self._tool_choices.append(tool.name)
 
-    def _get_tool_calls_that_take_control(
-        self, tool_calls: list[LanguageModelFunction]
-    ) -> list[LanguageModelFunction]:
-        tools = []
+    def does_a_tool_take_control(self, tool_calls: list[LanguageModelFunction]) -> bool:
         for tool_call in tool_calls:
             tool_instance = self.get_tool_by_name(tool_call.name)
             if tool_instance and tool_instance.takes_control():
-                tools.append(tool_call)
-
-        return tools
-
-    def does_a_tool_take_control(self, tool_calls: list[LanguageModelFunction]) -> bool:
-        return len(self._get_tool_calls_that_take_control(tool_calls)) > 0
+                return True
+        return False
 
     def get_evaluation_check_list(self) -> list[EvaluationMetricName]:
         return list(self._tool_evaluation_check_list)
 
-    def _take_control_conflict_response(
-        self,
-        tool_call: LanguageModelFunction,
-        take_control_names: list[str],
-    ) -> ToolCallResponse:
-        if tool_call.name in take_control_names:
-            content = (
-                f"ERROR: Tool `{tool_call.name}` directly returns its response to the user and therefore must be called on its own. "
-                "None of the tools in this batch were executed. "
-                f"You may recover by first calling the other tools, then calling `{tool_call.name}` alone."
-            )
-        else:
-            names = ", ".join(f"`{n}`" for n in take_control_names)
-            content = f"ERROR: Not executed because the following tool(s) must be called on their own: {names}"
-
-        return ToolCallResponse(
-            id=tool_call.id,
-            name=tool_call.name,
-            content=content,
-        )
-
     async def execute_selected_tools(
         self,
         tool_calls: list[LanguageModelFunction],
     ) -> list[ToolCallResponse]:
-        if len(tool_calls) > 1:
-            take_control_tools = [
-                t.name for t in self._get_tool_calls_that_take_control(tool_calls)
-            ]
-
-            if len(take_control_tools) > 0:
-                self._logger.warning(
-                    "Tool(s) %s take control and cannot be called alongside other tools. "
-                    "Returning error for all %d tool calls.",
-                    take_control_tools,
-                    len(tool_calls),
-                )
-                return [
-                    self._take_control_conflict_response(tool_call, take_control_tools)
-                    for tool_call in tool_calls
-                ]
-
         tool_call_responses = await self._execute_parallelized(tool_calls)
         return tool_call_responses
 

{unique_toolkit-2026.24.0 → unique_toolkit-2026.24.0.dev1}/unique_toolkit/app/schemas.py

@@ -4,17 +4,10 @@ import json
 from enum import StrEnum
 from logging import getLogger
 from pathlib import Path
-from typing import Annotated, Any, Generic, Optional, TypeVar, override
+from typing import Any, Generic, Optional, TypeVar, override
 
 from humps import camelize
-from pydantic import (
-    BaseModel,
-    BeforeValidator,
-    ConfigDict,
-    Field,
-    JsonValue,
-    field_validator,
-)
+from pydantic import BaseModel, ConfigDict, Field, JsonValue, field_validator
 from pydantic_settings import BaseSettings
 from typing_extensions import deprecated
 
@@ -23,8 +16,6 @@ from unique_toolkit.app.chat_event_filter_options_settings import (
     CHAT_EVENT_FILTER_OPTIONS_SETTINGS,
 )
 from unique_toolkit.app.unique_settings import UniqueChatEventFilterOptions
-from unique_toolkit.language_model.default_language_model import DEFAULT_LANGUAGE_MODEL
-from unique_toolkit.language_model.infos import LanguageModelInfo
 from unique_toolkit.smart_rules.compile import UniqueQL, parse_uniqueql
 
 FilterOptionsT = TypeVar("FilterOptionsT", bound=BaseSettings)
@@ -38,15 +29,6 @@ model_config = ConfigDict(
 _logger = getLogger(__name__)
 
 
-def _validate_language_model_info(v: Any) -> Any:
-    from unique_toolkit._common.validators import validate_and_init_language_model_info
-
-    return validate_and_init_language_model_info(v)
-
-
-LMI = Annotated[LanguageModelInfo, BeforeValidator(_validate_language_model_info)]
-
-
 class EventName(StrEnum):
     EXTERNAL_MODULE_CHOSEN = "unique.chat.external-module.chosen"
     USER_MESSAGE_CREATED = "unique.chat.user-message.created"
@@ -265,10 +247,6 @@ class ChatEventPayload(BaseModel):
     user_metadata: dict[str, Any] | None = Field(
         default_factory=dict,
     )
-    model_choice: LMI = Field(
-        default_factory=lambda: LanguageModelInfo.from_name(DEFAULT_LANGUAGE_MODEL),
-        description="The model choice the user has chosen to be used.",
-    )
     tool_choices: list[str] = Field(
         default_factory=list,
         description="A list containing the tool names the user has chosen to be activated.",
@@ -327,10 +305,6 @@ class ChatEventPayload(BaseModel):
         if value:
             return parse_uniqueql(value)
 
-    @property
-    def has_model_choice_override(self) -> bool:
-        return "model_choice" in self.model_fields_set
-
 
 @deprecated("""Use `ChatEventPayload` instead.
             This class will be removed in the next major version.""")