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

refactor: Update PromptComponent to return Message instead of Prompt

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 20:18:00 -03:00
commit 84e1d39b71
1 changed files with 1 additions and 2 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,5 +1,4 @@
from langflow.custom import Component from langflow.custom import Component
from langflow.field_typing.prompt import Prompt
from langflow.inputs import PromptInput from langflow.inputs import PromptInput
from langflow.schema.message import Message from langflow.schema.message import Message
from langflow.template import Output from langflow.template import Output
@ -27,7 +26,7 @@ class PromptComponent(Component):
async def build_prompt( async def build_prompt(
self, self,
) -> Prompt: ) -> Message:
kwargs = {k: v for k, v in self._arguments.items() if k != "template"} kwargs = {k: v for k, v in self._arguments.items() if k != "template"}
prompt = await Message.from_template_and_variables(self.template, kwargs) prompt = await Message.from_template_and_variables(self.template, kwargs)
self.status = prompt.format_text() self.status = prompt.format_text()