jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

style(test_llms_template.py): change boolean values to True in 'show' key for some template fields

Browse source 
feat(test_llms_template.py): make 'task' and 'huggingfacehub_api_token' fields required
feat(test_llms_template.py): add options to 'task' field and display name to 'huggingfacehub_api_token' field
This commit is contained in:
Gabriel Almeida 2023-04-08 18:13:57 -03:00 • committed by Gabriel Luiz Freitas Almeida
commit 2e3c0f3f51
1 changed files with 10 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/test_llms_template.py
View file

@ -53,7 +53,7 @@ def test_hugging_face_hub(client: TestClient):
assert template["repo_id"] == { assert template["repo_id"] == {
"required": False, "required": False,
"placeholder": "", "placeholder": "",
"show": False, "show": True,
"multiline": False, "multiline": False,
"value": "gpt2", "value": "gpt2",
"password": False, "password": False,
@ -62,19 +62,20 @@ def test_hugging_face_hub(client: TestClient):
"list": False, "list": False,
} }
assert template["task"] == { assert template["task"] == {
"required": False, "required": True,
"placeholder": "", "placeholder": "",
"show": False, "show": True,
"multiline": False, "multiline": False,
"password": False, "password": False,
"options": ["text-generation", "text2text-generation"],
"name": "task", "name": "task",
"type": "str", "type": "str",
"list": False, "list": True,
} }
assert template["model_kwargs"] == { assert template["model_kwargs"] == {
"required": False, "required": False,
"placeholder": "", "placeholder": "",
"show": False, "show": True,
"multiline": False, "multiline": False,
"password": False, "password": False,
"name": "model_kwargs", "name": "model_kwargs",
@ -82,12 +83,13 @@ def test_hugging_face_hub(client: TestClient):
"list": False, "list": False,
} }
assert template["huggingfacehub_api_token"] == { assert template["huggingfacehub_api_token"] == {
"required": False, "required": True,
"placeholder": "", "placeholder": "",
"show": True, "show": True,
"multiline": False, "multiline": False,
"password": True, "password": True,
"name": "huggingfacehub_api_token", "name": "huggingfacehub_api_token",
"display_name": "HuggingFace Hub API Token",
"type": "str", "type": "str",
"list": False, "list": False,
} }
@ -231,7 +233,7 @@ def test_openai(client: TestClient):
assert template["model_kwargs"] == { assert template["model_kwargs"] == {
"required": False, "required": False,
"placeholder": "", "placeholder": "",
"show": False, "show": True,
"multiline": False, "multiline": False,
"password": False, "password": False,
"name": "model_kwargs", "name": "model_kwargs",
@ -361,7 +363,7 @@ def test_chat_open_ai(client: TestClient):
assert template["model_kwargs"] == { assert template["model_kwargs"] == {
"required": False, "required": False,
"placeholder": "", "placeholder": "",
"show": False, "show": True,
"multiline": False, "multiline": False,
"password": False, "password": False,
"name": "model_kwargs", "name": "model_kwargs",