diff --git a/src/backend/langflow/template/frontend_node/chains.py b/src/backend/langflow/template/frontend_node/chains.py index b678dec3b..6cede781b 100644 --- a/src/backend/langflow/template/frontend_node/chains.py +++ b/src/backend/langflow/template/frontend_node/chains.py @@ -87,6 +87,8 @@ class ChainFrontendNode(FrontendNode): field.required = True field.show = True field.advanced = False + field.field_type = "BaseLanguageModel" # temporary fix + field.is_list = False if field.name == "return_source_documents": field.required = False diff --git a/tests/test_prompts_template.py b/tests/test_prompts_template.py index bc2e935e9..2fc161c6d 100644 --- a/tests/test_prompts_template.py +++ b/tests/test_prompts_template.py @@ -96,7 +96,7 @@ def test_prompt_template(client: TestClient, logged_in_headers): "placeholder": "", "show": False, "multiline": False, - "value": True, + "value": False, "password": False, "name": "validate_template", "type": "bool", diff --git a/tests/test_template.py b/tests/test_template.py index 81f2a6020..52ab955c3 100644 --- a/tests/test_template.py +++ b/tests/test_template.py @@ -237,7 +237,7 @@ def test_format_dict(): "password": False, "multiline": False, "options": CHAT_OPENAI_MODELS, - "value": "gpt-3.5-turbo-0613", + "value": "gpt-4-1106-preview", }, } assert format_dict(input_dict, "OpenAI") == expected_output_openai