From ada2d9d5a8d05606702a8c8f18b9125a6472508f Mon Sep 17 00:00:00 2001 From: yannj-fr <4557670+yannj-fr@users.noreply.github.com> Date: Wed, 29 Oct 2025 18:40:37 +0100 Subject: [PATCH] fix import for oauth2.py --- src/mcp/client/auth/__init__.py | 20 +++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) diff --git a/src/mcp/client/auth/__init__.py b/src/mcp/client/auth/__init__.py index c43a29ce0..987beb25d 100644 --- a/src/mcp/client/auth/__init__.py +++ b/src/mcp/client/auth/__init__.py @@ -4,4 +4,22 @@ Implements authorization code flow with PKCE and automatic token refresh. """ -from mcp.client.auth.oauth2 import * # noqa: F403 +from mcp.client.auth.oauth2 import ( + OAuthClientProvider, + OAuthContext, + OAuthFlowError, + OAuthRegistrationError, + OAuthTokenError, + PKCEParameters, + TokenStorage, +) + +__all__ = [ + "OAuthClientProvider", + "OAuthContext", + "OAuthFlowError", + "OAuthRegistrationError", + "OAuthTokenError", + "PKCEParameters", + "TokenStorage", +]