klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Bug fix] Fix test (#1255)

Browse Source
This commit is contained in:
Deshraj Yadav
2024-02-11 17:57:14 -08:00
committed by GitHub
parent 2f285ea00a
commit 38e212c721
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/llm/test_generate_prompt.py
View File

@@ -66,5 +66,5 @@ class TestGeneratePrompt(unittest.TestCase):
self.app.llm.set_history(["Past context 1", "Past context 2"]) self.app.llm.set_history(["Past context 1", "Past context 2"])
prompt = self.app.llm.generate_prompt("Test query", ["Test context"]) 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) self.assertEqual(prompt, expected_prompt)