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

refactor: Update get_artifact_type function in artifact.py

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 17:12:24 -03:00
commit 2cd0316506
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/base/langflow/schema/artifact.py
View file

@ -18,8 +18,12 @@ class ArtifactType(str, Enum):
def get_artifact_type(value, build_result=None) -> str:
result = ArtifactType.UNKNOWN
match value:
case Message():
enum_value = get_artifact_type(value.text)
result = ArtifactType(enum_value)
case Data():
result = ArtifactType.DATA
enum_value = get_artifact_type(value.data)
result = ArtifactType(enum_value)
case str():
result = ArtifactType.TEXT
@ -30,9 +34,6 @@ def get_artifact_type(value, build_result=None) -> str:
case list():
result = ArtifactType.ARRAY
case Message():
result = ArtifactType.MESSAGE
if result == ArtifactType.UNKNOWN:
if build_result and isinstance(build_result, Generator):
result = ArtifactType.STREAM