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

refactor: Add output_name parameter to stop method in CustomComponent

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 18:52:15 -03:00
commit 7f6aebb490
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/custom/custom_component/custom_component.py
View file

@ -91,7 +91,11 @@ class CustomComponent(BaseComponent):
except Exception as e: except Exception as e:
raise ValueError(f"Error updating state: {e}") raise ValueError(f"Error updating state: {e}")
def stop(self, output_name: str): def stop(self, output_name: str | None):
if not output_name and len(self.vertex.outputs) == 1:
output_name = self.vertex.outputs[0]["name"]
else:
raise ValueError("You must specify an output name to call stop")
if not self.vertex: if not self.vertex:
raise ValueError("Vertex is not set") raise ValueError("Vertex is not set")
try: try: