From c82aeb25369e39c4c68f290a5c91fa1299b9b991 Mon Sep 17 00:00:00 2001
From: Rodrigo Nader <rodrigosilvanader@gmail.com>
Date: Mon, 16 Dec 2024 11:39:33 -0300
Subject: [PATCH] feat: improve component configuration and input clarity
 (#5267)

* Add tool mode to structured output 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 <cristhian.lousa@gmail.com>
---
 .../langflow/components/helpers/structured_output.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/backend/base/langflow/components/helpers/structured_output.py b/src/backend/base/langflow/components/helpers/structured_output.py
index 2832898e5..f09dec332 100644
--- a/src/backend/base/langflow/components/helpers/structured_output.py
+++ b/src/backend/base/langflow/components/helpers/structured_output.py
@@ -18,6 +18,7 @@ class StructuredOutputComponent(Component):
         "Transforms LLM responses into **structured data formats**. Ideal for extracting specific information "
         "or creating consistent outputs."
     )
+    name = "StructuredOutput"
     icon = "braces"
 
     inputs = [
@@ -26,17 +27,25 @@ class StructuredOutputComponent(Component):
             display_name="Language Model",
             info="The language model to use to generate the structured output.",
             input_types=["LanguageModel"],
+            required=True,
+        ),
+        MessageTextInput(
+            name="input_value",
+            display_name="Input Message",
+            info="The input message to the language model.",
+            tool_mode=True,
         ),
-        MessageTextInput(name="input_value", display_name="Input message"),
         StrInput(
             name="schema_name",
             display_name="Schema Name",
             info="Provide a name for the output data schema.",
+            advanced=True,
         ),
         TableInput(
             name="output_schema",
             display_name="Output Schema",
             info="Define the structure and data types for the model's output.",
+            required=True,
             table_schema=[
                 {
                     "name": "name",
@@ -73,6 +82,7 @@ class StructuredOutputComponent(Component):
         ),
         BoolInput(
             name="multiple",
+            advanced=True,
             display_name="Generate Multiple",
             info="Set to True if the model should generate a list of outputs instead of a single output.",
         ),