From 8cdac3d2f8edfeef25a70abfea16b512a3243c52 Mon Sep 17 00:00:00 2001
From: Max Isbey <224885523+maxisbey@users.noreply.github.com>
Date: Wed, 29 Oct 2025 18:32:26 +0000
Subject: [PATCH] Fix pyright error and replace wildcard import with explicit
 imports (#1532)

---
 src/mcp/client/auth/__init__.py            | 18 +++++++++++++++++-
 src/mcp/client/auth/extensions/__init__.py |  0
 2 files changed, 17 insertions(+), 1 deletion(-)
 create mode 100644 src/mcp/client/auth/extensions/__init__.py

diff --git a/src/mcp/client/auth/__init__.py b/src/mcp/client/auth/__init__.py
index c43a29ce..a5c4b734 100644
--- a/src/mcp/client/auth/__init__.py
+++ b/src/mcp/client/auth/__init__.py
@@ -4,4 +4,20 @@ OAuth2 Authentication implementation for HTTPX.
 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,
+    OAuthFlowError,
+    OAuthRegistrationError,
+    OAuthTokenError,
+    PKCEParameters,
+    TokenStorage,
+)
+
+__all__ = [
+    "OAuthClientProvider",
+    "OAuthFlowError",
+    "OAuthRegistrationError",
+    "OAuthTokenError",
+    "PKCEParameters",
+    "TokenStorage",
+]
diff --git a/src/mcp/client/auth/extensions/__init__.py b/src/mcp/client/auth/extensions/__init__.py
new file mode 100644
index 00000000..e69de29b