Migrate from template to prompt arg while keeping backward compatibility (#1066)

2023-12-28 23:36:33 +05:30
parent 12e6eaf802
commit d9d529987e
9 changed files with 56 additions and 42 deletions
--- a/tests/helper_classes/test_json_serializable.py
+++ b/tests/helper_classes/test_json_serializable.py
@@ -76,4 +76,4 @@ class TestJsonSerializable(unittest.TestCase):
        config = BaseLlmConfig(template=Template("My custom template with $query, $context and $history."))
        s = config.serialize()
        new_config: BaseLlmConfig = BaseLlmConfig.deserialize(s)
-        self.assertEqual(config.template.template, new_config.template.template)
+        self.assertEqual(config.prompt.template, new_config.prompt.template)