test: fixup prompt

Author: lpm0073

models/tests/test_examples.py

@@ -15,7 +15,7 @@
 from models.prompt_templates import NetecPromptTemplates
 
 
-HUMAN_MESSAGE = 'return the word "SUCCESS" in upper case.'
+HUMAN_MESSAGE = "this is a test"
 
 
 class TestExamples:
@@ -25,7 +25,10 @@ class TestExamples:
     def test_prompt(self, mock_parse_args):
         """Test prompt example."""
         mock_args = MagicMock()
-        mock_args.system_prompt = "you are a helpful assistant"
+        mock_args.system_prompt = """you are a helpful assistant. If you are prompted,
+        'this is a test', then return the word 'SUCCESS' in upper case. Return only
+        this single word, in upper case. Do not embellish. do not further prompt
+        the user for any reason."""
         mock_args.human_prompt = HUMAN_MESSAGE
         mock_parse_args.return_value = mock_args