From c390a4750494699ac844d284639abb531030182d Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Mon, 20 Mar 2023 20:54:05 -0300
Subject: [PATCH] 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