From 5561b4ec52f1c8e62c74accabaae26bc4f4b910e Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 21:51:45 -0300 Subject: [PATCH] refactor: Update stop method in CustomComponent to allow for optional output_name parameter --- .../base/langflow/custom/custom_component/custom_component.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/base/langflow/custom/custom_component/custom_component.py b/src/backend/base/langflow/custom/custom_component/custom_component.py index aa2f7bff4..c9bc1bfdf 100644 --- a/src/backend/base/langflow/custom/custom_component/custom_component.py +++ b/src/backend/base/langflow/custom/custom_component/custom_component.py @@ -91,7 +91,7 @@ class CustomComponent(BaseComponent): except Exception as e: raise ValueError(f"Error updating state: {e}") - def stop(self, output_name: str | None): + def stop(self, output_name: str | None = None): if not output_name and self.vertex and len(self.vertex.outputs) == 1: output_name = self.vertex.outputs[0]["name"] else: