yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-09-19 21:23:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update PromptTemplate tests (#4131)

Browse Source
This commit is contained in:
Vladimir Blagojevic 2023-02-10 15:24:01 +01:00 committed by GitHub
parent 05950719ba
commit d839b9314f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
haystack/nodes/prompt/prompt_node.py
View File

@ -629,7 +629,7 @@ def get_predefined_prompt_templates() -> List[PromptTemplate]:
return [ return [
PromptTemplate( PromptTemplate(
name="question-answering", name="question-answering",
prompt_text="Given the contex please answer the question. Context: $documents; Question: " prompt_text="Given the context please answer the question. Context: $documents; Question: "
"$questions; Answer:", "$questions; Answer:",
), ),
PromptTemplate( PromptTemplate(

10
test/nodes/test_prompt_node.py
View File

@ -18,13 +18,13 @@ def is_openai_api_key_set(api_key: str):
def test_prompt_templates(): def test_prompt_templates():
p = PromptTemplate("t1", "Here is some fake template with variable $foo", ["foo"]) p = PromptTemplate("t1", "Here is some fake template with variable $foo", ["foo"])
with pytest.raises(ValueError, match="Number of parameters in"): with pytest.raises(ValueError, match="The number of parameters in prompt text"):
PromptTemplate("t2", "Here is some fake template with variable $foo and $bar", ["foo"]) PromptTemplate("t2", "Here is some fake template with variable $foo and $bar", ["foo"])
with pytest.raises(ValueError, match="Invalid parameter"): with pytest.raises(ValueError, match="Invalid parameter"):
PromptTemplate("t2", "Here is some fake template with variable $footur", ["foo"]) PromptTemplate("t2", "Here is some fake template with variable $footur", ["foo"])
with pytest.raises(ValueError, match="Number of parameters in"): with pytest.raises(ValueError, match="The number of parameters in prompt text"):
PromptTemplate("t2", "Here is some fake template with variable $foo and $bar", ["foo", "bar", "baz"]) PromptTemplate("t2", "Here is some fake template with variable $foo and $bar", ["foo", "bar", "baz"])
p = PromptTemplate("t3", "Here is some fake template with variable $for and $bar", ["for", "bar"]) p = PromptTemplate("t3", "Here is some fake template with variable $for and $bar", ["for", "bar"])
@ -133,7 +133,7 @@ def test_invalid_template(prompt_node):
name="custom-task", prompt_text="Custom task: $pram1 $param2", prompt_params=["param1", "param2"] name="custom-task", prompt_text="Custom task: $pram1 $param2", prompt_params=["param1", "param2"]
) )
with pytest.raises(ValueError, match="Number of parameters"): with pytest.raises(ValueError, match="The number of parameters in prompt text"):
PromptTemplate(name="custom-task", prompt_text="Custom task: $param1", prompt_params=["param1", "param2"]) PromptTemplate(name="custom-task", prompt_text="Custom task: $param1", prompt_params=["param1", "param2"])
@ -195,12 +195,12 @@ def test_has_supported_template_names(prompt_node):
def test_invalid_template_params(prompt_node): def test_invalid_template_params(prompt_node):
with pytest.raises(ValueError, match="Expected prompt params"): with pytest.raises(ValueError, match="Expected prompt parameters"):
prompt_node.prompt("question-answering", {"some_crazy_key": "Berlin is the capital of Germany."}) prompt_node.prompt("question-answering", {"some_crazy_key": "Berlin is the capital of Germany."})
def test_wrong_template_params(prompt_node): def test_wrong_template_params(prompt_node):
with pytest.raises(ValueError, match="Expected prompt params"): with pytest.raises(ValueError, match="Expected prompt parameters"):
# with don't have options param, multiple choice QA has # with don't have options param, multiple choice QA has
prompt_node.prompt("question-answering", options=["Berlin is the capital of Germany."]) prompt_node.prompt("question-answering", options=["Berlin is the capital of Germany."])