diff --git a/libs/core/tests/unit_tests/prompts/test_prompt.py b/libs/core/tests/unit_tests/prompts/test_prompt.py index 66df3bc5b84..8ae3156e82d 100644 --- a/libs/core/tests/unit_tests/prompts/test_prompt.py +++ b/libs/core/tests/unit_tests/prompts/test_prompt.py @@ -685,7 +685,10 @@ def test_prompt_with_template_variable_name_jinja2() -> None: def test_prompt_template_add_with_with_another_format() -> None: with pytest.raises(ValueError, match=r"Cannot add templates"): - PromptTemplate.from_template("This is a {template}") + PromptTemplate.from_template("So {{this}} is", template_format="mustache") + ( + PromptTemplate.from_template("This is a {template}") + + PromptTemplate.from_template("So {{this}} is", template_format="mustache") + ) @pytest.mark.parametrize(