From bd0e5694da8ac709e6e9d4be6820cb5f772317d6 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 23 Feb 2024 10:10:19 -0300 Subject: [PATCH] Update base_type in RoutingVertex constructor --- src/backend/langflow/graph/vertex/types.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/langflow/graph/vertex/types.py b/src/backend/langflow/graph/vertex/types.py index 957fbe78b..e80eef594 100644 --- a/src/backend/langflow/graph/vertex/types.py +++ b/src/backend/langflow/graph/vertex/types.py @@ -365,7 +365,7 @@ class ChatVertex(StatelessVertex): class RoutingVertex(StatelessVertex): def __init__(self, data: Dict, graph): - super().__init__(data, graph=graph, base_type="routing") + super().__init__(data, graph=graph, base_type="custom_components") self.use_result = True self.steps = [self._build, self._run]