Revise gpt prompt

ryansurf · Oct 8, 2024 · 5c6f39b · 5c6f39b
1 parent 1c93982
commit 5c6f39b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_gpt.py b/tests/test_gpt.py
@@ -14,7 +14,7 @@ def test_simple_gpt():
     "gpt works". If anything else is outputted, we can
     assume an error has occured
     """
-    surf_summary = "Please only output: 'gpt works!' "
+    surf_summary = "Please respond with the exact phrase 'gpt works'. Do not include any additional text or context."
     gpt_prompt = "This is for testing purposes"
     gpt_response = gpt.simple_gpt(surf_summary, gpt_prompt)
-    assert "gpt works" in gpt_response
+    assert "gpt works" in gpt_response, f"Expected 'gpt works', but got: {gpt_response}"