chore: Expose pydantic model from async tool

Author: anubhav756

packages/toolbox-core/src/toolbox_core/tool.py

@@ -19,6 +19,7 @@
 from warnings import warn
 
 from aiohttp import ClientSession
+from pydantic import BaseModel
 
 from .protocol import ParameterSchema
 from .utils import (
@@ -158,6 +159,10 @@ def _auth_service_token_getters(self) -> Mapping[str, Callable[[], str]]:
     def _client_headers(self) -> Mapping[str, Union[Callable, Coroutine, str]]:
         return MappingProxyType(self.__client_headers)
 
+    @property
+    def _pydantic_model(self) -> type[BaseModel]:
+        return self.__pydantic_model
+
     def __copy(
         self,
         session: Optional[ClientSession] = None,