request headers for no logging works

Author: winzamark123

jigsawstack/request.py

@@ -12,7 +12,7 @@
 class RequestConfig(TypedDict):
     api_url: str
     api_key: str
-    disable_request_logging: Union[bool, None] = False
+    headers: Dict[str, str]
 
 
 # This class wraps the HTTP request creation logic
@@ -34,7 +34,9 @@ def __init__(
         self.api_key = config.get("api_key")
         self.data = data
         self.headers = headers
-        self.disable_request_logging = config.get("disable_request_logging")
+        self.disable_request_logging = config.get("headers", {}).get(
+            "x-jigsaw-no-request-log"
+        )
         self.stream = stream
 
     def perform(self) -> Union[T, None]:
@@ -50,13 +52,16 @@ def perform(self) -> Union[T, None]:
         """
         resp = self.make_request(url=f"{self.api_url}{self.path}")
 
-        #for binary responses
+        # for binary responses
         if resp.status_code == 200:
             content_type = resp.headers.get("content-type", "")
-            if not resp.text or any(t in content_type for t in ["audio/", "image/", "application/octet-stream", "image/png"]):
+            if not resp.text or any(
+                t in content_type
+                for t in ["audio/", "image/", "application/octet-stream", "image/png"]
+            ):
                 return cast(T, resp.content)
 
-        #for json resposes.
+        # for json resposes.
         if resp.status_code != 200:
             try:
                 error = resp.json()
@@ -94,7 +99,6 @@ def perform_file(self) -> Union[T, None]:
             raise_for_code_and_type(
                 code=500,
                 message="Failed to parse JigsawStack API response. Please try again.",
-                error_type="InternalServerError",
             )
 
         if resp.status_code != 200:
@@ -105,7 +109,7 @@ def perform_file(self) -> Union[T, None]:
                 err=error.get("error"),
             )
 
-        #for binary responses
+        # for binary responses
         if resp.status_code == 200:
             content_type = resp.headers.get("content-type", "")
             if "application/json" not in content_type: