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

refactor: Update Component class to use correct variable for artifact_type

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 12:10:30 -03:00
commit 305bdda01f
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,6 +1,5 @@
import inspect
from typing import AsyncIterator, Awaitable, Callable, ClassVar, Generator, Iterator, List, Optional, Union
from uuid import UUID
import yaml
@ -107,7 +106,7 @@ class Component(CustomComponent):
custom_repr = result
if not isinstance(custom_repr, str):
custom_repr = str(custom_repr)
raw = self.status
raw = result
if hasattr(raw, "data") and raw is not None:
raw = raw.data
if raw is None:
@ -118,7 +117,7 @@ class Component(CustomComponent):
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.repr_value or raw, result)
artifact_type = get_artifact_type(self.status or raw, result)
raw = post_process_raw(raw, artifact_type)
artifact = {"repr": custom_repr, "raw": raw, "type": artifact_type}
_artifacts[output.name] = artifact