fix: Fix imports from mcp 1.20.0

Author: clareliguori

e2e_tests/python/server_clients/automated_oauth.py

@@ -16,7 +16,8 @@
 import httpx
 from botocore.exceptions import ClientError
 
-from mcp.client.auth import OAuthClientProvider, TokenStorage, OAuthContext
+from mcp.client.auth import OAuthClientProvider, TokenStorage
+from mcp.client.auth.oauth2 import OAuthContext
 from mcp.client.session import ClientSession
 from mcp.client.streamable_http import streamablehttp_client, MCP_PROTOCOL_VERSION
 from mcp.shared.auth import (