chore(identity):fix type hints for tool_context in auth mixins

Author: loveyana

veadk/integrations/ve_identity/auth_mixins.py

@@ -23,7 +23,7 @@
 
 import urllib.parse
 import asyncio
-from typing import Any, Callable, List, Literal, Optional, Union
+from typing import Any, Callable, List, Literal, Optional
 from abc import ABC, abstractmethod
 
 from google.adk.auth.auth_credential import (
@@ -111,7 +111,7 @@ def __init__(
 
     @abstractmethod
     async def _get_credential(
-        self, *, tool_context: Union[ToolContext | ReadonlyContext]
+        self, *, tool_context: ToolContext | ReadonlyContext
     ) -> AuthCredential:
         """Get or create authentication credential.
 
@@ -128,7 +128,7 @@ async def _execute_with_credential(
         self,
         *,
         args: dict[str, Any],
-        tool_context: Union[ToolContext | ReadonlyContext],
+        tool_context: ToolContext | ReadonlyContext,
         credential: AuthCredential,
     ) -> Any:
         """Execute the tool with the provided credential.
@@ -147,7 +147,7 @@ async def run_with_identity_auth(
         self,
         *,
         args: dict[str, Any],
-        tool_context: Union[ToolContext | ReadonlyContext],
+        tool_context: ToolContext | ReadonlyContext,
     ) -> Any:
         """Execute the tool with Identity authentication.
 
@@ -177,7 +177,7 @@ class ApiKeyAuthMixin(BaseAuthMixin):
     """
 
     async def _get_credential(
-        self, *, tool_context: Union[ToolContext | ReadonlyContext]
+        self, *, tool_context: ToolContext | ReadonlyContext
     ) -> AuthCredential:
         """Get or create API key credential.
 
@@ -225,7 +225,7 @@ async def _execute_with_credential(
         self,
         *,
         args: dict[str, Any],
-        tool_context: Union[ToolContext | ReadonlyContext],
+        tool_context: ToolContext | ReadonlyContext,
         credential: AuthCredential,
     ) -> Any:
         """Default implementation - should be overridden by concrete tool classes.
@@ -259,7 +259,7 @@ class WorkloadAuthMixin(BaseAuthMixin):
     """
 
     async def _get_credential(
-        self, *, tool_context: Union[ToolContext | ReadonlyContext]
+        self, *, tool_context: ToolContext | ReadonlyContext
     ) -> AuthCredential:
         """Get or create Workload Access Token credential.
 
@@ -301,7 +301,7 @@ async def _execute_with_credential(
         self,
         *,
         args: dict[str, Any],
-        tool_context: Union[ToolContext | ReadonlyContext],
+        tool_context: ToolContext | ReadonlyContext,
         credential: AuthCredential,
     ) -> Any:
         """Default implementation - should be overridden by concrete tool classes.
@@ -369,7 +369,7 @@ def __init__(
         self._oauth2_auth_poller = oauth2_auth_poller
 
     async def _get_oauth2_token_or_auth_url(
-        self, *, tool_context: Union[ToolContext | ReadonlyContext]
+        self, *, tool_context: ToolContext | ReadonlyContext
     ) -> OAuth2TokenResponse:
         """Retrieve OAuth2 token or authorization URL from identity service.
 
@@ -396,7 +396,7 @@ async def _get_oauth2_token_or_auth_url(
         )
 
     async def _get_credential(
-        self, *, tool_context: Union[ToolContext | ReadonlyContext]
+        self, *, tool_context: ToolContext | ReadonlyContext
     ) -> AuthCredential:
         """Get or create OAuth2 credential.
 
@@ -527,7 +527,7 @@ async def _execute_with_credential(
         self,
         *,
         args: dict[str, Any],
-        tool_context: Union[ToolContext | ReadonlyContext],
+        tool_context: ToolContext | ReadonlyContext,
         credential: AuthCredential,
     ) -> Any:
         """Default implementation - should be overridden by concrete tool classes.
@@ -611,7 +611,7 @@ def _create_auth_delegate(self, **kwargs):
             raise ValueError(f"Unsupported auth config type: {type(self._auth_config)}")
 
     async def _get_credential(
-        self, *, tool_context: Union[ToolContext | ReadonlyContext]
+        self, *, tool_context: ToolContext | ReadonlyContext
     ) -> AuthCredential:
         """Get or create authentication credential using the configured auth type."""
         return await self._auth_delegate._get_credential(tool_context=tool_context)
@@ -620,7 +620,7 @@ async def _execute_with_credential(
         self,
         *,
         args: dict[str, Any],
-        tool_context: Union[ToolContext | ReadonlyContext],
+        tool_context: ToolContext | ReadonlyContext,
         credential: AuthCredential,
     ) -> Any:
         """Default implementation - should be overridden by concrete tool classes.

veadk/integrations/ve_identity/models.py

@@ -198,7 +198,7 @@ def validate_discovery_url(cls, v: Optional[str]) -> Optional[str]:
 
 
 class WorkloadToken(BaseModel):
-    """workload access token and expiration time.
+    """Workload access token and expiration time.
 
     Attributes:
         workload_access_token: The workload access token.