From fa68a8fee6be156973e6074bb1c1a2890c1e23d4 Mon Sep 17 00:00:00 2001
From: ogabrielluiz <gabriel@langflow.org>
Date: Mon, 17 Jun 2024 19:35:18 -0300
Subject: [PATCH] refactor: Update Vertex class to store artifacts in a
 dictionary

---
 src/backend/base/langflow/graph/vertex/base.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/backend/base/langflow/graph/vertex/base.py b/src/backend/base/langflow/graph/vertex/base.py
index 0e3dfd186..736fc8f37 100644
--- 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