refactor: Update Vertex class to store artifacts in a dictionary

2024-06-17 19:35:18 -03:00 · 2024-06-17 19:35:18 -03:00 · fa68a8fee6
commit fa68a8fee6
parent 2efa10ed37
1 changed files with 1 additions and 0 deletions
--- a/src/backend/base/langflow/graph/vertex/base.py
+++ b/src/backend/base/langflow/graph/vertex/base.py
@ -661,6 +661,7 @@ class Vertex:
                self._custom_component, self._built_object, self.artifacts = result
                self.artifacts_raw = self.artifacts.get("raw", None)
                self.artifacts_type = self.artifacts.get("type", None) or ArtifactType.UNKNOWN.value
+                self.artifacts = {self.outputs[0]["name"]: self.artifacts}
                self.logs = build_logs(self)
        else:
            self._built_object = result