From 8d83c3a8bf6806114be1ac938d5c09e6b33014dc Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 14 Aug 2024 14:29:11 -0300 Subject: [PATCH] feat: Add method to set specific parameters on output object. (#3328) --- .../base/langflow/custom/custom_component/component.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/backend/base/langflow/custom/custom_component/component.py b/src/backend/base/langflow/custom/custom_component/component.py index 3f38ff45e..5c3c4b030 100644 --- a/src/backend/base/langflow/custom/custom_component/component.py +++ b/src/backend/base/langflow/custom/custom_component/component.py @@ -211,6 +211,13 @@ class Component(CustomComponent): return self._outputs[name] raise ValueError(f"Output {name} not found in {self.__class__.__name__}") + def set_on_output(self, name: str, **kwargs): + output = self.get_output(name) + for key, value in kwargs.items(): + if not hasattr(output, key): + raise ValueError(f"Output {name} does not have a method {key}") + setattr(output, key, value) + def set_output_value(self, name: str, value: Any): if name in self._outputs: self._outputs[name].value = value