Fix logout route signature

dudil · Dec 4, 2023 · 08d0573 · 08d0573
1 parent 2669da4
commit 08d0573
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/fastapi_msal/auth.py b/fastapi_msal/auth.py
@@ -76,11 +76,10 @@ async def _post_token_route(
         return BearerToken(access_token=token.id_token)
 
     async def _logout_route(
-        self, request: Request, callback_url: OptStr, referer: Annotated[OptStr, Header()] = None
+        self, request: Request, referer: Annotated[OptStr, Header()] = None, callback_url: OptStr = None
     ) -> RedirectResponse:
         # check if callback_url is set, if not try to get it from referer header
-        if not callback_url:
-            callback_url = referer if referer else str(self.return_to_path)
+        callback_url = referer if referer else str(self.return_to_path)
         return self.handler.logout(request=request, callback_url=callback_url)
 
     async def get_session_token(self, request: Request) -> Optional[AuthToken]: