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

PromptComponent now handles Record inputs

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-19 16:28:32 -03:00
commit d413b2b6fc
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/components/prompts/Prompt.py
View file

@ -1,6 +1,7 @@
from langchain_core.prompts import PromptTemplate from langchain_core.prompts import PromptTemplate
from langflow import CustomComponent from langflow import CustomComponent
from langflow.field_typing import Prompt, TemplateField, Text from langflow.field_typing import Prompt, TemplateField, Text
from langflow.schema import Record
class PromptComponent(CustomComponent): class PromptComponent(CustomComponent):
@ -20,6 +21,10 @@ class PromptComponent(CustomComponent):
**kwargs, **kwargs,
) -> Text: ) -> Text:
prompt_template = PromptTemplate.from_template(template) prompt_template = PromptTemplate.from_template(template)
for key, value in kwargs.copy().items():
if isinstance(value, Record):
kwargs[key] = value.text
try: try:
formated_prompt = prompt_template.format(**kwargs) formated_prompt = prompt_template.format(**kwargs)
except Exception as exc: except Exception as exc: