diff --git a/backend/chainlit/langchain/callbacks.py b/backend/chainlit/langchain/callbacks.py
index bfa3e1f98b..72254e57fe 100644
--- a/backend/chainlit/langchain/callbacks.py
+++ b/backend/chainlit/langchain/callbacks.py
@@ -241,9 +241,12 @@ def build_template_messages() -> List[PromptMessage]:
                 if "placeholder" in class_name.lower():
                     variable_name = lc_message.get(
                         "variable_name"
+                    ) or message_kwargs.get(
+                        "variable_name"
                     )  # type: Optional[str]
                     variable = inputs.get(variable_name, [])
                     placeholder_size = len(variable)
+
                     if placeholder_size:
                         template_messages += [
                             PromptMessage(placeholder_size=placeholder_size)