arcade-mcp/libs/arcade-tdk/arcade_tdk/error_adapters/utils.py

17 lines
560 B
Python

from arcade_tdk.auth import Google, Microsoft, ToolAuthorization
from arcade_tdk.error_adapters import ErrorAdapter, GoogleErrorAdapter, MicrosoftGraphErrorAdapter
def get_adapter_for_auth_provider(auth_provider: ToolAuthorization | None) -> ErrorAdapter | None:
"""
Get an error adapter from an auth provider.
"""
if not auth_provider:
return None
if isinstance(auth_provider, Google):
return GoogleErrorAdapter()
if isinstance(auth_provider, Microsoft):
return MicrosoftGraphErrorAdapter()
return None