diff --git a/main.py b/main.py index cc3fc19..f74e315 100644 --- a/main.py +++ b/main.py @@ -120,7 +120,7 @@ async def require_iap(request: Request, call_next): from google.auth.transport import requests as grequests id_token.verify_oauth2_token( token, grequests.Request(), - audience="https://osvauco-agent-zjbqp3prqq-uc.a.run.app" + audience=os.environ.get("SERVICE_URL", "https://osvauco-agent-357036551735.us-central1.run.app") ) return await call_next(request) except Exception as e: @@ -199,7 +199,7 @@ def require_auth(func): from google.auth.transport import requests as grequests idinfo = id_token.verify_oauth2_token( token, grequests.Request(), - audience="https://osvauco-agent-zjbqp3prqq-uc.a.run.app" + audience=os.environ.get("SERVICE_URL", "https://osvauco-agent-357036551735.us-central1.run.app") ) request.session['user'] = {'email': idinfo.get('email', 'service-account')} return await func(request, *args, **kwargs)