From f275471285ab1a2d4222b7b39334279432a8a280 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 10 Dec 2023 00:08:28 -0300 Subject: [PATCH] Remove redundant code and update default values for template fields --- tests/test_agents_template.py | 5 ----- tests/test_frontend_nodes.py | 4 ++-- 2 files changed, 2 insertions(+), 7 deletions(-) diff --git a/tests/test_agents_template.py b/tests/test_agents_template.py index a7919faac..01891ec05 100644 --- a/tests/test_agents_template.py +++ b/tests/test_agents_template.py @@ -29,7 +29,6 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers): "advanced": False, "info": "", "fileTypes": [], - "value": None, } # Additional assertions for other template variables @@ -46,7 +45,6 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers): "advanced": False, "info": "", "fileTypes": [], - "value": None, } assert template["llm"] == { "required": True, @@ -61,7 +59,6 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers): "advanced": False, "info": "", "fileTypes": [], - "value": None, } assert template["output_parser"] == { "required": False, @@ -76,7 +73,6 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers): "advanced": False, "info": "", "fileTypes": [], - "value": None, } assert template["input_variables"] == { "required": False, @@ -91,7 +87,6 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers): "advanced": False, "info": "", "fileTypes": [], - "value": None, } assert template["prefix"] == { "required": False, diff --git a/tests/test_frontend_nodes.py b/tests/test_frontend_nodes.py index 815cc13c2..e92ad1fe4 100644 --- a/tests/test_frontend_nodes.py +++ b/tests/test_frontend_nodes.py @@ -31,9 +31,9 @@ def test_template_field_defaults(sample_template_field: TemplateField): assert sample_template_field.is_list is False assert sample_template_field.show is True assert sample_template_field.multiline is False - assert sample_template_field.value is None + assert sample_template_field.value == "" assert sample_template_field.file_types == [] - assert sample_template_field.file_path is None + assert sample_template_field.file_path == "" assert sample_template_field.password is False assert sample_template_field.name == "test_field"