From 16276f18dd126a908bc57617f80b50c6b4ab3d2e Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Thu, 6 Apr 2023 15:46:20 -0300
Subject: [PATCH] fix linting

---
 src/backend/langflow/api/validate.py | 1 -
 src/backend/langflow/graph/graph.py  | 3 +--
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/backend/langflow/api/validate.py b/src/backend/langflow/api/validate.py
index 15600f9f7..a60bcc506 100644
--- a/src/backend/langflow/api/validate.py
+++ b/src/backend/langflow/api/validate.py
@@ -7,7 +7,6 @@ from langflow.api.base import (
     PromptValidationResponse,
     validate_prompt,
 )
-from langflow.graph.utils import extract_input_variables_from_prompt
 from langflow.utils.logger import logger
 from langflow.utils.validate import validate_code
 
diff --git a/src/backend/langflow/graph/graph.py b/src/backend/langflow/graph/graph.py
index 81c488881..57b9082bb 100644
--- a/src/backend/langflow/graph/graph.py
+++ b/src/backend/langflow/graph/graph.py
@@ -19,7 +19,6 @@ from langflow.interface.prompts.base import prompt_creator
 from langflow.interface.toolkits.base import toolkits_creator
 from langflow.interface.tools.base import tool_creator
 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.memories.base import memory_creator
 from langflow.utils import payload
@@ -109,7 +108,7 @@ class Graph:
         return edges
 
     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: AgentNode for t in agent_creator.to_list()},
             **{t: ChainNode for t in chain_creator.to_list()},