Fixing empty input variable crashing PromptTemplate validations (#14314)

- Fixes `input_variables=[""]` crashing validations with a template
`"{}"`
- Uses `__cause__` for proper `Exception` chaining in
`check_valid_template`
This commit is contained in:
James Braza
2023-12-05 16:13:08 -05:00
committed by GitHub
parent 0f02e94565
commit 8b0060184d
2 changed files with 17 additions and 11 deletions

View File

@@ -47,6 +47,12 @@ def test_prompt_missing_input_variables() -> None:
).input_variables == ["foo"]
def test_prompt_empty_input_variable() -> None:
"""Test error is raised when empty string input variable."""
with pytest.raises(ValueError):
PromptTemplate(input_variables=[""], template="{}", validate_template=True)
def test_prompt_extra_input_variables() -> None:
"""Test error is raised when there are too many input variables."""
template = "This is a {foo} test."