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

refactor: Improve handling of repr_value in CustomComponent

Browse source
This commit is contained in:
ogabrielluiz 2024-06-14 15:05:19 -03:00
commit 6f7efd01f2
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/base/langflow/interface/initialize/loading.py
View file

@ -160,7 +160,7 @@ async def build_custom_component(params: dict, custom_component: "CustomComponen
if raw is None and isinstance(build_result, (dict, Data, str)): if raw is None and isinstance(build_result, (dict, Data, str)):
raw = build_result.data if isinstance(build_result, Data) else build_result raw = build_result.data if isinstance(build_result, Data) else build_result
artifact_type = get_artifact_type(custom_component or raw, build_result) artifact_type = get_artifact_type(custom_component.repr_value or raw, build_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}
return custom_component, build_result, artifact return custom_component, build_result, artifact