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

refactor: Update Component class to include dynamic inputs in get_trace_as_inputs method

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 17:59:59 -03:00
commit 60642faf71
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/custom/custom_component/component.py
View file

@ -99,11 +99,14 @@ class Component(CustomComponent):
setattr(self, output.name, output) setattr(self, output.name, output)
def get_trace_as_inputs(self): def get_trace_as_inputs(self):
return { predefined_inputs = {
input_.name: input_.value input_.name: input_.value
for input_ in self.inputs for input_ in self.inputs
if hasattr(input_, "trace_as_input") and input_.trace_as_input 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): def get_trace_as_metadata(self):
return { return {