From f4e642ec477f0a1ed1521dc16b7e9c8dde8c0bb8 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 6 Nov 2023 21:02:22 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(chains.py):=20temporarily=20?= =?UTF-8?q?fix=20field=5Ftype=20for=20BaseLanguageModel=20to=20improve=20c?= =?UTF-8?q?ompatibility=20=F0=9F=90=9B=20fix(test=5Fprompts=5Ftemplate.py)?= =?UTF-8?q?:=20fix=20value=20of=20validate=5Ftemplate=20to=20False=20to=20?= =?UTF-8?q?align=20with=20expected=20behavior=20=F0=9F=90=9B=20fix(test=5F?= =?UTF-8?q?template.py):=20update=20value=20of=20options=20to=20"gpt-4-110?= =?UTF-8?q?6-preview"=20to=20reflect=20the=20latest=20version?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/template/frontend_node/chains.py | 2 ++ tests/test_prompts_template.py | 2 +- tests/test_template.py | 2 +- 3 files changed, 4 insertions(+), 2 deletions(-) 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