From 121b7c6e39eb651b7bdabf2e3e764661a8dc3a89 Mon Sep 17 00:00:00 2001 From: Rodrigo Nader Date: Fri, 6 Dec 2024 15:03:49 -0300 Subject: [PATCH] feat: Add tool mode to the prompt component (#5113) * Add tool mode to the prompt component * [autofix.ci] apply automated fixes --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Cristhian Zanforlin Lousa --- src/backend/base/langflow/components/prompts/prompt.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/components/prompts/prompt.py b/src/backend/base/langflow/components/prompts/prompt.py index 01646b14d..fa0fe3810 100644 --- a/src/backend/base/langflow/components/prompts/prompt.py +++ b/src/backend/base/langflow/components/prompts/prompt.py @@ -1,7 +1,7 @@ from langflow.base.prompts.api_utils import process_prompt_template from langflow.custom import Component from langflow.inputs.inputs import DefaultPromptField -from langflow.io import Output, PromptInput +from langflow.io import MessageTextInput, Output, PromptInput from langflow.schema.message import Message from langflow.template.utils import update_template_values @@ -15,6 +15,13 @@ class PromptComponent(Component): inputs = [ PromptInput(name="template", display_name="Template"), + MessageTextInput( + name="tool_placeholder", + display_name="Tool Placeholder", + tool_mode=True, + advanced=True, + info="A placeholder input for tool mode.", + ), ] outputs = [