From 79fe41f65a2ee017de9aa8492313704285e7d456 Mon Sep 17 00:00:00 2001 From: Max Schaefer Date: Thu, 11 Nov 2021 11:57:56 +0000 Subject: [PATCH] Fix recording of reponse headers in exceptions. These were passed as positional arguments, but in the wrong position, which meant they were passed into the `json_body` parameter of the `OpenAIError` constructor. --- openai/api_requestor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/openai/api_requestor.py b/openai/api_requestor.py index 1923e9bac9..a79f8e0450 100644 --- a/openai/api_requestor.py +++ b/openai/api_requestor.py @@ -300,7 +300,7 @@ def _interpret_response_line( ) -> OpenAIResponse: if rcode == 503: raise error.ServiceUnavailableError( - "The server is overloaded or not ready yet.", rbody, rcode, rheaders + "The server is overloaded or not ready yet.", rbody, rcode, headers=rheaders ) try: if hasattr(rbody, "decode"): @@ -308,7 +308,7 @@ def _interpret_response_line( data = json.loads(rbody) except (JSONDecodeError, UnicodeDecodeError): raise error.APIError( - f"HTTP code {rcode} from API ({rbody})", rbody, rcode, rheaders + f"HTTP code {rcode} from API ({rbody})", rbody, rcode, headers=rheaders ) resp = OpenAIResponse(data, rheaders) # In the future, we might add a "status" parameter to errors