unique_toolkit 1.67.1__tar.gz → 1.67.3__tar.gz

{unique_toolkit-1.67.1 → unique_toolkit-1.67.3}/PKG-INFO

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

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

{unique_toolkit-1.67.1 → unique_toolkit-1.67.3}/unique_toolkit/agentic/tools/a2a/tool/config.py

@@ -2,7 +2,7 @@ import re
 from enum import StrEnum
 from typing import Annotated, Generic, Literal, TypeVar
 
-from pydantic import Field
+from pydantic import Field, field_validator
 from pydantic.main import BaseModel
 
 from unique_toolkit._common.pydantic.rjsf_tags import RJSFMetaTag
@@ -114,11 +114,16 @@ class SubAgentToolConfig(BaseToolConfig):
         default=True,
         description="Whether this sub agent's references should be used in the main agent's response.",
     )
-    forced_tools: list[str] | None = Field(
-        default=None,
+    forced_tools: list[str] = Field(
+        default=[],
         description="The list of tool names that will be forced to be called for this sub-agent.",
     )
 
+    @field_validator("forced_tools", mode="before")
+    @classmethod
+    def _coerce_forced_tools_none(cls, v: list[str] | None) -> list[str]:
+        return v if v is not None else []
+
     tool_description_for_system_prompt: Annotated[
         str, RJSFMetaTag.StringWidget.textarea(rows=5)
     ] = Field(
@@ -167,12 +172,18 @@ class SubAgentToolConfig(BaseToolConfig):
         description="The condition that will be used to stop the polling for the sub-agent response.",
     )
 
-    tool_input_json_schema: (
-        Annotated[str, RJSFMetaTag.StringWidget.textarea(rows=5)] | None
-    ) = Field(
-        default=None,
-        description="A custom JSON schema to send to the llm as the tool input schema.",
+    tool_input_json_schema: Annotated[
+        str, RJSFMetaTag.StringWidget.textarea(rows=5)
+    ] = Field(
+        default="",
+        description="Optional: A custom JSON schema to send to the llm as the tool input schema.",
     )
+
+    @field_validator("tool_input_json_schema", mode="before")
+    @classmethod
+    def _coerce_tool_input_json_schema_none(cls, v: str | None) -> str:
+        return v if v is not None else ""
+
     returns_content_chunks: bool = Field(
         default=False,
         description="If set, the sub-agent response will be interpreted as a list of content chunks.",

{unique_toolkit-1.67.1 → unique_toolkit-1.67.3}/unique_toolkit/agentic/tools/a2a/tool/service.py

@@ -110,7 +110,7 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
 
     @override
     def tool_description(self) -> LanguageModelToolDescription:
-        if self.config.tool_input_json_schema is not None:
+        if self.config.tool_input_json_schema:
             return LanguageModelToolDescription(
                 name=self.name,
                 description=self.config.tool_description,
@@ -163,7 +163,7 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
         active_message_log: MessageLog | None = None
 
         try:
-            if self.config.tool_input_json_schema is not None:
+            if self.config.tool_input_json_schema:
                 tool_input = json.dumps(tool_call.arguments)
             else:
                 tool_input = SubAgentToolInput.model_validate(
@@ -396,7 +396,7 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
                 text=tool_user_message,
                 chat_id=chat_id,
                 poll_interval=self.config.poll_interval,
-                tool_choices=self.config.forced_tools,
+                tool_choices=self.config.forced_tools or None,
                 max_wait=self.config.max_wait,
                 stop_condition=self.config.stop_condition,
                 correlation=Space.Correlation(

{unique_toolkit-1.67.1 → unique_toolkit-1.67.3}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py

@@ -166,8 +166,12 @@ class DisplayCodeInterpreterFilesPostProcessor(
             loop_response.message.references = []
         if loop_response.message.text is None:
             loop_response.message.text = ""
+
         ref_number = _get_next_ref_number(loop_response.message.references)
         changed = False
+        fence_ff_on = feature_flags.enable_code_execution_fence_un_17972.is_enabled(
+            self._company_id
+        )
 
         for filename, content_id in self._content_map.items():
             if content_id is None:
@@ -181,9 +185,6 @@ class DisplayCodeInterpreterFilesPostProcessor(
 
             is_image = (guess_type(filename)[0] or "").startswith("image/")
             is_html = (guess_type(filename)[0] or "") == "text/html"
-            fence_ff_on = feature_flags.enable_code_execution_fence_un_17972.is_enabled(
-                self._company_id
-            )
 
             # Images
             if is_image:
@@ -229,7 +230,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
                     self._company_id
                 )
             )
-            if replaced and not is_image and not is_html_rendered:
+            if replaced and not (is_image or is_html_rendered or fence_ff_on):
                 loop_response.message.references.append(
                     ContentReference(
                         sequence_number=ref_number,
@@ -241,9 +242,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 )
                 ref_number += 1
 
-        if feature_flags.enable_code_execution_fence_un_17972.is_enabled(
-            self._company_id
-        ):
+        if fence_ff_on:
             code_blocks = _build_code_blocks(loop_response, self._content_map)
             _warn_unmatched_code_blocks(self._content_map, code_blocks)
             text_before = loop_response.message.text
@@ -259,19 +258,6 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 loop_response.message.text = (
                     loop_response.message.text.rstrip() + "\n\n" + orphan_fences
                 )
-                ref_number = _get_next_ref_number(loop_response.message.references)
-                for block in self._orphan_code_blocks:
-                    for file in block.files:
-                        loop_response.message.references.append(
-                            ContentReference(
-                                sequence_number=ref_number,
-                                source_id=file.content_id,
-                                source="node-ingestion-chunks",
-                                url=f"unique://content/{file.content_id}",
-                                name=file.filename,
-                            )
-                        )
-                        ref_number += 1
                 changed = True
 
         _warn_missing_content_ids(loop_response.message.text, self._content_map)