test(test_keys.py): fix test

2024-01-31 11:26:45 -08:00 · 2024-01-31 11:26:45 -08:00 · 92ebe132e8
commit 92ebe132e8
parent a6d7e54cda
1 changed files with 5 additions and 2 deletions
--- a/tests/test_keys.py
+++ b/tests/test_keys.py
@ -122,8 +122,11 @@ async def chat_completion(session, key, model="gpt-4"):
                    )

                return await response.json()
-        except:
-            pass
+        except Exception as e:
+            if "Request did not return a 200 status code" in str(e):
+                raise e
+            else:
+                pass


 async def chat_completion_streaming(session, key, model="gpt-4"):