test: change default model and prompt

nalgeon · Jul 19, 2024 · 8dc4c80 · 8dc4c80
1 parent 077abbb
commit 8dc4c80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_config.py b/tests/test_config.py
@@ -19,7 +19,7 @@ def test_init(self):
 
         self.assertEqual(config.openai.api_key, "oa-1234")
         self.assertEqual(config.openai.model, "gpt-4")
-        self.assertTrue(config.openai.prompt.startswith("Your primary goal"))
+        self.assertTrue(config.openai.prompt, "You are an AI assistant.")
         self.assertEqual(config.openai.params["temperature"], 0.7)
         self.assertEqual(config.openai.params["presence_penalty"], 0)
         self.assertEqual(config.openai.params["frequency_penalty"], 0)