From 18a04ad170edf76ef65611176b993c208a21dfde Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 3 Jul 2023 21:56:49 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=80=20refactor(constants.py):=20refact?= =?UTF-8?q?or=20VERTEX=5FTYPE=5FMAP=20to=20use=20explicit=20import=20paths?= =?UTF-8?q?=20for=20better=20readability=20and=20maintainability=20The=20V?= =?UTF-8?q?ERTEX=5FTYPE=5FMAP=20dictionary=20in=20constants.py=20has=20bee?= =?UTF-8?q?n=20refactored=20to=20use=20explicit=20import=20paths=20for=20e?= =?UTF-8?q?ach=20vertex=20type.=20This=20improves=20the=20readability=20an?= =?UTF-8?q?d=20maintainability=20of=20the=20code=20by=20making=20it=20clea?= =?UTF-8?q?r=20where=20each=20vertex=20type=20is=20coming=20from.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/graph/graph/constants.py | 24 +++++++++---------- 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/src/backend/langflow/graph/graph/constants.py b/src/backend/langflow/graph/graph/constants.py index 40d77e31c..a2fd287eb 100644 --- a/src/backend/langflow/graph/graph/constants.py +++ b/src/backend/langflow/graph/graph/constants.py @@ -19,18 +19,18 @@ from typing import Dict, Type VERTEX_TYPE_MAP: Dict[str, Type[Vertex]] = { - **{t: types.types.PromptVertex for t in prompt_creator.to_list()}, - **{t: types.types.AgentVertex for t in agent_creator.to_list()}, - **{t: types.types.ChainVertex for t in chain_creator.to_list()}, - **{t: types.types.ToolVertex for t in tool_creator.to_list()}, - **{t: types.types.ToolkitVertex for t in toolkits_creator.to_list()}, - **{t: types.types.WrapperVertex for t in wrapper_creator.to_list()}, - **{t: types.types.LLMVertex for t in llm_creator.to_list()}, - **{t: types.types.MemoryVertex for t in memory_creator.to_list()}, - **{t: types.types.EmbeddingVertex for t in embedding_creator.to_list()}, - **{t: types.types.VectorStoreVertex for t in vectorstore_creator.to_list()}, - **{t: types.types.DocumentLoaderVertex for t in documentloader_creator.to_list()}, - **{t: types.types.TextSplitterVertex for t in textsplitter_creator.to_list()}, + **{t: types.PromptVertex for t in prompt_creator.to_list()}, + **{t: types.AgentVertex for t in agent_creator.to_list()}, + **{t: types.ChainVertex for t in chain_creator.to_list()}, + **{t: types.ToolVertex for t in tool_creator.to_list()}, + **{t: types.ToolkitVertex for t in toolkits_creator.to_list()}, + **{t: types.WrapperVertex for t in wrapper_creator.to_list()}, + **{t: types.LLMVertex for t in llm_creator.to_list()}, + **{t: types.MemoryVertex for t in memory_creator.to_list()}, + **{t: types.EmbeddingVertex for t in embedding_creator.to_list()}, + **{t: types.VectorStoreVertex for t in vectorstore_creator.to_list()}, + **{t: types.DocumentLoaderVertex for t in documentloader_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.RetrieverVertex for t in retriever_creator.to_list()}, }