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/llm/test_base_llm.py
+++ b/tests/llm/test_base_llm.py
@@ -25,7 +25,7 @@ def test_is_stream_bool():
 def test_template_string_gets_converted_to_Template_instance():
    config = BaseLlmConfig(template="test value $query $context")
    llm = BaseLlm(config=config)
-    assert isinstance(llm.config.template, Template)
+    assert isinstance(llm.config.prompt, Template)


 def test_is_get_llm_model_answer_implemented():