fix lc placeholder with lcel (#568)

Chainlit · Nov 26, 2023 · 3c44500 · 3c44500
1 parent 60c96f0
commit 3c44500
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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)