tests: Raise an exception if the response comes back empty

YeetCodeDevs · Mar 18, 2024 · 33e39d2 · 33e39d2
1 parent b9103a4
commit 33e39d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/test_bing.py b/tests/test_bing.py
@@ -31,6 +31,7 @@ def generate_response() -> str:
         )
     except:
         print("ERROR: Could not create a prompt!")
+        raise Exception("ERROR: Could not create a prompt!")
 
     return response.choices[0].message.content
 

diff --git a/tests/test_freechatgpt.py b/tests/test_freechatgpt.py
@@ -15,6 +15,7 @@ def generate_response() -> str:
         )
     except:
         print("ERROR: Could not create a prompt!")
+        raise Exception("ERROR: Could not create a prompt!")
 
     return response.choices[0].message.content
-Original file line number
+Diff line change
@@ Expand Up / @@ -31,6 +31,7 @@ def generate_response() -> str: @@
             )
         except:
             print("ERROR: Could not create a prompt!")
+            raise Exception("ERROR: Could not create a prompt!")
         return response.choices[0].message.content
@@ Expand Down @@