test(test_keys.py): fix test

2026-08-18 06:26:16 +00:00 · 2024-01-31 11:26:49 -08:00
parent a6d7e54cda
commit 92ebe132e8
1 changed files with 5 additions and 2 deletions
@@ -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"):