From 04b98ba36e952eaa4572c87879989e403046e7f7 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 21 Feb 2024 12:11:07 -0300 Subject: [PATCH] Add build time tracking to Vertex class --- src/backend/langflow/graph/vertex/base.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py index 47c9bf2ec..e76bd714e 100644 --- a/src/backend/langflow/graph/vertex/base.py +++ b/src/backend/langflow/graph/vertex/base.py @@ -57,6 +57,14 @@ class Vertex: self.is_interface_component = False self.use_result = False + self.build_times: List[float] = [] + + @property + def avg_build_time(self): + return sum(self.build_times) / len(self.build_times) if self.build_times else 0 + + def add_build_time(self, time): + self.build_times.append(time) # Build a result dict for each edge # like so: {edge.target.id: {edge.target_param: self._built_object}}