Display 401 error messages

Author: w0rp

neural_sources/chatgpt.py

@@ -191,7 +191,7 @@ def main() -> None:
     try:
         get_chatgpt_completion(config, input_data["prompt"])
     except urllib.error.HTTPError as error:
-        if error.code == 400:
+        if error.code == 400 or error.code == 401:
             message = get_error_message(error)
             sys.exit('Neural error: OpenAI request failure: ' + message)
         elif error.code == 429:

neural_sources/openai.py

@@ -181,7 +181,7 @@ def main() -> None:
     try:
         get_openai_completion(config, input_data["prompt"])
     except urllib.error.HTTPError as error:
-        if error.code == 400:
+        if error.code == 400 or error.code == 401:
             message = get_error_message(error)
             sys.exit('Neural error: OpenAI request failure: ' + message)
         elif error.code == 429:

test/python/test_chatgpt.py

@@ -175,6 +175,16 @@ def test_main_function_bad_config():
             'OpenAI request failure: Too much text for a request!',
             id="too_much_text",
         ),
+        pytest.param(
+            401,
+            json.dumps({
+                'error': {
+                    'message': "Bad authentication error",
+                },
+            }),
+            'OpenAI request failure: Bad authentication error',
+            id="unauthorised_failure",
+        ),
     )
 )
 def test_api_error(

test/python/test_openai.py

@@ -168,6 +168,16 @@ def test_main_function_bad_config():
             'OpenAI request failure: Too much text for a request!',
             id="too_much_text",
         ),
+        pytest.param(
+            401,
+            json.dumps({
+                'error': {
+                    'message': "Bad authentication error",
+                },
+            }),
+            'OpenAI request failure: Bad authentication error',
+            id="unauthorised_failure",
+        ),
     )
 )
 def test_api_error(