From ba2331ba777ad12d78d2eb6be782645b6c7df5ab Mon Sep 17 00:00:00 2001
From: ogabrielluiz <gabriel@langflow.org>
Date: Mon, 17 Jun 2024 14:58:05 -0300
Subject: [PATCH] refactor: Add TextInput and MessageInput to InputTypes in
 inputs.py

---
 src/backend/base/langflow/inputs/inputs.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/backend/base/langflow/inputs/inputs.py b/src/backend/base/langflow/inputs/inputs.py
index c84feed20..ee0e913f9 100644
--- a/src/backend/base/langflow/inputs/inputs.py
+++ b/src/backend/base/langflow/inputs/inputs.py
@@ -140,4 +140,6 @@ InputTypes = Union[
     PromptInput,
     SecretStrInput,
     StrInput,
+    TextInput,
+    MessageInput,
 ]