unique_toolkit 1.1.5__py3-none-any.whl → 1.1.6__py3-none-any.whl

unique_toolkit/agentic/tools/config.py

@@ -1,5 +1,6 @@
+import json
 from enum import StrEnum
-from typing import Any
+from typing import Any, Dict
 
 from pydantic import (
     BaseModel,
@@ -98,10 +99,46 @@ class ToolBuildConfig(BaseModel):
         value["configuration"] = config
         return value
 
-
-def _rebuild_config_model():
-    """Rebuild the ToolBuildConfig model to resolve forward references."""
-    # Import here to avoid circular imports
-    from unique_toolkit.agentic.tools.schemas import BaseToolConfig  # noqa: F401
-
-    ToolBuildConfig.model_rebuild()
+    def model_dump(self) -> Dict[str, Any]:
+        """
+        Returns a dict representation of the tool config that preserves
+        subclass fields from `configuration` by delegating to its own
+        model_dump. This prevents `{}` when `configuration` is typed
+        as `BaseToolConfig` but holds a subclass instance.
+        """
+        data: Dict[str, Any] = {
+            "name": self.name,
+            "configuration": self.configuration.model_dump()
+            if self.configuration
+            else None,
+            "display_name": self.display_name,
+            "icon": self.icon,
+            "selection_policy": self.selection_policy,
+            "is_exclusive": self.is_exclusive,
+            "is_sub_agent": self.is_sub_agent,
+            "is_enabled": self.is_enabled,
+        }
+        return data
+
+    def model_dump_json(self) -> str:
+        """
+        Returns a JSON string representation of the tool config.
+        Ensures `configuration` is fully serialized by using the
+        subclass's `model_dump_json()` when available.
+        """
+        config_json = (
+            self.configuration.model_dump_json() if self.configuration else None
+        )
+        config = json.loads(config_json) if config_json else None
+
+        data: Dict[str, Any] = {
+            "name": self.name,
+            "configuration": config,
+            "display_name": self.display_name,
+            "icon": self.icon,
+            "selection_policy": self.selection_policy,
+            "is_exclusive": self.is_exclusive,
+            "is_sub_agent": self.is_sub_agent,
+            "is_enabled": self.is_enabled,
+        }
+        return json.dumps(data)

unique_toolkit/agentic/tools/tool_manager.py

@@ -6,7 +6,7 @@ from pydantic import BaseModel, Field
 
 from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricName
 from unique_toolkit.agentic.tools.a2a.manager import A2AManager
-from unique_toolkit.agentic.tools.config import ToolBuildConfig, _rebuild_config_model
+from unique_toolkit.agentic.tools.config import ToolBuildConfig
 from unique_toolkit.agentic.tools.factory import ToolFactory
 from unique_toolkit.agentic.tools.mcp.manager import MCPManager
 from unique_toolkit.agentic.tools.schemas import ToolCallResponse, ToolPrompts
@@ -23,9 +23,6 @@ from unique_toolkit.language_model.schemas import (
     LanguageModelToolDescription,
 )
 
-# Rebuild the config model now that all imports are resolved
-_rebuild_config_model()
-
 
 class ForcedToolOption:
     type: str = "function"

{unique_toolkit-1.1.5.dist-info → unique_toolkit-1.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.1.5
+Version: 1.1.6
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,7 +118,10 @@ 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.1.4] - 2025-09-23
+## [1.1.6] - 2025-09-23
+- Fix model_dump for `ToolBuildConfig`
+
+## [1.1.5] - 2025-09-23
 - Fix a circular import and add tests for `ToolBuildConfig`
 
 ## [1.1.4] - 2025-09-23

{unique_toolkit-1.1.5.dist-info → unique_toolkit-1.1.6.dist-info}/RECORD

@@ -56,7 +56,7 @@ unique_toolkit/agentic/tools/a2a/memory.py,sha256=4VFBzITCv5E_8YCc4iF4Y6FhzplS2C
 unique_toolkit/agentic/tools/a2a/schema.py,sha256=T1l5z6trtPE5nhqPzt5tvfRNDhqL_ST1Wj7_lBWJ58g,304
 unique_toolkit/agentic/tools/a2a/service.py,sha256=7n7BkJxTR-hpoQczfHLK8eS0rtATL7-YARH-4netbbI,5339
 unique_toolkit/agentic/tools/agent_chunks_hanlder.py,sha256=x32Dp1Z8cVW5i-XzXbaMwX2KHPcNGmqEU-FB4AV9ZGo,1909
-unique_toolkit/agentic/tools/config.py,sha256=NWFp8Kr8LOSoOk0cy207tFnenomu3HXSjEqMvi7tJJQ,3221
+unique_toolkit/agentic/tools/config.py,sha256=B9Tl0uJ81XB4SjSke4kSydame4glGT2s69ZXfr0ViZU,4634
 unique_toolkit/agentic/tools/factory.py,sha256=Wt0IGSbLg8ZTq5PU9p_JTW0LtNATWLpc3336irJKXlM,1277
 unique_toolkit/agentic/tools/mcp/__init__.py,sha256=RLF_p-LDRC7GhiB3fdCi4u3bh6V9PY_w26fg61BLyco,122
 unique_toolkit/agentic/tools/mcp/manager.py,sha256=DPYwwDe6RSZyuPaxn-je49fP_qOOs0ZV46EM6GZcV4c,2748
@@ -66,7 +66,7 @@ unique_toolkit/agentic/tools/schemas.py,sha256=0ZR8xCdGj1sEdPE0lfTIG2uSR5zqWoprU
 unique_toolkit/agentic/tools/test/test_mcp_manager.py,sha256=9F7FjpYKeOkg2Z4bt2H1WGaxu9fzB-1iiE-b7g3KzQk,15724
 unique_toolkit/agentic/tools/test/test_tool_progress_reporter.py,sha256=dod5QPqgGUInVAGXAbsAKNTEypIi6pUEWhDbJr9YfUU,6307
 unique_toolkit/agentic/tools/tool.py,sha256=m56VLxiHuKU2_J5foZp00xhm5lTxWEW7zRLGbIE9ssU,6744
-unique_toolkit/agentic/tools/tool_manager.py,sha256=RWz6hKzuTTqzry5KpaeSfsO84zMp6RqVfX59YWFYblc,11034
+unique_toolkit/agentic/tools/tool_manager.py,sha256=E1eKZuc7eHPeGFqrGfdMGPNLqu3NYhrfKDjSJxMSuvU,10925
 unique_toolkit/agentic/tools/tool_progress_reporter.py,sha256=ixud9VoHey1vlU1t86cW0-WTvyTwMxNSWBon8I11SUk,7955
 unique_toolkit/agentic/tools/utils/__init__.py,sha256=iD1YYzf9LcJFv95Z8BqCAFSewNBabybZRZyvPKGfvro,27
 unique_toolkit/agentic/tools/utils/execution/__init__.py,sha256=OHiKpqBnfhBiEQagKVWJsZlHv8smPp5OI4dFIexzibw,37
@@ -128,7 +128,7 @@ unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJ
 unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBuE9sI2o9Aajqjxg,8884
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=cxWjb2dxEI2HGsakKdVCkSNi7VK9mr08w5sDcFCQyWI,9553
-unique_toolkit-1.1.5.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.1.5.dist-info/METADATA,sha256=J5VKFRheot-bjwdjbOoXzTR4NlgWvq5dNZ1ffBE7sVg,32806
-unique_toolkit-1.1.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.1.5.dist-info/RECORD,,
+unique_toolkit-1.1.6.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.1.6.dist-info/METADATA,sha256=sMJptXKLbf2ULpa6UcCWdX_mi7W-B5rQLNvHq0X1Q5A,32870
+unique_toolkit-1.1.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.1.6.dist-info/RECORD,,