From d413b2b6fc70254ee4f42fb0b5699799483fa542 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 19 Feb 2024 16:28:32 -0300 Subject: [PATCH] PromptComponent now handles Record inputs --- src/backend/langflow/components/prompts/Prompt.py | 5 +++++ 1 file changed, 5 insertions(+) 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: