diff --git a/gateway/common/authorization.py b/gateway/common/authorization.py index 12a68bc..124b8d3 100644 --- a/gateway/common/authorization.py +++ b/gateway/common/authorization.py @@ -12,7 +12,7 @@ def extract_guardrail_service_authorization_from_headers( request: Request, ) -> Tuple[Optional[str], Optional[str]]: """ - Extracts the optional Invariant-Guardrail-Service authorization header from the request. + Extracts the optional Invariant-Guardrails-Authorization authorization header from the request. This header can be specifified to use a different API key for guardrailing compared to Explorer interactions. diff --git a/gateway/common/request_context.py b/gateway/common/request_context.py index 618cf4d..68cb1d7 100644 --- a/gateway/common/request_context.py +++ b/gateway/common/request_context.py @@ -7,7 +7,7 @@ import fastapi from common.config_manager import GatewayConfig from common.guardrails import GuardrailRuleSet, Guardrail, GuardrailAction -from gateway.common.authorization import ( +from common.authorization import ( extract_guardrail_service_authorization_from_headers, )