[Improvements] Improve the default prompt and data loader util functions (#1272)

2024-02-18 14:06:32 -08:00
parent 9a11683003
commit 6c12bc9044
19 changed files with 79 additions and 62 deletions
--- a/tests/llm/test_chat.py
+++ b/tests/llm/test_chat.py
@@ -64,7 +64,7 @@ class TestApp(unittest.TestCase):
            self.assertEqual(len(app.llm.history), 1)
            history = app.llm.history
            dry_run = app.chat("Test query 2", dry_run=True)
-            self.assertIn("History:", dry_run)
+            self.assertIn("Conversation history:", dry_run)
            self.assertEqual(history, app.llm.history)
            self.assertEqual(len(app.llm.history), 1)