diff --git a/src/maggma/api/resource/read_resource.py b/src/maggma/api/resource/read_resource.py index 5e2703041..8f429df66 100644 --- a/src/maggma/api/resource/read_resource.py +++ b/src/maggma/api/resource/read_resource.py @@ -173,7 +173,10 @@ def get_by_key( response = Response(orjson.dumps(response, default=serialization_helper)) # type: ignore if self.header_processor is not None: - self.header_processor.process_header(response, request) + if self.disable_validation: + self.header_processor.process_header(response, request) + else: + self.header_processor.process_header(temp_response, request) return response @@ -269,7 +272,10 @@ def search(**queries: Dict[str, STORE_PARAMS]) -> Union[Dict, Response]: response = Response(orjson.dumps(response, default=serialization_helper)) # type: ignore if self.header_processor is not None: - self.header_processor.process_header(response, request) + if self.disable_validation: + self.header_processor.process_header(response, request) + else: + self.header_processor.process_header(temp_response, request) return response diff --git a/src/maggma/api/resource/s3_url.py b/src/maggma/api/resource/s3_url.py index c4d1176d1..0eaf899e0 100644 --- a/src/maggma/api/resource/s3_url.py +++ b/src/maggma/api/resource/s3_url.py @@ -123,7 +123,10 @@ def get_by_key( response = Response(orjson.dumps(response, default=serialization_helper)) # type: ignore if self.header_processor is not None: - self.header_processor.process_header(response, request) + if self.disable_validation: + self.header_processor.process_header(response, request) + else: + self.header_processor.process_header(temp_response, request) return response