unique_toolkit 1.66.0__tar.gz → 1.67.0__tar.gz

{unique_toolkit-1.66.0 → unique_toolkit-1.67.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.66.0
+Version: 1.67.0
 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.66.0 → unique_toolkit-1.67.0}/pyproject.toml

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

{unique_toolkit-1.66.0 → unique_toolkit-1.67.0}/unique_toolkit/agentic/feature_flags/feature_flags.py

@@ -87,6 +87,11 @@ class FeatureFlags(BaseSettings):
         description="Persist tool calls and reconstruct tool call history across turns (UN-15977). Can be 'true' or comma-separated company IDs.",
     )
 
+    enable_web_search_argument_screening_un_18741: FeatureFlag = Field(
+        default=FeatureFlag(False),
+        description="Enable argument screening agent for web search tool calls (UN-18741). Can be 'true' or comma-separated company IDs.",
+    )
+
     model_config = SettingsConfigDict(
         extra="ignore",
         env_prefix="FEATURE_FLAG_",

{unique_toolkit-1.66.0 → unique_toolkit-1.67.0}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/service.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from typing import Any, override
 
@@ -197,10 +199,19 @@ class OpenAICodeInterpreterTool(OpenAIBuiltInTool[CodeInterpreter]):
         user_id: str,
         chat_id: str,
         is_exclusive: bool = False,
-    ) -> "OpenAICodeInterpreterTool":
+        force_auto_container: bool = False,
+    ) -> OpenAICodeInterpreterTool:
+        if force_auto_container:
+            config = config.model_copy(update={"use_auto_container": True})
+
         if config.use_auto_container:
             logger.info("Using `auto` container setting")
-            return cls(config=config, container_id=None, company_id=company_id)
+            return cls(
+                config=config,
+                container_id=None,
+                company_id=company_id,
+                is_exclusive=is_exclusive,
+            )
 
         memory_manager = _get_container_code_execution_short_term_memory_manager(
             company_id=company_id,

{unique_toolkit-1.66.0 → unique_toolkit-1.67.0}/unique_toolkit/agentic/tools/openai_builtin/manager.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from openai import AsyncOpenAI
 from openai.types.responses import ResponseIncludable
 
@@ -33,6 +35,7 @@ class OpenAIBuiltInToolManager:
         chat_id: str,
         client: AsyncOpenAI,
         tool_config: ToolBuildConfig,
+        force_auto_container: bool = False,
     ) -> OpenAIBuiltInTool:
         if tool_config.name == OpenAIBuiltInToolName.CODE_INTERPRETER:
             assert isinstance(tool_config.configuration, CodeInterpreterExtendedConfig)
@@ -45,6 +48,7 @@ class OpenAIBuiltInToolManager:
                 user_id=user_id,
                 chat_id=chat_id,
                 is_exclusive=tool_config.is_exclusive,
+                force_auto_container=force_auto_container,
             )
             return tool
         else:
@@ -60,7 +64,8 @@ class OpenAIBuiltInToolManager:
         chat_id: str,
         client: AsyncOpenAI,
         tool_configs: list[ToolBuildConfig],
-    ) -> "OpenAIBuiltInToolManager":
+        force_auto_container: bool = False,
+    ) -> OpenAIBuiltInToolManager:
         builtin_tools = []
         for tool_config in tool_configs:
             if tool_config.name in OpenAIBuiltInToolName and tool_config.is_enabled:
@@ -73,6 +78,7 @@ class OpenAIBuiltInToolManager:
                         chat_id,
                         client,
                         tool_config,
+                        force_auto_container,
                     )
                 )
 

{unique_toolkit-1.66.0 → unique_toolkit-1.67.0}/unique_toolkit/language_model/infos.py

@@ -212,6 +212,7 @@ class LanguageModelProvider(StrEnum):
 
 
 class ModelCapabilities(StrEnum):
+    AUTO_CONTAINER_ONLY = "auto_container_only"
     CHAT_COMPLETIONS_API = "chat_completions_api"
     FUNCTION_CALLING = "function_calling"
     PARALLEL_FUNCTION_CALLING = "parallel_function_calling"
@@ -804,6 +805,7 @@ class LanguageModelInfo(BaseModel):
                     version="2026-03-05",
                     encoder_name=EncoderName.O200K_BASE,
                     capabilities=[
+                        ModelCapabilities.AUTO_CONTAINER_ONLY,
                         ModelCapabilities.FUNCTION_CALLING,
                         ModelCapabilities.PARALLEL_FUNCTION_CALLING,
                         ModelCapabilities.REASONING,