From 60642faf7127679b82efd9d2d8d3b03e4103a44d Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Date: Thu, 20 Jun 2024 17:59:59 -0300
Subject: [PATCH] refactor: Update Component class to include dynamic inputs in
 get_trace_as_inputs method

---
 .../base/langflow/custom/custom_component/component.py       | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/backend/base/langflow/custom/custom_component/component.py b/src/backend/base/langflow/custom/custom_component/component.py
index a5e51801b..9a01e62ff 100644
--- a/src/backend/base/langflow/custom/custom_component/component.py
+++ b/src/backend/base/langflow/custom/custom_component/component.py
@@ -99,11 +99,14 @@ class Component(CustomComponent):
             setattr(self, output.name, output)
 
     def get_trace_as_inputs(self):
-        return {
+        predefined_inputs = {
             input_.name: input_.value
             for input_ in self.inputs
             if hasattr(input_, "trace_as_input") and input_.trace_as_input
         }
+        # Dynamic inputs
+        dynamic_inputs = {key: value for key, value in self._attributes.items() if key not in predefined_inputs}
+        return {**predefined_inputs, **dynamic_inputs}
 
     def get_trace_as_metadata(self):
         return {