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

Refactor CustomComponent repr_value handling (#2173)

Browse source 
* Refactor CustomComponent to handle repr_value more efficiently
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-14 09:48:35 -07:00 • committed by GitHub
commit 9daf66993f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -159,11 +159,11 @@ class CustomComponent(Component):
if self.repr_value == "":
self.repr_value = self.status
if isinstance(self.repr_value, dict):
return yaml.dump(self.repr_value)
if isinstance(self.repr_value, str):
return self.repr_value
if isinstance(self.repr_value, BaseModel) and not isinstance(self.repr_value, Record):
return str(self.repr_value)
self.repr_value = yaml.dump(self.repr_value)
if isinstance(self.repr_value, BaseModel) and not isinstance(self.repr_value, Data):
self.repr_value = str(self.repr_value)
elif hasattr(self.repr_value, "to_json"):
self.repr_value = self.repr_value.to_json()
return self.repr_value
def build_config(self):