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

feat: added constants file to support model types and others

Browse source
This commit is contained in:
Gabriel Almeida 2023-03-20 20:54:05 -03:00
commit c390a47504
2 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/utils/constants.py Normal file
View file

@ -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"]

6
src/backend/langflow/utils/util.py
View file

@ -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