Align response format

NREL · Dec 15, 2023 · 8a86503 · 8a86503
1 parent d21c09c
commit 8a86503
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/alfalfa_client/alfalfa_client.py b/alfalfa_client/alfalfa_client.py
@@ -81,10 +81,13 @@ def _request(self, endpoint: str, method="POST", parameters=None) -> requests.Re
         else:
             response = requests.request(method=method, url=self.url + endpoint)
 
-        if response.status_code == 400 or response.status_code == 500:
+        if response.status_code >= 400:
             try:
                 body = response.json()
-                raise AlfalfaAPIException(body["error"])
+                exception = AlfalfaAPIException(body["message"])
+                if "payload" in body:
+                    exception.add_note(json.dumps(body["payload"]))
+                raise exception
             except json.JSONDecodeError:
                 pass
         response.raise_for_status()