From 38e212c721a1aa5cab090c16f0f142d45c92355b Mon Sep 17 00:00:00 2001 From: Deshraj Yadav Date: Sun, 11 Feb 2024 17:57:14 -0800 Subject: [PATCH] [Bug fix] Fix test (#1255) --- tests/llm/test_generate_prompt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/llm/test_generate_prompt.py b/tests/llm/test_generate_prompt.py index 31f5fddd..e283c3f8 100644 --- a/tests/llm/test_generate_prompt.py +++ b/tests/llm/test_generate_prompt.py @@ -66,5 +66,5 @@ class TestGeneratePrompt(unittest.TestCase): self.app.llm.set_history(["Past context 1", "Past context 2"]) prompt = self.app.llm.generate_prompt("Test query", ["Test context"]) - expected_prompt = "Context: Test context | Query: Test query | History: ['Past context 1', 'Past context 2']" + expected_prompt = "Context: Test context | Query: Test query | History: Past context 1\nPast context 2" self.assertEqual(prompt, expected_prompt)