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

🐛 fix(test_llms_template.py): remove unused test_llms_settings function to improve code readability and maintainability

Browse source 
🐛 fix(test_template.py): update type annotation in test_format_dict function to reflect the correct type of the field
🐛 fix(test_prompts_template.py): update type annotation in test_prompt_template function to reflect the correct type of the field
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-19 14:08:12 -03:00
commit 1243c0b644
3 changed files with 7 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/test_llms_template.py
View file

@ -2,15 +2,6 @@ from fastapi.testclient import TestClient
from langflow.services.utils import get_settings_manager from langflow.services.utils import get_settings_manager
def test_llms_settings(client: TestClient, logged_in_headers):
settings_manager = get_settings_manager()
response = client.get("api/v1/all", headers=logged_in_headers)
assert response.status_code == 200
json_response = response.json()
llms = json_response["llms"]
assert set(llms.keys()) == set(settings_manager.settings.LLMS)
# def test_hugging_face_hub(client: TestClient): # def test_hugging_face_hub(client: TestClient):
# response = client.get("api/v1/all") # response = client.get("api/v1/all")
# assert response.status_code == 200 # assert response.status_code == 200
@ -85,7 +76,7 @@ def test_llms_settings(client: TestClient, logged_in_headers):
# "multiline": False, # "multiline": False,
# "password": False, # "password": False,
# "name": "model_kwargs", # "name": "model_kwargs",
# "type": "code", # "type": "dict" ,
# "list": False, # "list": False,
# "advanced": True, # "advanced": True,
# } # }
@ -279,7 +270,7 @@ def test_openai(client: TestClient, logged_in_headers):
"multiline": False, "multiline": False,
"password": False, "password": False,
"name": "model_kwargs", "name": "model_kwargs",
"type": "code", "type": "dict",
"list": False, "list": False,
"advanced": True, "advanced": True,
"info": "", "info": "",
@ -334,7 +325,7 @@ def test_openai(client: TestClient, logged_in_headers):
"multiline": False, "multiline": False,
"password": False, "password": False,
"name": "logit_bias", "name": "logit_bias",
"type": "code", "type": "dict",
"list": False, "list": False,
"advanced": False, "advanced": False,
"info": "", "info": "",
@ -451,7 +442,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
"multiline": False, "multiline": False,
"password": False, "password": False,
"name": "model_kwargs", "name": "model_kwargs",
"type": "code", "type": "dict",
"list": False, "list": False,
"advanced": True, "advanced": True,
"info": "", "info": "",

2
tests/test_prompts_template.py
View file

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

4
tests/test_template.py
View file

@ -135,7 +135,7 @@ def test_format_dict():
} }
expected_output = { expected_output = {
"field1": { "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, "required": False,
"list": False, "list": False,
"show": False, "show": False,
@ -249,7 +249,7 @@ def test_format_dict():
} }
expected_output = { expected_output = {
"field1": { "field1": {
"type": "code", "type": "Dict[str, int]",
"required": False, "required": False,
"list": False, "list": False,
"show": False, "show": False,