Skip to content

Commit

Permalink
Merge pull request #147 from macnult/main
Browse files Browse the repository at this point in the history
Revise GPT prompt
  • Loading branch information
ryansurf authored Oct 21, 2024
2 parents 679b475 + 463cd47 commit f7f9676
Showing 1 changed file with 11 additions and 2 deletions.
13 changes: 11 additions & 2 deletions tests/test_gpt.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,16 @@ 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
expected_response = "gpt works"

assert gpt_response == expected_response, (
f"Expected '{expected_response}', but got: {gpt_response}"
)

0 comments on commit f7f9676

Please sign in to comment.