Type chcking enhancements for pydantic tools

Author: shashi-stackone

stackone_ai/models.py

@@ -496,7 +496,7 @@ def implementation(**kwargs: Any) -> Any:
         return Tool.from_schema(
             function=implementation,
             name=self.name,
-            description=self.description or "",
+            description=self.description,
             json_schema=json_schema,
         )
 

stackone_ai/toolset.py

@@ -11,7 +11,7 @@
 from collections.abc import Coroutine, Sequence
 from dataclasses import dataclass
 from importlib import metadata
-from typing import Any, Literal, TypedDict, TypeVar
+from typing import TYPE_CHECKING, Any, Literal, TypedDict, TypeVar
 
 from pydantic import BaseModel, Field, PrivateAttr, ValidationError, field_validator
 
@@ -32,6 +32,9 @@
 )
 from stackone_ai.utils.normalize import _normalize_action_name
 
+if TYPE_CHECKING:
+    from pydantic_ai.tools import Tool as PydanticAITool
+
 logger = logging.getLogger("stackone.tools")
 
 SearchMode = Literal["auto", "semantic", "local"]
@@ -757,7 +760,7 @@ def pydantic_ai(
         *,
         mode: Literal["search_and_execute"] | None = None,
         account_ids: list[str] | None = None,
-    ) -> list[Any]:
+    ) -> list[PydanticAITool]:
         """Get tools as Pydantic AI ``Tool`` instances.
 
         Args: