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

refactor: Update Component class to handle status and artifact value separately

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 15:46:52 -03:00
commit 43381b2820
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -124,8 +124,12 @@ class Component(CustomComponent):
raw = raw.model_dump()
if raw is None and isinstance(result, (dict, Data, str)):
raw = result.data if isinstance(result, Data) else result
artifact_type = get_artifact_type(self.status or raw, 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)
raw = post_process_raw(raw, artifact_type)
artifact = {"repr": custom_repr, "raw": raw, "type": artifact_type}
_artifacts[output.name] = artifact