mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-05 12:48:12 +00:00
Harrison/clean up language (#179)
dynamic prompts are no longer a thing
This commit is contained in:
parent
736b6ee65c
commit
4334ffa6f9
@ -1,4 +1,4 @@
|
|||||||
"""Test functionality related to dynamic prompts."""
|
"""Test functionality related to length based selector."""
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from langchain.prompts.example_selector.length_based import LengthBasedExampleSelector
|
from langchain.prompts.example_selector.length_based import LengthBasedExampleSelector
|
||||||
@ -22,15 +22,15 @@ def selector() -> LengthBasedExampleSelector:
|
|||||||
return selector
|
return selector
|
||||||
|
|
||||||
|
|
||||||
def test_dynamic_prompt_valid(selector: LengthBasedExampleSelector) -> None:
|
def test_selector_valid(selector: LengthBasedExampleSelector) -> None:
|
||||||
"""Test dynamic prompt can be successfully constructed from examples."""
|
"""Test LengthBasedExampleSelector can select examples.."""
|
||||||
short_question = "Short question?"
|
short_question = "Short question?"
|
||||||
output = selector.select_examples({"question": short_question})
|
output = selector.select_examples({"question": short_question})
|
||||||
assert output == EXAMPLES
|
assert output == EXAMPLES
|
||||||
|
|
||||||
|
|
||||||
def test_dynamic_prompt_add_example(selector: LengthBasedExampleSelector) -> None:
|
def test_selector_add_example(selector: LengthBasedExampleSelector) -> None:
|
||||||
"""Test dynamic prompt can add an example."""
|
"""Test LengthBasedExampleSelector can add an example."""
|
||||||
new_example = {"question": "Question: what are you?\nAnswer: bar"}
|
new_example = {"question": "Question: what are you?\nAnswer: bar"}
|
||||||
selector.add_example(new_example)
|
selector.add_example(new_example)
|
||||||
short_question = "Short question?"
|
short_question = "Short question?"
|
||||||
@ -38,18 +38,18 @@ def test_dynamic_prompt_add_example(selector: LengthBasedExampleSelector) -> Non
|
|||||||
assert output == EXAMPLES + [new_example]
|
assert output == EXAMPLES + [new_example]
|
||||||
|
|
||||||
|
|
||||||
def test_dynamic_prompt_trims_one_example(selector: LengthBasedExampleSelector) -> None:
|
def test_selector_trims_one_example(selector: LengthBasedExampleSelector) -> None:
|
||||||
"""Test dynamic prompt can trim one example."""
|
"""Test LengthBasedExampleSelector can trim one example."""
|
||||||
long_question = """I am writing a really long question,
|
long_question = """I am writing a really long question,
|
||||||
this probably is going to affect the example right?"""
|
this probably is going to affect the example right?"""
|
||||||
output = selector.select_examples({"question": long_question})
|
output = selector.select_examples({"question": long_question})
|
||||||
assert output == EXAMPLES[:1]
|
assert output == EXAMPLES[:1]
|
||||||
|
|
||||||
|
|
||||||
def test_dynamic_prompt_trims_all_examples(
|
def test_selector_trims_all_examples(
|
||||||
selector: LengthBasedExampleSelector,
|
selector: LengthBasedExampleSelector,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test dynamic prompt can trim all examples."""
|
"""Test LengthBasedExampleSelector can trim all examples."""
|
||||||
longest_question = """This question is super super super,
|
longest_question = """This question is super super super,
|
||||||
super super super super super super super super super super super,
|
super super super super super super super super super super super,
|
||||||
super super super super long, this will affect the example right?"""
|
super super super super long, this will affect the example right?"""
|
||||||
|
Loading…
Reference in New Issue
Block a user