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

♻️ (Prompt.py): refactor build_prompt method by simplifying the code and separating concerns

Browse source
This commit is contained in:
ogabrielluiz 2024-06-18 11:09:44 -03:00
commit f5a150aa13
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -20,8 +20,9 @@ class PromptComponent(Component):
async def build_prompt( async def build_prompt(
self, self,
) -> Message: ) -> Message:
kwargs = {k: v for k, v in self._arguments.items() if k != "template"} prompt = await Message.from_template_and_variables(**self._arguments)
prompt = await Message.from_template_and_variables(self.template, kwargs) kwargs = self._arguments.copy()
prompt_message = Message(text=prompt.format_text(), **kwargs) kwargs["text"] = prompt.format_text()
prompt_message = Message(**kwargs)
self.status = prompt_message self.status = prompt_message
return prompt return prompt