From 2e3c0f3f51f62f3b9ce46bcfe3313cff05232215 Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Sat, 8 Apr 2023 18:13:57 -0300
Subject: [PATCH] style(test_llms_template.py): change boolean values to True
 in 'show' key for some template fields 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

---
 tests/test_llms_template.py | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

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