From 5c6f39bf10120a6bdf4da88899ceea9dc26f3ca8 Mon Sep 17 00:00:00 2001 From: macnult <113482585+macnult@users.noreply.github.com> Date: Tue, 8 Oct 2024 11:52:45 -0400 Subject: [PATCH] Revise gpt prompt --- tests/test_gpt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_gpt.py b/tests/test_gpt.py index c683423..fe8d2ef 100644 --- 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}"