tests: bing: Add error handling during prompt creation

YeetCodeDevs · Mar 17, 2024 · 5d46daa · 5d46daa
1 parent afa6853
commit 5d46daa
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tests/test_bing.py b/tests/test_bing.py
@@ -21,10 +21,13 @@
 def generate_response() -> str:
     client = Client(provider=Bing)
 
-    response = client.chat.completions.create(
-        model="gpt-4.0-turbo",
-        messages=[{"role": "user", "content": "Say hi, with your response starting with START and ending with END"}],
-    )
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4.0-turbo",
+            messages=[{"role": "user", "content": "Say hi, with your response starting with START and ending with END"}],
+        )
+    except:
+        print("ERROR: Could not create a prompt!")
 
     return response.choices[0].message.content