jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add routing component to vertex types

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-22 18:46:21 -03:00
commit 0003e4b7e9
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/graph/graph/constants.py
View file

@ -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):