From f21ddc230663858bde16a236176e5ad74e525b7b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 7 Feb 2024 21:00:03 -0300
Subject: [PATCH] Refactor OpenAIModelComponent configuration fields

---
 src/backend/langflow/components/models/OpenAI.py | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/src/backend/langflow/components/models/OpenAI.py b/src/backend/langflow/components/models/OpenAI.py
index 52aeea956..45fa12e23 100644
--- a/src/backend/langflow/components/models/OpenAI.py
+++ b/src/backend/langflow/components/models/OpenAI.py
@@ -11,21 +11,19 @@ class OpenAIModelComponent(CustomComponent):
 
     def build_config(self):
         return {
+            "inputs": {"display_name": "Input"},
             "max_tokens": {
                 "display_name": "Max Tokens",
-                "field_type": "int",
                 "advanced": False,
                 "required": False,
             },
             "model_kwargs": {
                 "display_name": "Model Kwargs",
-                "field_type": "NestedDict",
                 "advanced": True,
                 "required": False,
             },
             "model_name": {
                 "display_name": "Model Name",
-                "field_type": "str",
                 "advanced": False,
                 "required": False,
                 "options": [
@@ -39,7 +37,6 @@ class OpenAIModelComponent(CustomComponent):
             },
             "openai_api_base": {
                 "display_name": "OpenAI API Base",
-                "field_type": "str",
                 "advanced": False,
                 "required": False,
                 "info": (
@@ -49,14 +46,12 @@ class OpenAIModelComponent(CustomComponent):
             },
             "openai_api_key": {
                 "display_name": "OpenAI API Key",
-                "field_type": "str",
                 "advanced": False,
                 "required": False,
                 "password": True,
             },
             "temperature": {
                 "display_name": "Temperature",
-                "field_type": "float",
                 "advanced": False,
                 "required": False,
                 "value": 0.7,
@@ -85,4 +80,6 @@ class OpenAIModelComponent(CustomComponent):
         )
 
         message = model.invoke(inputs)
-        return message.content if hasattr(message, "content") else message
+        result = message.content if hasattr(message, "content") else message
+        self.status = result
+        return result