tests: increase max tokens for full system prompt significantly

ErikBjare · Aug 20, 2024 · c785cfe · c785cfe
1 parent 3423474
commit c785cfe
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_prompts.py b/tests/test_prompts.py
@@ -1,10 +1,11 @@
-from gptme.prompts import get_prompt
 from gptme.message import len_tokens
+from gptme.prompts import get_prompt
 
 
 def test_get_prompt():
     prompt = get_prompt("full")
-    assert 1200 < len_tokens(prompt.content) < 2000
+    # TODO: lower this significantly by selectively removing examples from the full prompt
+    assert 1200 < len_tokens(prompt.content) < 3000
 
     prompt = get_prompt("short")
     assert 700 < len_tokens(prompt.content) < 900