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

refactor: Fix type hint for _logs attribute in CustomComponent class

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

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

@ -81,7 +81,7 @@ class CustomComponent(BaseComponent):
status: Optional[Any] = None status: Optional[Any] = None
"""The status of the component. This is displayed on the frontend. Defaults to None.""" """The status of the component. This is displayed on the frontend. Defaults to None."""
_flows_data: Optional[List[Data]] = None _flows_data: Optional[List[Data]] = None
_logs: Optional[List[Log]] = [] _logs: List[Log] = []
def update_state(self, name: str, value: Any): def update_state(self, name: str, value: Any):
if not self.vertex: if not self.vertex:
@ -92,7 +92,7 @@ class CustomComponent(BaseComponent):
raise ValueError(f"Error updating state: {e}") raise ValueError(f"Error updating state: {e}")
def stop(self, output_name: str | None): def stop(self, output_name: str | None):
if not output_name and len(self.vertex.outputs) == 1: if not output_name and self.vertex and len(self.vertex.outputs) == 1:
output_name = self.vertex.outputs[0]["name"] output_name = self.vertex.outputs[0]["name"]
else: else:
raise ValueError("You must specify an output name to call stop") raise ValueError("You must specify an output name to call stop")