unique_toolkit 1.50.4__tar.gz → 1.51.0__tar.gz

{unique_toolkit-1.50.4 → unique_toolkit-1.51.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.50.4
+Version: 1.51.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.50.4 → unique_toolkit-1.51.0}/pyproject.toml

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

{unique_toolkit-1.50.4 → unique_toolkit-1.51.0}/unique_toolkit/agentic/tools/config.py

@@ -1,5 +1,5 @@
 from enum import StrEnum
-from typing import Annotated, Any
+from typing import Annotated, Any, Generic
 
 from pydantic import (
     BaseModel,
@@ -9,10 +9,13 @@ from pydantic import (
     field_serializer,
     model_validator,
 )
+from typing_extensions import TypeVar
 
 from unique_toolkit._common.pydantic_helpers import get_configuration_dict
 from unique_toolkit.agentic.tools.schemas import BaseToolConfig
 
+T = TypeVar("T", bound=BaseToolConfig, default=BaseToolConfig)
+
 
 class ToolIcon(StrEnum):
     ANALYTICS = "IconAnalytics"
@@ -45,12 +48,12 @@ def handle_undefined_icon(value: Any) -> ToolIcon:
         return ToolIcon.BOOK
 
 
-class ToolBuildConfig(BaseModel):
+class ToolBuildConfig(BaseModel, Generic[T]):
     model_config = get_configuration_dict()
     """Main tool configuration"""
 
     name: str
-    configuration: BaseToolConfig
+    configuration: T
     display_name: str = ""
     icon: Annotated[ToolIcon, BeforeValidator(handle_undefined_icon)] = Field(
         default=ToolIcon.BOOK,
@@ -124,5 +127,5 @@ class ToolBuildConfig(BaseModel):
         return value
 
     @field_serializer("configuration")
-    def serialize_config(self, value: BaseToolConfig) -> dict[str, Any]:
+    def serialize_config(self, value: T) -> dict[str, Any]:
         return value.__class__.model_dump(value)