diff --git a/src/backend/langflow/components/prompts/Prompt.py b/src/backend/langflow/components/prompts/Prompt.py index 803a9170b..c8d04f7d3 100644 --- a/src/backend/langflow/components/prompts/Prompt.py +++ b/src/backend/langflow/components/prompts/Prompt.py @@ -1,6 +1,7 @@ from langchain_core.prompts import PromptTemplate from langflow import CustomComponent from langflow.field_typing import Prompt, TemplateField, Text +from langflow.schema import Record class PromptComponent(CustomComponent): @@ -20,6 +21,10 @@ class PromptComponent(CustomComponent): **kwargs, ) -> Text: prompt_template = PromptTemplate.from_template(template) + + for key, value in kwargs.copy().items(): + if isinstance(value, Record): + kwargs[key] = value.text try: formated_prompt = prompt_template.format(**kwargs) except Exception as exc: