From bb6a2ceb58174fd67627f04acec317340379c61b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 8 Mar 2024 11:58:56 -0300
Subject: [PATCH] Refactor PromptComponent and add import statement for
 PromptComponent

---
 src/backend/langflow/components/inputs/Prompt.py   | 7 +++++--
 src/backend/langflow/components/inputs/__init__.py | 3 ++-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/backend/langflow/components/inputs/Prompt.py b/src/backend/langflow/components/inputs/Prompt.py
index f56b48b4e..a0b03e369 100644
--- a/src/backend/langflow/components/inputs/Prompt.py
+++ b/src/backend/langflow/components/inputs/Prompt.py
@@ -1,7 +1,6 @@
 from langchain_core.prompts import PromptTemplate
 
 from langflow import CustomComponent
-from langflow.base.prompts.utils import dict_values_to_string
 from langflow.field_typing import Prompt, TemplateField, Text
 
 
@@ -21,9 +20,13 @@ class PromptComponent(CustomComponent):
         template: Prompt,
         **kwargs,
     ) -> Text:
+        from langflow.base.prompts.utils import dict_values_to_string
+
         prompt_template = PromptTemplate.from_template(Text(template))
         kwargs = dict_values_to_string(kwargs)
-        kwargs = {k: "\n".join(v) if isinstance(v, list) else v for k, v in kwargs.items()}
+        kwargs = {
+            k: "\n".join(v) if isinstance(v, list) else v for k, v in kwargs.items()
+        }
         try:
             formated_prompt = prompt_template.format(**kwargs)
         except Exception as exc:
diff --git a/src/backend/langflow/components/inputs/__init__.py b/src/backend/langflow/components/inputs/__init__.py
index c8b4f1645..d909a091f 100644
--- a/src/backend/langflow/components/inputs/__init__.py
+++ b/src/backend/langflow/components/inputs/__init__.py
@@ -1,4 +1,5 @@
 from .ChatInput import ChatInput
+from .Prompt import PromptComponent
 from .TextInput import TextInput
 
-__all__ = ["ChatInput", "TextInput"]
+__all__ = ["ChatInput", "TextInput", "PromptComponent"]