add response header check back

materialsproject · Aug 9, 2024 · c2c0796 · c2c0796
1 parent 5a472b1
commit c2c0796
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/maggma/api/resource/read_resource.py b/src/maggma/api/resource/read_resource.py
@@ -281,6 +281,12 @@ def search(**queries: dict[str, STORE_PARAMS]) -> Union[dict, Response]:
             if self.disable_validation:
                 response = Response(orjson.dumps(response, default=serialization_helper))  # type: ignore
 
+            if self.header_processor is not None:
+                if self.disable_validation:
+                    self.header_processor.process_header(response, request)
+                else:
+                    self.header_processor.process_header(temp_response, request)
+
             return response
 
         self.router.get(