refactor(mcp): Make create_mcp_oauth_handler public and use top-level import

This commit is contained in:
Zamil Majdy
2026-02-10 17:08:03 +04:00
parent 79d6e8e2d7
commit cb7a0cbdd7
2 changed files with 7 additions and 8 deletions

View File

@@ -40,7 +40,10 @@ from backend.data.user import get_user_integrations
from backend.executor.utils import add_graph_execution
from backend.integrations.ayrshare import AyrshareClient, SocialPlatform
from backend.integrations.credentials_store import provider_matches
from backend.integrations.creds_manager import IntegrationCredentialsManager
from backend.integrations.creds_manager import (
IntegrationCredentialsManager,
create_mcp_oauth_handler,
)
from backend.integrations.oauth import CREDENTIALS_BY_PROVIDER, HANDLERS_BY_NAME
from backend.integrations.providers import ProviderName
from backend.integrations.webhooks import get_webhook_manager
@@ -351,11 +354,7 @@ async def delete_credentials(
if isinstance(creds, OAuth2Credentials):
if provider_matches(provider.value, ProviderName.MCP.value):
# MCP uses dynamic per-server OAuth — create handler from metadata
from backend.integrations.creds_manager import (
_create_mcp_oauth_handler,
)
handler = _create_mcp_oauth_handler(creds)
handler = create_mcp_oauth_handler(creds)
else:
handler = _get_provider_oauth_handler(request, provider)
tokens_revoked = await handler.revoke_tokens(creds)

View File

@@ -141,7 +141,7 @@ class IntegrationCredentialsManager:
) -> OAuth2Credentials:
async with self._locked(user_id, credentials.id, "refresh"):
if provider_matches(credentials.provider, ProviderName.MCP.value):
oauth_handler = _create_mcp_oauth_handler(credentials)
oauth_handler = create_mcp_oauth_handler(credentials)
else:
oauth_handler = await _get_provider_oauth_handler(credentials.provider)
if oauth_handler.needs_refresh(credentials):
@@ -244,7 +244,7 @@ async def _get_provider_oauth_handler(provider_name_str: str) -> "BaseOAuthHandl
)
def _create_mcp_oauth_handler(
def create_mcp_oauth_handler(
credentials: OAuth2Credentials,
) -> "BaseOAuthHandler":
"""Create an MCPOAuthHandler from credential metadata for token refresh.