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

🔀 refactor(constants.py): refactor VERTEX_TYPE_MAP to use explicit import paths for better readability and maintainability

Browse source 
The VERTEX_TYPE_MAP dictionary in constants.py has been refactored to use explicit import paths for each vertex type. This improves the readability and maintainability of the code by making it clear where each vertex type is coming from.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-03 21:56:49 -03:00
commit 18a04ad170
1 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -19,18 +19,18 @@ from typing import Dict, Type
VERTEX_TYPE_MAP: Dict[str, Type[Vertex]] = { VERTEX_TYPE_MAP: Dict[str, Type[Vertex]] = {
**{t: types.types.PromptVertex for t in prompt_creator.to_list()}, **{t: types.PromptVertex for t in prompt_creator.to_list()},
**{t: types.types.AgentVertex for t in agent_creator.to_list()}, **{t: types.AgentVertex for t in agent_creator.to_list()},
**{t: types.types.ChainVertex for t in chain_creator.to_list()}, **{t: types.ChainVertex for t in chain_creator.to_list()},
**{t: types.types.ToolVertex for t in tool_creator.to_list()}, **{t: types.ToolVertex for t in tool_creator.to_list()},
**{t: types.types.ToolkitVertex for t in toolkits_creator.to_list()}, **{t: types.ToolkitVertex for t in toolkits_creator.to_list()},
**{t: types.types.WrapperVertex for t in wrapper_creator.to_list()}, **{t: types.WrapperVertex for t in wrapper_creator.to_list()},
**{t: types.types.LLMVertex for t in llm_creator.to_list()}, **{t: types.LLMVertex for t in llm_creator.to_list()},
**{t: types.types.MemoryVertex for t in memory_creator.to_list()}, **{t: types.MemoryVertex for t in memory_creator.to_list()},
**{t: types.types.EmbeddingVertex for t in embedding_creator.to_list()}, **{t: types.EmbeddingVertex for t in embedding_creator.to_list()},
**{t: types.types.VectorStoreVertex for t in vectorstore_creator.to_list()}, **{t: types.VectorStoreVertex for t in vectorstore_creator.to_list()},
**{t: types.types.DocumentLoaderVertex for t in documentloader_creator.to_list()}, **{t: types.DocumentLoaderVertex for t in documentloader_creator.to_list()},
**{t: types.types.TextSplitterVertex for t in textsplitter_creator.to_list()}, **{t: types.TextSplitterVertex for t in textsplitter_creator.to_list()},
**{t: types.OutputParserVertex for t in output_parser_creator.to_list()}, **{t: types.OutputParserVertex for t in output_parser_creator.to_list()},
**{t: types.RetrieverVertex for t in retriever_creator.to_list()}, **{t: types.RetrieverVertex for t in retriever_creator.to_list()},
} }