fix linting

This commit is contained in:
Gabriel Almeida 2023-04-06 15:46:20 -03:00
commit 16276f18dd
2 changed files with 1 additions and 3 deletions

View file

@ -7,7 +7,6 @@ from langflow.api.base import (
PromptValidationResponse, PromptValidationResponse,
validate_prompt, validate_prompt,
) )
from langflow.graph.utils import extract_input_variables_from_prompt
from langflow.utils.logger import logger from langflow.utils.logger import logger
from langflow.utils.validate import validate_code from langflow.utils.validate import validate_code

View file

@ -19,7 +19,6 @@ from langflow.interface.prompts.base import prompt_creator
from langflow.interface.toolkits.base import toolkits_creator from langflow.interface.toolkits.base import toolkits_creator
from langflow.interface.tools.base import tool_creator from langflow.interface.tools.base import tool_creator
from langflow.interface.tools.constants import FILE_TOOLS from langflow.interface.tools.constants import FILE_TOOLS
from langflow.interface.tools.util import get_tools_dict
from langflow.interface.wrappers.base import wrapper_creator from langflow.interface.wrappers.base import wrapper_creator
from langflow.interface.memories.base import memory_creator from langflow.interface.memories.base import memory_creator
from langflow.utils import payload from langflow.utils import payload
@ -109,7 +108,7 @@ class Graph:
return edges return edges
def _get_node_class(self, node_type: str, node_lc_type: str) -> Type[Node]: def _get_node_class(self, node_type: str, node_lc_type: str) -> Type[Node]:
node_type_map = { node_type_map: Dict[str, Type[Node]] = {
**{t: PromptNode for t in prompt_creator.to_list()}, **{t: PromptNode for t in prompt_creator.to_list()},
**{t: AgentNode for t in agent_creator.to_list()}, **{t: AgentNode for t in agent_creator.to_list()},
**{t: ChainNode for t in chain_creator.to_list()}, **{t: ChainNode for t in chain_creator.to_list()},