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

core[minor]: Implement aformat for FewShotPromptWithTemplates (#20039)

Browse Source
This commit is contained in:
Christophe Bornet
2024-04-09 21:58:41 +02:00
committed by GitHub
parent 855ba46f80
commit 19001e6cb9
2 changed files with 48 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/core/tests/unit_tests/prompts/test_few_shot_with_templates.py
View File

@@ -10,7 +10,7 @@ EXAMPLE_PROMPT = PromptTemplate(
)
def test_prompttemplate_prefix_suffix() -> None:
async def test_prompttemplate_prefix_suffix() -> None:
"""Test that few shot works when prefix and suffix are PromptTemplates."""
prefix = PromptTemplate(
input_variables=["content"], template="This is a test about {content}."
@@ -32,13 +32,15 @@ def test_prompttemplate_prefix_suffix() -> None:
example_prompt=EXAMPLE_PROMPT,
example_separator="\n",
)
output = prompt.format(content="animals", new_content="party")
expected_output = (
"This is a test about animals.\n"
"foo: bar\n"
"baz: foo\n"
"Now you try to talk about party."
)
output = prompt.format(content="animals", new_content="party")
assert output == expected_output
output = await prompt.aformat(content="animals", new_content="party")
assert output == expected_output