fix tests to comply with updates

This commit is contained in:
Gabriel Almeida 2023-04-29 19:48:59 -03:00
commit 2b6f70fdb4
2 changed files with 8 additions and 8 deletions

View file

@ -60,7 +60,7 @@ def test_prompt_template(client: TestClient):
"name": "template",
"type": "prompt",
"list": False,
"advanced": True,
"advanced": False,
}
assert template["template_format"] == {
"required": False,
@ -106,7 +106,7 @@ def test_few_shot_prompt_template(client: TestClient):
"name": "examples",
"type": "prompt",
"list": True,
"advanced": True,
"advanced": False,
}
assert template["example_selector"] == {
"required": False,
@ -139,7 +139,7 @@ def test_few_shot_prompt_template(client: TestClient):
"name": "suffix",
"type": "prompt",
"list": False,
"advanced": True,
"advanced": False,
}
assert template["example_separator"] == {
"required": False,
@ -163,7 +163,7 @@ def test_few_shot_prompt_template(client: TestClient):
"name": "prefix",
"type": "prompt",
"list": False,
"advanced": True,
"advanced": False,
}