diff --git a/api/core/variables/variables.py b/api/core/variables/variables.py
index 1fe0e36a47..54aeec61e9 100644
--- a/api/core/variables/variables.py
+++ b/api/core/variables/variables.py
@@ -1,5 +1,5 @@
 from collections.abc import Sequence
-from typing import cast
+from typing import Any, cast
 from uuid import uuid4
 
 from pydantic import BaseModel, Field
@@ -104,7 +104,7 @@ class RAGPipelineVariable(BaseModel):
     max_length: int | None = Field(
         description="max length, applicable to text-input, paragraph, and file-list", default=0
     )
-    default_value: str | None = Field(description="default value", default="")
+    default_value: Any = Field(description="default value", default="")
     placeholder: str | None = Field(description="placeholder", default="")
     unit: str | None = Field(description="unit, applicable to Number", default="")
     tooltips: str | None = Field(description="helpful text", default="")
diff --git a/api/fields/workflow_fields.py b/api/fields/workflow_fields.py
index c138266b14..36249d2ae9 100644
--- a/api/fields/workflow_fields.py
+++ b/api/fields/workflow_fields.py
@@ -47,6 +47,7 @@ pipeline_variable_fields = {
     "belong_to_node_id": fields.String,
     "max_length": fields.Integer,
     "required": fields.Boolean,
+    "unit": fields.String,
     "default_value": fields.Raw,
     "options": fields.List(fields.String),
     "placeholder": fields.String,