diff --git a/src/backend/langflow/template/field/base.py b/src/backend/langflow/template/field/base.py
index a9c18ff63..b73a516b4 100644
--- a/src/backend/langflow/template/field/base.py
+++ b/src/backend/langflow/template/field/base.py
@@ -21,6 +21,7 @@ class TemplateFieldCreator(BaseModel, ABC):
     name: str = ""
     display_name: Optional[str] = None
     advanced: bool = False
+    input_types: list[str] = []
 
     def to_dict(self):
         result = self.dict()
diff --git a/src/frontend/src/types/api/index.ts b/src/frontend/src/types/api/index.ts
index 7fc10f309..ecd79240e 100644
--- a/src/frontend/src/types/api/index.ts
+++ b/src/frontend/src/types/api/index.ts
@@ -23,6 +23,7 @@ export type TemplateVariableType = {
   show: boolean;
   multiline?: boolean;
   value?: any;
+  input_types?: Array<string>;
   [key: string]: any;
 };
 export type sendAllProps = {