Bladeren bron

[Bug fix] Fix test (#1255)

Deshraj Yadav 1 jaar geleden
bovenliggende
commit
38e212c721
1 gewijzigde bestanden met toevoegingen van 1 en 1 verwijderingen
  1. 1 1
      tests/llm/test_generate_prompt.py

+ 1 - 1
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)