diff --git a/src/oidc-test/main.py b/src/oidc-test/main.py index e0f999c..a53083a 100644 --- a/src/oidc-test/main.py +++ b/src/oidc-test/main.py @@ -13,6 +13,7 @@ from fastapi.security import ( OAuth2AuthorizationCodeBearer, ) from starlette.middleware.sessions import SessionMiddleware +from authlib.integrations.starlette_client.apps import StarletteOAuth2App from authlib.integrations.starlette_client import OAuth, OAuthError from pydantic import BaseModel from pydantic_settings import ( @@ -174,7 +175,7 @@ async def current_user(request: Request, token: str | None = Depends(fastapi_oau async def login(request: Request, provider: str) -> RedirectResponse: redirect_uri = request.url_for("auth", provider=provider) try: - provider_ = getattr(authlib_oauth, provider) + provider_: StarletteOAuth2App = getattr(authlib_oauth, provider) except AttributeError: raise HTTPException(500, "") try: @@ -186,7 +187,7 @@ async def login(request: Request, provider: str) -> RedirectResponse: @app.get("/auth/{provider}") async def auth(request: Request, provider: str) -> RedirectResponse: try: - provider_ = getattr(authlib_oauth, provider) + provider_: StarletteOAuth2App = getattr(authlib_oauth, provider) except AttributeError: raise HTTPException(500, "") try: