Merge remote-tracking branch 'origin/dev' into add_cli_options_superuser

tests/test_llms_template.py

@@ -1,98 +1,6 @@
 from fastapi.testclient import TestClient
 
 
-# def test_hugging_face_hub(client: TestClient):
-#     response = client.get("api/v1/all")
-#     assert response.status_code == 200
-#     json_response = response.json()
-#     language_models = json_response["llms"]
-
-#     model = language_models["HuggingFaceHub"]
-#     template = model["template"]
-
-#     assert template["cache"] == {
-#         "required": False,
-#         "placeholder": "",
-#         "show": False,
-#         "multiline": False,
-#         "password": False,
-#         "name": "cache",
-#         "type": "bool",
-#         "list": False,
-#         "advanced": False,
-#     }
-#     assert template["verbose"] == {
-#         "required": False,
-#         "placeholder": "",
-#         "show": False,
-#         "multiline": False,
-#         "value": False,
-#         "password": False,
-#         "name": "verbose",
-#         "type": "bool",
-#         "list": False,
-#         "advanced": False,
-#     }
-#     assert template["client"] == {
-#         "required": False,
-#         "placeholder": "",
-#         "show": False,
-#         "multiline": False,
-#         "password": False,
-#         "name": "client",
-#         "type": "Any",
-#         "list": False,
-#         "advanced": False,
-#     }
-#     assert template["repo_id"] == {
-#         "required": False,
-#         "placeholder": "",
-#         "show": True,
-#         "multiline": False,
-#         "value": "gpt2",
-#         "password": False,
-#         "name": "repo_id",
-#         "type": "str",
-#         "list": False,
-#         "advanced": False,
-#     }
-#     assert template["task"] == {
-#         "required": True,
-#         "placeholder": "",
-#         "show": True,
-#         "multiline": False,
-#         "password": False,
-#         "options": ["text-generation", "text2text-generation"],
-#         "name": "task",
-#         "type": "str",
-#         "list": True,
-#         "advanced": True,
-#     }
-#     assert template["model_kwargs"] == {
-#         "required": False,
-#         "placeholder": "",
-#         "show": True,
-#         "multiline": False,
-#         "password": False,
-#         "name": "model_kwargs",
-#         "type": "code",
-#         "list": False,
-#         "advanced": True,
-#     }
-#     assert template["huggingfacehub_api_token"] == {
-#         "required": False,
-#         "placeholder": "",
-#         "show": True,
-#         "multiline": False,
-#         "password": True,
-#         "name": "huggingfacehub_api_token",
-#         "display_name": "HuggingFace Hub API Token",
-#         "type": "str",
-#         "list": False,
-#         "advanced": False,
-#     }
-
-
 def test_openai(client: TestClient, logged_in_headers):
     response = client.get("api/v1/all", headers=logged_in_headers)
     assert response.status_code == 200
@@ -269,7 +177,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "multiline": False,
         "password": False,
         "name": "model_kwargs",
-        "type": "code",
+        "type": "dict",
         "list": False,
         "advanced": True,
         "info": "",
@@ -324,7 +232,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "multiline": False,
         "password": False,
         "name": "logit_bias",
-        "type": "code",
+        "type": "dict",
         "list": False,
         "advanced": False,
         "info": "",
@@ -441,7 +349,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "multiline": False,
         "password": False,
         "name": "model_kwargs",
-        "type": "code",
+        "type": "dict",
         "list": False,
         "advanced": True,
         "info": "",

tests/test_prompts_template.py

@@ -55,7 +55,7 @@ def test_prompt_template(client: TestClient, logged_in_headers):
         "multiline": False,
         "password": False,
         "name": "partial_variables",
-        "type": "code",
+        "type": "dict",
         "list": False,
         "advanced": False,
         "info": "",

tests/test_template.py

@@ -135,7 +135,7 @@ def test_format_dict():
     }
     expected_output = {
         "field1": {
-            "type": "code",  # Mapping type is replaced with dict which is replaced with code
+            "type": "dict[str, int]",  # Mapping type is replaced with dict which is replaced with code
             "required": False,
             "list": False,
             "show": False,
@@ -249,7 +249,7 @@ def test_format_dict():
     }
     expected_output = {
         "field1": {
-            "type": "code",
+            "type": "Dict[str, int]",
             "required": False,
             "list": False,
             "show": False,