From 77286d8bbaad33744d2ae258d9480351e09b63a3 Mon Sep 17 00:00:00 2001
From: Max Isbey <224885523+maxisbey@users.noreply.github.com>
Date: Wed, 29 Oct 2025 18:24:22 +0000
Subject: [PATCH] Fix pyright error and replace wildcard import with explicit
 imports

This addresses issues introduced in PR #1247:

1. Fixed pyright reportMissingTypeStubs error by adding __init__.py to the
   extensions directory, making it a proper Python package
2. Replaced wildcard import in auth/__init__.py with explicit imports as
   requested in code review

The changes ensure type checking passes and maintain backward compatibility
with existing code that imports from mcp.client.auth.
---
 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 c43a29ce08..a5c4b73464 100644
--- a/src/mcp/client/auth/__init__.py
+++ b/src/mcp/client/auth/__init__.py
@@ -4,4 +4,20 @@
 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 0000000000..e69de29bb2