jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update PromptComponent to use Message instead of Prompt for generating prompts

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 15:47:05 -03:00
commit 600ab2d3b9
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/base/langflow/components/prompts/Prompt.py
View file

@ -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