From 8dc4c80e66ca51637d675973c69718924d3e46d9 Mon Sep 17 00:00:00 2001 From: Anton Date: Fri, 19 Jul 2024 08:28:35 +0500 Subject: [PATCH] test: change default model and prompt --- tests/test_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_config.py b/tests/test_config.py index f4c1d4c..989b405 100644 --- 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)