From 92ebe132e84ebed58d80d661af3eb4755d874b7e Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 31 Jan 2024 11:26:45 -0800
Subject: [PATCH] test(test_keys.py): fix test

---
 tests/test_keys.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/test_keys.py b/tests/test_keys.py
index a3bd3f0c6..2cc93cf73 100644
--- 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"):