From 0003e4b7e9bbebeccd02b9d927892a743e77a46d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 22 Feb 2024 18:46:21 -0300 Subject: [PATCH] Add routing component to vertex types --- src/backend/langflow/graph/graph/constants.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/graph/graph/constants.py b/src/backend/langflow/graph/graph/constants.py index f85761ed4..4317ac99d 100644 --- a/src/backend/langflow/graph/graph/constants.py +++ b/src/backend/langflow/graph/graph/constants.py @@ -15,7 +15,9 @@ from langflow.interface.tools.base import tool_creator from langflow.interface.wrappers.base import wrapper_creator from langflow.utils.lazy_load import LazyLoadDictBase -chat_components = ["ChatInput", "ChatOutput", "TextInput", "SessionID"] +chat_components = ["ChatInput", "ChatOutput", "TextInput"] +# For now the only routing we support is to stop or run the next component +routing_components = ["ShouldRunNext"] class VertexTypesDict(LazyLoadDictBase): @@ -51,6 +53,7 @@ class VertexTypesDict(LazyLoadDictBase): **{t: types.CustomComponentVertex for t in custom_component_creator.to_list()}, **{t: types.RetrieverVertex for t in retriever_creator.to_list()}, **{t: types.ChatVertex for t in chat_components}, + **{t: types.RoutingVertex for t in routing_components}, } def get_custom_component_vertex_type(self):