fix(tool): fix mypy errors using cast and schema corrections

Author: anubhav756

packages/toolbox-adk/src/toolbox_adk/tool.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, Awaitable, Callable, Dict, Optional
+from typing import Any, Awaitable, Callable, Dict, Optional, cast
 
 import toolbox_core
 from fastapi.openapi.models import (
@@ -127,21 +127,23 @@ async def run_async(
                     oauth2=OAuth2Auth(
                         client_id=self._auth_config.client_id,
                         client_secret=self._auth_config.client_secret,
-                        scopes=scopes,
                     ),
                 ),
             )
 
             # Check if we already have credentials from a previous exchange
             try:
                 # get_auth_response returns AuthCredential if found
-                creds = tool_context.get_auth_response(auth_config_adk)
+                # Cast to Any because ReadonlyContext definition might be missing these methods in stubs
+                # but valid at runtime for strict ADK compliance.
+                ctx_any = cast(Any, tool_context)
+                creds = ctx_any.get_auth_response(auth_config_adk)
                 if creds and creds.oauth2 and creds.oauth2.access_token:
                     reset_token = USER_TOKEN_CONTEXT_VAR.set(creds.oauth2.access_token)
                 else:
                     # Request credentials. This will signal the runner to pause.
                     # We return None (or raise) to stop current execution.
-                    tool_context.request_credential(auth_config_adk)
+                    ctx_any.request_credential(auth_config_adk)
                     # Returning None here. The ADK runner will see the requested_auth_configs
                     # in the tool_context/event and trigger the client event.
                     return None
@@ -158,7 +160,8 @@ async def run_async(
                 ):  # Loose check, but safest is to re-raise
                     raise e
                 # Fallback to request logic if it was a lookup error?
-                tool_context.request_credential(auth_config_adk)
+                ctx_any = cast(Any, tool_context)
+                ctx_any.request_credential(auth_config_adk)
                 return None
 
         try:

packages/toolbox-adk/src/toolbox_adk/toolset.py

@@ -41,7 +41,7 @@ def __init__(
         bound_params: Optional[Mapping[str, Union[Callable[[], Any], Any]]] = None,
         pre_hook: Optional[Callable[[ToolboxContext], Awaitable[None]]] = None,
         post_hook: Optional[Callable[[ToolboxContext], Awaitable[None]]] = None,
-        **kwargs,
+        **kwargs: Any,
     ):
         """
         Args: