Update base_type in RoutingVertex constructor

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-23 10:10:19 -03:00
commit bd0e5694da

View file

@ -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]