update test references

rgbkrk · Mar 24, 2023 · 8e75747 · 8e75747
1 parent fe23bd9
commit 8e75747
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tests/test_magics.py b/tests/test_magics.py
@@ -1,6 +1,6 @@
 from unittest import mock
 
-from genai.prompts import assist_prompt
+from genai.prompts import PromptStore
 from genai.context import PastAssists
 
 
@@ -27,7 +27,7 @@ def test_assist_magic(create, ip):
         messages=[
             {
                 "role": "system",
-                "content": assist_prompt,
+                "content": PromptStore.assist_prompt,
             },
             {
                 "role": "user",
@@ -66,7 +66,7 @@ def test_assist_magic_with_args(create, ip):
         messages=[
             {
                 "role": "system",
-                "content": assist_prompt,
+                "content": PromptStore.assist_prompt,
             },
             {
                 "role": "user",
@@ -108,7 +108,7 @@ def test_assist_magic_with_fresh_arg(create, ip):
         messages=[
             {
                 "role": "system",
-                "content": assist_prompt,
+                "content": PromptStore.assist_prompt,
             },
             # Note that there is zero other context, due to running with --fresh
             {

diff --git a/tests/test_suggestions.py b/tests/test_suggestions.py
@@ -59,7 +59,7 @@ def test_custom_exc(create, display, ip):
     assert len(kwargs["messages"]) == 3
     assert kwargs["messages"][0] == {
         "role": "system",
-        "content": error_prompt,
+        "content": PromptStore.exception_prompt,
     }
     assert kwargs["messages"][1] == {
         "role": "user",