diff --git a/lambda_proxy/proxy.py b/lambda_proxy/proxy.py index c16a363..d62929f 100644 --- a/lambda_proxy/proxy.py +++ b/lambda_proxy/proxy.py @@ -494,7 +494,7 @@ def new_func(*args, **kwargs) -> Callable: def setup_docs(self) -> None: """Add default documentation routes.""" - openapi_url = f"/openapi.json" + openapi_url = "/openapi.json" def _openapi() -> Tuple[str, str, str]: """Return OpenAPI json.""" @@ -704,15 +704,18 @@ def __call__(self, event, context): json.dumps({"errorMessage": str(err)}), ) - return self.response( - response[0], - response[1], - response[2], - cors=route_entry.cors, - accepted_methods=route_entry.methods, - accepted_compression=self.event["headers"].get("accept-encoding", ""), - compression=route_entry.compression, - b64encode=route_entry.b64encode, - ttl=route_entry.ttl, - cache_control=route_entry.cache_control, - ) + custom_response_params = response[3] if len(response) > 3 else {} + response_params = { + **dict( + cors=route_entry.cors, + accepted_methods=route_entry.methods, + accepted_compression=self.event["headers"].get("accept-encoding", ""), + compression=route_entry.compression, + b64encode=route_entry.b64encode, + ttl=route_entry.ttl, + cache_control=route_entry.cache_control, + ), + **custom_response_params, + } + + return self.response(response[0], response[1], response[2], **response_params)