From 84e1d39b71e01f0224190155a0cc9b53ffd1532b Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 17 Jun 2024 20:18:00 -0300 Subject: [PATCH] refactor: Update PromptComponent to return Message instead of Prompt --- src/backend/base/langflow/components/prompts/Prompt.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/prompts/Prompt.py b/src/backend/base/langflow/components/prompts/Prompt.py index 720830fbb..75d6f21c4 100644 --- a/src/backend/base/langflow/components/prompts/Prompt.py +++ b/src/backend/base/langflow/components/prompts/Prompt.py @@ -1,5 +1,4 @@ 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 @@ -27,7 +26,7 @@ class PromptComponent(Component): async def build_prompt( self, - ) -> Prompt: + ) -> Message: kwargs = {k: v for k, v in self._arguments.items() if k != "template"} prompt = await Message.from_template_and_variables(self.template, kwargs) self.status = prompt.format_text()