From ad25be1815e293ba016af43f2f82c2ee924d71be Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Thu, 25 Jan 2024 11:43:14 -0300
Subject: [PATCH] Add logger import and update input_types in
 add_new_variables_to_template function

---
 src/backend/langflow/api/v1/validate.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/backend/langflow/api/v1/validate.py b/src/backend/langflow/api/v1/validate.py
index 51d0f0d76..1cebeedb6 100644
--- a/src/backend/langflow/api/v1/validate.py
+++ b/src/backend/langflow/api/v1/validate.py
@@ -1,4 +1,6 @@
 from fastapi import APIRouter, HTTPException
+from loguru import logger
+
 from langflow.api.v1.base import (
     Code,
     CodeValidationResponse,
@@ -8,7 +10,6 @@ from langflow.api.v1.base import (
 )
 from langflow.template.field.base import TemplateField
 from langflow.utils.validate import validate_code
-from loguru import logger
 
 # build router
 router = APIRouter(prefix="/validate", tags=["Validate"])
@@ -81,7 +82,7 @@ def add_new_variables_to_template(input_variables, prompt_request):
                 show=True,
                 advanced=False,
                 multiline=True,
-                input_types=["Document", "BaseOutputParser"],
+                input_types=["Document", "BaseOutputParser", "Text"],
                 value="",  # Set the value to empty string
             )
             if variable in prompt_request.frontend_node.template: