From f5a150aa132ab663cb39018a6f515270744f28cd Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Tue, 18 Jun 2024 11:09:44 -0300 Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20(Prompt.py):=20refactor=20?= =?UTF-8?q?build=5Fprompt=20method=20by=20simplifying=20the=20code=20and?= =?UTF-8?q?=20separating=20concerns?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/base/langflow/components/prompts/Prompt.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/components/prompts/Prompt.py b/src/backend/base/langflow/components/prompts/Prompt.py index 8cbc5cfed..65bcb4581 100644 --- a/src/backend/base/langflow/components/prompts/Prompt.py +++ b/src/backend/base/langflow/components/prompts/Prompt.py @@ -20,8 +20,9 @@ class PromptComponent(Component): async def build_prompt( self, ) -> Message: - kwargs = {k: v for k, v in self._arguments.items() if k != "template"} - prompt = await Message.from_template_and_variables(self.template, kwargs) - prompt_message = Message(text=prompt.format_text(), **kwargs) + prompt = await Message.from_template_and_variables(**self._arguments) + kwargs = self._arguments.copy() + kwargs["text"] = prompt.format_text() + prompt_message = Message(**kwargs) self.status = prompt_message return prompt