From 8a865037ff8f1578e825efb32a99643ce1e9638b Mon Sep 17 00:00:00 2001 From: TShapinsky Date: Fri, 15 Dec 2023 09:45:00 -0700 Subject: [PATCH] Align response format --- alfalfa_client/alfalfa_client.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/alfalfa_client/alfalfa_client.py b/alfalfa_client/alfalfa_client.py index a5c7129..02c3e29 100644 --- 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()