From c390a4750494699ac844d284639abb531030182d Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Mon, 20 Mar 2023 20:54:05 -0300 Subject: [PATCH 1/2] feat: added constants file to support model types and others --- src/backend/langflow/utils/constants.py | 8 ++++++++ src/backend/langflow/utils/util.py | 6 ++++-- 2 files changed, 12 insertions(+), 2 deletions(-) create mode 100644 src/backend/langflow/utils/constants.py diff --git a/src/backend/langflow/utils/constants.py b/src/backend/langflow/utils/constants.py new file mode 100644 index 000000000..73a50ce40 --- /dev/null +++ b/src/backend/langflow/utils/constants.py @@ -0,0 +1,8 @@ +OPENAI_MODELS = [ + "text-davinci-003", + "text-davinci-002", + "text-curie-001", + "text-babbage-001", + "text-ada-001", +] +CHAT_OPENAI_MODELS = ["gpt-3.5-turbo", "gpt-4", "gpt-4-32k"] diff --git a/src/backend/langflow/utils/util.py b/src/backend/langflow/utils/util.py index 5d13e931d..18fdf658d 100644 --- a/src/backend/langflow/utils/util.py +++ b/src/backend/langflow/utils/util.py @@ -11,6 +11,8 @@ from langchain.agents.load_tools import ( ) from typing import Optional, Dict +from backend.langflow.utils.constants import CHAT_OPENAI_MODELS, OPENAI_MODELS + def build_template_from_function(name: str, type_to_loader_dict: Dict): classes = [ @@ -296,9 +298,9 @@ def format_dict(d, name: Optional[str] = None): # Add options to openai if name == "OpenAI" and key == "model_name": - value["options"] = ["text-davinci-003", "text-davinci-002"] + value["options"] = OPENAI_MODELS elif name == "OpenAIChat" and key == "model_name": - value["options"] = ["gpt-3.5-turbo", "gpt-4"] + value["options"] = CHAT_OPENAI_MODELS return d From 049ab2b39beabbd4adff9a4719f685ea3712d426 Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Mon, 20 Mar 2023 20:56:15 -0300 Subject: [PATCH 2/2] fix: changed import and linting --- src/backend/langflow/__main__.py | 1 - src/backend/langflow/utils/util.py | 6 +++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/__main__.py b/src/backend/langflow/__main__.py index 6f848dca6..3da26667d 100644 --- a/src/backend/langflow/__main__.py +++ b/src/backend/langflow/__main__.py @@ -1,6 +1,5 @@ import multiprocessing import platform -import re from langflow.main import create_app diff --git a/src/backend/langflow/utils/util.py b/src/backend/langflow/utils/util.py index 18fdf658d..dae450f0c 100644 --- a/src/backend/langflow/utils/util.py +++ b/src/backend/langflow/utils/util.py @@ -11,7 +11,7 @@ from langchain.agents.load_tools import ( ) from typing import Optional, Dict -from backend.langflow.utils.constants import CHAT_OPENAI_MODELS, OPENAI_MODELS +from langflow.utils import constants def build_template_from_function(name: str, type_to_loader_dict: Dict): @@ -298,9 +298,9 @@ def format_dict(d, name: Optional[str] = None): # Add options to openai if name == "OpenAI" and key == "model_name": - value["options"] = OPENAI_MODELS + value["options"] = constants.OPENAI_MODELS elif name == "OpenAIChat" and key == "model_name": - value["options"] = CHAT_OPENAI_MODELS + value["options"] = constants.CHAT_OPENAI_MODELS return d