From 600ab2d3b9e2258dce658e3093f1a6fc18fc79f5 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 17 Jun 2024 15:47:05 -0300 Subject: [PATCH] refactor: Update PromptComponent to use Message instead of Prompt for generating prompts --- src/backend/base/langflow/components/prompts/Prompt.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/components/prompts/Prompt.py b/src/backend/base/langflow/components/prompts/Prompt.py index bb3b9e47c..720830fbb 100644 --- a/src/backend/base/langflow/components/prompts/Prompt.py +++ b/src/backend/base/langflow/components/prompts/Prompt.py @@ -1,6 +1,7 @@ from langflow.custom import Component from langflow.field_typing.prompt import Prompt from langflow.inputs import PromptInput +from langflow.schema.message import Message from langflow.template import Output @@ -28,6 +29,6 @@ class PromptComponent(Component): self, ) -> Prompt: kwargs = {k: v for k, v in self._arguments.items() if k != "template"} - prompt = await Prompt.from_template_and_variables(self.template, kwargs) + prompt = await Message.from_template_and_variables(self.template, kwargs) self.status = prompt.format_text() return prompt