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

refactor: Update artifact handling in Component class

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 17:12:19 -03:00
commit 33331dced8
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/base/langflow/custom/custom_component/component.py
View file

@ -114,6 +114,12 @@ class Component(CustomComponent):
if not isinstance(custom_repr, str): if not isinstance(custom_repr, str):
custom_repr = str(custom_repr) custom_repr = str(custom_repr)
raw = result raw = result
if self.status is None:
artifact_value = raw
else:
artifact_value = self.status
raw = self.status
if hasattr(raw, "data") and raw is not None: if hasattr(raw, "data") and raw is not None:
raw = raw.data raw = raw.data
if raw is None: if raw is None:
@ -123,11 +129,6 @@ class Component(CustomComponent):
raw = raw.model_dump() raw = raw.model_dump()
if raw is None and isinstance(result, (dict, Data, str)): if raw is None and isinstance(result, (dict, Data, str)):
raw = result.data if isinstance(result, Data) else result raw = result.data if isinstance(result, Data) else result
if self.status is None:
artifact_value = raw
else:
artifact_value = self.status
raw = self.status
artifact_type = get_artifact_type(artifact_value, result) artifact_type = get_artifact_type(artifact_value, result)
raw = post_process_raw(raw, artifact_type) raw = post_process_raw(raw, artifact_type)
artifact = {"repr": custom_repr, "raw": raw, "type": artifact_type} artifact = {"repr": custom_repr, "raw": raw, "type": artifact_type}