diff --git a/src/ad_seller/clients/freewheel_adapter.py b/src/ad_seller/clients/freewheel_adapter.py index ca25952..23a8e28 100644 --- a/src/ad_seller/clients/freewheel_adapter.py +++ b/src/ad_seller/clients/freewheel_adapter.py @@ -27,7 +27,6 @@ BookingResult, ) from .freewheel_mcp_client import FreeWheelMCPClient -from .freewheel_oauth import FreeWheelOAuthManager, build_oauth_mcp_url from .freewheel_normalizer import ( micros_to_dollars, normalize_audience_segments, @@ -35,6 +34,7 @@ normalize_deal, normalize_inventory, ) +from .freewheel_oauth import FreeWheelOAuthManager, build_oauth_mcp_url logger = logging.getLogger(__name__) @@ -65,16 +65,12 @@ def _get_settings(self) -> Any: def _get_sh_oauth_manager(self) -> FreeWheelOAuthManager: if self._sh_oauth_manager is None: - self._sh_oauth_manager = FreeWheelOAuthManager.for_provider( - self._get_settings(), "sh" - ) + self._sh_oauth_manager = FreeWheelOAuthManager.for_provider(self._get_settings(), "sh") return self._sh_oauth_manager def _get_bc_oauth_manager(self) -> FreeWheelOAuthManager: if self._bc_oauth_manager is None: - self._bc_oauth_manager = FreeWheelOAuthManager.for_provider( - self._get_settings(), "bc" - ) + self._bc_oauth_manager = FreeWheelOAuthManager.for_provider(self._get_settings(), "bc") return self._bc_oauth_manager # ========================================================================= diff --git a/src/ad_seller/clients/freewheel_oauth.py b/src/ad_seller/clients/freewheel_oauth.py index 3ddb7f0..c6739ae 100644 --- a/src/ad_seller/clients/freewheel_oauth.py +++ b/src/ad_seller/clients/freewheel_oauth.py @@ -354,7 +354,9 @@ async def bootstrap( webbrowser.open(authorize_url) try: - callback_params = await asyncio.to_thread(callback_server.wait_for_result, timeout_seconds) + callback_params = await asyncio.to_thread( + callback_server.wait_for_result, timeout_seconds + ) finally: callback_server.server_close() @@ -442,4 +444,4 @@ def _state_from_token_response( expires_at=expires_at.isoformat(), scope=token_response.get("scope", self.config.scope), token_type=token_response.get("token_type", "Bearer"), - ) \ No newline at end of file + )