From 112a208b74c127ee13aca869d5449c77d793b987 Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Wed, 5 Apr 2023 16:15:03 -0300
Subject: [PATCH] fix: add display_name for OpenAI models

---
 src/backend/langflow/template/base.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/backend/langflow/template/base.py b/src/backend/langflow/template/base.py
index 0f3c816e3..c131be085 100644
--- a/src/backend/langflow/template/base.py
+++ b/src/backend/langflow/template/base.py
@@ -217,6 +217,9 @@ class FrontendNode(BaseModel):
         if name == "OpenAI" and key == "model_name":
             field.options = constants.OPENAI_MODELS
             field.is_list = True
-        elif name == "ChatOpenAI" and key == "model_name":
-            field.options = constants.CHAT_OPENAI_MODELS
-            field.is_list = True
+        elif name == "ChatOpenAI":
+            if key == "model_name":
+                field.options = constants.CHAT_OPENAI_MODELS
+                field.is_list = True
+        if "api_key" in key and "OpenAI" in str(name):
+            field.display_name = "OpenAI API Key"