github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-12 22:28:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply lint

Browse Source
This commit is contained in:
Sadra Barikbin 2025-07-27 02:13:14 +03:30
parent 4b74e35532
commit d42865a58f
1 changed files with 34 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -683,28 +683,48 @@ def test_prompt_with_template_variable_name_jinja2() -> None:
assert prompt.invoke({"template": "bar"}).to_string() == "This is a bar test." assert prompt.invoke({"template": "bar"}).to_string() == "This is a bar test."
def test_prompt_template_add_with_with_another_format(): def test_prompt_template_add_with_with_another_format() -> None:
with pytest.raises(match=r"Cannot add templates"): 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( @pytest.mark.parametrize(
"template_format, prompt1, prompt2", ("template_format", "prompt1", "prompt2"),
[ [
("f-string", "This is a {variable}", ". This is {another_variable}"), ("f-string", "This is a {variable}", ". This is {another_variable}"),
pytest.param("jinja2", "This is a {{variable}}", ". This is {{another_variable}}", marks=[pytest.mark.requires("jinja2")]), pytest.param(
("mustache", "This is a {{variable}}", ". This is {{another_variable}}"), "jinja2",
] "This is a {{variable}}",
". This is {{another_variable}}",
marks=[pytest.mark.requires("jinja2")]
),
("mustache", "This is a {{variable}}", ". This is {{another_variable}}"),
]
) )
def test_prompt_template_add(template_format: str, prompt1: str, prompt2: str) -> None: def test_prompt_template_add(template_format: str, prompt1: str, prompt2: str) -> None:
first_prompt = PromptTemplate.from_template(prompt1, template_format=template_format) first_prompt = PromptTemplate.from_template(
second_prompt = PromptTemplate.from_template(prompt2, template_format=template_format) prompt1,
template_format=template_format,
)
second_prompt = PromptTemplate.from_template(
prompt2,
template_format=template_format,
)
concated_prompt = first_prompt + second_prompt concated_prompt = first_prompt + second_prompt
prompt_of_concated = PromptTemplate.from_template(prompt1 + prompt2, template_format=template_format) prompt_of_concated = PromptTemplate.from_template(
prompt1 + prompt2,
template_format=template_format,
)
assert concated_prompt.input_variables == prompt_of_concated.input_variables assert concated_prompt.input_variables == prompt_of_concated.input_variables
assert ( assert (
concated_prompt.format(variable="template", another_variable="other_template") == concated_prompt.format(
prompt_of_concated.format(variable="template", another_variable="other_template") variable="template",
another_variable="other_template",
) ==
prompt_of_concated.format(
variable="template",
another_variable="other_template",
)
) )