github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-03 20:16:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

core: prompt variable error msg (#25787)

Browse Source
This commit is contained in:
Erick Friis
2024-08-28 15:54:00 -07:00
committed by GitHub
parent ff168aaec0
commit c8b8335b82
2 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
libs/core/tests/unit_tests/prompts/test_prompt.py
View File

@@ -702,3 +702,15 @@ def test_prompt_falsy_vars(
expected if not isinstance(expected, dict) else expected[template_format]
)
assert result.to_string() == expected_output
def test_prompt_missing_vars_error() -> None:
prompt = PromptTemplate.from_template("This is a {foo} {goingtobemissing} test.")
with pytest.raises(KeyError) as e:
prompt.invoke({"foo": "bar"})
# Check that the error message contains the missing variable
assert "{'goingtobemissing'}" in str(e.value.args[0])
# Check helper text has right number of braces
assert "'{{goingtobemissing}}'" in str(e.value.args[0])