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

🐛 fix(types.py): add conditional check for task_id and is_task to improve object representation in CustomComponentVertex

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-17 10:15:21 -03:00
commit ab6515515d
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/graph/vertex/types.py
View file

@ -255,5 +255,10 @@ class CustomComponentVertex(Vertex):
super().__init__(data, base_type="custom_components", is_task=True)
def _built_object_repr(self):
if self.task_id and self.is_task:
if task := self.get_task():
return str(task.info)
else:
return f"Task {self.task_id} is not running"
if self.artifacts and "repr" in self.artifacts:
return self.artifacts["repr"] or super()._built_object_repr()