unique_toolkit 1.67.1__tar.gz → 1.67.2__tar.gz

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

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

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

{unique_toolkit-1.67.1 → unique_toolkit-1.67.2}/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.2}/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(