From bd8c709870d82428f8ee42fada3bc9c4c6d08aee Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 8 Nov 2023 12:44:46 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=84=20chore(constants.py):=20update=20?= =?UTF-8?q?CHAT=5FOPENAI=5FMODELS=20list=20to=20reflect=20the=20latest=20a?= =?UTF-8?q?vailable=20models?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 🔄 chore(test_llms_template.py): update test case to use the latest model name in the CHAT_OPENAI_MODELS list --- src/backend/langflow/utils/constants.py | 10 ++++------ tests/test_llms_template.py | 11 ++++------- 2 files changed, 8 insertions(+), 13 deletions(-) diff --git a/src/backend/langflow/utils/constants.py b/src/backend/langflow/utils/constants.py index 0c97b56a2..283f44406 100644 --- a/src/backend/langflow/utils/constants.py +++ b/src/backend/langflow/utils/constants.py @@ -6,16 +6,14 @@ OPENAI_MODELS = [ "text-ada-001", ] CHAT_OPENAI_MODELS = [ - "gpt-3.5-turbo-0613", - "gpt-3.5-turbo", - "gpt-3.5-turbo-16k-0613", - "gpt-3.5-turbo-16k", - "gpt-4-0613", - "gpt-4-32k-0613", + "gpt-4-1106-preview", "gpt-4", "gpt-4-32k", + "gpt-3.5-turbo", + "gpt-3.5-turbo-16k", ] + ANTHROPIC_MODELS = [ # largest model, ideal for a wide range of more complex tasks. "claude-v1", diff --git a/tests/test_llms_template.py b/tests/test_llms_template.py index 0a30a825e..78131cb05 100644 --- a/tests/test_llms_template.py +++ b/tests/test_llms_template.py @@ -309,17 +309,14 @@ def test_chat_open_ai(client: TestClient, logged_in_headers): "placeholder": "", "show": True, "multiline": False, - "value": "gpt-3.5-turbo-0613", + "value": "gpt-4-1106-preview", "password": False, "options": [ - "gpt-3.5-turbo-0613", - "gpt-3.5-turbo", - "gpt-3.5-turbo-16k-0613", - "gpt-3.5-turbo-16k", - "gpt-4-0613", - "gpt-4-32k-0613", + "gpt-4-1106-preview", "gpt-4", "gpt-4-32k", + "gpt-3.5-turbo", + "gpt-3.5-turbo-16k", ], "name": "model_name", "type": "str",