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

fix union-attr error reported by mypy

Browse source
This commit is contained in:
italojohnny 2024-06-20 21:06:54 -03:00
commit 628e78959d
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -166,6 +166,10 @@ async def build_custom_component(params: dict, custom_component: "CustomComponen
artifact_type = get_artifact_type(custom_component.repr_value or raw, build_result)
raw = post_process_raw(raw, artifact_type)
artifact = {"repr": custom_repr, "raw": raw, "type": artifact_type}
custom_component._artifacts = {custom_component.vertex.outputs[0].get("name"): artifact}
custom_component._results = {custom_component.vertex.outputs[0].get("name"): build_result}
return custom_component, build_result, artifact
if custom_component.vertex is not None:
custom_component._artifacts = {custom_component.vertex.outputs[0].get("name"): artifact}
custom_component._results = {custom_component.vertex.outputs[0].get("name"): build_result}
return custom_component, build_result, artifact
raise ValueError("Custom component does not have a vertex")