jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

✨ (openai_constants.py): add a new file to define constant MODEL_NAMES for OpenAI models (#1858)

Browse source 
* ✨ (openai_constants.py): add a new file to define constant MODEL_NAMES for OpenAI models
♻️ (ChatOpenAISpecs.py, OpenAIModel.py): refactor to use MODEL_NAMES constant from openai_constants.py for consistency and easier maintenance

* ✨ (IDGenerator.py): update key name from 'real_time_refresh' to 'refresh_button' for better clarity and consistency
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-05-07 22:47:21 -03:00 • committed by GitHub
commit 28c2585415
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 7 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/backend/base/langflow/base/models/openai_constants.py Normal file
View file

@ -0,0 +1 @@
MODEL_NAMES = ["gpt-4-turbo", "gpt-4-turbo-preview", "gpt-3.5-turbo", "gpt-3.5-turbo-0125"]

2
src/backend/base/langflow/components/helpers/IDGenerator.py
View file

@ -23,7 +23,7 @@ class UUIDGeneratorComponent(CustomComponent):
return { return {
"unique_id": { "unique_id": {
"display_name": "Value", "display_name": "Value",
"real_time_refresh": True, "refresh_button": True,
} }
} }

18
src/backend/base/langflow/components/model_specs/ChatOpenAISpecs.py
View file

@ -1,9 +1,9 @@
from typing import Optional from typing import Optional
from langflow.field_typing import BaseLanguageModel
from langchain_community.chat_models.openai import ChatOpenAI from langchain_community.chat_models.openai import ChatOpenAI
from langflow.field_typing import NestedDict from langflow.base.models.openai_constants import MODEL_NAMES
from langflow.field_typing import BaseLanguageModel, NestedDict
from langflow.interface.custom.custom_component import CustomComponent from langflow.interface.custom.custom_component import CustomComponent
@ -24,19 +24,7 @@ class ChatOpenAIComponent(CustomComponent):
"advanced": True, "advanced": True,
"required": False, "required": False,
}, },
"model_name": { "model_name": {"display_name": "Model Name", "advanced": False, "options": MODEL_NAMES},
"display_name": "Model Name",
"advanced": False,
"required": False,
"options": [
"gpt-4-turbo-preview",
"gpt-4-0125-preview",
"gpt-4-1106-preview",
"gpt-4-vision-preview",
"gpt-3.5-turbo-0125",
"gpt-3.5-turbo-1106",
],
},
"openai_api_base": { "openai_api_base": {
"display_name": "OpenAI API Base", "display_name": "OpenAI API Base",
"advanced": False, "advanced": False,

12
src/backend/base/langflow/components/models/OpenAIModel.py
View file

@ -5,6 +5,7 @@ from pydantic.v1 import SecretStr
from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.constants import STREAM_INFO_TEXT
from langflow.base.models.model import LCModelComponent from langflow.base.models.model import LCModelComponent
from langflow.base.models.openai_constants import MODEL_NAMES
from langflow.field_typing import NestedDict, Text from langflow.field_typing import NestedDict, Text
@ -39,16 +40,7 @@ class OpenAIModelComponent(LCModelComponent):
"model_name": { "model_name": {
"display_name": "Model Name", "display_name": "Model Name",
"advanced": False, "advanced": False,
"options": [ "options": MODEL_NAMES,
"gpt-4-turbo-2024-04-09",
"gpt-4-turbo-preview",
"gpt-3.5-turbo",
"gpt-4-0125-preview",
"gpt-4-1106-preview",
"gpt-4-vision-preview",
"gpt-3.5-turbo-0125",
"gpt-3.5-turbo-1106",
],
"value": "gpt-4-turbo-preview", "value": "gpt-4-turbo-preview",
}, },
"openai_api_base": { "openai_api_base": {